Add support for a %quit command

diff --git a/lm-echo.c b/lm-echo.c
index e723599f0b1d6c223190e7c686e7f50cc70afb76..d92188bd14510e66de361b55ad92a660bbda0667 100644 (file)
--- a/lm-echo.c
+++ b/lm-echo.c
@@ -26,9 +26,13 @@
 #endif
  
 typedef struct {
-    gchar *name;
-    gchar *passwd;
+    gchar      *name;
+    gchar      *passwd;
 } UserInfo;
+
+typedef struct {
+    GMainLoop  *main_loop;
+} msg_data_t;
  
 static void
 free_user_info (UserInfo *info)
@@ -39,10 +43,11 @@ free_user_info (UserInfo *info)
     g_free (info);
 }
  
- 
 static void
 authentication_cb (LmConnection *connection, gboolean result, gpointer ud)
 {
+    gchar *string;
+
     g_print ("Auth: %d\n", result);
     free_user_info ((UserInfo *) ud);
  
@@ -52,7 +57,9 @@ authentication_cb (LmConnection *connection, gboolean result, gpointer ud)
        m = lm_message_new_with_sub_type (NULL,
                                          LM_MESSAGE_TYPE_PRESENCE,
                                          LM_MESSAGE_SUB_TYPE_AVAILABLE);
-       g_print (":: %s\n", lm_message_node_to_string (m->node));
+       string = lm_message_node_to_string (m->node);
+       g_print (":: %s\n", string);
+       free (string);
                  
        lm_connection_send (connection, m, NULL);
        lm_message_unref (m);
@@ -69,26 +76,19 @@ connection_open_cb (LmConnection *connection, gboolean result, UserInfo *info)
                                authentication_cb, info, FALSE,  NULL);
     g_print ("Sent auth message\n");
 }
- 
-static LmHandlerResult
-handle_messages (LmMessageHandler *handler,
-                 LmConnection     *connection,
-                 LmMessage        *m,
-                 gpointer          user_data)
+
+static void
+send_reply (LmConnection       *connection,
+           const char          *peer,
+           const char          *message)
 {
     LmMessage *reply;
     gboolean result;
     GError *error = NULL;
-    LmMessageNode *body;
 
-    g_print ("Incoming message from: %s\n",
-            lm_message_node_get_attribute (m->node, "from"));
+    reply = lm_message_new (peer, LM_MESSAGE_TYPE_MESSAGE);
 
-    reply = lm_message_new (lm_message_node_get_attribute (m->node, "from"),
-                           LM_MESSAGE_TYPE_MESSAGE);
-    body = lm_message_node_get_child (m->node, "body");
-    lm_message_node_add_child (reply->node, "body",
-                              lm_message_node_get_value (body));
+    lm_message_node_add_child (reply->node, "body", message);
 
     result = lm_connection_send (connection, reply, &error);
     lm_message_unref (reply);
@@ -96,19 +96,65 @@ handle_messages (LmMessageHandler *handler,
     if (! result) {
        g_error ("lm_connection_send failed");
     }
+}
+
+static void
+handle_command (LmConnection   *connection,
+               const char      *peer,
+               const char      *command,
+               gpointer         closure)
+{
+    msg_data_t *data = closure;
+    char *error;
+
+    if (strcmp (command, "quit") == 0) {
+       g_main_loop_quit (data->main_loop);
+       return;
+    }
+
+    error = g_strdup_printf ("Unknown command: '%s'", command);
+    send_reply (connection, peer, error);
+    free (error);
+}
  
+static LmHandlerResult
+handle_messages (LmMessageHandler *handler,
+                 LmConnection     *connection,
+                 LmMessage        *m,
+                 gpointer          closure)
+{
+    LmMessageNode *body;
+    const char *peer;
+    const char *body_str;
+
+    peer = lm_message_node_get_attribute (m->node, "from");
+
+
+    g_print ("Incoming message from: %s\n", peer);
+
+    body = lm_message_node_get_child (m->node, "body");
+    if (body) {
+       body_str = lm_message_node_get_value (body);
+
+       if (body_str && body_str[0] == '%')
+           handle_command (connection, peer, body_str + 1, closure);
+       else
+           send_reply (connection, peer, body_str);
+    }
+       
     return LM_HANDLER_RESULT_REMOVE_MESSAGE;
 }
 int
 main (int argc, char **argv)
 {
-    GMainLoop        *main_loop;
     LmConnection     *connection;
     LmMessageHandler *handler;
     gboolean          result;
     UserInfo         *info;
     gchar            *jid;
-                                                                                
+    msg_data_t       data;
+    GError          *error;
+
     if (argc < 6) {
        g_print ("Usage: %s <server> <username> <password> <connectserver> <connectport>\n", argv[0]);
        return 1;
@@ -122,7 +168,9 @@ main (int argc, char **argv)
 
     lm_connection_set_port (connection, strtol (argv[5], (char **) NULL, 10));
 
-    handler = lm_message_handler_new (handle_messages, NULL, NULL);
+    data.main_loop = g_main_loop_new (NULL, FALSE);
+
+    handler = lm_message_handler_new (handle_messages, &data, NULL);
     lm_connection_register_message_handler (connection, handler,
                                            LM_MESSAGE_TYPE_MESSAGE,
                                            LM_HANDLER_PRIORITY_NORMAL);
@@ -143,8 +191,12 @@ main (int argc, char **argv)
        g_print ("Returned from the connection_open\n");
     }
                                                                                 
-    main_loop = g_main_loop_new (NULL, FALSE);
-    g_main_loop_run (main_loop);
+    g_main_loop_run (data.main_loop);
+
+    g_main_loop_unref (data.main_loop);
+
+    lm_connection_close (connection, &error);
+    lm_connection_unref (connection);
                                                                                 
     return 0;
 }