moparisthebest/hexchat
1
0
Fork 0
mirror of https://github.com/moparisthebest/hexchat synced 2024-11-15 22:05:01 -05:00
Code Issues Releases Wiki Activity
605c3dea36
hexchat/plugins/python/python.c

2258 lines
57 KiB
C
Raw Normal View History

add xchat r1489
2011-02-23 22:14:30 -05:00
/*
* Copyright (c) 2002-2003 Gustavo Niemeyer <niemeyer@conectiva.com>
*
* XChat Python Plugin Interface
*
* Xchat Python Plugin Interface 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.
*
* pybot 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 file; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
/* Thread support
* ==============
*
* The python interpreter has a global interpreter lock. Any thread
* executing must acquire it before working with data accessible from
* python code. Here we must also care about xchat not being
* thread-safe. We do this by using an xchat lock, which protects
* xchat instructions from being executed out of time (when this
* plugin is not "active").
*
* When xchat calls python code:
* - Change the current_plugin for the executing plugin;
* - Release xchat lock
* - Acquire the global interpreter lock
* - Make the python call
* - Release the global interpreter lock
* - Acquire xchat lock
*
* When python code calls xchat:
* - Release the global interpreter lock
* - Acquire xchat lock
* - Restore context, if necessary
* - Make the xchat call
* - Release xchat lock
* - Acquire the global interpreter lock
*
* Inside a timer, so that individual threads have a chance to run:
* - Release xchat lock
* - Go ahead threads. Have a nice time!
* - Acquire xchat lock
*
*/
#include <glib.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <dirent.h>
#include "xchat-plugin.h"
#include "Python.h"
#include "structmember.h"
#include "pythread.h"
#define VERSION_MAJOR 0
#define VERSION_MINOR 1
#ifdef WIN32
#undef WITH_THREAD /* Thread support locks up xchat on Win32. */
#define VERSION "0.8/2.4" /* Linked to python24.dll */
#else
#define VERSION "0.8"
#endif
#define NONE 0
#define ALLOW_THREADS 1
#define RESTORE_CONTEXT 2
#ifdef WITH_THREAD
#define ACQUIRE_XCHAT_LOCK() PyThread_acquire_lock(xchat_lock, 1)
#define RELEASE_XCHAT_LOCK() PyThread_release_lock(xchat_lock)
#define BEGIN_XCHAT_CALLS(x) \
do { \
PyObject *calls_plugin = NULL; \
PyThreadState *calls_thread; \
if ((x) & RESTORE_CONTEXT) \
calls_plugin = Plugin_GetCurrent(); \
calls_thread = PyEval_SaveThread(); \
ACQUIRE_XCHAT_LOCK(); \
if (!((x) & ALLOW_THREADS)) { \
PyEval_RestoreThread(calls_thread); \
calls_thread = NULL; \
} \
if (calls_plugin) \
xchat_set_context(ph, \
Plugin_GetContext(calls_plugin)); \
while (0)
#define END_XCHAT_CALLS() \
RELEASE_XCHAT_LOCK(); \
if (calls_thread) \
PyEval_RestoreThread(calls_thread); \
} while(0)
#else
#define ACQUIRE_XCHAT_LOCK()
#define RELEASE_XCHAT_LOCK()
#define BEGIN_XCHAT_CALLS(x)
#define END_XCHAT_CALLS()
#endif
#ifdef WITH_THREAD
#define BEGIN_PLUGIN(plg) \
do { \
xchat_context *begin_plugin_ctx = xchat_get_context(ph); \
RELEASE_XCHAT_LOCK(); \
Plugin_AcquireThread(plg); \
Plugin_SetContext(plg, begin_plugin_ctx); \
} while (0)
#define END_PLUGIN(plg) \
do { \
Plugin_ReleaseThread(plg); \
ACQUIRE_XCHAT_LOCK(); \
} while (0)
#else /* !WITH_THREAD (win32) */
static PyThreadState *pTempThread;
#define BEGIN_PLUGIN(plg) \
do { \
xchat_context *begin_plugin_ctx = xchat_get_context(ph); \
RELEASE_XCHAT_LOCK(); \
PyEval_AcquireLock(); \
pTempThread = PyThreadState_Swap(((PluginObject *)(plg))->tstate); \
Plugin_SetContext(plg, begin_plugin_ctx); \
} while (0)
#define END_PLUGIN(plg) \
do { \
((PluginObject *)(plg))->tstate = PyThreadState_Swap(pTempThread); \
PyEval_ReleaseLock(); \
ACQUIRE_XCHAT_LOCK(); \
} while (0)
#endif /* !WITH_THREAD */
#define Plugin_Swap(x) \
PyThreadState_Swap(((PluginObject *)(x))->tstate)
#define Plugin_AcquireThread(x) \
PyEval_AcquireThread(((PluginObject *)(x))->tstate)
#define Plugin_ReleaseThread(x) \
Util_ReleaseThread(((PluginObject *)(x))->tstate)
#define Plugin_GetFilename(x) \
(((PluginObject *)(x))->filename)
#define Plugin_GetName(x) \
(((PluginObject *)(x))->name)
#define Plugin_GetVersion(x) \
(((PluginObject *)(x))->version)
#define Plugin_GetDesc(x) \
(((PluginObject *)(x))->description)
#define Plugin_GetHooks(x) \
(((PluginObject *)(x))->hooks)
#define Plugin_GetContext(x) \
(((PluginObject *)(x))->context)
#define Plugin_SetFilename(x, y) \
((PluginObject *)(x))->filename = (y);
#define Plugin_SetName(x, y) \
((PluginObject *)(x))->name = (y);
#define Plugin_SetVersion(x, y) \
((PluginObject *)(x))->version = (y);
#define Plugin_SetDescription(x, y) \
((PluginObject *)(x))->description = (y);
#define Plugin_SetHooks(x, y) \
((PluginObject *)(x))->hooks = (y);
#define Plugin_SetContext(x, y) \
((PluginObject *)(x))->context = (y);
#define HOOK_XCHAT 1
#define HOOK_UNLOAD 2
/* ===================================================================== */
/* Object definitions */
typedef struct {
PyObject_HEAD
int softspace; /* We need it for print support. */
} XChatOutObject;
typedef struct {
PyObject_HEAD
xchat_context *context;
} ContextObject;
typedef struct {
PyObject_HEAD
const char *listname;
PyObject *dict;
} ListItemObject;
typedef struct {
PyObject_HEAD
char *name;
char *version;
char *filename;
char *description;
GSList *hooks;
PyThreadState *tstate;
xchat_context *context;
void *gui;
} PluginObject;
typedef struct {
int type;
PyObject *plugin;
PyObject *callback;
PyObject *userdata;
void *data; /* A handle, when type == HOOK_XCHAT */
} Hook;
/* ===================================================================== */
/* Function declarations */
static PyObject *Util_BuildList(char *word[]);
static void Util_Autoload();
static char *Util_Expand(char *filename);
static int Callback_Command(char *word[], char *word_eol[], void *userdata);
static int Callback_Print(char *word[], void *userdata);
static int Callback_Timer(void *userdata);
static int Callback_ThreadTimer(void *userdata);
static PyObject *XChatOut_New();
static PyObject *XChatOut_write(PyObject *self, PyObject *args);
static void XChatOut_dealloc(PyObject *self);
static void Context_dealloc(PyObject *self);
static PyObject *Context_set(ContextObject *self, PyObject *args);
static PyObject *Context_command(ContextObject *self, PyObject *args);
static PyObject *Context_prnt(ContextObject *self, PyObject *args);
static PyObject *Context_get_info(ContextObject *self, PyObject *args);
static PyObject *Context_get_list(ContextObject *self, PyObject *args);
static PyObject *Context_compare(ContextObject *a, ContextObject *b, int op);
static PyObject *Context_FromContext(xchat_context *context);
static PyObject *Context_FromServerAndChannel(char *server, char *channel);
static PyObject *Plugin_New(char *filename, PyMethodDef *xchat_methods,
PyObject *xcoobj);
static PyObject *Plugin_GetCurrent();
static PluginObject *Plugin_ByString(char *str);
static Hook *Plugin_AddHook(int type, PyObject *plugin, PyObject *callback,
PyObject *userdata, void *data);
static void Plugin_RemoveHook(PyObject *plugin, Hook *hook);
static void Plugin_RemoveAllHooks(PyObject *plugin);
static PyObject *Module_xchat_command(PyObject *self, PyObject *args);
static PyObject *Module_xchat_prnt(PyObject *self, PyObject *args);
static PyObject *Module_xchat_get_context(PyObject *self, PyObject *args);
static PyObject *Module_xchat_find_context(PyObject *self, PyObject *args,
PyObject *kwargs);
static PyObject *Module_xchat_get_info(PyObject *self, PyObject *args);
static PyObject *Module_xchat_hook_command(PyObject *self, PyObject *args,
PyObject *kwargs);
static PyObject *Module_xchat_hook_server(PyObject *self, PyObject *args,
PyObject *kwargs);
static PyObject *Module_xchat_hook_print(PyObject *self, PyObject *args,
PyObject *kwargs);
static PyObject *Module_xchat_hook_timer(PyObject *self, PyObject *args,
PyObject *kwargs);
static PyObject *Module_xchat_unhook(PyObject *self, PyObject *args);
static PyObject *Module_xchat_get_info(PyObject *self, PyObject *args);
static PyObject *Module_xchat_get_list(PyObject *self, PyObject *args);
static PyObject *Module_xchat_get_lists(PyObject *self, PyObject *args);
static PyObject *Module_xchat_nickcmp(PyObject *self, PyObject *args);
static PyObject *Module_xchat_strip(PyObject *self, PyObject *args);
static void IInterp_Exec(char *command);
static int IInterp_Cmd(char *word[], char *word_eol[], void *userdata);
static void Command_PyList();
static void Command_PyLoad(char *filename);
static void Command_PyUnload(char *name);
static void Command_PyReload(char *name);
static void Command_PyAbout();
static int Command_Py(char *word[], char *word_eol[], void *userdata);
/* ===================================================================== */
/* Static declarations and definitions */
staticforward PyTypeObject Plugin_Type;
staticforward PyTypeObject XChatOut_Type;
staticforward PyTypeObject Context_Type;
staticforward PyTypeObject ListItem_Type;
static PyThreadState *main_tstate = NULL;
static void *thread_timer = NULL;
static xchat_plugin *ph;
static GSList *plugin_list = NULL;
static PyObject *interp_plugin = NULL;
static PyObject *xchatout = NULL;
#ifdef WITH_THREAD
static PyThread_type_lock xchat_lock = NULL;
#endif
static const char usage[] = "\
Usage: /PY LOAD <filename>\n\
UNLOAD <filename|name>\n\
RELOAD <filename|name>\n\
LIST\n\
EXEC <command>\n\
CONSOLE\n\
ABOUT\n\
\n";
static const char about[] = "\
\n\
X-Chat Python Interface " VERSION "\n\
\n\
Copyright (c) 2002-2003 Gustavo Niemeyer <niemeyer@conectiva.com>\n\
\n";
/* ===================================================================== */
/* Utility functions */
static PyObject *
Util_BuildList(char *word[])
{
PyObject *list;
int listsize = 0;
int i;
while (word[listsize] && word[listsize][0])
listsize++;
list = PyList_New(listsize);
if (list == NULL) {
PyErr_Print();
return NULL;
}
for (i = 0; i != listsize; i++) {
PyObject *o = PyString_FromString(word[i]);
if (o == NULL) {
Py_DECREF(list);
PyErr_Print();
return NULL;
}
PyList_SetItem(list, i, o);
}
return list;
}
static void
Util_Autoload_from (const char *dir_name)
{
#ifndef PATH_MAX
#define PATH_MAX 1024 /* Hurd doesn't define it */
#endif
char oldcwd[PATH_MAX];
struct dirent *ent;
DIR *dir;
if (getcwd(oldcwd, PATH_MAX) == NULL)
return;
if (chdir(dir_name) != 0)
return;
dir = opendir(".");
if (dir == NULL)
return;
while ((ent = readdir(dir))) {
int len = strlen(ent->d_name);
if (len > 3 && strcmp(".py", ent->d_name+len-3) == 0)
Command_PyLoad(ent->d_name);
}
closedir(dir);
chdir(oldcwd);
}
static void
Util_Autoload()
{
/* we need local filesystem encoding for chdir, opendir etc */
/* auto-load from ~/.xchat2/ or %APPDATA%\X-Chat 2\ */
Util_Autoload_from(xchat_get_info(ph, "xchatdirfs"));
#ifdef WIN32 /* also auto-load C:\Program Files\XChat\Plugins\*.py */
Util_Autoload_from(XCHATLIBDIR"/plugins");
#endif
}
static char *
Util_Expand(char *filename)
{
char *expanded;
/* Check if this is an absolute path. */
if (g_path_is_absolute(filename)) {
if (g_file_test(filename, G_FILE_TEST_EXISTS))
return g_strdup(filename);
else
return NULL;
}
/* Check if it starts with ~/ and expand the home if positive. */
if (*filename == '~' && *(filename+1) == '/') {
expanded = g_build_filename(g_get_home_dir(),
filename+2, NULL);
if (g_file_test(expanded, G_FILE_TEST_EXISTS))
return expanded;
else {
g_free(expanded);
return NULL;
}
}
/* Check if it's in the current directory. */
expanded = g_build_filename(g_get_current_dir(),
filename, NULL);
if (g_file_test(expanded, G_FILE_TEST_EXISTS))
return expanded;
g_free(expanded);
/* Check if ~/.xchat2/<filename> exists. */
expanded = g_build_filename(xchat_get_info(ph, "xchatdir"),
filename, NULL);
if (g_file_test(expanded, G_FILE_TEST_EXISTS))
return expanded;
g_free(expanded);
return NULL;
}
/* Similar to PyEval_ReleaseThread, but accepts NULL thread states. */
static void
Util_ReleaseThread(PyThreadState *tstate)
{
PyThreadState *old_tstate;
if (tstate == NULL)
Py_FatalError("PyEval_ReleaseThread: NULL thread state");
old_tstate = PyThreadState_Swap(NULL);
if (old_tstate != tstate && old_tstate != NULL)
Py_FatalError("PyEval_ReleaseThread: wrong thread state");
PyEval_ReleaseLock();
}
/* ===================================================================== */
/* Hookable functions. These are the entry points to python code, besides
* the load function, and the hooks for interactive interpreter. */
static int
Callback_Command(char *word[], char *word_eol[], void *userdata)
{
Hook *hook = (Hook *) userdata;
PyObject *retobj;
PyObject *word_list, *word_eol_list;
int ret = 0;
BEGIN_PLUGIN(hook->plugin);
word_list = Util_BuildList(word+1);
if (word_list == NULL) {
END_PLUGIN(hook->plugin);
return 0;
}
word_eol_list = Util_BuildList(word_eol+1);
if (word_eol_list == NULL) {
Py_DECREF(word_list);
END_PLUGIN(hook->plugin);
return 0;
}
retobj = PyObject_CallFunction(hook->callback, "(OOO)", word_list,
word_eol_list, hook->userdata);
Py_DECREF(word_list);
Py_DECREF(word_eol_list);
if (retobj == Py_None) {
ret = XCHAT_EAT_NONE;
Py_DECREF(retobj);
} else if (retobj) {
ret = PyInt_AsLong(retobj);
Py_DECREF(retobj);
} else {
PyErr_Print();
}
END_PLUGIN(hook->plugin);
return ret;
}
/* No Callback_Server() here. We use Callback_Command() as well. */
static int
Callback_Print(char *word[], void *userdata)
{
Hook *hook = (Hook *) userdata;
PyObject *retobj;
PyObject *word_list;
PyObject *word_eol_list;
char **word_eol;
char *word_eol_raw;
int listsize = 0;
int next = 0;
int i;
int ret = 0;
/* Cut off the message identifier. */
word += 1;
/* XChat doesn't provide a word_eol for print events, so we
* build our own here. */
while (word[listsize] && word[listsize][0])
listsize++;
word_eol = (char **) g_malloc(sizeof(char*)*(listsize+1));
if (word_eol == NULL) {
xchat_print(ph, "Not enough memory to alloc word_eol "
"for python plugin callback.");
return 0;
}
/* First build a word clone, but NULL terminated. */
memcpy(word_eol, word, listsize*sizeof(char*));
word_eol[listsize] = NULL;
/* Then join it. */
word_eol_raw = g_strjoinv(" ", word_eol);
if (word_eol_raw == NULL) {
xchat_print(ph, "Not enough memory to alloc word_eol_raw "
"for python plugin callback.");
return 0;
}
/* And rebuild the real word_eol. */
for (i = 0; i != listsize; i++) {
word_eol[i] = word_eol_raw+next;
next += strlen(word[i])+1;
}
word_eol[i] = "";
BEGIN_PLUGIN(hook->plugin);
word_list = Util_BuildList(word);
if (word_list == NULL) {
g_free(word_eol_raw);
g_free(word_eol);
END_PLUGIN(hook->plugin);
return 0;
}
word_eol_list = Util_BuildList(word_eol);
if (word_eol_list == NULL) {
g_free(word_eol_raw);
g_free(word_eol);
Py_DECREF(word_list);
END_PLUGIN(hook->plugin);
return 0;
}
retobj = PyObject_CallFunction(hook->callback, "(OOO)", word_list,
word_eol_list, hook->userdata);
Py_DECREF(word_list);
Py_DECREF(word_eol_list);
g_free(word_eol_raw);
g_free(word_eol);
if (retobj == Py_None) {
ret = XCHAT_EAT_NONE;
Py_DECREF(retobj);
} else if (retobj) {
ret = PyInt_AsLong(retobj);
Py_DECREF(retobj);
} else {
PyErr_Print();
}
END_PLUGIN(hook->plugin);
return ret;
}
static int
Callback_Timer(void *userdata)
{
Hook *hook = (Hook *) userdata;
PyObject *retobj;
int ret = 0;
PyObject *plugin;
plugin = hook->plugin;
BEGIN_PLUGIN(hook->plugin);
retobj = PyObject_CallFunction(hook->callback, "(O)", hook->userdata);
if (retobj) {
ret = PyObject_IsTrue(retobj);
Py_DECREF(retobj);
} else {
PyErr_Print();
}
/* Returning 0 for this callback unhooks itself. */
if (ret == 0)
Plugin_RemoveHook(plugin, hook);
END_PLUGIN(plugin);
return ret;
}
#ifdef WITH_THREAD
static int
Callback_ThreadTimer(void *userdata)
{
RELEASE_XCHAT_LOCK();
usleep(1);
ACQUIRE_XCHAT_LOCK();
return 1;
}
#endif
/* ===================================================================== */
/* XChatOut object */
/* We keep this information global, so we can reset it when the
* deinit function is called. */
/* XXX This should be somehow bound to the printing context. */
static char *xchatout_buffer = NULL;
static int xchatout_buffer_size = 0;
static int xchatout_buffer_pos = 0;
static PyObject *
XChatOut_New()
{
XChatOutObject *xcoobj;
xcoobj = PyObject_New(XChatOutObject, &XChatOut_Type);
if (xcoobj != NULL)
xcoobj->softspace = 0;
return (PyObject *) xcoobj;
}
static void
XChatOut_dealloc(PyObject *self)
{
self->ob_type->tp_free((PyObject *)self);
}
/* This is a little bit complex because we have to buffer data
* until a \n is received, since xchat breaks the line automatically.
* We also crop the last \n for this reason. */
static PyObject *
XChatOut_write(PyObject *self, PyObject *args)
{
int new_buffer_pos, data_size, print_limit, add_space;
char *data, *pos;
if (!PyArg_ParseTuple(args, "s#:write", &data, &data_size))
return NULL;
if (!data_size) {
Py_INCREF(Py_None);
return Py_None;
}
BEGIN_XCHAT_CALLS(RESTORE_CONTEXT|ALLOW_THREADS);
if (((XChatOutObject *)self)->softspace) {
add_space = 1;
((XChatOutObject *)self)->softspace = 0;
} else {
add_space = 0;
}
if (xchatout_buffer_size-xchatout_buffer_pos < data_size+add_space) {
char *new_buffer;
/* This buffer grows whenever needed, and does not
* shrink. If we ever implement unloading of the
* python interface, we must find some way to free
* this buffer as well. */
xchatout_buffer_size += data_size*2+16;
new_buffer = g_realloc(xchatout_buffer, xchatout_buffer_size);
if (new_buffer == NULL) {
xchat_print(ph, "Not enough memory to print");
/* The system is out of resources. Let's help. */
g_free(xchatout_buffer);
xchatout_buffer = NULL;
xchatout_buffer_size = 0;
xchatout_buffer_pos = 0;
/* Return something valid, since we have
* already warned the user, and he probably
* won't be able to notice this exception. */
goto exit;
}
xchatout_buffer = new_buffer;
}
memcpy(xchatout_buffer+xchatout_buffer_pos, data, data_size);
print_limit = new_buffer_pos = xchatout_buffer_pos+data_size;
pos = xchatout_buffer+print_limit;
if (add_space && *(pos-1) != '\n') {
*pos = ' ';
*(pos+1) = 0;
new_buffer_pos++;
}
while (*pos != '\n' && print_limit > xchatout_buffer_pos) {
pos--;
print_limit--;
}
if (*pos == '\n') {
/* Crop it, inserting the string limiter there. */
*pos = 0;
xchat_print(ph, xchatout_buffer);
if (print_limit < new_buffer_pos) {
/* There's still data to be printed. */
print_limit += 1; /* Include the limiter. */
xchatout_buffer_pos = new_buffer_pos-print_limit;
memmove(xchatout_buffer, xchatout_buffer+print_limit,
xchatout_buffer_pos);
} else {
xchatout_buffer_pos = 0;
}
} else {
xchatout_buffer_pos = new_buffer_pos;
}
exit:
END_XCHAT_CALLS();
Py_INCREF(Py_None);
return Py_None;
}
#define OFF(x) offsetof(XChatOutObject, x)
static PyMemberDef XChatOut_members[] = {
{"softspace", T_INT, OFF(softspace), 0},
{0}
};
static PyMethodDef XChatOut_methods[] = {
{"write", XChatOut_write, METH_VARARGS},
{NULL, NULL}
};
statichere PyTypeObject XChatOut_Type = {
PyObject_HEAD_INIT(NULL)
0, /*ob_size*/
"xchat.XChatOut", /*tp_name*/
sizeof(XChatOutObject), /*tp_basicsize*/
0, /*tp_itemsize*/
XChatOut_dealloc, /*tp_dealloc*/
0, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
PyObject_GenericGetAttr,/*tp_getattro*/
PyObject_GenericSetAttr,/*tp_setattro*/
0, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT, /*tp_flags*/
0, /*tp_doc*/
0, /*tp_traverse*/
0, /*tp_clear*/
0, /*tp_richcompare*/
0, /*tp_weaklistoffset*/
0, /*tp_iter*/
0, /*tp_iternext*/
XChatOut_methods, /*tp_methods*/
XChatOut_members, /*tp_members*/
0, /*tp_getset*/
0, /*tp_base*/
0, /*tp_dict*/
0, /*tp_descr_get*/
0, /*tp_descr_set*/
0, /*tp_dictoffset*/
0, /*tp_init*/
PyType_GenericAlloc, /*tp_alloc*/
PyType_GenericNew, /*tp_new*/
_PyObject_Del, /*tp_free*/
0, /*tp_is_gc*/
};
/* ===================================================================== */
/* Context object */
static void
Context_dealloc(PyObject *self)
{
self->ob_type->tp_free((PyObject *)self);
}
static PyObject *
Context_set(ContextObject *self, PyObject *args)
{
PyObject *plugin = Plugin_GetCurrent();
Plugin_SetContext(plugin, self->context);
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
Context_command(ContextObject *self, PyObject *args)
{
char *text;
if (!PyArg_ParseTuple(args, "s:command", &text))
return NULL;
BEGIN_XCHAT_CALLS(ALLOW_THREADS);
xchat_set_context(ph, self->context);
xchat_command(ph, text);
END_XCHAT_CALLS();
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
Context_prnt(ContextObject *self, PyObject *args)
{
char *text;
if (!PyArg_ParseTuple(args, "s:prnt", &text))
return NULL;
BEGIN_XCHAT_CALLS(ALLOW_THREADS);
xchat_set_context(ph, self->context);
xchat_print(ph, text);
END_XCHAT_CALLS();
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
Context_emit_print(ContextObject *self, PyObject *args)
{
char *argv[10];
char *name;
int res;
memset(&argv, 0, sizeof(char*)*10);
if (!PyArg_ParseTuple(args, "s|ssssss:print_event", &name,
&argv[0], &argv[1], &argv[2],
&argv[3], &argv[4], &argv[5],
&argv[6], &argv[7], &argv[8]))
return NULL;
BEGIN_XCHAT_CALLS(ALLOW_THREADS);
xchat_set_context(ph, self->context);
res = xchat_emit_print(ph, name, argv[0], argv[1], argv[2],
argv[3], argv[4], argv[5],
argv[6], argv[7], argv[8]);
END_XCHAT_CALLS();
return PyInt_FromLong(res);
}
static PyObject *
Context_get_info(ContextObject *self, PyObject *args)
{
const char *info;
char *name;
if (!PyArg_ParseTuple(args, "s:get_info", &name))
return NULL;
BEGIN_XCHAT_CALLS(NONE);
xchat_set_context(ph, self->context);
info = xchat_get_info(ph, name);
END_XCHAT_CALLS();
if (info == NULL) {
Py_INCREF(Py_None);
return Py_None;
}
return PyString_FromString(info);
}
static PyObject *
Context_get_list(ContextObject *self, PyObject *args)
{
PyObject *plugin = Plugin_GetCurrent();
xchat_context *saved_context = Plugin_GetContext(plugin);
PyObject *ret;
Plugin_SetContext(plugin, self->context);
ret = Module_xchat_get_list((PyObject*)self, args);
Plugin_SetContext(plugin, saved_context);
return ret;
}
/* needed to make context1 == context2 work */
static PyObject *
Context_compare(ContextObject *a, ContextObject *b, int op)
{
PyObject *ret;
/* check for == */
if (op == Py_EQ)
ret = (a->context == b->context ? Py_True : Py_False);
/* check for != */
else if (op == Py_NE)
ret = (a->context != b->context ? Py_True : Py_False);
/* only makes sense as == and != */
else
{
PyErr_SetString(PyExc_TypeError, "contexts are either equal or not equal");
ret = NULL;
}
return ret;
}
static PyMethodDef Context_methods[] = {
{"set", (PyCFunction) Context_set, METH_NOARGS},
{"command", (PyCFunction) Context_command, METH_VARARGS},
{"prnt", (PyCFunction) Context_prnt, METH_VARARGS},
{"emit_print", (PyCFunction) Context_emit_print, METH_VARARGS},
{"get_info", (PyCFunction) Context_get_info, METH_VARARGS},
{"get_list", (PyCFunction) Context_get_list, METH_VARARGS},
{NULL, NULL}
};
statichere PyTypeObject Context_Type = {
PyObject_HEAD_INIT(NULL)
0, /*ob_size*/
"xchat.Context", /*tp_name*/
sizeof(ContextObject), /*tp_basicsize*/
0, /*tp_itemsize*/
Context_dealloc, /*tp_dealloc*/
0, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
PyObject_GenericGetAttr,/*tp_getattro*/
PyObject_GenericSetAttr,/*tp_setattro*/
0, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT, /*tp_flags*/
0, /*tp_doc*/
0, /*tp_traverse*/
0, /*tp_clear*/
(richcmpfunc)Context_compare, /*tp_richcompare*/
0, /*tp_weaklistoffset*/
0, /*tp_iter*/
0, /*tp_iternext*/
Context_methods, /*tp_methods*/
0, /*tp_members*/
0, /*tp_getset*/
0, /*tp_base*/
0, /*tp_dict*/
0, /*tp_descr_get*/
0, /*tp_descr_set*/
0, /*tp_dictoffset*/
0, /*tp_init*/
PyType_GenericAlloc, /*tp_alloc*/
PyType_GenericNew, /*tp_new*/
_PyObject_Del, /*tp_free*/
0, /*tp_is_gc*/
};
static PyObject *
Context_FromContext(xchat_context *context)
{
ContextObject *ctxobj = PyObject_New(ContextObject, &Context_Type);
if (ctxobj != NULL)
ctxobj->context = context;
return (PyObject *) ctxobj;
}
static PyObject *
Context_FromServerAndChannel(char *server, char *channel)
{
ContextObject *ctxobj;
xchat_context *context;
BEGIN_XCHAT_CALLS(NONE);
context = xchat_find_context(ph, server, channel);
END_XCHAT_CALLS();
if (context == NULL)
return NULL;
ctxobj = PyObject_New(ContextObject, &Context_Type);
if (ctxobj == NULL)
return NULL;
ctxobj->context = context;
return (PyObject *) ctxobj;
}
/* ===================================================================== */
/* ListItem object */
#undef OFF
#define OFF(x) offsetof(ListItemObject, x)
static PyMemberDef ListItem_members[] = {
{"__dict__", T_OBJECT, OFF(dict), 0},
{0}
};
static void
ListItem_dealloc(PyObject *self)
{
Py_DECREF(((ListItemObject*)self)->dict);
self->ob_type->tp_free((PyObject *)self);
}
static PyObject *
ListItem_repr(PyObject *self)
{
return PyString_FromFormat("<%s list item at %p>",
((ListItemObject*)self)->listname, self);
}
statichere PyTypeObject ListItem_Type = {
PyObject_HEAD_INIT(NULL)
0, /*ob_size*/
"xchat.ListItem", /*tp_name*/
sizeof(ListItemObject), /*tp_basicsize*/
0, /*tp_itemsize*/
ListItem_dealloc, /*tp_dealloc*/
0, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
ListItem_repr, /*tp_repr*/
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
PyObject_GenericGetAttr,/*tp_getattro*/
PyObject_GenericSetAttr,/*tp_setattro*/
0, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT, /*tp_flags*/
0, /*tp_doc*/
0, /*tp_traverse*/
0, /*tp_clear*/
0, /*tp_richcompare*/
0, /*tp_weaklistoffset*/
0, /*tp_iter*/
0, /*tp_iternext*/
0, /*tp_methods*/
ListItem_members, /*tp_members*/
0, /*tp_getset*/
0, /*tp_base*/
0, /*tp_dict*/
0, /*tp_descr_get*/
0, /*tp_descr_set*/
OFF(dict), /*tp_dictoffset*/
0, /*tp_init*/
PyType_GenericAlloc, /*tp_alloc*/
PyType_GenericNew, /*tp_new*/
_PyObject_Del, /*tp_free*/
0, /*tp_is_gc*/
};
static PyObject *
ListItem_New(const char *listname)
{
ListItemObject *item;
item = PyObject_New(ListItemObject, &ListItem_Type);
if (item != NULL) {
/* listname parameter must be statically allocated. */
item->listname = listname;
item->dict = PyDict_New();
if (item->dict == NULL) {
Py_DECREF(item);
item = NULL;
}
}
return (PyObject *) item;
}
/* ===================================================================== */
/* Plugin object */
#define GET_MODULE_DATA(x, force) \
o = PyObject_GetAttrString(m, "__module_" #x "__"); \
if (o == NULL) { \
if (force) { \
xchat_print(ph, "Module has no __module_" #x "__ " \
"defined"); \
goto error; \
} \
plugin->x = g_strdup(""); \
} else {\
if (!PyString_Check(o)) { \
xchat_print(ph, "Variable __module_" #x "__ " \
"must be a string"); \
goto error; \
} \
plugin->x = g_strdup(PyString_AsString(o)); \
if (plugin->x == NULL) { \
xchat_print(ph, "Not enough memory to allocate " #x); \
goto error; \
} \
}
static PyObject *
Plugin_New(char *filename, PyMethodDef *xchat_methods, PyObject *xcoobj)
{
PluginObject *plugin = NULL;
PyObject *m, *o;
char *argv[] = {"<xchat>", 0};
if (filename) {
char *old_filename = filename;
filename = Util_Expand(filename);
if (filename == NULL) {
xchat_printf(ph, "File not found: %s", old_filename);
return NULL;
}
}
/* Allocate plugin structure. */
plugin = PyObject_New(PluginObject, &Plugin_Type);
if (plugin == NULL) {
xchat_print(ph, "Can't create plugin object");
goto error;
}
Plugin_SetName(plugin, NULL);
Plugin_SetVersion(plugin, NULL);
Plugin_SetFilename(plugin, NULL);
Plugin_SetDescription(plugin, NULL);
Plugin_SetHooks(plugin, NULL);
Plugin_SetContext(plugin, xchat_get_context(ph));
/* Start a new interpreter environment for this plugin. */
PyEval_AcquireLock();
plugin->tstate = Py_NewInterpreter();
if (plugin->tstate == NULL) {
xchat_print(ph, "Can't create interpreter state");
goto error;
}
PySys_SetArgv(1, argv);
PySys_SetObject("__plugin__", (PyObject *) plugin);
/* Set stdout and stderr to xchatout. */
Py_INCREF(xcoobj);
PySys_SetObject("stdout", xcoobj);
Py_INCREF(xcoobj);
PySys_SetObject("stderr", xcoobj);
/* Add xchat module to the environment. */
m = Py_InitModule("xchat", xchat_methods);
if (m == NULL) {
xchat_print(ph, "Can't create xchat module");
goto error;
}
PyModule_AddIntConstant(m, "EAT_NONE", XCHAT_EAT_NONE);
PyModule_AddIntConstant(m, "EAT_XCHAT", XCHAT_EAT_XCHAT);
PyModule_AddIntConstant(m, "EAT_PLUGIN", XCHAT_EAT_PLUGIN);
PyModule_AddIntConstant(m, "EAT_ALL", XCHAT_EAT_ALL);
PyModule_AddIntConstant(m, "PRI_HIGHEST", XCHAT_PRI_HIGHEST);
PyModule_AddIntConstant(m, "PRI_HIGH", XCHAT_PRI_HIGH);
PyModule_AddIntConstant(m, "PRI_NORM", XCHAT_PRI_NORM);
PyModule_AddIntConstant(m, "PRI_LOW", XCHAT_PRI_LOW);
PyModule_AddIntConstant(m, "PRI_LOWEST", XCHAT_PRI_LOWEST);
o = Py_BuildValue("(ii)", VERSION_MAJOR, VERSION_MINOR);
if (o == NULL) {
xchat_print(ph, "Can't create version tuple");
goto error;
}
PyObject_SetAttrString(m, "__version__", o);
if (filename) {
#ifdef WIN32
PyObject* PyFileObject = PyFile_FromString(filename, "r");
if (PyFileObject == NULL) {
xchat_printf(ph, "Can't open file %s: %s\n",
filename, strerror(errno));
goto error;
}
if (PyRun_SimpleFile(PyFile_AsFile(PyFileObject), filename) != 0) {
xchat_printf(ph, "Error loading module %s\n",
filename);
goto error;
}
plugin->filename = filename;
filename = NULL;
#else
FILE *fp;
plugin->filename = filename;
/* It's now owned by the plugin. */
filename = NULL;
/* Open the plugin file. */
fp = fopen(plugin->filename, "r");
if (fp == NULL) {
xchat_printf(ph, "Can't open file %s: %s\n",
plugin->filename, strerror(errno));
goto error;
}
/* Run the plugin. */
if (PyRun_SimpleFile(fp, plugin->filename) != 0) {
xchat_printf(ph, "Error loading module %s\n",
plugin->filename);
fclose(fp);
goto error;
}
fclose(fp);
#endif
m = PyDict_GetItemString(PyImport_GetModuleDict(),
"__main__");
if (m == NULL) {
xchat_print(ph, "Can't get __main__ module");
goto error;
}
GET_MODULE_DATA(name, 1);
GET_MODULE_DATA(version, 0);
GET_MODULE_DATA(description, 0);
plugin->gui = xchat_plugingui_add(ph, plugin->filename,
plugin->name,
plugin->description,
plugin->version, NULL);
}
PyEval_ReleaseThread(plugin->tstate);
return (PyObject *) plugin;
error:
g_free(filename);
if (plugin) {
if (plugin->tstate)
Py_EndInterpreter(plugin->tstate);
Py_DECREF(plugin);
}
PyEval_ReleaseLock();
return NULL;
}
static PyObject *
Plugin_GetCurrent()
{
PyObject *plugin;
plugin = PySys_GetObject("__plugin__");
if (plugin == NULL)
PyErr_SetString(PyExc_RuntimeError, "lost sys.__plugin__");
return plugin;
}
static PluginObject *
Plugin_ByString(char *str)
{
GSList *list;
PluginObject *plugin;
char *basename;
list = plugin_list;
while (list != NULL) {
plugin = (PluginObject *) list->data;
basename = g_path_get_basename(plugin->filename);
if (basename == NULL)
break;
if (strcasecmp(plugin->name, str) == 0 ||
strcasecmp(plugin->filename, str) == 0 ||
strcasecmp(basename, str) == 0) {
g_free(basename);
return plugin;
}
g_free(basename);
list = list->next;
}
return NULL;
}
static Hook *
Plugin_AddHook(int type, PyObject *plugin, PyObject *callback,
PyObject *userdata, void *data)
{
Hook *hook = (Hook *) g_malloc(sizeof(Hook));
if (hook == NULL) {
PyErr_NoMemory();
return NULL;
}
hook->type = type;
hook->plugin = plugin;
Py_INCREF(callback);
hook->callback = callback;
Py_INCREF(userdata);
hook->userdata = userdata;
hook->data = NULL;
Plugin_SetHooks(plugin, g_slist_append(Plugin_GetHooks(plugin),
hook));
return hook;
}
static void
Plugin_RemoveHook(PyObject *plugin, Hook *hook)
{
GSList *list;
/* Is this really a hook of the running plugin? */
list = g_slist_find(Plugin_GetHooks(plugin), hook);
if (list) {
/* Ok, unhook it. */
if (hook->type == HOOK_XCHAT) {
/* This is an xchat hook. Unregister it. */
BEGIN_XCHAT_CALLS(NONE);
xchat_unhook(ph, (xchat_hook*)hook->data);
END_XCHAT_CALLS();
}
Plugin_SetHooks(plugin,
g_slist_remove(Plugin_GetHooks(plugin),
hook));
Py_DECREF(hook->callback);
Py_DECREF(hook->userdata);
g_free(hook);
}
}
static void
Plugin_RemoveAllHooks(PyObject *plugin)
{
GSList *list = Plugin_GetHooks(plugin);
while (list) {
Hook *hook = (Hook *) list->data;
if (hook->type == HOOK_XCHAT) {
/* This is an xchat hook. Unregister it. */
BEGIN_XCHAT_CALLS(NONE);
xchat_unhook(ph, (xchat_hook*)hook->data);
END_XCHAT_CALLS();
}
Py_DECREF(hook->callback);
Py_DECREF(hook->userdata);
g_free(hook);
list = list->next;
}
Plugin_SetHooks(plugin, NULL);
}
static void
Plugin_Delete(PyObject *plugin)
{
PyThreadState *tstate = ((PluginObject*)plugin)->tstate;
GSList *list = Plugin_GetHooks(plugin);
while (list) {
Hook *hook = (Hook *) list->data;
if (hook->type == HOOK_UNLOAD) {
PyObject *retobj;
retobj = PyObject_CallFunction(hook->callback, "(O)",
hook->userdata);
if (retobj) {
Py_DECREF(retobj);
} else {
PyErr_Print();
PyErr_Clear();
}
}
list = list->next;
}
Plugin_RemoveAllHooks(plugin);
xchat_plugingui_remove(ph, ((PluginObject *)plugin)->gui);
Py_DECREF(plugin);
/*PyThreadState_Swap(tstate); needed? */
Py_EndInterpreter(tstate);
}
static void
Plugin_dealloc(PluginObject *self)
{
g_free(self->filename);
g_free(self->name);
g_free(self->version);
g_free(self->description);
self->ob_type->tp_free((PyObject *)self);
}
statichere PyTypeObject Plugin_Type = {
PyObject_HEAD_INIT(NULL)
0, /*ob_size*/
"xchat.Plugin", /*tp_name*/
sizeof(PluginObject), /*tp_basicsize*/
0, /*tp_itemsize*/
(destructor)Plugin_dealloc, /*tp_dealloc*/
0, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
PyObject_GenericGetAttr,/*tp_getattro*/
PyObject_GenericSetAttr,/*tp_setattro*/
0, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT, /*tp_flags*/
0, /*tp_doc*/
0, /*tp_traverse*/
0, /*tp_clear*/
0, /*tp_richcompare*/
0, /*tp_weaklistoffset*/
0, /*tp_iter*/
0, /*tp_iternext*/
0, /*tp_methods*/
0, /*tp_members*/
0, /*tp_getset*/
0, /*tp_base*/
0, /*tp_dict*/
0, /*tp_descr_get*/
0, /*tp_descr_set*/
0, /*tp_dictoffset*/
0, /*tp_init*/
PyType_GenericAlloc, /*tp_alloc*/
PyType_GenericNew, /*tp_new*/
_PyObject_Del, /*tp_free*/
0, /*tp_is_gc*/
};
/* ===================================================================== */
/* XChat module */
static PyObject *
Module_xchat_command(PyObject *self, PyObject *args)
{
char *text;
if (!PyArg_ParseTuple(args, "s:command", &text))
return NULL;
BEGIN_XCHAT_CALLS(RESTORE_CONTEXT|ALLOW_THREADS);
xchat_command(ph, text);
END_XCHAT_CALLS();
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
Module_xchat_prnt(PyObject *self, PyObject *args)
{
char *text;
if (!PyArg_ParseTuple(args, "s:prnt", &text))
return NULL;
BEGIN_XCHAT_CALLS(RESTORE_CONTEXT|ALLOW_THREADS);
xchat_print(ph, text);
END_XCHAT_CALLS();
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
Module_xchat_emit_print(PyObject *self, PyObject *args)
{
char *argv[10];
char *name;
int res;
memset(&argv, 0, sizeof(char*)*10);
if (!PyArg_ParseTuple(args, "s|ssssss:print_event", &name,
&argv[0], &argv[1], &argv[2],
&argv[3], &argv[4], &argv[5],
&argv[6], &argv[7], &argv[8]))
return NULL;
BEGIN_XCHAT_CALLS(RESTORE_CONTEXT|ALLOW_THREADS);
res = xchat_emit_print(ph, name, argv[0], argv[1], argv[2],
argv[3], argv[4], argv[5],
argv[6], argv[7], argv[8]);
END_XCHAT_CALLS();
return PyInt_FromLong(res);
}
static PyObject *
Module_xchat_get_info(PyObject *self, PyObject *args)
{
const char *info;
char *name;
if (!PyArg_ParseTuple(args, "s:get_info", &name))
return NULL;
BEGIN_XCHAT_CALLS(RESTORE_CONTEXT);
info = xchat_get_info(ph, name);
END_XCHAT_CALLS();
if (info == NULL) {
Py_INCREF(Py_None);
return Py_None;
}
return PyString_FromString(info);
}
static PyObject *
Module_xchat_get_prefs(PyObject *self, PyObject *args)
{
PyObject *res;
const char *info;
int integer;
char *name;
int type;
if (!PyArg_ParseTuple(args, "s:get_prefs", &name))
return NULL;
BEGIN_XCHAT_CALLS(NONE);
type = xchat_get_prefs(ph, name, &info, &integer);
END_XCHAT_CALLS();
switch (type) {
case 0:
Py_INCREF(Py_None);
res = Py_None;
break;
case 1:
res = PyString_FromString((char*)info);
break;
case 2:
case 3:
res = PyInt_FromLong(integer);
break;
default:
PyErr_Format(PyExc_RuntimeError,
"unknown get_prefs type (%d), "
"please report", type);
res = NULL;
break;
}
return res;
}
static PyObject *
Module_xchat_get_context(PyObject *self, PyObject *args)
{
PyObject *plugin;
PyObject *ctxobj;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
ctxobj = Context_FromContext(Plugin_GetContext(plugin));
if (ctxobj == NULL) {
Py_INCREF(Py_None);
return Py_None;
}
return ctxobj;
}
static PyObject *
Module_xchat_find_context(PyObject *self, PyObject *args, PyObject *kwargs)
{
char *server = NULL;
char *channel = NULL;
PyObject *ctxobj;
char *kwlist[] = {"server", "channel", 0};
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|zz:find_context",
kwlist, &server, &channel))
return NULL;
ctxobj = Context_FromServerAndChannel(server, channel);
if (ctxobj == NULL) {
Py_INCREF(Py_None);
return Py_None;
}
return ctxobj;
}
static PyObject *
Module_xchat_hook_command(PyObject *self, PyObject *args, PyObject *kwargs)
{
char *name;
PyObject *callback;
PyObject *userdata = Py_None;
int priority = XCHAT_PRI_NORM;
char *help = NULL;
PyObject *plugin;
Hook *hook;
char *kwlist[] = {"name", "callback", "userdata",
"priority", "help", 0};
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "sO|Oiz:hook_command",
kwlist, &name, &callback, &userdata,
&priority, &help))
return NULL;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
if (!PyCallable_Check(callback)) {
PyErr_SetString(PyExc_TypeError, "callback is not callable");
return NULL;
}
hook = Plugin_AddHook(HOOK_XCHAT, plugin, callback, userdata, NULL);
if (hook == NULL)
return NULL;
BEGIN_XCHAT_CALLS(NONE);
hook->data = (void*)xchat_hook_command(ph, name, priority,
Callback_Command, help, hook);
END_XCHAT_CALLS();
return PyInt_FromLong((long)hook);
}
static PyObject *
Module_xchat_hook_server(PyObject *self, PyObject *args, PyObject *kwargs)
{
char *name;
PyObject *callback;
PyObject *userdata = Py_None;
int priority = XCHAT_PRI_NORM;
PyObject *plugin;
Hook *hook;
char *kwlist[] = {"name", "callback", "userdata", "priority", 0};
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "sO|Oi:hook_server",
kwlist, &name, &callback, &userdata,
&priority))
return NULL;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
if (!PyCallable_Check(callback)) {
PyErr_SetString(PyExc_TypeError, "callback is not callable");
return NULL;
}
hook = Plugin_AddHook(HOOK_XCHAT, plugin, callback, userdata, NULL);
if (hook == NULL)
return NULL;
BEGIN_XCHAT_CALLS(NONE);
hook->data = (void*)xchat_hook_server(ph, name, priority,
Callback_Command, hook);
END_XCHAT_CALLS();
return PyInt_FromLong((long)hook);
}
static PyObject *
Module_xchat_hook_print(PyObject *self, PyObject *args, PyObject *kwargs)
{
char *name;
PyObject *callback;
PyObject *userdata = Py_None;
int priority = XCHAT_PRI_NORM;
PyObject *plugin;
Hook *hook;
char *kwlist[] = {"name", "callback", "userdata", "priority", 0};
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "sO|Oi:hook_print",
kwlist, &name, &callback, &userdata,
&priority))
return NULL;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
if (!PyCallable_Check(callback)) {
PyErr_SetString(PyExc_TypeError, "callback is not callable");
return NULL;
}
hook = Plugin_AddHook(HOOK_XCHAT, plugin, callback, userdata, NULL);
if (hook == NULL)
return NULL;
BEGIN_XCHAT_CALLS(NONE);
hook->data = (void*)xchat_hook_print(ph, name, priority,
Callback_Print, hook);
END_XCHAT_CALLS();
return PyInt_FromLong((long)hook);
}
static PyObject *
Module_xchat_hook_timer(PyObject *self, PyObject *args, PyObject *kwargs)
{
int timeout;
PyObject *callback;
PyObject *userdata = Py_None;
PyObject *plugin;
Hook *hook;
char *kwlist[] = {"timeout", "callback", "userdata", 0};
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "iO|O:hook_timer",
kwlist, &timeout, &callback,
&userdata))
return NULL;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
if (!PyCallable_Check(callback)) {
PyErr_SetString(PyExc_TypeError, "callback is not callable");
return NULL;
}
hook = Plugin_AddHook(HOOK_XCHAT, plugin, callback, userdata, NULL);
if (hook == NULL)
return NULL;
BEGIN_XCHAT_CALLS(NONE);
hook->data = (void*)xchat_hook_timer(ph, timeout,
Callback_Timer, hook);
END_XCHAT_CALLS();
return PyInt_FromLong((long)hook);
}
static PyObject *
Module_xchat_hook_unload(PyObject *self, PyObject *args, PyObject *kwargs)
{
PyObject *callback;
PyObject *userdata = Py_None;
PyObject *plugin;
Hook *hook;
char *kwlist[] = {"callback", "userdata", 0};
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O|O:hook_unload",
kwlist, &callback, &userdata))
return NULL;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
if (!PyCallable_Check(callback)) {
PyErr_SetString(PyExc_TypeError, "callback is not callable");
return NULL;
}
hook = Plugin_AddHook(HOOK_UNLOAD, plugin, callback, userdata, NULL);
if (hook == NULL)
return NULL;
return PyInt_FromLong((long)hook);
}
static PyObject *
Module_xchat_unhook(PyObject *self, PyObject *args)
{
PyObject *plugin;
Hook *hook;
if (!PyArg_ParseTuple(args, "l:unhook", &hook))
return NULL;
plugin = Plugin_GetCurrent();
if (plugin == NULL)
return NULL;
Plugin_RemoveHook(plugin, hook);
Py_INCREF(Py_None);
return Py_None;
}
static PyObject *
Module_xchat_get_list(PyObject *self, PyObject *args)
{
xchat_list *list;
PyObject *l;
const char *name;
const char *const *fields;
int i;
if (!PyArg_ParseTuple(args, "s:get_list", &name))
return NULL;
/* This function is thread safe, and returns statically
* allocated data. */
fields = xchat_list_fields(ph, "lists");
for (i = 0; fields[i]; i++) {
if (strcmp(fields[i], name) == 0) {
/* Use the static allocated one. */
name = fields[i];
break;
}
}
if (fields[i] == NULL) {
PyErr_SetString(PyExc_KeyError, "list not available");
return NULL;
}
l = PyList_New(0);
if (l == NULL)
return NULL;
BEGIN_XCHAT_CALLS(RESTORE_CONTEXT);
list = xchat_list_get(ph, (char*)name);
if (list == NULL)
goto error;
fields = xchat_list_fields(ph, (char*)name);
while (xchat_list_next(ph, list)) {
PyObject *o = ListItem_New(name);
if (o == NULL || PyList_Append(l, o) == -1) {
Py_XDECREF(o);
goto error;
}
Py_DECREF(o); /* l is holding a reference */
for (i = 0; fields[i]; i++) {
const char *fld = fields[i]+1;
PyObject *attr = NULL;
const char *sattr;
int iattr;
switch(fields[i][0]) {
case 's':
sattr = xchat_list_str(ph, list, (char*)fld);
attr = PyString_FromString(sattr?sattr:"");
break;
case 'i':
iattr = xchat_list_int(ph, list, (char*)fld);
attr = PyInt_FromLong((long)iattr);
break;
case 'p':
sattr = xchat_list_str(ph, list, (char*)fld);
if (strcmp(fld, "context") == 0) {
attr = Context_FromContext(
(xchat_context*)sattr);
break;
}
default: /* ignore unknown (newly added?) types */
continue;
}
if (attr == NULL)
goto error;
PyObject_SetAttrString(o, (char*)fld, attr); /* add reference on attr in o */
Py_DECREF(attr); /* make o own attr */
}
}
xchat_list_free(ph, list);
goto exit;
error:
if (list)
xchat_list_free(ph, list);
Py_DECREF(l);
l = NULL;
exit:
END_XCHAT_CALLS();
return l;
}
static PyObject *
Module_xchat_get_lists(PyObject *self, PyObject *args)
{
PyObject *l, *o;
const char *const *fields;
int i;
/* This function is thread safe, and returns statically
* allocated data. */
fields = xchat_list_fields(ph, "lists");
l = PyList_New(0);
if (l == NULL)
return NULL;
for (i = 0; fields[i]; i++) {
o = PyString_FromString(fields[i]);
if (o == NULL || PyList_Append(l, o) == -1) {
Py_DECREF(l);
Py_XDECREF(o);
return NULL;
}
Py_DECREF(o); /* l is holding a reference */
}
return l;
}
static PyObject *
Module_xchat_nickcmp(PyObject *self, PyObject *args)
{
char *s1, *s2;
if (!PyArg_ParseTuple(args, "ss:nickcmp", &s1, &s2))
return NULL;
return PyInt_FromLong((long) xchat_nickcmp(ph, s1, s2));
}
static PyObject *
Module_xchat_strip(PyObject *self, PyObject *args)
{
PyObject *result;
char *str, *str2;
int len = -1, flags = 1 | 2;
if (!PyArg_ParseTuple(args, "s|ii:strip", &str, &len, &flags))
return NULL;
str2 = xchat_strip(ph, str, len, flags);
result = PyString_FromString(str2);
xchat_free(ph, str2);
return result;
}
static PyMethodDef Module_xchat_methods[] = {
{"command", Module_xchat_command,
METH_VARARGS},
{"prnt", Module_xchat_prnt,
METH_VARARGS},
{"emit_print", Module_xchat_emit_print,
METH_VARARGS},
{"get_info", Module_xchat_get_info,
METH_VARARGS},
{"get_prefs", Module_xchat_get_prefs,
METH_VARARGS},
{"get_context", Module_xchat_get_context,
METH_NOARGS},
{"find_context", (PyCFunction)Module_xchat_find_context,
METH_VARARGS|METH_KEYWORDS},
{"hook_command", (PyCFunction)Module_xchat_hook_command,
METH_VARARGS|METH_KEYWORDS},
{"hook_server", (PyCFunction)Module_xchat_hook_server,
METH_VARARGS|METH_KEYWORDS},
{"hook_print", (PyCFunction)Module_xchat_hook_print,
METH_VARARGS|METH_KEYWORDS},
{"hook_timer", (PyCFunction)Module_xchat_hook_timer,
METH_VARARGS|METH_KEYWORDS},
{"hook_unload", (PyCFunction)Module_xchat_hook_unload,
METH_VARARGS|METH_KEYWORDS},
{"unhook", Module_xchat_unhook,
METH_VARARGS},
{"get_list", Module_xchat_get_list,
METH_VARARGS},
{"get_lists", Module_xchat_get_lists,
METH_NOARGS},
{"nickcmp", Module_xchat_nickcmp,
METH_VARARGS},
{"strip", Module_xchat_strip,
METH_VARARGS},
{NULL, NULL}
};
/* ===================================================================== */
/* Python interactive interpreter functions */
static void
IInterp_Exec(char *command)
{
PyObject *m, *d, *o;
char *buffer;
int len;
BEGIN_PLUGIN(interp_plugin);
m = PyImport_AddModule("__main__");
if (m == NULL) {
xchat_print(ph, "Can't get __main__ module");
goto fail;
}
d = PyModule_GetDict(m);
len = strlen(command);
buffer = (char *) g_malloc(len+2);
if (buffer == NULL) {
xchat_print(ph, "Not enough memory for command buffer");
goto fail;
}
memcpy(buffer, command, len);
buffer[len] = '\n';
buffer[len+1] = 0;
o = PyRun_StringFlags(buffer, Py_single_input, d, d, NULL);
g_free(buffer);
if (o == NULL) {
PyErr_Print();
goto fail;
}
Py_DECREF(o);
if (Py_FlushLine())
PyErr_Clear();
fail:
END_PLUGIN(interp_plugin);
return;
}
static int
IInterp_Cmd(char *word[], char *word_eol[], void *userdata)
{
char *channel = (char *) xchat_get_info(ph, "channel");
if (channel && channel[0] == '>' && strcmp(channel, ">>python<<") == 0) {
xchat_printf(ph, ">>> %s\n", word_eol[1]);
IInterp_Exec(word_eol[1]);
return 1;
}
return 0;
}
/* ===================================================================== */
/* Python command handling */
static void
Command_PyList()
{
GSList *list;
list = plugin_list;
if (list == NULL) {
xchat_print(ph, "No python modules loaded");
} else {
xchat_print(ph,
"Name Version Filename Description\n"
"---- ------- -------- -----------\n");
while (list != NULL) {
PluginObject *plg = (PluginObject *) list->data;
char *basename = g_path_get_basename(plg->filename);
xchat_printf(ph, "%-12s %-8s %-20s %-10s\n",
plg->name,
*plg->version ? plg->version
: "<none>",
basename,
*plg->description ? plg->description
: "<none>");
g_free(basename);
list = list->next;
}
xchat_print(ph, "\n");
}
}
static void
Command_PyLoad(char *filename)
{
PyObject *plugin;
RELEASE_XCHAT_LOCK();
plugin = Plugin_New(filename, Module_xchat_methods, xchatout);
ACQUIRE_XCHAT_LOCK();
if (plugin)
plugin_list = g_slist_append(plugin_list, plugin);
}
static void
Command_PyUnload(char *name)
{
PluginObject *plugin = Plugin_ByString(name);
if (!plugin) {
xchat_print(ph, "Can't find a python plugin with that name");
} else {
BEGIN_PLUGIN(plugin);
Plugin_Delete((PyObject*)plugin);
END_PLUGIN(plugin);
plugin_list = g_slist_remove(plugin_list, plugin);
}
}
static void
Command_PyReload(char *name)
{
PluginObject *plugin = Plugin_ByString(name);
if (!plugin) {
xchat_print(ph, "Can't find a python plugin with that name");
} else {
char *filename = strdup(plugin->filename);
Command_PyUnload(filename);
Command_PyLoad(filename);
g_free(filename);
}
}
static void
Command_PyAbout()
{
xchat_print(ph, about);
}
static int
Command_Py(char *word[], char *word_eol[], void *userdata)
{
char *cmd = word[2];
int ok = 0;
if (strcasecmp(cmd, "LIST") == 0) {
ok = 1;
Command_PyList();
} else if (strcasecmp(cmd, "EXEC") == 0) {
if (word[3][0]) {
ok = 1;
IInterp_Exec(word_eol[3]);
}
} else if (strcasecmp(cmd, "LOAD") == 0) {
if (word[3][0]) {
ok = 1;
Command_PyLoad(word[3]);
}
} else if (strcasecmp(cmd, "UNLOAD") == 0) {
if (word[3][0]) {
ok = 1;
Command_PyUnload(word[3]);
}
} else if (strcasecmp(cmd, "RELOAD") == 0) {
if (word[3][0]) {
ok = 1;
Command_PyReload(word[3]);
}
} else if (strcasecmp(cmd, "CONSOLE") == 0) {
ok = 1;
xchat_command(ph, "QUERY >>python<<");
} else if (strcasecmp(cmd, "ABOUT") == 0) {
ok = 1;
Command_PyAbout();
}
if (!ok)
xchat_print(ph, usage);
return XCHAT_EAT_ALL;
}
static int
Command_Load(char *word[], char *word_eol[], void *userdata)
{
int len = strlen(word[2]);
if (len > 3 && strcasecmp(".py", word[2]+len-3) == 0) {
Command_PyLoad(word[2]);
return XCHAT_EAT_XCHAT;
}
return XCHAT_EAT_NONE;
}
static int
Command_Unload(char *word[], char *word_eol[], void *userdata)
{
int len = strlen(word[2]);
if (len > 3 && strcasecmp(".py", word[2]+len-3) == 0) {
Command_PyUnload(word[2]);
return XCHAT_EAT_XCHAT;
}
return XCHAT_EAT_NONE;
}
/* ===================================================================== */
/* Autoload function */
/* ===================================================================== */
/* (De)initialization functions */
static int initialized = 0;
static int reinit_tried = 0;
void
xchat_plugin_get_info(char **name, char **desc, char **version, void **reserved)
{
*name = "Python";
*version = VERSION;
*desc = "Python scripting interface";
if (reserved)
*reserved = NULL;
}
int
xchat_plugin_init(xchat_plugin *plugin_handle,
char **plugin_name,
char **plugin_desc,
char **plugin_version,
char *arg)
{
char *argv[] = {"<xchat>", 0};
ph = plugin_handle;
/* Block double initalization. */
if (initialized != 0) {
xchat_print(ph, "Python interface already loaded");
/* deinit is called even when init fails, so keep track
* of a reinit failure. */
reinit_tried++;
return 0;
}
initialized = 1;
*plugin_name = "Python";
*plugin_version = VERSION;
*plugin_desc = "Python scripting interface";
/* Initialize python. */
Py_SetProgramName("xchat");
Py_Initialize();
PySys_SetArgv(1, argv);
Plugin_Type.ob_type = &PyType_Type;
Context_Type.ob_type = &PyType_Type;
XChatOut_Type.ob_type = &PyType_Type;
xchatout = XChatOut_New();
if (xchatout == NULL) {
xchat_print(ph, "Can't allocate xchatout object");
return 0;
}
#ifdef WITH_THREAD
PyEval_InitThreads();
xchat_lock = PyThread_allocate_lock();
if (xchat_lock == NULL) {
xchat_print(ph, "Can't allocate xchat lock");
Py_DECREF(xchatout);
xchatout = NULL;
return 0;
}
#endif
main_tstate = PyEval_SaveThread();
interp_plugin = Plugin_New(NULL, Module_xchat_methods, xchatout);
if (interp_plugin == NULL) {
xchat_print(ph, "Plugin_New() failed.\n");
#ifdef WITH_THREAD
PyThread_free_lock(xchat_lock);
#endif
Py_DECREF(xchatout);
xchatout = NULL;
return 0;
}
xchat_hook_command(ph, "", XCHAT_PRI_NORM, IInterp_Cmd, 0, 0);
xchat_hook_command(ph, "PY", XCHAT_PRI_NORM, Command_Py, usage, 0);
xchat_hook_command(ph, "LOAD", XCHAT_PRI_NORM, Command_Load, 0, 0);
xchat_hook_command(ph, "UNLOAD", XCHAT_PRI_NORM, Command_Unload, 0, 0);
#ifdef WITH_THREAD
thread_timer = xchat_hook_timer(ph, 300, Callback_ThreadTimer, NULL);
#endif
xchat_print(ph, "Python interface loaded\n");
Util_Autoload();
return 1;
}
int
xchat_plugin_deinit()
{
GSList *list;
/* A reinitialization was tried. Just give up and live the
* environment as is. We are still alive. */
if (reinit_tried) {
reinit_tried--;
return 1;
}
list = plugin_list;
while (list != NULL) {
PyObject *plugin = (PyObject *) list->data;
BEGIN_PLUGIN(plugin);
Plugin_Delete(plugin);
END_PLUGIN(plugin);
list = list->next;
}
g_slist_free(plugin_list);
plugin_list = NULL;
/* Reset xchatout buffer. */
g_free(xchatout_buffer);
xchatout_buffer = NULL;
xchatout_buffer_size = 0;
xchatout_buffer_pos = 0;
if (interp_plugin) {
Py_DECREF(interp_plugin);
interp_plugin = NULL;
}
/* Switch back to the main thread state. */
if (main_tstate) {
PyThreadState_Swap(main_tstate);
main_tstate = NULL;
}
Py_Finalize();
#ifdef WITH_THREAD
if (thread_timer != NULL) {
xchat_unhook(ph, thread_timer);
thread_timer = NULL;
}
PyThread_free_lock(xchat_lock);
#endif
xchat_print(ph, "Python interface unloaded\n");
initialized = 0;
return 1;
}
Reference in New Issue Copy Permalink