+
+extern struct userNode *global, *chanserv, *opserv, *nickserv, *spamserv;
+
+/*
+Some hooks you can call from modpython.py to interact with the
+service. These emb_* functions are available as _svc.* in python. */
+
+struct _tuple_dict_extra {
+ PyObject* data;
+ size_t* extra;
+};
+
+static int _dict_iter_get_users(char const* key, UNUSED_ARG(void* data), void* extra) {
+ PyObject* tmp;
+ struct _tuple_dict_extra* real_extra = (struct _tuple_dict_extra*)extra;
+
+ if ((tmp = PyString_FromString(key)) == NULL)
+ return 1;
+
+ if (PyTuple_SetItem(real_extra->data, *(int*)real_extra->extra, tmp))
+ return 1;
+
+ *real_extra->extra = *real_extra->extra + 1;
+
+ return 0;
+}
+
+static int _dict_iter_get_channels(char const* key, UNUSED_ARG(void* data), void* extra) {
+ PyObject* tmp;
+ struct _tuple_dict_extra* real_extra = (struct _tuple_dict_extra*)extra;
+
+ if ((tmp = PyString_FromString(key)) == NULL)
+ return 1;
+
+ if (PyTuple_SetItem(real_extra->data, *(int*)real_extra->extra, tmp))
+ return 1;
+
+ *real_extra->extra = *real_extra->extra + 1;
+ return 0;
+}
+
+static int _dict_iter_get_servers(char const* key, UNUSED_ARG(void* data), void* extra) {
+ struct _tuple_dict_extra* real_extra = (struct _tuple_dict_extra*)extra;
+
+ PyTuple_SetItem(real_extra->data, *(int*)real_extra->extra,
+ PyString_FromString(key));
+ *real_extra->extra = *real_extra->extra + 1;
+ return 0;
+}
+
+/* get a tuple with all users in it */
+static PyObject*
+emb_get_users(UNUSED_ARG(PyObject *self), PyObject *args) {
+ PyObject* retval;
+ PyObject* tmp;
+ size_t num_clients, n = 0, i;
+ struct _tuple_dict_extra extra;
+
+ if (!PyArg_ParseTuple(args, ""))
+ return NULL;
+
+ num_clients = dict_size(clients);
+ retval = PyTuple_New(num_clients);
+ if (retval == NULL)
+ return NULL;
+
+ extra.extra = &n;
+ extra.data = retval;
+
+ if (dict_foreach(clients, _dict_iter_get_users, (void*)&extra) != NULL) {
+ for (i = 0; i < n; ++i) {
+ tmp = PyTuple_GetItem(retval, i);
+ PyTuple_SET_ITEM(retval, i, NULL);
+ Py_DECREF(tmp);
+ }
+ Py_DECREF(retval);
+ return NULL;
+ }
+
+ return retval;
+}
+
+/* get a tuple with all channels in it */
+static PyObject*
+emb_get_channels(UNUSED_ARG(PyObject* self), PyObject* args) {
+ PyObject* retval;
+ PyObject* tmp;
+ size_t num_channels, n = 0, i;
+ struct _tuple_dict_extra extra;
+
+ if (!PyArg_ParseTuple(args, ""))
+ return NULL;
+
+ num_channels = dict_size(channels);
+ retval = PyTuple_New(num_channels);
+ if (retval == NULL)
+ return NULL;
+
+ extra.extra = &n;
+ extra.data = retval;
+
+ if (dict_foreach(channels, _dict_iter_get_channels, (void*)&extra) != NULL) {
+ for (i = 0; i < n; ++i) {
+ tmp = PyTuple_GetItem(retval, i);
+ PyTuple_SET_ITEM(retval, i, NULL);
+ Py_DECREF(tmp);
+ }
+ Py_DECREF(retval);
+ return NULL;
+ }
+
+ return retval;
+}
+
+static PyObject*
+emb_get_servers(UNUSED_ARG(PyObject* self), PyObject* args) {
+ PyObject* retval;
+ size_t n = 0;
+ struct _tuple_dict_extra extra;
+
+ if (!PyArg_ParseTuple(args, ""))
+ return NULL;
+
+ retval = PyTuple_New(dict_size(servers));
+
+ extra.extra = &n;
+ extra.data = retval;
+
+ dict_foreach(servers, _dict_iter_get_servers, (void*)&extra);
+
+ return retval;
+}
+
+static PyObject*
+emb_dump(UNUSED_ARG(PyObject *self), PyObject *args)
+{
+ /* Dump a raw string into the socket
+ usage: _svc.dump(<P10 string>)
+ */
+ char *buf;
+ int ret = 0;
+ char linedup[MAXLEN];
+
+
+ if(!PyArg_ParseTuple(args, "s:dump", &buf ))
+ return NULL;
+
+ safestrncpy(linedup, buf, sizeof(linedup));
+
+ if(parse_line(linedup, 1)) {
+ irc_raw(buf);
+ ret = 1;
+ } else {
+ PyErr_SetString(PyExc_Exception, "invalid protocol message");
+ return NULL;
+ }
+
+ return Py_BuildValue("i", ret);
+}
+
+static PyObject*
+emb_send_target_privmsg(UNUSED_ARG(PyObject *self), PyObject *args)
+{
+ /* Send a privmsg
+ usage: _svc.send_target_privmsg(<servicenick_from>, <nick_to>, <message>)
+ */
+ int ret = 0;
+ char *servicenick;
+ char *channel;
+ char *buf;
+
+ struct service *service;
+
+
+ if(!PyArg_ParseTuple(args, "sss:reply", &servicenick, &channel, &buf ))
+ return NULL;
+
+ if (buf == NULL || strlen(buf) == 0) {
+ PyErr_SetString(PyExc_Exception, "invalid empty message");
+ return NULL;
+ }
+
+ if(!(service = service_find(servicenick))) {
+ PyErr_SetString(PyExc_Exception, "no such service nick");
+ return NULL;
+ }
+
+ ret = send_target_message(5, channel, service->bot, "%s", buf);
+ return Py_BuildValue("i", ret);
+}
+
+static PyObject*
+emb_send_target_notice(UNUSED_ARG(PyObject *self), PyObject *args)
+{
+ /* send a notice
+ usage: _svc.send_target_notice(<servicenick_from>, <nick_to>, <message>)
+ */
+ int ret = 0;
+ char *servicenick;
+ char *target;
+ char *buf;
+
+ struct service *service;
+
+ if(!PyArg_ParseTuple(args, "sss:reply", &servicenick, &target, &buf ))
+ return NULL;
+
+ if (buf == NULL || strlen(buf) == 0) {
+ PyErr_SetString(PyExc_Exception, "invalid empty message");
+ return NULL;
+ }
+
+ if(!(service = service_find(servicenick))) {
+ PyErr_SetString(PyExc_Exception, "no such service nick");
+ return NULL;
+ }
+
+ ret = send_target_message(4, target, service->bot, "%s", buf);
+
+ return Py_BuildValue("i", ret);
+}
+
+static PyObject*
+pyobj_from_usernode(struct userNode* user) {
+ unsigned int n;
+ struct modeNode *mn;
+ PyObject* pChanList = PyTuple_New(user->channels.used);
+
+ for (n=0; n < user->channels.used; n++) {
+ mn = user->channels.list[n];
+ PyTuple_SetItem(pChanList, n, Py_BuildValue("s", mn->channel->name));
+ }
+
+ return Py_BuildValue("{"
+ "s: s, " /* nick */
+ "s: s, " /* ident */
+ "s: s, " /* info */
+ "s: s, " /* hostname */
+ "s: s, " /* ip */
+ "s: s, " /* fakehost */
+ "s: s, " /* sethost */
+ "s: s, " /* crypthost */
+ "s: s, " /* cryptip */
+ "s: s, " /* numeric */
+ "s: i, " /* loc */
+ "s: i, " /* no_notice */
+ "s: s, " /* mark */
+ "s: s, " /* version_reply */
+ "s: s, " /* account */
+ "s: O}", /* channels */
+ "nick", user->nick,
+ "ident", user->ident,
+ "info", user->info,
+ "hostname", user->hostname,
+ "ip", irc_ntoa(&user->ip),
+ "fakehost", user->fakehost,
+ "sethost", user->sethost,
+ "crypthost", user->crypthost,
+ "cryptip", user->cryptip,
+ "numeric", user->numeric,
+ "loc", user->loc,
+ "no_notice", user->no_notice,
+ "mark", user->mark,
+ "version_reply", user->version_reply,
+ "account", user->handle_info ? user->handle_info->handle : NULL,
+ "channels", pChanList);
+}
+
+static PyObject*
+emb_get_user(UNUSED_ARG(PyObject *self), PyObject *args)
+{
+ /* Get a python object containing everything x3 knows about a user, by nick.
+ usage: _svc.get_user(<nick>)
+ */
+ char const* nick;
+ struct userNode *user;
+
+ if(!PyArg_ParseTuple(args, "s", &nick))
+ return NULL;
+
+ if(!(user = GetUserH(nick))) {
+ PyErr_SetString(PyExc_Exception, "no such user");
+ return NULL;
+ }
+
+ return pyobj_from_usernode(user);
+}
+
+static PyObject*
+pyobj_from_server(struct server* srv) {
+ size_t n, idx;
+ PyObject* tmp = NULL;
+ PyObject* retval = NULL;
+ PyObject* users = PyTuple_New(srv->clients);
+
+ if (users == NULL)
+ return NULL;
+
+ idx = 0;
+ for (n = 0; n < srv->num_mask; ++n) {
+ if (srv->users[n] == NULL)
+ continue;
+
+ tmp = PyString_FromString(srv->users[n]->nick);
+ if (tmp == NULL)
+ goto cleanup;
+
+ if (PyTuple_SetItem(users, idx++, tmp))
+ goto cleanup;
+ }
+
+ retval = Py_BuildValue("{"
+ "s:s," /* name */
+ "s:l," /* boot */
+ "s:l," /* link_time */
+ "s:s," /* description */
+ "s:s," /* numeric */
+ "s:I," /* num_mask */
+ "s:I," /* hops */
+ "s:I," /* clients */
+ "s:I," /* max_clients */
+ "s:I," /* burst */
+ "s:I," /* self_burst */
+ "s:s" /* uplink */
+ "s:O" /* users */
+ /* TODO: Children */
+ "}",
+ "name", srv->name,
+ "boot", srv->boot,
+ "link_time", srv->link_time,
+ "description", srv->description,
+ "numeric", srv->numeric,
+ "num_mask", srv->num_mask,
+ "hops", srv->hops,
+ "clients", srv->clients,
+ "max_clients", srv->max_clients,
+ "burst", srv->burst,
+ "self_burst", srv->self_burst,
+ "uplink", srv->uplink ? srv->uplink->name : NULL,
+ "users", users
+ );
+
+ if (retval == NULL)
+ goto cleanup;
+
+ return retval;
+
+cleanup:
+ Py_XDECREF(retval);
+
+ for (n = 0; n < idx; ++n) {
+ tmp = PyTuple_GetItem(users, n);
+ PyTuple_SetItem(users, n, NULL);
+ Py_DECREF(tmp);
+ }
+ Py_DECREF(users);
+
+ return NULL;
+}
+
+static PyObject*
+emb_get_server(UNUSED_ARG(PyObject* self), PyObject* args) {
+ struct server* srv;
+ char const* name;
+
+ if (!PyArg_ParseTuple(args, "s", &name))
+ return NULL;
+
+ if (name == NULL || strlen(name) == 0) {
+ PyErr_SetString(PyExc_Exception, "invalid server name");
+ return NULL;
+ }
+
+ if ((srv = GetServerH(name)) == NULL) {
+ PyErr_SetString(PyExc_Exception, "unknown server");
+ return NULL;
+ }
+
+ return pyobj_from_server(srv);
+}
+
+static PyObject*
+emb_get_channel(UNUSED_ARG(PyObject *self), PyObject *args)
+{
+ /* Returns a python dict object with all sorts of info about a channel.
+ usage: _svc.get_channel(<name>)
+ */
+ char *name;
+ struct chanNode *channel;
+ unsigned int n;
+ PyObject *pChannelMembers;
+ PyObject *pChannelBans;
+ PyObject *pChannelExempts;
+
+
+ if(!PyArg_ParseTuple(args, "s", &name))
+ return NULL;
+
+ if(!(channel = GetChannel(name))) {
+ PyErr_SetString(PyExc_Exception, "unknown channel");
+ return NULL;
+ }
+
+ /* build tuple of nicks in channel */
+ pChannelMembers = PyTuple_New(channel->members.used);
+ for(n=0;n < channel->members.used;n++) {
+ struct modeNode *mn = channel->members.list[n];
+ PyTuple_SetItem(pChannelMembers, n, Py_BuildValue("s", mn->user->nick));
+ }
+
+ /* build tuple of bans */
+ pChannelBans = PyTuple_New(channel->banlist.used);
+ for(n=0; n < channel->banlist.used;n++) {
+ struct banNode *bn = channel->banlist.list[n];
+ PyTuple_SetItem(pChannelBans, n,
+ Py_BuildValue("{s:s,s:s,s:i}",
+ "ban", bn->ban,
+ "who", bn->who,
+ "set", bn->set)
+ );
+ }
+
+ /* build tuple of exempts */
+ pChannelExempts = PyTuple_New(channel->exemptlist.used);
+ for(n=0; n < channel->exemptlist.used;n++) {
+ struct exemptNode *en = channel->exemptlist.list[n];
+ PyTuple_SetItem(pChannelExempts, n,
+ Py_BuildValue("{s:s,s:s,s:i}",
+ "ban", en->exempt,
+ "who", en->who,
+ "set", en->set)
+ );
+ }
+
+ return Py_BuildValue("{s:s,s:s,s:s,s:i"
+ ",s:i,s:i,s:O,s:O,s:O}",
+
+ "name", channel->name,
+ "topic", channel->topic,
+ "topic_nick", channel->topic_nick,
+ "topic_time", channel->topic_time,
+
+ "timestamp", channel->timestamp,
+ "modes", channel->modes,
+ "members", pChannelMembers,
+ "bans", pChannelBans,
+ "exempts", pChannelExempts
+ );
+}
+
+static PyObject*
+emb_get_account(UNUSED_ARG(PyObject *self), PyObject *args)
+{
+ /* Returns a python dict object with all sorts of info about an account.
+ usage: _svc.get_account(<account name>)
+ */
+ char *name;
+ struct handle_info *hi;
+
+
+ if(!PyArg_ParseTuple(args, "s", &name))
+ return NULL;
+
+ hi = get_handle_info(name);
+
+ if(!hi) {
+ PyErr_SetString(PyExc_Exception, "unknown account name");
+ return NULL;
+ }
+
+ return Py_BuildValue("{s:s,s:i,s:s,s:s,s:s"
+ ",s:s,s:s}",
+
+ "account", hi->handle,
+ "registered", hi->registered,
+ "last_seen", hi->lastseen,
+ "infoline", hi->infoline ? hi->infoline : "",
+ "email", hi->email_addr ? hi->email_addr : "",
+
+ "fakehost", hi->fakehost ? hi->fakehost : "",
+ "last_quit_host", hi->last_quit_host
+
+ /* TODO: */
+ /* users online authed to this account */
+ /* cookies */
+ /* nicks (nickserv nets only?) */
+ /* masks */
+ /* ignores */
+ /* channels */
+ );
+}
+
+static PyObject*
+emb_get_info(UNUSED_ARG(PyObject *self), UNUSED_ARG(PyObject *args))