1
0
mirror of https://github.com/profanity-im/profanity.git synced 2024-12-04 14:46:46 -05:00
profanity/src/plugins/python_plugins.c

944 lines
31 KiB
C
Raw Normal View History

2016-02-23 19:31:55 -05:00
/*
* python_plugins.c
2019-11-13 06:11:05 -05:00
* vim: expandtab:ts=4:sts=4:sw=4
2016-02-23 19:31:55 -05:00
*
2019-01-22 05:31:45 -05:00
* Copyright (C) 2012 - 2019 James Booth <boothj5@gmail.com>
2016-02-23 19:31:55 -05:00
*
* This file is part of Profanity.
*
* Profanity 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 3 of the License, or
* (at your option) any later version.
*
* Profanity 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
2016-07-23 20:14:49 -04:00
* along with Profanity. If not, see <https://www.gnu.org/licenses/>.
2016-02-23 19:31:55 -05:00
*
* In addition, as a special exception, the copyright holders give permission to
* link the code of portions of this program with the OpenSSL library under
* certain conditions as described in each individual source file, and
* distribute linked combinations including the two.
*
* You must obey the GNU General Public License in all respects for all of the
* code used other than OpenSSL. If you modify file(s) with this exception, you
* may extend this exception to your version of the file(s), but you are not
* obligated to do so. If you do not wish to do so, delete this exception
* statement from your version. If you delete this exception statement from all
* source files in the program, then also delete it here.
*
*/
#undef _XOPEN_SOURCE
2016-02-23 19:31:55 -05:00
#include <Python.h>
#include "log.h"
2016-07-13 19:00:46 -04:00
#include "config.h"
#include "config/preferences.h"
#include "config/files.h"
2016-02-23 19:31:55 -05:00
#include "plugins/api.h"
#include "plugins/callbacks.h"
#include "plugins/disco.h"
2016-02-23 19:31:55 -05:00
#include "plugins/plugins.h"
#include "plugins/python_api.h"
#include "plugins/python_plugins.h"
#include "ui/ui.h"
2020-07-07 08:18:57 -04:00
static PyThreadState* thread_state;
static GHashTable* loaded_modules;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
static void _python_undefined_error(ProfPlugin* plugin, char* hook, char* type);
static void _python_type_error(ProfPlugin* plugin, char* hook, char* type);
2020-07-07 08:18:57 -04:00
static char* _handle_string_or_none_result(ProfPlugin* plugin, PyObject* result, char* hook);
static gboolean _handle_boolean_result(ProfPlugin* plugin, PyObject* result, char* hook);
2016-02-23 19:31:55 -05:00
void
allow_python_threads()
{
thread_state = PyEval_SaveThread();
2016-02-23 19:31:55 -05:00
}
void
disable_python_threads()
{
PyEval_RestoreThread(thread_state);
2016-02-23 19:31:55 -05:00
}
2016-07-09 18:42:09 -04:00
static void
2020-07-07 08:18:57 -04:00
_unref_module(PyObject* module)
2016-07-09 18:42:09 -04:00
{
Py_XDECREF(module);
}
2016-07-23 21:12:56 -04:00
const char*
python_get_version_string(void)
2016-07-23 21:12:56 -04:00
{
return Py_GetVersion();
}
gchar*
python_get_version_number(void)
{
2020-07-07 08:18:57 -04:00
const char* version_str = Py_GetVersion();
gchar** split = g_strsplit(version_str, " ", 0);
gchar* version_number = g_strdup(split[0]);
g_strfreev(split);
return version_number;
}
2016-02-23 19:31:55 -05:00
void
python_env_init(void)
{
2016-07-09 18:42:09 -04:00
loaded_modules = g_hash_table_new_full(g_str_hash, g_str_equal, free, (GDestroyNotify)_unref_module);
2016-07-18 18:24:26 -04:00
python_init_prof();
2016-07-17 20:27:23 -04:00
2020-07-07 08:18:57 -04:00
char* plugins_dir = files_get_data_path(DIR_PLUGINS);
GString* path = g_string_new("import sys\n");
g_string_append(path, "sys.path.append(\"");
2016-02-23 19:31:55 -05:00
g_string_append(path, plugins_dir);
g_string_append(path, "/\")\n");
2016-02-23 19:31:55 -05:00
PyRun_SimpleString(path->str);
2016-07-17 20:27:23 -04:00
python_check_error();
2016-02-23 19:31:55 -05:00
g_string_free(path, TRUE);
2016-07-17 20:27:23 -04:00
g_free(plugins_dir);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
2016-03-30 18:18:12 -04:00
ProfPlugin*
2020-07-07 08:18:57 -04:00
python_plugin_create(const char* const filename)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_module = g_hash_table_lookup(loaded_modules, filename);
if (p_module) {
p_module = PyImport_ReloadModule(p_module);
} else {
2020-07-07 08:18:57 -04:00
gchar* module_name = g_strndup(filename, strlen(filename) - 3);
p_module = PyImport_ImportModule(module_name);
2016-07-17 20:27:23 -04:00
if (p_module) {
g_hash_table_insert(loaded_modules, strdup(filename), p_module);
}
2016-07-09 18:42:09 -04:00
g_free(module_name);
}
2016-02-23 19:31:55 -05:00
python_check_error();
if (p_module) {
2020-07-07 08:18:57 -04:00
ProfPlugin* plugin = malloc(sizeof(ProfPlugin));
2016-04-09 20:15:11 -04:00
plugin->name = strdup(filename);
2016-02-23 19:31:55 -05:00
plugin->lang = LANG_PYTHON;
plugin->module = p_module;
plugin->init_func = python_init_hook;
2017-01-22 13:08:29 -05:00
plugin->contains_hook = python_contains_hook;
2016-02-23 19:31:55 -05:00
plugin->on_start_func = python_on_start_hook;
plugin->on_shutdown_func = python_on_shutdown_hook;
2016-07-05 17:03:14 -04:00
plugin->on_unload_func = python_on_unload_hook;
2016-02-23 19:31:55 -05:00
plugin->on_connect_func = python_on_connect_hook;
plugin->on_disconnect_func = python_on_disconnect_hook;
plugin->pre_chat_message_display = python_pre_chat_message_display_hook;
plugin->post_chat_message_display = python_post_chat_message_display_hook;
plugin->pre_chat_message_send = python_pre_chat_message_send_hook;
plugin->post_chat_message_send = python_post_chat_message_send_hook;
plugin->pre_room_message_display = python_pre_room_message_display_hook;
plugin->post_room_message_display = python_post_room_message_display_hook;
plugin->pre_room_message_send = python_pre_room_message_send_hook;
plugin->post_room_message_send = python_post_room_message_send_hook;
2016-04-07 19:11:33 -04:00
plugin->on_room_history_message = python_on_room_history_message_hook;
2016-02-23 19:31:55 -05:00
plugin->pre_priv_message_display = python_pre_priv_message_display_hook;
plugin->post_priv_message_display = python_post_priv_message_display_hook;
plugin->pre_priv_message_send = python_pre_priv_message_send_hook;
plugin->post_priv_message_send = python_post_priv_message_send_hook;
2016-03-26 11:50:16 -04:00
plugin->on_message_stanza_send = python_on_message_stanza_send_hook;
2016-03-27 16:36:29 -04:00
plugin->on_message_stanza_receive = python_on_message_stanza_receive_hook;
2016-03-26 11:50:16 -04:00
plugin->on_presence_stanza_send = python_on_presence_stanza_send_hook;
2016-03-27 16:36:29 -04:00
plugin->on_presence_stanza_receive = python_on_presence_stanza_receive_hook;
2016-03-26 11:50:16 -04:00
plugin->on_iq_stanza_send = python_on_iq_stanza_send_hook;
2016-03-27 16:36:29 -04:00
plugin->on_iq_stanza_receive = python_on_iq_stanza_receive_hook;
2016-03-30 18:18:12 -04:00
plugin->on_contact_offline = python_on_contact_offline_hook;
plugin->on_contact_presence = python_on_contact_presence_hook;
2016-04-07 17:06:14 -04:00
plugin->on_chat_win_focus = python_on_chat_win_focus_hook;
2016-04-07 17:15:03 -04:00
plugin->on_room_win_focus = python_on_room_win_focus_hook;
2016-02-23 19:31:55 -05:00
allow_python_threads();
return plugin;
} else {
allow_python_threads();
return NULL;
}
}
void
2020-07-07 08:18:57 -04:00
python_init_hook(ProfPlugin* plugin, const char* const version, const char* const status, const char* const account_name,
const char* const fulljid)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ssss", version, status, account_name, fulljid);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_init")) {
p_function = PyObject_GetAttrString(p_module, "prof_init");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
2017-01-22 13:08:29 -05:00
gboolean
2020-07-07 08:18:57 -04:00
python_contains_hook(ProfPlugin* plugin, const char* const hook)
2017-01-22 13:08:29 -05:00
{
disable_python_threads();
gboolean res = FALSE;
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2017-01-22 13:08:29 -05:00
if (PyObject_HasAttrString(p_module, hook)) {
res = TRUE;
}
allow_python_threads();
return res;
}
2016-02-23 19:31:55 -05:00
void
2020-07-07 08:18:57 -04:00
python_on_start_hook(ProfPlugin* plugin)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_on_start")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_start");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, NULL);
python_check_error();
Py_XDECREF(p_function);
}
}
allow_python_threads();
}
void
2020-07-07 08:18:57 -04:00
python_on_shutdown_hook(ProfPlugin* plugin)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_on_shutdown")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_shutdown");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, NULL);
python_check_error();
Py_XDECREF(p_function);
}
}
allow_python_threads();
}
2016-07-05 17:03:14 -04:00
void
2020-07-07 08:18:57 -04:00
python_on_unload_hook(ProfPlugin* plugin)
2016-07-05 17:03:14 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_function;
2016-07-05 17:03:14 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-07-05 17:03:14 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_unload")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_unload");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, NULL);
python_check_error();
Py_XDECREF(p_function);
}
}
allow_python_threads();
}
2016-02-23 19:31:55 -05:00
void
2020-07-07 08:18:57 -04:00
python_on_connect_hook(ProfPlugin* plugin, const char* const account_name, const char* const fulljid)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ss", account_name, fulljid);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_on_connect")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_connect");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
void
2020-07-07 08:18:57 -04:00
python_on_disconnect_hook(ProfPlugin* plugin, const char* const account_name, const char* const fulljid)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ss", account_name, fulljid);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_on_disconnect")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_disconnect");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
char*
2020-07-07 08:18:57 -04:00
python_pre_chat_message_display_hook(ProfPlugin* plugin, const char* const barejid, const char* const resource,
const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, resource, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_pre_chat_message_display")) {
p_function = PyObject_GetAttrString(p_module, "prof_pre_chat_message_display");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-02-23 19:31:55 -05:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_pre_chat_message_display");
2016-02-23 19:31:55 -05:00
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
return NULL;
}
void
2020-07-07 08:18:57 -04:00
python_post_chat_message_display_hook(ProfPlugin* plugin, const char* const barejid, const char* const resource, const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, resource, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_post_chat_message_display")) {
p_function = PyObject_GetAttrString(p_module, "prof_post_chat_message_display");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
char*
2020-07-07 08:18:57 -04:00
python_pre_chat_message_send_hook(ProfPlugin* plugin, const char* const barejid, const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ss", barejid, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_pre_chat_message_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_pre_chat_message_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-02-23 19:31:55 -05:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_pre_chat_message_send");
2016-02-23 19:31:55 -05:00
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
return NULL;
}
void
2020-07-07 08:18:57 -04:00
python_post_chat_message_send_hook(ProfPlugin* plugin, const char* const barejid, const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ss", barejid, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_post_chat_message_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_post_chat_message_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
char*
2020-07-07 08:18:57 -04:00
python_pre_room_message_display_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick, const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, nick, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_pre_room_message_display")) {
p_function = PyObject_GetAttrString(p_module, "prof_pre_room_message_display");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-02-23 19:31:55 -05:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_pre_room_message_display");
2016-02-23 19:31:55 -05:00
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
return NULL;
}
void
2020-07-07 08:18:57 -04:00
python_post_room_message_display_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick,
const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, nick, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_post_room_message_display")) {
p_function = PyObject_GetAttrString(p_module, "prof_post_room_message_display");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
char*
2020-07-07 08:18:57 -04:00
python_pre_room_message_send_hook(ProfPlugin* plugin, const char* const barejid, const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ss", barejid, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_pre_room_message_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_pre_room_message_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-02-23 19:31:55 -05:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_pre_room_message_send");
2016-02-23 19:31:55 -05:00
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
return NULL;
}
void
2020-07-07 08:18:57 -04:00
python_post_room_message_send_hook(ProfPlugin* plugin, const char* const barejid, const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ss", barejid, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_post_room_message_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_post_room_message_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
2016-04-07 19:11:33 -04:00
void
2020-07-07 08:18:57 -04:00
python_on_room_history_message_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick,
const char* const message, const char* const timestamp)
2016-04-07 19:11:33 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ssss", barejid, nick, message, timestamp);
PyObject* p_function;
2016-04-07 19:11:33 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-04-07 19:11:33 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_room_history_message")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_room_history_message");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-04-07 19:11:33 -04:00
allow_python_threads();
}
2016-02-23 19:31:55 -05:00
char*
2020-07-07 08:18:57 -04:00
python_pre_priv_message_display_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick,
const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, nick, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_pre_priv_message_display")) {
p_function = PyObject_GetAttrString(p_module, "prof_pre_priv_message_display");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-02-23 19:31:55 -05:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_pre_priv_message_display");
2016-02-23 19:31:55 -05:00
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
return NULL;
}
void
2020-07-07 08:18:57 -04:00
python_post_priv_message_display_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick,
const char* message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, nick, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_post_priv_message_display")) {
p_function = PyObject_GetAttrString(p_module, "prof_post_priv_message_display");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
char*
2020-07-07 08:18:57 -04:00
python_pre_priv_message_send_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick,
const char* const message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, nick, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_pre_priv_message_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_pre_priv_message_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-02-23 19:31:55 -05:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_pre_priv_message_send");
2016-02-23 19:31:55 -05:00
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
return NULL;
}
void
2020-07-07 08:18:57 -04:00
python_post_priv_message_send_hook(ProfPlugin* plugin, const char* const barejid, const char* const nick,
const char* const message)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, nick, message);
PyObject* p_function;
2016-02-23 19:31:55 -05:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-02-23 19:31:55 -05:00
if (PyObject_HasAttrString(p_module, "prof_post_priv_message_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_post_priv_message_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-02-23 19:31:55 -05:00
allow_python_threads();
}
2016-03-26 11:50:16 -04:00
char*
2020-07-07 08:18:57 -04:00
python_on_message_stanza_send_hook(ProfPlugin* plugin, const char* const text)
2016-03-26 11:50:16 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", text);
PyObject* p_function;
2016-03-26 11:50:16 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-26 11:50:16 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_message_stanza_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_message_stanza_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-03-26 11:50:16 -04:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_on_message_stanza_send");
2016-03-26 11:50:16 -04:00
}
}
Py_XDECREF(p_args);
2016-03-26 11:50:16 -04:00
allow_python_threads();
return NULL;
}
2016-03-27 16:36:29 -04:00
gboolean
2020-07-07 08:18:57 -04:00
python_on_message_stanza_receive_hook(ProfPlugin* plugin, const char* const text)
2016-03-27 16:36:29 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", text);
PyObject* p_function;
2016-03-27 16:36:29 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-27 16:36:29 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_message_stanza_receive")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_message_stanza_receive");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-03-27 16:36:29 -04:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_boolean_result(plugin, result, "prof_on_message_stanza_receive");
2016-03-27 16:36:29 -04:00
}
}
Py_XDECREF(p_args);
allow_python_threads();
2016-03-27 16:36:29 -04:00
return TRUE;
}
2016-03-26 11:50:16 -04:00
char*
2020-07-07 08:18:57 -04:00
python_on_presence_stanza_send_hook(ProfPlugin* plugin, const char* const text)
2016-03-26 11:50:16 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", text);
PyObject* p_function;
2016-03-26 11:50:16 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-26 11:50:16 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_presence_stanza_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_presence_stanza_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-03-26 11:50:16 -04:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_on_presence_stanza_send");
2016-03-26 11:50:16 -04:00
}
}
Py_XDECREF(p_args);
2016-03-26 11:50:16 -04:00
allow_python_threads();
return NULL;
}
2016-03-27 16:36:29 -04:00
gboolean
2020-07-07 08:18:57 -04:00
python_on_presence_stanza_receive_hook(ProfPlugin* plugin, const char* const text)
2016-03-27 16:36:29 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", text);
PyObject* p_function;
2016-03-27 16:36:29 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-27 16:36:29 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_presence_stanza_receive")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_presence_stanza_receive");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-03-27 16:36:29 -04:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_boolean_result(plugin, result, "prof_on_presence_stanza_receive");
2016-03-27 16:36:29 -04:00
}
}
Py_XDECREF(p_args);
allow_python_threads();
2016-03-27 16:36:29 -04:00
return TRUE;
}
2016-03-26 11:50:16 -04:00
char*
2020-07-07 08:18:57 -04:00
python_on_iq_stanza_send_hook(ProfPlugin* plugin, const char* const text)
2016-03-26 11:50:16 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", text);
PyObject* p_function;
2016-03-26 11:50:16 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-26 11:50:16 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_iq_stanza_send")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_iq_stanza_send");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-03-26 11:50:16 -04:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_string_or_none_result(plugin, result, "prof_on_iq_stanza_send");
2016-03-26 11:50:16 -04:00
}
}
Py_XDECREF(p_args);
2016-03-26 11:50:16 -04:00
allow_python_threads();
return NULL;
}
2016-03-27 16:36:29 -04:00
gboolean
2020-07-07 08:18:57 -04:00
python_on_iq_stanza_receive_hook(ProfPlugin* plugin, const char* const text)
2016-03-27 16:36:29 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", text);
PyObject* p_function;
2016-03-27 16:36:29 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-27 16:36:29 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_iq_stanza_receive")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_iq_stanza_receive");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
2020-07-07 08:18:57 -04:00
PyObject* result = PyObject_CallObject(p_function, p_args);
2016-03-27 16:36:29 -04:00
python_check_error();
Py_XDECREF(p_function);
Py_XDECREF(p_args);
return _handle_boolean_result(plugin, result, "prof_on_iq_stanza_receive");
2016-03-27 16:36:29 -04:00
}
}
Py_XDECREF(p_args);
allow_python_threads();
2016-03-27 16:36:29 -04:00
return TRUE;
}
2016-03-30 18:18:12 -04:00
void
2020-07-07 08:18:57 -04:00
python_on_contact_offline_hook(ProfPlugin* plugin, const char* const barejid, const char* const resource,
const char* const status)
2016-03-30 18:18:12 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("sss", barejid, resource, status);
PyObject* p_function;
2016-03-30 18:18:12 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-30 18:18:12 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_contact_offline")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_contact_offline");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-03-30 18:18:12 -04:00
allow_python_threads();
}
void
2020-07-07 08:18:57 -04:00
python_on_contact_presence_hook(ProfPlugin* plugin, const char* const barejid, const char* const resource,
const char* const presence, const char* const status, const int priority)
2016-03-30 18:18:12 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("ssssi", barejid, resource, presence, status, priority);
PyObject* p_function;
2016-03-30 18:18:12 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-03-30 18:18:12 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_contact_presence")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_contact_presence");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-03-30 18:18:12 -04:00
allow_python_threads();
}
2016-04-07 17:06:14 -04:00
void
2020-07-07 08:18:57 -04:00
python_on_chat_win_focus_hook(ProfPlugin* plugin, const char* const barejid)
2016-04-07 17:06:14 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", barejid);
PyObject* p_function;
2016-04-07 17:06:14 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-04-07 17:06:14 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_chat_win_focus")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_chat_win_focus");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-04-07 17:06:14 -04:00
allow_python_threads();
}
2016-04-07 17:15:03 -04:00
void
2020-07-07 08:18:57 -04:00
python_on_room_win_focus_hook(ProfPlugin* plugin, const char* const barejid)
2016-04-07 17:15:03 -04:00
{
disable_python_threads();
2020-07-07 08:18:57 -04:00
PyObject* p_args = Py_BuildValue("(s)", barejid);
PyObject* p_function;
2016-04-07 17:15:03 -04:00
2020-07-07 08:18:57 -04:00
PyObject* p_module = plugin->module;
2016-04-07 17:15:03 -04:00
if (PyObject_HasAttrString(p_module, "prof_on_room_win_focus")) {
p_function = PyObject_GetAttrString(p_module, "prof_on_room_win_focus");
python_check_error();
if (p_function && PyCallable_Check(p_function)) {
PyObject_CallObject(p_function, p_args);
python_check_error();
Py_XDECREF(p_function);
}
}
Py_XDECREF(p_args);
2016-04-07 17:15:03 -04:00
allow_python_threads();
}
2016-02-23 19:31:55 -05:00
void
python_check_error(void)
{
if (PyErr_Occurred()) {
PyErr_Print();
2016-07-17 20:27:23 -04:00
PyRun_SimpleString("import sys\nsys.stdout.flush()");
2016-02-23 19:31:55 -05:00
PyErr_Clear();
}
}
void
2020-07-07 08:18:57 -04:00
python_plugin_destroy(ProfPlugin* plugin)
2016-02-23 19:31:55 -05:00
{
disable_python_threads();
callbacks_remove(plugin->name);
disco_remove_features(plugin->name);
2016-02-23 19:31:55 -05:00
free(plugin->name);
free(plugin);
allow_python_threads();
}
void
python_shutdown(void)
{
disable_python_threads();
2016-07-09 18:42:09 -04:00
g_hash_table_destroy(loaded_modules);
2016-02-23 19:31:55 -05:00
Py_Finalize();
}
static void
2020-07-07 08:18:57 -04:00
_python_undefined_error(ProfPlugin* plugin, char* hook, char* type)
{
2020-07-07 08:18:57 -04:00
GString* err_msg = g_string_new("Plugin error - ");
char* module_name = g_strndup(plugin->name, strlen(plugin->name) - 2);
g_string_append(err_msg, module_name);
free(module_name);
g_string_append(err_msg, hook);
g_string_append(err_msg, "(): return value undefined, expected ");
g_string_append(err_msg, type);
log_error(err_msg->str);
cons_show_error(err_msg->str);
g_string_free(err_msg, TRUE);
}
static void
2020-07-07 08:18:57 -04:00
_python_type_error(ProfPlugin* plugin, char* hook, char* type)
{
2020-07-07 08:18:57 -04:00
GString* err_msg = g_string_new("Plugin error - ");
char* module_name = g_strndup(plugin->name, strlen(plugin->name) - 2);
g_string_append(err_msg, module_name);
free(module_name);
g_string_append(err_msg, hook);
g_string_append(err_msg, "(): incorrect return type, expected ");
g_string_append(err_msg, type);
log_error(err_msg->str);
cons_show_error(err_msg->str);
g_string_free(err_msg, TRUE);
}
static char*
2020-07-07 08:18:57 -04:00
_handle_string_or_none_result(ProfPlugin* plugin, PyObject* result, char* hook)
{
if (result == NULL) {
allow_python_threads();
_python_undefined_error(plugin, hook, "string, unicode or None");
return NULL;
}
#if PY_MAJOR_VERSION >= 3
if (result != Py_None && !PyUnicode_Check(result) && !PyBytes_Check(result)) {
allow_python_threads();
_python_type_error(plugin, hook, "string, unicode or None");
return NULL;
}
#else
if (result != Py_None && !PyUnicode_Check(result) && !PyString_Check(result)) {
allow_python_threads();
_python_type_error(plugin, hook, "string, unicode or None");
return NULL;
}
#endif
2020-07-07 08:18:57 -04:00
char* result_str = python_str_or_unicode_to_string(result);
allow_python_threads();
return result_str;
}
static gboolean
2020-07-07 08:18:57 -04:00
_handle_boolean_result(ProfPlugin* plugin, PyObject* result, char* hook)
{
if (result == NULL) {
allow_python_threads();
_python_undefined_error(plugin, hook, "boolean");
return TRUE;
}
if (PyObject_IsTrue(result)) {
allow_python_threads();
return TRUE;
}
allow_python_threads();
return FALSE;
}