source: protocols/python/src/bpython.c @ 26f3b5f

/*
 * bpython.c - Python plugin for bitlbee
 * 
 * Copyright (c) 2013 Nick Murdoch
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

#include <bitlbee.h>
#include <Python.h>
#include "bytesobject.h"

/* Python module classes: */

typedef struct {
    PyObject_HEAD

} bpython_ProtocolObject;

static PyTypeObject bpython_ProtocolType = {
    PyObject_HEAD_INIT(NULL)
    0,                         /*ob_size*/
    "bpython.Protocol",  /*tp_name*/
    sizeof(bpython_ProtocolObject), /*tp_basicsize*/
    0,                         /*tp_itemsize*/
    0,                         /*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*/
    0,                         /*tp_getattro*/
    0,                         /*tp_setattro*/
    0,                         /*tp_as_buffer*/
    Py_TPFLAGS_DEFAULT,        /*tp_flags*/
    "Protocol plugin objects",           /* tp_doc */
};

/* Python module functions: */

static PyObject * bpython_register_protocol(PyObject *self, PyObject *args)
{
    const char *command;
    int sts;

    if (!PyArg_ParseTuple(args, "s", &command))
        return NULL;
    sts = system(command);
    return Py_BuildValue("i", sts);
}

static PyMethodDef BpythonMethods[] = {
    {"register_protocol",  bpython_register_protocol, METH_VARARGS,
     "Register a protocol."},
    {NULL, NULL, 0, NULL}        /* Sentinel */
};

PyMODINIT_FUNC initbpython(void)
{
    PyObject * m;

    bpython_ProtocolType.tp_new = PyType_GenericNew;
    if (PyType_Ready(&bpython_ProtocolType) < 0) {
        return;
    }

    m = Py_InitModule3("bpython", BpythonMethods, "Bitlbee Plugin module");

    Py_INCREF(&bpython_ProtocolType);
    PyModule_AddObject(m, "Protocol", (PyObject *)&bpython_ProtocolType);
}


static void load_pyfile(char * path, PyObject * main_dict) {
    PyObject * err;
    struct prpl *ret;
    ret = g_new0(struct prpl, 1);
    
    PyObject * pluginname = PyDict_GetItemString(main_dict, "name");
    if ((err = PyErr_Occurred()) || !pluginname) {
        printf("No plugin name\n");
        PyErr_Print();
        g_free(ret);
        return;
    }
    PyObject * pluginname_unicode = PyObject_Unicode(pluginname);
    PyObject * pluginname_encoded = PyUnicode_AsEncodedString(pluginname_unicode, "ascii", "ignore");
    if ((err = PyErr_Occurred())) {
        printf("Error encoding plugin name\n");
        PyErr_Print();
        g_free(ret);
        return;
    }
    char * pluginname_tmp; /* reference, do not modify */
    Py_ssize_t length;
    PyBytes_AsStringAndSize(pluginname_encoded, &pluginname_tmp, &length);
    if ((err = PyErr_Occurred())) {
        printf("Bad plugin name\n");
        PyErr_Print();
        g_free(ret);
        return;
    }
    ret->name = g_malloc0(length + 1);
    memmove(ret->name, pluginname_tmp, length);

    Py_DECREF(pluginname_encoded);
    Py_DECREF(pluginname_unicode);

    register_protocol(ret);
}

/* Bitlbee plugin functions: */

void init_plugin() {
    GDir *dir;
    GError *error = NULL;

    printf("Initialising Python... ");
    Py_Initialize();
    PyRun_SimpleString("print 'Python initialised!'\n");
    
    /* Add our static module */
    initbpython();

    /* Get a reference to the main module. */
    PyObject* main_module = PyImport_AddModule("__main__");

    /* Get the main module's dictionary */
    PyObject* main_dict = PyModule_GetDict(main_module);

    dir = g_dir_open(global.conf->plugindir, 0, &error);
    if (dir) {
        const gchar *entry;
        char * path;
        while ((entry = g_dir_read_name(dir))) {
            path = g_build_filename(global.conf->plugindir, entry, NULL);
            if (!path) {
                log_message(LOGLVL_WARNING, "Can't build path for %s\n", entry);
                continue;
            }
            
            if (g_str_has_suffix(path, ".py")) {
                FILE * pyfile;
                printf("Loading python file %s\n", path);
                pyfile = fopen(path, "r");
                
                /* Copy main dict to make sure that separate plugins
                   run in separate environments */
                PyObject * main_dict_copy = PyDict_Copy(main_dict);

                /* Run the python file */
                PyRun_File(pyfile, path, Py_file_input, main_dict_copy, main_dict_copy);

            }

            g_free(path);
        }
    }

    // Py_Finalize(); // TODO - there's no plugin_unload for us to do this in.
}

int main(int argc, char ** argv) {
    printf("This is a bitlbee plugin, you should place it in bitlbee's plugins directory.");
    return 0;
}