Add new plugin "relay", new function "string_format_size" in plugin API

1 files changed, 227 insertions, 0 deletions

diff --git a/src/plugins/relay/relay-buffer.c b/src/plugins/relay/relay-buffer.c
new file mode 100644
index 000000000..36b30bc1f
--- /dev/null
+++ b/src/plugins/relay/relay-buffer.c
@@ -0,0 +1,227 @@
+/*
+ * Copyright (c) 2003-2008 by FlashCode <flashcode@flashtux.org>
+ * See README for License detail, AUTHORS for developers list.
+ *
+ * This program 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.
+ *
+ * This program 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
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+/* relay-buffer.c: display clients list on relay buffer */
+
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <time.h>
+
+#include "../weechat-plugin.h"
+#include "relay.h"
+#include "relay-client.h"
+#include "relay-config.h"
+
+
+struct t_gui_buffer *relay_buffer = NULL;
+int relay_buffer_selected_line = 0;
+
+
+/*
+ * relay_buffer_refresh: update a client in buffer and update hotlist for
+ *                       relay buffer
+ */
+
+void
+relay_buffer_refresh (const char *hotlist)
+{
+    struct t_relay_client *ptr_client, *client_selected;
+    char str_color[256], status[64], date_start[128];
+    char *str_recv, *str_sent;
+    int i, length, line;
+    struct tm *date_tmp;
+    
+    if (relay_buffer)
+    {
+        weechat_buffer_clear (relay_buffer);
+        line = 0;
+        client_selected = relay_client_search_by_number (relay_buffer_selected_line);
+        if (client_selected)
+        {
+            weechat_printf_y (relay_buffer, 0,
+                              "%s%s%s%s%s%s%s",
+                              weechat_color("green"),
+                              _("Actions (letter+enter):"),
+                              weechat_color("lightgreen"),
+                              /* disconnect */
+                              (RELAY_CLIENT_HAS_ENDED(client_selected->status)) ?
+                              "" : _("  [D] Disconnect"),
+                              /* remove */
+                              (RELAY_CLIENT_HAS_ENDED(client_selected->status)) ?
+                              _("  [R] Remove") : "",
+                              /* purge old */
+                              _("  [P] Purge finished"),
+                              /* quit */
+                              _("  [Q] Close this buffer"));
+        }
+        for (ptr_client = relay_clients; ptr_client;
+             ptr_client = ptr_client->next_client)
+        {
+            snprintf (str_color, sizeof (str_color),
+                      "%s,%s",
+                      (line == relay_buffer_selected_line) ?
+                      weechat_config_string (relay_config_color_text_selected) :
+                      weechat_config_string (relay_config_color_text),
+                      weechat_config_string (relay_config_color_text_bg));
+            
+            snprintf (status, sizeof (status),
+                      "%s", _(relay_client_status_string[ptr_client->status]));
+            length = weechat_utf8_strlen_screen (status);
+            if (length < 20)
+            {
+                for (i = 0; i < 20 - length; i++)
+                {
+                    strcat (status, " ");
+                }
+            }
+            
+            date_tmp = localtime (&(ptr_client->start_time));
+            strftime (date_start, sizeof (date_start),
+                      "%a, %d %b %Y %H:%M:%S", date_tmp);
+            
+            /* first line with status and start time */
+            weechat_printf_y (relay_buffer, (line * 2) + 2,
+                              _("%s%s[%s%s%s%s] %s (started on: %s)"),
+                              weechat_color(str_color),
+                              (line == relay_buffer_selected_line) ?
+                              "*** " : "    ",
+                              weechat_color(weechat_config_string (relay_config_color_status[ptr_client->status])),
+                              status,
+                              weechat_color ("reset"),
+                              weechat_color (str_color),
+                              ptr_client->address,
+                              date_start);
+            
+            /* second line with bytes recv/sent */
+            str_recv = weechat_string_format_size (ptr_client->bytes_recv);
+            str_sent = weechat_string_format_size (ptr_client->bytes_sent);
+            weechat_printf_y (relay_buffer, (line * 2) + 3,
+                              _("%s%-26s received: %s, sent: %s"),
+                              weechat_color(str_color),
+                              " ",
+                              (str_recv) ? str_recv : "?",
+                              (str_sent) ? str_sent : "?");
+            if (str_recv)
+                free (str_recv);
+            if (str_sent)
+                free (str_sent);
+            
+            line++;
+        }
+        if (hotlist)
+            weechat_buffer_set (relay_buffer, "hotlist", hotlist);
+    }
+}
+
+/*
+ * relay_buffer_input_cb: callback called when user send data to client list
+ *                        buffer
+ */
+
+int
+relay_buffer_input_cb (void *data, struct t_gui_buffer *buffer,
+                       const char *input_data)
+{
+    struct t_relay_client *client, *ptr_client, *next_client;
+    
+    /* make C compiler happy */
+    (void) data;
+    
+    client = relay_client_search_by_number (relay_buffer_selected_line);
+    
+    /* disconnect client */
+    if (weechat_strcasecmp (input_data, "d") == 0)
+    {
+        if (client && !RELAY_CLIENT_HAS_ENDED(client->status))
+        {
+            relay_client_disconnect (client);
+            relay_buffer_refresh (WEECHAT_HOTLIST_MESSAGE);
+        }
+    }
+    /* purge old clients */
+    else if (weechat_strcasecmp (input_data, "p") == 0)
+    {
+        ptr_client = relay_clients;
+        while (ptr_client)
+        {
+            next_client = ptr_client->next_client;
+            if (RELAY_CLIENT_HAS_ENDED(ptr_client->status))
+                relay_client_free (ptr_client);
+            ptr_client = next_client;
+        }
+        relay_buffer_refresh (WEECHAT_HOTLIST_MESSAGE);
+    }
+    /* quit relay buffer (close it) */
+    else if (weechat_strcasecmp (input_data, "q") == 0)
+    {
+        weechat_buffer_close (buffer);
+    }
+    /* remove client */
+    else if (weechat_strcasecmp (input_data, "r") == 0)
+    {
+        if (client && RELAY_CLIENT_HAS_ENDED(client->status))
+        {
+            relay_client_free (client);
+            relay_buffer_refresh (WEECHAT_HOTLIST_MESSAGE);
+        }
+    }
+    
+    return WEECHAT_RC_OK;
+}
+
+/*
+ * relay_buffer_close_cb: callback called when relay buffer is closed
+ */
+
+int
+relay_buffer_close_cb (void *data, struct t_gui_buffer *buffer)
+{
+    /* make C compiler happy */
+    (void) data;
+    (void) buffer;
+    
+    relay_buffer = NULL;
+    
+    return WEECHAT_RC_OK;
+}
+
+/*
+ * relay_buffer_open: open relay buffer (to display list of clients)
+ */
+
+void
+relay_buffer_open ()
+{
+    if (!relay_buffer)
+    {
+        relay_buffer = weechat_buffer_new ("relay.list",
+                                           &relay_buffer_input_cb, NULL,
+                                           &relay_buffer_close_cb, NULL);
+        
+        /* failed to create buffer ? then exit */
+        if (!relay_buffer)
+            return;
+        
+        weechat_buffer_set (relay_buffer, "type", "free");
+        weechat_buffer_set (relay_buffer, "title", _("List of clients for relay"));
+        weechat_buffer_set (relay_buffer, "key_bind_meta2-A", "/relay up");
+        weechat_buffer_set (relay_buffer, "key_bind_meta2-B", "/relay down");
+    }
+}