inital commit

main.cpp

@@ -0,0 +1,116 @@
+#include <iostream>
+#include <string.h>
+#include <libnotify/notify.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include "Socket.h"
+#include "argpopt.h"
+
+void sigHandler(int sig)
+{
+    exit(0);
+}
+
+bool showMessage(const std::string& title, const std::string& body = "")
+{
+    NotifyNotification* message = notify_notification_new (title.c_str(), body.c_str(), nullptr);
+    bool ret = notify_notification_show (message, NULL);
+    g_object_unref(G_OBJECT(message));
+    return ret;
+}
+
+void processLineBuffer(std::string& buffer)
+{
+    if(buffer.find("bcst:") == 0) buffer.erase(0, 6);
+    if(buffer.size() > 5 && buffer.find("MESG") == 0)
+    {
+        size_t titleEnd  = buffer.size();
+        size_t bodyBegin = buffer.find("BODY");
+        if(bodyBegin != std::string::npos && bodyBegin > 5)
+        {
+          titleEnd = bodyBegin - 1;
+        }
+        
+        const std::string body = (bodyBegin == std::string::npos && buffer.begin()+bodyBegin+5 < buffer.end()) ? 
+        "" : std::string(buffer.begin()+bodyBegin+5, buffer.end());
+        
+        const std::string title(buffer.begin()+5, buffer.begin()+titleEnd);
+        showMessage(title, body);
+    }
+    
+}
+
+static TCPSocket* socket = nullptr;
+
+void exitHandler()
+{
+    notify_uninit();
+    if(socket)delete socket;
+}
+
+bool reconnect(std::string host,  unsigned short port)
+{
+    try
+    {
+        if(socket)delete socket;
+        socket = new TCPSocket(host, port);
+    }
+    catch(SocketException &e)
+    {
+        std::cerr<<e.what()<<std::endl;
+        return false;
+    }
+    return true;
+}
+
+int main(int argc, char* argv[])
+{
+	Config config;
+    
+    notify_init("Message");
+    
+    if( atexit(exitHandler) != 0 ) 
+    {
+        std::cerr<<"can not register atexit handler\n";
+        return -1;
+    }    
+    argp_parse(&argp, argc, argv, 0, 0, &config);
+    
+    std::cout<<argp_program_version<<std::endl;
+	
+    std::string lineBuffer;
+    
+    signal(SIGINT, sigHandler);
+    signal(SIGTERM, sigHandler);
+    char buffer[256];
+    
+    if(!reconnect(config.hostName, config.port)) return 1;
+	
+	while(true)
+	{
+        int len = socket->recv(buffer, sizeof(buffer));
+        if(len <= 0) 
+        {
+            std::cerr<<"socket error\n";
+            int i = 0;
+            for(; i < 30 && reconnect(config.hostName, config.port) == false; ++i) sleep(10);
+            if(i >= 10)
+            {
+                std::cerr<<"reconnect failed\n";
+                return 1;
+            }
+        }
+        for(int i = 0; i<len; ++i)
+        {
+            if(buffer[i] != '\n') lineBuffer.push_back(buffer[i]);
+            else 
+            {
+                processLineBuffer(lineBuffer);
+                lineBuffer.clear();
+            }
+            
+        }
+	}
+	
+	return 0;
+}