In case of a stream error, print the stanza.
[cascardo/pubsub-bot.git] / status.c
index 78ccd6f..6af0d44 100644 (file)
--- a/status.c
+++ b/status.c
 
 #include <iksemel.h>
 #include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
 
-struct udata
+iks *
+createiq (char *type, char *to, char *qnam, char *xmlns, iks **query)
 {
-  iksparser *parser;
-  int auth;
-};
+  static int id = 0;
+  char sid[32];;
+  iks *iq;
+  snprintf (sid, 32, "ps%d", id++);
+  iq = iks_new ("iq");
+  iks_insert_attrib (iq, "type", type);
+  iks_insert_attrib (iq, "to", to);
+  iks_insert_attrib (iq, "id", sid);
+  *query = iks_insert (iq, qnam);
+  iks_insert_attrib (*query, "xmlns", xmlns);
+  return iq;
+}
 
-enum
+void
+catnode (iksparser *parser, char *node)
 {
-  AUTH_NONE,
-  AUTH_TLS,
-  AUTH_SASL,
-  AUTH_BIND,
-  AUTH_SESSION
-} AUTH_STATE;
+  iks *iq;
+  iks *query;
+  iq = createiq ("get", "pubsub.jabber-br.org", "query",
+                 "http://jabber.org/protocol/disco#info", &query);
+  if (node != NULL)
+    iks_insert_attrib (query, "node", node);
+  iks_send (parser, iq);
+  iks_delete (iq);
+}
 
-static int
-hook (void *data, int type, iks *node)
+void
+listnode (iksparser *parser, char *node)
+{
+  iks *iq;
+  iks *query;
+  iq = createiq ("get", "pubsub.jabber-br.org", "query",
+                 "http://jabber.org/protocol/disco#items", &query);
+  if (node != NULL)
+    iks_insert_attrib (query, "node", node);
+  iks_send (parser, iq);
+  iks_delete (iq);
+}
+
+void
+createnode (iksparser *parser, char *node)
+{
+  iks *iq;
+  iks *query;
+  iq = createiq ("set", "pubsub.jabber-br.org", "pubsub",
+                 "http://jabber.org/protocol/pubsub", &query);
+  iks_insert_attrib (iks_insert (query, "create"), "node", node);
+  iks_send (parser, iq);
+  iks_delete (iq);
+}
+
+void
+getnode (iksparser *parser, char *node)
+{
+  iks *iq;
+  iks *query;
+  iq = createiq ("get", "pubsub.jabber-br.org", "pubsub",
+                 "http://jabber.org/protocol/pubsub", &query);
+  iks_insert_attrib (iks_insert (query, "items"), "node", node);
+  iks_send (parser, iq);
+  iks_delete (iq);
+}
+
+void
+vcard (iksparser *parser)
+{
+  iks *iq;
+  iks *query;
+  iq = createiq ("get", "pubsub.jabber-br.org", "vCard", "vcard-temp", &query);
+  iks_send (parser, iq);
+  iks_delete (iq);
+}
+
+iks *
+createmood (char *line)
+{
+  iks *mood;
+  mood = iks_new ("mood");
+  iks_insert_attrib (mood, "xmlns", "http://jabber.org/protocol/mood");
+  iks_insert (mood, line);
+  return mood;
+}
+
+void
+pushmood (iksparser *parser, char *node, char *line)
+{
+  iks *iq;
+  iks *query;
+  iks *publish;
+  iks *item;
+  iks *mood;
+  iq = createiq ("set", "pubsub.jabber-br.org", "pubsub",
+                 "http://jabber.org/protocol/pubsub", &query);
+  publish = iks_insert (query, "publish");
+  iks_insert_attrib (publish, "node", node);
+  item = iks_insert (publish, "item");
+  mood = createmood (line);
+  iks_insert_node (item, mood);
+  iks_send (parser, iq);
+  iks_delete (iq);
+}
+
+void
+process_mood (iksparser *parser, char *cmdline)
+{
+  char *cmd;
+  char *orig_cmdline;
+  orig_cmdline = cmdline = strdup (cmdline);
+  cmd = strsep (&cmdline, " ");
+  if (!strcmp (cmd, "ls"))
+    listnode (parser, cmdline);
+  else if (!strcmp (cmd, "cat"))
+    catnode (parser, cmdline);
+  else if (!strcmp (cmd, "vcard"))
+    vcard (parser);
+  else if (!strcmp (cmd, "create"))
+    createnode (parser, cmdline);
+  else if (!strcmp (cmd, "get"))
+    getnode (parser, cmdline);
+  else if (!strcmp (cmd, "push"))
+    {
+      char *node;
+      node = strsep (&cmdline, " ");
+      pushmood (parser, node, cmdline);
+    }
+  free (orig_cmdline);
+}
+
+int
+xmpp_session_hook (iksparser *parser, iks *node)
+{
+  iks *iq;
+  iq = iks_new ("iq");
+  iks_insert_attrib (iq, "type", "set");
+  iks_insert_attrib (iq, "id", "session1");
+  iks_insert_attrib (iks_insert (iq, "session"),
+                     "xmlns", "urn:ietf:params:xml:ns:xmpp-session");
+  iks_send (parser, iq);
+  iks_delete (iq);
+  return 0;
+}
+
+int
+xmpp_initial_presence_hook (iksparser *parser, iks *node)
 {
-  iks *sub;
-  struct udata *udata;
-  udata = (struct udata *) data;
-  if (type == IKS_NODE_START)
+  iks *pres;
+  pres = iks_make_pres (IKS_SHOW_AVAILABLE, "Microblogging here!");
+  iks_send (parser, pres);
+  iks_delete (pres);
+  return 0;
+}
+
+int
+xmpp_id_hook (iksparser *parser, iks *node, char *id)
+{
+  if (!iks_strcmp (id, "bind1"))
+    {
+      xmpp_session_hook (parser, node);
+      return 0;
+    }
+  else if (!iks_strcmp (id, "session1"))
     {
-      if (!iks_is_secure (udata->parser) || udata->auth == AUTH_NONE)
+      xmpp_initial_presence_hook (parser, node);
+      return 0;
+    }
+  else if (!iks_strncmp (id, "ps", 2))
+    {
+      printf ("%s\n", iks_string (iks_stack (node), node));
+    }
+  return 1;
+}
+
+int
+xmpp_ns_hook (iksparser *parser, iks *node, char *ns)
+{
+  return 1;
+}
+
+int
+xmpp_iq_error (iksparser *parser, iks *node)
+{
+  iks *enode;
+  char *to;
+  char *from;
+  if (!iks_strcmp (iks_find_attrib (node, "type"), "error"))
+    return 1;
+  to = iks_find_attrib (node, "to");
+  from = iks_find_attrib (node, "from");
+  if (to)
+    iks_insert_attrib (node, "from", to);
+  else
+    iks_insert_attrib (node, "from", NULL);
+  if (from)
+    iks_insert_attrib (node, "to", from);
+  else
+    iks_insert_attrib (node, "to", NULL);
+  iks_insert_attrib (node, "type", "error");
+  enode = iks_insert (node, "error");
+  iks_insert_attrib (enode, "type", "cancel");
+  iks_insert_attrib (iks_insert (enode, "feature-not-implemented"),
+                     "xmlns", "urn:ietf:params:xml:ns:xmpp-stanzas");
+  iks_send (parser, node);
+  return 0;
+}
+
+int
+xmpp_tls_hook (iksparser *parser, iks *node)
+{
+  iks_start_tls (parser);
+  return 0;
+}
+
+int
+xmpp_sasl_hook (iksparser *parser, iks* node)
+{
+  iks_start_sasl (parser, IKS_SASL_DIGEST_MD5, "pubsub", "pubsub");
+  return 0;
+}
+
+int
+xmpp_bind_hook (iksparser *parser, iks *node)
+{
+  iks *iq;
+  iq = iks_new ("iq");
+  iks_insert_attrib (iq, "type", "set");
+  iks_insert_attrib (iq, "id", "bind1");
+  iks_insert_attrib (iks_insert (iq, "bind"),
+                     "xmlns", "urn:ietf:params:xml:ns:xmpp-bind");
+  iks_send (parser, iq);
+  iks_delete (iq);
+  return 0;
+}
+
+int
+xmpp_features_hook (iksparser *parser, iks *node)
+{
+  iks *feat;
+  char *ns;
+  for (feat = iks_child (node); feat != NULL; feat = iks_next (feat))
+    {
+      ns = iks_find_attrib (feat, "xmlns");
+      if (!iks_strcmp (ns, "urn:ietf:params:xml:ns:xmpp-tls"))
         {
-          iks_start_tls (udata->parser);
-          udata->auth = AUTH_TLS;
+          xmpp_tls_hook (parser, node);
+          return 0;
         }
-      else if (udata->auth == AUTH_TLS)
+      else if (!iks_strcmp (ns, "urn:ietf:params:xml:ns:xmpp-sasl"))
         {
-          iks_start_sasl (udata->parser, IKS_SASL_PLAIN, "pubsub", "pubsub");
+          xmpp_sasl_hook (parser, node);
+          return 0;
         }
-      else if (udata->auth == AUTH_SASL)
+      else if (!iks_strcmp (ns, "urn:ietf:params:xml:ns:xmpp-bind"))
         {
-          iks_send_raw (udata->parser, "<iq type='set' id='boo'><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/></iq>");
+          xmpp_bind_hook (parser, node);
+          return 0;
         }
     }
-  else
+  return 1;
+}
+
+int
+xmpp_other_hook (iksparser *parser, iks *node, char *ns)
+{
+  if (!iks_strcmp (ns, "urn:ietf:params:xml:ns:xmpp-sasl"))
     {
-      if (!iks_strcmp (iks_find_attrib (node, "id"), "boo"))
-        {
-          udata->auth = AUTH_BIND;
-          iks_send_raw (udata->parser, "<iq type='set' id='goo'><session xmlns='urn:ietf:params:xml:ns:xmpp-session'/></iq>");
-        }
-      else if (!iks_strcmp (iks_find_attrib (node, "id"), "goo"))
+      if (!iks_strcmp (iks_name (node), "success"))
+        iks_send_header (parser, "jabber-br.org");
+      return 0;
+    }
+  return 1;
+}
+
+const char * jid = "cascardo@jabber-br.org";
+
+static int
+hook (void *data, int type, iks *node)
+{
+  char *name;
+  char *id;
+  char *ns;
+  char *iqns;
+  iksparser *parser;
+  parser = *(iksparser **) data;
+  name = iks_name (node);
+  id = iks_find_attrib (node, "id");
+  ns = iks_find_attrib (node, "xmlns"); 
+  iqns = iks_find_attrib (iks_child (node), "xmlns"); 
+  if (!iks_strcmp (name, "message"))
+    {
+      char *from;
+      from = iks_find_attrib (node, "from");
+      if (!iks_strncmp (from, jid, iks_strlen (jid)))
         {
-          printf ("goooooo\n");
-          sub = iks_make_s10n (IKS_TYPE_SUBSCRIBED, "cascardo@jabber-br.org", "");
-          iks_send (udata->parser, sub);
-          iks_delete (sub);
-          sub = iks_make_pres (IKS_SHOW_AVAILABLE, "here");
-          iks_insert_attrib (sub, "to", "cascardo@jabber-br.org");
-          iks_send (udata->parser, sub);
-          iks_delete (sub);
+          char *body = iks_find_cdata (node, "body");
+          if (body != NULL)
+            process_mood (parser, body);
         }
-      if (!iks_strcmp (iks_name (node), "success"))
+      else
         {
-          iks_send_header (udata->parser, "jabber-br.org");
-          udata->auth = AUTH_SASL;
+          printf ("%s is not authorized\n", from);
         }
-      printf ("%s\n", iks_string (iks_stack (node), node));
+      return IKS_OK;
+    }
+  else if (!iks_strcmp (name, "presence"))
+    {
+      char *from;
+      from = iks_find_attrib (node, "from");
+      printf ("I sense a disturbance in the force: %s!\n", from);
+    }
+  else if (!iks_strcmp (name, "iq"))
+    {
+      if (xmpp_id_hook (parser, node, id) == 0)
+        return IKS_OK;
+      if (xmpp_ns_hook (parser, node, iqns) == 0)
+        return IKS_OK;
+      xmpp_iq_error (parser, node);
+    }
+  else if (!iks_strcmp (name, "stream:features"))
+    {
+      if (xmpp_features_hook (parser, node) == 0)
+        return IKS_OK;
+    }
+  else if (!iks_strcmp (name, "stream:error"))
+    {
+      printf ("streamerror: %s\n", iks_string (iks_stack (node), node));
+    }
+  else
+    {
+      if (xmpp_other_hook (parser, node, ns) == 0)
+        return IKS_OK;
+      printf ("No no: %s\n", name);
     }
   return IKS_OK;
 }
@@ -88,11 +363,10 @@ hook (void *data, int type, iks *node)
 int
 main (int argc, char **argv)
 {
-  struct udata udata;
-  udata.auth = 0;
-  udata.parser = iks_stream_new ("jabber:client", &udata, hook);
-  iks_connect_tcp (udata.parser, "jabber-br.org", 5222);
+  iksparser *parser;
+  parser = iks_stream_new ("jabber:client", &parser, hook);
+  iks_connect_tcp (parser, "jabber-br.org", 5222);
   while (1)
-    iks_recv (udata.parser, -1);
+    iks_recv (parser, -1);
   return 0;
 }