mirror of
https://github.com/weechat/weechat.git
synced 2026-06-23 03:16:37 +02:00
New plugin interface (loads dynamic C libraries)
This commit is contained in:
Sebastien Helleu
@@ -0,0 +1,855 @@
|
||||
/*
|
||||
* Copyright (c) 2003-2005 by FlashCode <flashcode@flashtux.org>
|
||||
* See README for License detail, AUTHORS for developers list.
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
*/
|
||||
|
||||
/* weechat-python.c: Python plugin support for WeeChat */
|
||||
|
||||
|
||||
#include <Python.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#undef _
|
||||
#include "../../weechat-plugin.h"
|
||||
#include "../weechat-script.h"
|
||||
|
||||
|
||||
char plugin_name[] = "Python";
|
||||
char plugin_version[] = "0.1";
|
||||
char plugin_description[] = "Python scripts support";
|
||||
|
||||
t_weechat_plugin *python_plugin;
|
||||
|
||||
t_plugin_script *python_scripts = NULL;
|
||||
t_plugin_script *python_current_script = NULL;
|
||||
char *python_current_script_filename = NULL;
|
||||
PyThreadState *python_mainThreadState = NULL;
|
||||
|
||||
|
||||
/*
|
||||
* weechat_python_exec: execute a Python script
|
||||
*/
|
||||
|
||||
int
|
||||
weechat_python_exec (t_weechat_plugin *plugin,
|
||||
t_plugin_script *script,
|
||||
char *function, char *server, char *arguments)
|
||||
{
|
||||
PyObject *evMain;
|
||||
PyObject *evDict;
|
||||
PyObject *evFunc;
|
||||
|
||||
PyThreadState_Swap (NULL);
|
||||
|
||||
PyEval_AcquireLock ();
|
||||
|
||||
PyThreadState_Swap (script->interpreter);
|
||||
|
||||
evMain = PyImport_AddModule ((char *) "__main__");
|
||||
evDict = PyModule_GetDict (evMain);
|
||||
evFunc = PyDict_GetItemString (evDict, function);
|
||||
|
||||
if ( !(evFunc && PyCallable_Check (evFunc)) )
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to run function \"%s\"",
|
||||
function);
|
||||
PyEval_ReleaseLock();
|
||||
return 1;
|
||||
}
|
||||
|
||||
PyObject_CallFunction(evFunc, "ss", server == NULL ? "" : server, arguments == NULL ? "" : arguments);
|
||||
|
||||
PyEval_ReleaseLock();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_python_handler: general message and command handler for Python
|
||||
*/
|
||||
|
||||
int
|
||||
weechat_python_handler (t_weechat_plugin *plugin,
|
||||
char *server, char *command, char *arguments,
|
||||
char *handler_args, void *handler_pointer)
|
||||
{
|
||||
/* make gcc happy */
|
||||
(void) command;
|
||||
|
||||
weechat_python_exec (plugin, (t_plugin_script *)handler_pointer,
|
||||
handler_args, server, arguments);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.register: startup function for all WeeChat Python scripts
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_register (PyObject *self, PyObject *args)
|
||||
{
|
||||
char *name, *version, *shutdown_func, *description;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
name = NULL;
|
||||
version = NULL;
|
||||
shutdown_func = NULL;
|
||||
description = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "ssss", &name, &version, &shutdown_func, &description))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"register\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (weechat_script_search (python_plugin, &python_scripts, name))
|
||||
{
|
||||
/* error: another scripts already exists with this name! */
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: unable to register "
|
||||
"\"%s\" script (another script "
|
||||
"already exists with this name)",
|
||||
name);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* register script */
|
||||
python_current_script = weechat_script_add (python_plugin,
|
||||
&python_scripts,
|
||||
(python_current_script_filename) ?
|
||||
python_current_script_filename : "",
|
||||
name, version, shutdown_func,
|
||||
description);
|
||||
if (python_current_script)
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python: registered script \"%s\", "
|
||||
"version %s (%s)",
|
||||
name, version, description);
|
||||
}
|
||||
else
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: unable to load script "
|
||||
"\"%s\" (not enough memory)",
|
||||
name);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
Py_INCREF (Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.print: print message into a buffer (current or specified one)
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_print (PyObject *self, PyObject *args)
|
||||
{
|
||||
char *message, *channel_name, *server_name;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
message = NULL;
|
||||
channel_name = NULL;
|
||||
server_name = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "s|ss", &message, &channel_name, &server_name))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"print\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
python_plugin->printf (python_plugin,
|
||||
server_name, channel_name,
|
||||
"%s", message);
|
||||
return Py_BuildValue ("i", 1);
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.print_infobar: print message to infobar
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_print_infobar (PyObject *self, PyObject *args)
|
||||
{
|
||||
int delay;
|
||||
char *message;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
delay = 1;
|
||||
message = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "is", &delay, &message))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"print_infobar\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
python_plugin->infobar_printf (python_plugin, delay, message);
|
||||
|
||||
Py_INCREF (Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.command: send command to server
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_command (PyObject *self, PyObject *args)
|
||||
{
|
||||
char *command, *channel_name, *server_name;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
command = NULL;
|
||||
channel_name = NULL;
|
||||
server_name = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "s|ss", &command, &channel_name, &server_name))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"command\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
python_plugin->exec_command (python_plugin,
|
||||
server_name, channel_name,
|
||||
command);
|
||||
return Py_BuildValue ("i", 1);
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.add_message_handler: add handler for messages
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_add_message_handler (PyObject *self, PyObject *args)
|
||||
{
|
||||
char *message, *function;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
message = NULL;
|
||||
function = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "ss", &message, &function))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"add_message_handler\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (python_current_script)
|
||||
python_plugin->msg_handler_add (python_plugin, message,
|
||||
weechat_python_handler, function,
|
||||
(void *)python_current_script);
|
||||
else
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: unable to add message handler, "
|
||||
"script not initialized");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
Py_INCREF (Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.add_command_handler: define/redefines commands
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_add_command_handler (PyObject *self, PyObject *args)
|
||||
{
|
||||
char *command, *function, *description, *arguments, *arguments_description;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
command = NULL;
|
||||
function = NULL;
|
||||
description = NULL;
|
||||
arguments = NULL;
|
||||
arguments_description = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "ss|sss", &command, &function,
|
||||
&description, &arguments, &arguments_description))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"add_command_handler\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (python_current_script)
|
||||
python_plugin->cmd_handler_add (python_plugin,
|
||||
command,
|
||||
description,
|
||||
arguments,
|
||||
arguments_description,
|
||||
weechat_python_handler,
|
||||
function,
|
||||
(void *)python_current_script);
|
||||
else
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: unable to add command handler, "
|
||||
"script not initialized");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
Py_INCREF (Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.get_info: get various infos
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_get_info (PyObject *self, PyObject *args)
|
||||
{
|
||||
char *arg, *server_name, *channel_name, *info;
|
||||
PyObject *object;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
|
||||
arg = NULL;
|
||||
server_name = NULL;
|
||||
channel_name = NULL;
|
||||
|
||||
if (!PyArg_ParseTuple (args, "s|ss", &arg, &server_name, &channel_name))
|
||||
{
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: wrong parameters for "
|
||||
"\"get_info\" function");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (arg)
|
||||
{
|
||||
info = python_plugin->get_info (python_plugin, arg, server_name, channel_name);
|
||||
|
||||
if (info)
|
||||
{
|
||||
object = Py_BuildValue ("s", info);
|
||||
free (info);
|
||||
return object;
|
||||
}
|
||||
else
|
||||
return Py_BuildValue ("s", "");
|
||||
}
|
||||
|
||||
return Py_BuildValue ("i", 1);
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat.get_dcc_info: get infos about DCC
|
||||
*/
|
||||
|
||||
static PyObject *
|
||||
weechat_python_get_dcc_info (PyObject *self, PyObject *args)
|
||||
{
|
||||
t_plugin_dcc_info *dcc_info, *ptr_dcc;
|
||||
int dcc_count;
|
||||
PyObject *list, *listvalue;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) self;
|
||||
(void) args;
|
||||
|
||||
dcc_info = python_plugin->get_dcc_info (python_plugin);
|
||||
dcc_count = 0;
|
||||
|
||||
if (!dcc_info)
|
||||
return Py_None;
|
||||
|
||||
for (ptr_dcc = dcc_info; ptr_dcc; ptr_dcc = ptr_dcc->next_dcc)
|
||||
{
|
||||
dcc_count++;
|
||||
}
|
||||
|
||||
list = PyList_New (dcc_count);
|
||||
|
||||
if (!list)
|
||||
{
|
||||
python_plugin->free_dcc_info (python_plugin, dcc_info);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
dcc_count = 0;
|
||||
for(ptr_dcc = dcc_info; ptr_dcc; ptr_dcc = ptr_dcc->next_dcc)
|
||||
{
|
||||
listvalue = Py_BuildValue ("{s:s,s:s,s:i,s:i,s:k,s:k,s:k,s:i,s:s,s:s,"
|
||||
"s:s,s:s,s:k,s:k,s:k,s:k}",
|
||||
"server", ptr_dcc->server,
|
||||
"channel", ptr_dcc->channel,
|
||||
"type", ptr_dcc->type,
|
||||
"status", ptr_dcc->status,
|
||||
"start_time", ptr_dcc->start_time,
|
||||
"start_transfer", ptr_dcc->start_transfer,
|
||||
"address", ptr_dcc->addr,
|
||||
"port", ptr_dcc->port,
|
||||
"nick", ptr_dcc->nick,
|
||||
"remote_file", ptr_dcc->filename,
|
||||
"local_file", ptr_dcc->local_filename,
|
||||
"filename_suffix", ptr_dcc->filename_suffix,
|
||||
"size", ptr_dcc->size,
|
||||
"pos", ptr_dcc->pos,
|
||||
"start_resume", ptr_dcc->start_resume,
|
||||
"cps", ptr_dcc->bytes_per_sec);
|
||||
if (listvalue)
|
||||
{
|
||||
if (PyList_SetItem (list, dcc_count, listvalue) != 0)
|
||||
{
|
||||
PyMem_Free (listvalue);
|
||||
PyMem_Free (list);
|
||||
python_plugin->free_dcc_info (python_plugin, dcc_info);
|
||||
return Py_None;
|
||||
}
|
||||
PyMem_Free (listvalue);
|
||||
}
|
||||
else
|
||||
{
|
||||
python_plugin->free_dcc_info (python_plugin, dcc_info);
|
||||
return Py_None;
|
||||
}
|
||||
dcc_count++;
|
||||
}
|
||||
|
||||
python_plugin->free_dcc_info (python_plugin, dcc_info);
|
||||
|
||||
return list;
|
||||
}
|
||||
|
||||
/*
|
||||
* Python subroutines
|
||||
*/
|
||||
|
||||
static
|
||||
PyMethodDef weechat_funcs[] = {
|
||||
{ "register", weechat_python_register, METH_VARARGS, "" },
|
||||
{ "prnt", weechat_python_print, METH_VARARGS, "" },
|
||||
{ "print_infobar", weechat_python_print_infobar, METH_VARARGS, "" },
|
||||
{ "command", weechat_python_command, METH_VARARGS, "" },
|
||||
{ "add_message_handler", weechat_python_add_message_handler, METH_VARARGS, "" },
|
||||
{ "add_command_handler", weechat_python_add_command_handler, METH_VARARGS, "" },
|
||||
{ "get_info", weechat_python_get_info, METH_VARARGS, "" },
|
||||
{ "get_dcc_info", weechat_python_get_dcc_info, METH_VARARGS, "" },
|
||||
{ NULL, NULL, 0, NULL }
|
||||
};
|
||||
|
||||
/*
|
||||
* weechat_python_load: load a Python script
|
||||
*/
|
||||
|
||||
int
|
||||
weechat_python_load (t_weechat_plugin *plugin, char *filename)
|
||||
{
|
||||
FILE *fp;
|
||||
PyThreadState *python_current_interpreter;
|
||||
|
||||
plugin->printf_server (plugin, "Loading Python script \"%s\"", filename);
|
||||
|
||||
if ((fp = fopen (filename, "r")) == NULL)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to open file \"%s\"",
|
||||
filename);
|
||||
return 0;
|
||||
}
|
||||
|
||||
python_current_script = NULL;
|
||||
|
||||
PyThreadState_Swap(NULL);
|
||||
|
||||
PyEval_AcquireLock();
|
||||
|
||||
python_current_interpreter = PyThreadState_New (python_mainThreadState->interp);
|
||||
|
||||
if (python_current_interpreter == NULL)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to create new sub-interpreter");
|
||||
PyEval_ReleaseLock();
|
||||
fclose (fp);
|
||||
return 0;
|
||||
}
|
||||
|
||||
PyThreadState_Swap (python_current_interpreter);
|
||||
|
||||
if (Py_InitModule ("weechat", weechat_funcs) == NULL)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to initialize WeeChat module");
|
||||
PyThreadState_Swap (NULL);
|
||||
PyThreadState_Clear (python_current_interpreter);
|
||||
PyThreadState_Delete (python_current_interpreter);
|
||||
PyEval_ReleaseLock ();
|
||||
fclose (fp);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (PyRun_SimpleString (
|
||||
"import weechat, sys, string\n"
|
||||
|
||||
"class weechatStdout:\n"
|
||||
"\tdef write(self, str):\n"
|
||||
"\t\tstr = string.strip(str)\n"
|
||||
"\t\tif str != \"\":\n"
|
||||
"\t\t\tweechat.prnt(\"Python stdout : \" + str, \"\")\n"
|
||||
|
||||
"class weechatStderr:\n"
|
||||
"\tdef write(self, str):\n"
|
||||
"\t\tstr = string.strip(str)\n"
|
||||
"\t\tif str != \"\":\n"
|
||||
"\t\t\tweechat.prnt(\"Python stderr : \" + str, \"\")\n"
|
||||
|
||||
"sys.stdout = weechatStdout()\n"
|
||||
"sys.stderr = weechatStderr()\n"
|
||||
) != 0)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python warning: unable to redirect stdout and stderr");
|
||||
}
|
||||
|
||||
python_current_script_filename = strdup (filename);
|
||||
|
||||
if (PyRun_SimpleFile (fp, filename) != 0)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to parse file \"%s\"",
|
||||
filename);
|
||||
free (python_current_script_filename);
|
||||
PyThreadState_Swap (NULL);
|
||||
PyThreadState_Clear (python_current_interpreter);
|
||||
PyThreadState_Delete (python_current_interpreter);
|
||||
PyEval_ReleaseLock ();
|
||||
fclose (fp);
|
||||
return 0;
|
||||
}
|
||||
|
||||
fclose (fp);
|
||||
free (python_current_script_filename);
|
||||
|
||||
if (python_current_script == NULL)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: function \"register\" not found "
|
||||
"in file \"%s\"",
|
||||
filename);
|
||||
PyThreadState_Swap (NULL);
|
||||
PyThreadState_Clear (python_current_interpreter);
|
||||
PyThreadState_Delete (python_current_interpreter);
|
||||
PyEval_ReleaseLock ();
|
||||
return 0;
|
||||
}
|
||||
|
||||
python_current_script->interpreter = (PyThreadState *) python_current_interpreter;
|
||||
PyThreadState_Swap (NULL);
|
||||
PyEval_ReleaseLock ();
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_python_unload: unload a Python script
|
||||
*/
|
||||
|
||||
void
|
||||
weechat_python_unload (t_weechat_plugin *plugin, t_plugin_script *script)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Unloading Python script \"%s\"",
|
||||
script->name);
|
||||
|
||||
if (script->shutdown_func[0])
|
||||
weechat_python_exec (plugin, script, script->shutdown_func, "", "");
|
||||
|
||||
PyThreadState_Swap (NULL);
|
||||
PyThreadState_Clear (script->interpreter);
|
||||
PyThreadState_Delete (script->interpreter);
|
||||
|
||||
weechat_script_remove (plugin, &python_scripts, script);
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_python_unload_name: unload a Python script by name
|
||||
*/
|
||||
|
||||
void
|
||||
weechat_python_unload_name (t_weechat_plugin *plugin, char *name)
|
||||
{
|
||||
t_plugin_script *ptr_script;
|
||||
|
||||
ptr_script = weechat_script_search (plugin, &python_scripts, name);
|
||||
if (ptr_script)
|
||||
{
|
||||
weechat_python_unload (plugin, ptr_script);
|
||||
plugin->printf_server (plugin,
|
||||
"Python script \"%s\" unloaded",
|
||||
name);
|
||||
}
|
||||
else
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: script \"%s\" not loaded",
|
||||
name);
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_python_unload_all: unload all Python scripts
|
||||
*/
|
||||
|
||||
void
|
||||
weechat_python_unload_all (t_weechat_plugin *plugin)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Unloading all Python scripts");
|
||||
while (python_scripts)
|
||||
weechat_python_unload (plugin, python_scripts);
|
||||
|
||||
plugin->printf_server (plugin,
|
||||
"Python scripts unloaded");
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_python_cmd: /python command handler
|
||||
*/
|
||||
|
||||
int
|
||||
weechat_python_cmd (t_weechat_plugin *plugin,
|
||||
char *server, char *command, char *arguments,
|
||||
char *handler_args, void *handler_pointer)
|
||||
{
|
||||
int argc, path_length, handler_found;
|
||||
char **argv, *path_script, *dir_home;
|
||||
t_plugin_script *ptr_plugin_script;
|
||||
t_plugin_msg_handler *ptr_msg_handler;
|
||||
t_plugin_cmd_handler *ptr_cmd_handler;
|
||||
|
||||
/* make gcc happy */
|
||||
(void) server;
|
||||
(void) command;
|
||||
(void) handler_args;
|
||||
(void) handler_pointer;
|
||||
|
||||
argv = plugin->explode_string (plugin, arguments, " ", 0, &argc);
|
||||
|
||||
switch (argc)
|
||||
{
|
||||
case 0:
|
||||
/* list registered Python scripts */
|
||||
plugin->printf_server (plugin, "");
|
||||
plugin->printf_server (plugin, "Registered Python scripts:");
|
||||
if (python_scripts)
|
||||
{
|
||||
for (ptr_plugin_script = python_scripts; ptr_plugin_script;
|
||||
ptr_plugin_script = ptr_plugin_script->next_script)
|
||||
{
|
||||
plugin->printf_server (plugin, " %s v%s%s%s",
|
||||
ptr_plugin_script->name,
|
||||
ptr_plugin_script->version,
|
||||
(ptr_plugin_script->description[0]) ? " - " : "",
|
||||
ptr_plugin_script->description);
|
||||
}
|
||||
}
|
||||
else
|
||||
plugin->printf_server (plugin, " (none)");
|
||||
|
||||
/* list Python message handlers */
|
||||
plugin->printf_server (plugin, "");
|
||||
plugin->printf_server (plugin, "Python message handlers:");
|
||||
handler_found = 0;
|
||||
for (ptr_msg_handler = plugin->msg_handlers; ptr_msg_handler;
|
||||
ptr_msg_handler = ptr_msg_handler->next_handler)
|
||||
{
|
||||
if (ptr_msg_handler->msg_handler_args)
|
||||
{
|
||||
handler_found = 1;
|
||||
plugin->printf_server (plugin, " IRC(%s) => Python(%s)",
|
||||
ptr_msg_handler->irc_command,
|
||||
ptr_msg_handler->msg_handler_args);
|
||||
}
|
||||
}
|
||||
if (!handler_found)
|
||||
plugin->printf_server (plugin, " (none)");
|
||||
|
||||
/* list Python command handlers */
|
||||
plugin->printf_server (plugin, "");
|
||||
plugin->printf_server (plugin, "Python command handlers:");
|
||||
handler_found = 0;
|
||||
for (ptr_cmd_handler = plugin->cmd_handlers; ptr_cmd_handler;
|
||||
ptr_cmd_handler = ptr_cmd_handler->next_handler)
|
||||
{
|
||||
if (ptr_cmd_handler->cmd_handler_args)
|
||||
{
|
||||
handler_found = 1;
|
||||
plugin->printf_server (plugin, " /%s => Python(%s)",
|
||||
ptr_cmd_handler->command,
|
||||
ptr_cmd_handler->cmd_handler_args);
|
||||
}
|
||||
}
|
||||
if (!handler_found)
|
||||
plugin->printf_server (plugin, " (none)");
|
||||
break;
|
||||
case 1:
|
||||
if (plugin->ascii_strcasecmp (plugin, argv[0], "autoload") == 0)
|
||||
weechat_script_auto_load (plugin, "python", weechat_python_load);
|
||||
else if (plugin->ascii_strcasecmp (plugin, argv[0], "reload") == 0)
|
||||
{
|
||||
weechat_python_unload_all (plugin);
|
||||
weechat_script_auto_load (plugin, "python", weechat_python_load);
|
||||
}
|
||||
else if (plugin->ascii_strcasecmp (plugin, argv[0], "unload") == 0)
|
||||
weechat_python_unload_all (plugin);
|
||||
break;
|
||||
case 2:
|
||||
if (plugin->ascii_strcasecmp (plugin, argv[0], "load") == 0)
|
||||
{
|
||||
/* load Python script */
|
||||
if ((strstr (argv[1], "/")) || (strstr (argv[1], "\\")))
|
||||
path_script = NULL;
|
||||
else
|
||||
{
|
||||
dir_home = plugin->get_info (plugin, "weechat_dir", NULL, NULL);
|
||||
if (dir_home)
|
||||
{
|
||||
path_length = strlen (dir_home) + strlen (argv[1]) + 16;
|
||||
path_script = (char *) malloc (path_length * sizeof (char));
|
||||
if (path_script)
|
||||
snprintf (path_script, path_length, "%s/python/%s",
|
||||
dir_home, argv[1]);
|
||||
else
|
||||
path_script = NULL;
|
||||
free (dir_home);
|
||||
}
|
||||
else
|
||||
path_script = NULL;
|
||||
}
|
||||
weechat_python_load (plugin, (path_script) ? path_script : argv[1]);
|
||||
if (path_script)
|
||||
free (path_script);
|
||||
}
|
||||
else if (plugin->ascii_strcasecmp (plugin, argv[0], "unload") == 0)
|
||||
{
|
||||
/* unload Python script */
|
||||
weechat_python_unload_name (plugin, argv[1]);
|
||||
}
|
||||
else
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unknown option for "
|
||||
"\"python\" command");
|
||||
}
|
||||
break;
|
||||
default:
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: wrong argument count for \"python\" command");
|
||||
}
|
||||
plugin->free_exploded_string (plugin, argv);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_plugin_init: initialize Python plugin
|
||||
*/
|
||||
|
||||
int
|
||||
weechat_plugin_init (t_weechat_plugin *plugin)
|
||||
{
|
||||
python_plugin = plugin;
|
||||
|
||||
plugin->printf_server (plugin, "Loading Python module \"weechat\"");
|
||||
|
||||
Py_Initialize ();
|
||||
if (Py_IsInitialized () == 0)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to launch global interpreter");
|
||||
return 0;
|
||||
}
|
||||
|
||||
PyEval_InitThreads();
|
||||
|
||||
python_mainThreadState = PyThreadState_Get();
|
||||
|
||||
if (python_mainThreadState == NULL)
|
||||
{
|
||||
plugin->printf_server (plugin,
|
||||
"Python error: unable to get current interpreter state");
|
||||
return 0;
|
||||
}
|
||||
|
||||
PyEval_ReleaseLock ();
|
||||
|
||||
plugin->cmd_handler_add (plugin, "python",
|
||||
"list/load/unload Python scripts",
|
||||
"[load filename] | [autoload] | [reload] | [unload]",
|
||||
"filename: Python script (file) to load\n\n"
|
||||
"Without argument, /python command lists all loaded Python scripts.",
|
||||
weechat_python_cmd, NULL, NULL);
|
||||
|
||||
plugin->mkdir_home (plugin, "python");
|
||||
plugin->mkdir_home (plugin, "python/autoload");
|
||||
|
||||
weechat_script_auto_load (plugin, "python", weechat_python_load);
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
/*
|
||||
* weechat_plugin_end: shutdown Python interface
|
||||
*/
|
||||
|
||||
void
|
||||
weechat_plugin_end (t_weechat_plugin *plugin)
|
||||
{
|
||||
/* unload all scripts */
|
||||
weechat_python_unload_all (plugin);
|
||||
|
||||
/* free Python interpreter */
|
||||
/*Py_Finalize ();
|
||||
if (Py_IsInitialized () != 0)
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python error: unable to free interpreter");
|
||||
|
||||
python_plugin->printf_server (python_plugin,
|
||||
"Python plugin ended");*/
|
||||
}
|
||||
Reference in New Issue
Block a user