1 files changed, 306 insertions, 0 deletions
diff --git a/pyloader.c b/pyloader.c
new file mode 100644
index 0000000..6c7fdfe
--- /dev/null
+++ b/pyloader.c
@@ -0,0 +1,306 @@
+#include <Python.h>
+#include <string.h>
+#include "pyirssi.h"
+#include "pyloader.h"
+#include "pyutils.h"
+#include "pyscript-object.h"
+
+/* List of loaded modules */
+static PyObject *script_modules;
+
+/* List of load paths for scripts */
+static GSList *script_paths = NULL;
+
+static PyObject *py_get_script(const char *name, int *id);
+static int py_load_module(PyObject *module, const char *path);
+static char *py_find_script(const char *name);
+
+/* Add to the list of script load paths */
+void pyloader_add_script_path(const char *path)
+{
+    PyObject *ppath = PySys_GetObject("path");
+    if (ppath)
+    {
+        PyList_Append(ppath, PyString_FromString(path));
+        script_paths = g_slist_append(script_paths, g_strdup(path));
+    }
+}
+
+/* Loads a file into a module; it is not inserted into sys.modules */
+static int py_load_module(PyObject *module, const char *path) 
+{
+    PyObject *dict, *ret, *fp;
+
+    if (PyModule_AddStringConstant(module, "__file__", (char *)path) < 0)
+        return 0;
+    
+    dict = PyModule_GetDict(module);
+    
+    if (PyDict_SetItemString(dict, "__builtins__", PyEval_GetBuiltins()) < 0)
+        return 0;
+
+    /* Dont use the standard library to avoid incompatabilities with 
+       the FILE structure and Python */
+    fp = PyFile_FromString((char *)path, "r");
+    if (!fp)
+        return 0;
+
+    ret = PyRun_File(PyFile_AsFile(fp), path, Py_file_input, dict, dict);
+    Py_DECREF(fp);  /* XXX: I assume that the file is closed when refs drop to zero? */ 
+    if (!ret)
+        return 0;
+
+    Py_DECREF(ret);
+    return 1;
+
+}
+
+/* looks up name in Irssi script directories
+   returns full path or NULL if not found */
+static char *py_find_script(const char *name)
+{
+    GSList *node;
+    char *fname;
+    char *path = NULL;
+
+    //XXX: what if there's another ext?
+    if (!file_has_ext(name, "py"))
+        fname = g_strdup_printf("%s.py", name);
+    else
+        fname = (char *)name;
+    
+    /*XXX: use case insensitive path search? */
+    for (node = script_paths; node != NULL && !path; node = node->next)
+    {
+        path = g_strdup_printf("%s/%s", (char *)node->data, fname);
+
+        if (!g_file_test(path, G_FILE_TEST_IS_REGULAR))
+        { 
+            g_free(path);
+            path = NULL;
+        }
+    }
+
+    if (fname != name)
+        g_free(fname);
+
+    return path;
+}
+
+/* Load a script manually using PyRun_File.
+ * This expects a null terminated array of strings 
+ * (such as from g_strsplit) of the command line.
+ * The array needs at least one item
+ */
+int pyloader_load_script_argv(char **argv)
+{
+    PyObject *module = NULL, *script = NULL;
+    char *name = NULL, *path = NULL;
+
+    if (py_get_script(argv[0], NULL) != NULL)
+    {
+        printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "script %s already loaded", argv[0]); 
+        return 0;
+    }
+   
+    path = py_find_script(argv[0]);
+    if (!path)
+    {
+        printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "script %s does not exist", argv[0]); 
+        return 0;
+    }
+
+    name = file_get_filename(path);
+    module = PyModule_New(name);
+    g_free(name);
+
+    if (!module)
+        goto error;
+
+    script = pyscript_new(module, argv);
+    Py_DECREF(module);
+
+    if (!script)
+        goto error;
+   
+    /* insert script obj into module dict, load file */
+    if (PyModule_AddObject(module, "_script", script) < 0)
+        goto error;
+    Py_INCREF(script);
+    
+    if (!py_load_module(module, path))
+        goto error;
+    
+    PyList_Append(script_modules, script);
+    Py_DECREF(script);
+    g_free(path);
+
+    /* PySys_WriteStdout("load %s, script -> 0x%x\n", argv[0], script); */
+
+    printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "loaded script %s", argv[0]); 
+    return 1;
+
+error:
+    if (PyErr_Occurred())
+        PyErr_Print();
+    else
+        printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "error loading script %s", argv[0]); 
+
+    Py_XDECREF(script);
+    g_free(path);
+    
+    return 0;
+}
+
+int pyloader_load_script(char *name)
+{
+    char *argv[2];
+
+    argv[0] = name;
+    argv[1] = NULL;
+
+    return pyloader_load_script_argv(argv);
+}
+
+static PyObject *py_get_script(const char *name, int *id)
+{
+    int i;
+
+    g_return_val_if_fail(script_modules != NULL, NULL);
+    
+    for (i = 0; i < PyList_Size(script_modules); i++)
+    {
+        PyObject *script;
+        char *sname;
+
+        script = PyList_GET_ITEM(script_modules, i);
+        sname = pyscript_get_name(script);
+
+        if (sname && !strcmp(sname, name))
+        {
+            if (id)
+                *id = i;
+            return script;
+        }
+    }
+
+    return NULL;
+}
+
+int pyloader_unload_script(const char *name)
+{
+    int id;
+    PyObject *script = py_get_script(name, &id);
+
+    if (!script)
+    {
+        printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "%s is not loaded", name); 
+        return 0;
+    }
+
+    PySys_WriteStdout("unload %s, script -> 0x%x\n", name, script);
+    
+    pyscript_remove_signals(script);
+    pyscript_clear_modules(script);
+
+    if (PySequence_DelItem(script_modules, id) < 0)
+    {
+        PyErr_Print();
+        printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "error unloading script %s", name); 
+        return 0;
+    }
+
+    /* Probably a good time to call the garbage collecter to clean up reference cycles */
+    PyGC_Collect();
+    printtext(NULL, NULL, MSGLEVEL_CLIENTERROR, "unloaded script %s", name); 
+    
+    return 1; 
+}
+
+GSList *pyloader_list(void)
+{
+    int i;
+    GSList *list = NULL;
+
+    g_return_val_if_fail(script_modules != NULL, NULL);
+
+    for (i = 0; i < PyList_Size(script_modules); i++)
+    {
+        PyObject *scr;
+        char *name, *file;
+
+        scr = PyList_GET_ITEM(script_modules, i);
+        name = pyscript_get_name(scr);
+        file = pyscript_get_filename(scr);
+
+        if (name && file)
+        {
+            PY_LIST_REC *rec;
+            rec = g_new0(PY_LIST_REC, 1);
+
+            rec->name = g_strdup(name);
+            rec->file = g_strdup(file);
+            list = g_slist_append(list, rec); 
+        }
+    }
+
+    return list;
+}
+
+void pyloader_list_destroy(GSList **list)
+{
+    GSList *node;
+
+    if (*list == NULL)
+        return;
+    
+    for (node = *list; node != NULL; node = node->next)
+    {
+        PY_LIST_REC *rec = node->data;
+
+        g_free(rec->name);
+        g_free(rec->file);
+        g_free(rec);
+    }
+   
+    g_slist_free(*list);
+    
+    *list = NULL;
+}
+
+int pyloader_init(void)
+{
+    char *pyhome;
+
+    g_return_val_if_fail(script_paths == NULL, 0);
+    g_return_val_if_fail(script_modules == NULL, 0);
+
+    script_modules = PyList_New(0);
+    if (!script_modules)
+        return 0;
+    
+    /* XXX: load autorun scripts here */
+    /* Add script location to the load path (add more paths later) */
+    pyhome = g_strdup_printf("%s/scripts", get_irssi_dir());
+    pyloader_add_script_path(pyhome);
+    g_free(pyhome);
+
+    return 1;
+}
+
+void pyloader_deinit(void)
+{
+    GSList *node;
+
+    g_return_if_fail(script_paths != NULL);
+    g_return_if_fail(script_modules != NULL);
+
+    for (node = script_paths; node != NULL; node = node->next)
+        g_free(node->data);
+    g_slist_free(script_paths);
+    script_paths = NULL;
+
+    /* script specific resources (cmd + signal handlers) should be removed
+       by the object's destructor */
+    Py_DECREF(script_modules);
+}