Merge branch 'master' of git://localhost/acess2
[tpg/acess2.git] / Usermode / Applications / irc_src / main.c
index 3f46d15..09e1164 100644 (file)
@@ -7,6 +7,7 @@
 #include <string.h>\r
 #include <net.h>\r
 #include <readline.h>\r
+#include <acess/devices/pty.h>\r
 \r
 // === TYPES ===\r
 typedef struct sServer {\r
@@ -64,7 +65,7 @@ void  Redraw_Screen(void);
 \r
  int   ProcessIncoming(tServer *Server);\r
 // --- Helpers\r
- int   SetCursorPos(int Row, int Col);\r
+void   SetCursorPos(int Row, int Col);\r
  int   writef(int FD, const char *Format, ...);\r
  int   OpenTCP(const char *AddressString, short PortNumber);\r
 char   *GetValue(char *Str, int *Ofs);\r
@@ -78,7 +79,7 @@ char  *gsNickname = "acess";
 tServer        *gpServers;\r
 tWindow        gWindow_Status = {\r
        NULL, NULL, NULL,       // No next, empty list, no server\r
-       0, ""   // No activity, empty name (rendered as status)\r
+       0, {""} // No activity, empty name (rendered as status)\r
 };\r
 tWindow        *gpWindows = &gWindow_Status;\r
 tWindow        *gpCurrentWindow = &gWindow_Status;\r
@@ -102,8 +103,27 @@ int main(int argc, const char *argv[], const char *envp[])
        \r
        atexit(ExitHandler);\r
        \r
-       giTerminal_Width = ioctl(1, 5, NULL);   // getset_width\r
-       giTerminal_Height = ioctl(1, 6, NULL);  // getset_height\r
+<<<<<<< HEAD
+       if( _SysIOCtl(1, DRV_IOCTL_TYPE, NULL) != DRV_TYPE_TERMINAL ) {\r
+               printf(stderr, "note: assuming 80x25, can't get terminal dimensions\n");\r
+               giTerminal_Width = 80;\r
+               giTerminal_Height = 25;\r
+       }\r
+       else {\r
+               struct ptydims  dims;\r
+               _SysIOCtl(1, PTY_IOCTL_GETDIMS, &dims);\r
+=======
+       {\r
+               struct ptydims  dims;\r
+               if( _SysIOCtl(1, PTY_IOCTL_GETDIMS, &dims) ) {\r
+                       perror("Can't get terminal dimensions");\r
+                       return 1;\r
+               }\r
+               \r
+>>>>>>> e7a76b0d8a0cc6aa77966509780973a6f8216ef7
+               giTerminal_Width = dims.W;\r
+               giTerminal_Height = dims.H;\r
+       }\r
        \r
        printf("\x1B[?1047h");\r
        printf("\x1B[%i;%ir", 0, giTerminal_Height-1);\r
@@ -113,16 +133,19 @@ int main(int argc, const char *argv[], const char *envp[])
        \r
        // HACK: Static server entry\r
        // UCC (University [of Western Australia] Computer Club) IRC Server\r
-//     gWindow_Status.Server = Server_Connect( "UCC", "130.95.13.18", 6667 );\r
-       gWindow_Status.Server = Server_Connect( "Freenode", "89.16.176.16", 6667 );\r
+       gWindow_Status.Server = Server_Connect( "UCC", "130.95.13.18", 6667 );\r
+       // Freenode (#osdev)\r
+//     gWindow_Status.Server = Server_Connect( "Freenode", "89.16.176.16", 6667 );\r
 //     gWindow_Status.Server = Server_Connect( "Host", "10.0.2.2", 6667 );\r
-//     gWindow_Status.Server = Server_Connect( "BitlBee", "192.168.1.34", 6667 );\r
+       // Local server\r
+//     gWindow_Status.Server = Server_Connect( "BitlBee", "192.168.1.39", 6667 );\r
        \r
        if( !gWindow_Status.Server )\r
                return -1;\r
        \r
        SetCursorPos(giTerminal_Height-1, 0);\r
        printf("[(status)] ");\r
+       fflush(stdout);\r
        readline_info = Readline_Init(1);\r
        \r
        for( ;; )\r
@@ -144,7 +167,7 @@ int main(int argc, const char *argv[], const char *envp[])
                                maxFD = srv->FD;\r
                }\r
                \r
-               rv = select(maxFD+1, &readfds, 0, &errorfds, NULL);\r
+               rv = _SysSelect(maxFD+1, &readfds, 0, &errorfds, NULL, 0);\r
                if( rv == -1 )  break;\r
                \r
                if(FD_ISSET(0, &readfds))\r
@@ -165,6 +188,7 @@ int main(int argc, const char *argv[], const char *envp[])
                                        printf("[%s:%s] ", gpCurrentWindow->Server->Name, gpCurrentWindow->Name);\r
                                else\r
                                        printf("[(status)] ");\r
+                               fflush(stdout);\r
                        }\r
                }\r
                \r
@@ -192,7 +216,7 @@ int main(int argc, const char *argv[], const char *envp[])
        {\r
                tServer *srv;\r
                for( srv = gpServers; srv; srv = srv->Next )\r
-                       close(srv->FD);\r
+                       _SysClose(srv->FD);\r
        }\r
        return 0;\r
 }\r
@@ -205,6 +229,82 @@ int ParseArguments(int argc, const char *argv[])
        return 0;\r
 }\r
 \r
+\r
+void Cmd_join(char *ArgString)\r
+{\r
+        int    pos=0;\r
+       char    *channel_name = GetValue(ArgString, &pos);\r
+       \r
+       if( gpCurrentWindow->Server )\r
+       {\r
+               writef(gpCurrentWindow->Server->FD, "JOIN :%s\n", channel_name);\r
+       }\r
+}\r
+\r
+void Cmd_quit(char *ArgString)\r
+{\r
+       const char *quit_message = ArgString;\r
+       if( quit_message == NULL || quit_message[0] == '\0' )\r
+               quit_message = "/quit - Acess2 IRC Client";\r
+       \r
+       for( tServer *srv = gpServers; srv; srv = srv->Next )\r
+       {\r
+               writef(srv->FD, "QUIT :%s\n", quit_message);\r
+       }\r
+       \r
+       exit(0);\r
+}\r
+\r
+void Cmd_window(char *ArgString)\r
+{\r
+        int    pos = 0;\r
+       char    *window_id = GetValue(ArgString, &pos);\r
+        int    window_num = atoi(window_id);\r
+       \r
+       if( window_num > 0 )\r
+       {\r
+               tWindow *win;\r
+               window_num --;  // Move to base 0\r
+               // Get `window_num`th window\r
+               for( win = gpWindows; win && window_num--; win = win->Next );\r
+               if( win ) {\r
+                       gpCurrentWindow = win;\r
+                       Redraw_Screen();\r
+               }\r
+               // Otherwise, silently ignore\r
+       }\r
+       else\r
+       {\r
+               window_num = 1;\r
+               for( tWindow *win = gpWindows; win; win = win->Next, window_num ++ )\r
+               {\r
+                       if( win->Name[0] ) {\r
+                               Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "",\r
+                                       "%i: %s/%s", window_num, win->Server->Name, win->Name);\r
+                       }\r
+                       else {\r
+                               Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "",\r
+                                       "%i: (status)", window_num);\r
+                       }\r
+               }\r
+       }\r
+}\r
+\r
+const struct {\r
+       const char *Name;\r
+       void    (*Fcn)(char *ArgString);\r
+} caCommands[] = {\r
+       {"join", Cmd_join},\r
+       {"quit", Cmd_quit},\r
+       {"window", Cmd_window},\r
+       {"win",    Cmd_window},\r
+       {"w",      Cmd_window},\r
+};\r
+const int ciNumCommands = sizeof(caCommands)/sizeof(caCommands[0]);\r
+\r
+/**\r
+ * \brief Handle a line from the prompt\r
+ */\r
 int ParseUserCommand(char *String)\r
 {\r
        if( String[0] == '/' )\r
@@ -212,65 +312,19 @@ int ParseUserCommand(char *String)
                char    *command;\r
                 int    pos = 0;\r
                \r
-               command = GetValue(String, &pos);\r
-               \r
-               if( strcmp(command, "/join") == 0 )\r
+               command = GetValue(String, &pos)+1;\r
+\r
+               // TODO: Prefix matches\r
+                int    cmdIdx = -1;\r
+               for( int i = 0; i < ciNumCommands; i ++ )\r
                {\r
-                       char    *channel_name = GetValue(String, &pos);\r
-                       \r
-                       if( gpCurrentWindow->Server )\r
-                       {\r
-                               writef(gpCurrentWindow->Server->FD, "JOIN :%s\n",  channel_name);\r
+                       if( strcmp(command, caCommands[i].Name) == 0 ) {\r
+                               cmdIdx = i;\r
+                               break;\r
                        }\r
                }\r
-               else if( strcmp(command, "/quit") == 0 )\r
-               {\r
-                       char    *quit_message = String + pos;\r
-                       tServer *srv;\r
-                       \r
-                       if( quit_message == NULL || quit_message[0] == '\0' )\r
-                               quit_message = "/quit - Acess2 IRC Client";\r
-                       \r
-                       for( srv = gpServers; srv; srv = srv->Next )\r
-                       {\r
-                               writef(srv->FD, "QUIT :%s\n", quit_message);\r
-                       }\r
-                       \r
-                       exit(0);\r
-               }\r
-               else if( strcmp(command, "/window") == 0 || strcmp(command, "/win") == 0 || strcmp(command, "/w") == 0 )\r
-               {\r
-                       char    *window_id = GetValue(String, &pos);\r
-                        int    window_num = atoi(window_id);\r
-                       \r
-                       if( window_num > 0 )\r
-                       {\r
-                               tWindow *win;\r
-                               window_num --;  // Move to base 0\r
-                               // Get `window_num`th window\r
-                               for( win = gpWindows; win && window_num--; win = win->Next );\r
-                               if( win ) {\r
-                                       gpCurrentWindow = win;\r
-                                       Redraw_Screen();\r
-                               }\r
-                               // Otherwise, silently ignore\r
-                       }\r
-                       else\r
-                       {\r
-                               tWindow *win;\r
-                               window_num = 1;\r
-                               for( win = gpWindows; win; win = win->Next, window_num ++ )\r
-                               {\r
-                                       if( win->Name[0] ) {\r
-                                               Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "",\r
-                                                       "%i: %s/%s", window_num, win->Server->Name, win->Name);\r
-                                       }\r
-                                       else {\r
-                                               Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "",\r
-                                                       "%i: (status)", window_num);\r
-                                       }\r
-                               }\r
-                       }\r
+               if( cmdIdx != -1 ) {\r
+                       caCommands[cmdIdx].Fcn(String+pos);\r
                }\r
                else\r
                {\r
@@ -335,7 +389,7 @@ tMessage *Message_AppendF(tServer *Server, int Type, const char *Source, const c
        va_list args;\r
         int    len;\r
        va_start(args, Message);\r
-       len = vsnprintf(NULL, 1000, Message, args);\r
+       len = vsnprintf(NULL, 0, Message, args);\r
        {\r
                char    buf[len+1];\r
                vsnprintf(buf, len+1, Message, args);\r
@@ -345,16 +399,15 @@ tMessage *Message_AppendF(tServer *Server, int Type, const char *Source, const c
 \r
 tMessage *Message_Append(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message)\r
 {\r
-       tMessage        *ret;\r
        tWindow *win = NULL;\r
         int    msgLen = strlen(Message);\r
        \r
-       // NULL servers are internal messages\r
+       // Server==NULL indicates an internal message\r
        if( Server == NULL || Source[0] == '\0' )\r
        {\r
                win = &gWindow_Status;\r
        }\r
-       // Determine if it's a channel or PM message\r
+       // Determine if it's a channel or PM\r
        else if( Dest[0] == '#' || Dest[0] == '&' )     // TODO: Better determining here\r
        {\r
                for(win = gpWindows; win; win = win->Next)\r
@@ -406,7 +459,10 @@ tMessage *Message_Append(tServer *Server, int Type, const char *Source, const ch
                        win = Window_Create(Server, Dest);\r
                }\r
        }\r
-       \r
+\r
+       // Create message cache \r
+       _SysDebug("Win (%s) msg: <%s> %s", win->Name, Source, Message);\r
+       tMessage        *ret;\r
        ret = malloc( sizeof(tMessage) + msgLen + 1 + strlen(Source) + 1 );\r
        ret->Source = ret->Data + msgLen + 1;\r
        strcpy(ret->Source, Source);\r
@@ -414,29 +470,28 @@ tMessage *Message_Append(tServer *Server, int Type, const char *Source, const ch
        ret->Type = Type;\r
        ret->Server = Server;\r
        \r
-       // TODO: Append to window message list\r
+       // Append to window message list\r
        ret->Next = win->Messages;\r
        win->Messages = ret;\r
        \r
-       //TODO: Set location\r
-       \r
+       // Print now if current window\r
+       if( win == gpCurrentWindow )\r
        {\r
-               int pos = SetCursorPos(giTerminal_Height-2, 0);\r
-               if( win == gpCurrentWindow ) {\r
-                        int    prefixlen = strlen(Source) + 3;\r
-                        int    avail = giTerminal_Width - prefixlen;\r
-                        int    msglen = strlen(Message);\r
-                       printf("\x1B[T");       // Scroll down 1 (free space below)\r
-                       printf("[%s] %.*s\n", Source, avail, Message);\r
-                       while( msglen > avail ) {\r
-                               msglen -= avail;\r
-                               Message += avail;\r
-                               printf("\x1B[T");\r
-                               SetCursorPos(giTerminal_Height-2, prefixlen);\r
-                               printf("%.*s\n", avail, Message);\r
-                       }\r
+               printf("\x1b[s");\r
+               SetCursorPos(giTerminal_Height-2, 0);\r
+                int    prefixlen = strlen(Source) + 3;\r
+                int    avail = giTerminal_Width - prefixlen;\r
+                int    msglen = strlen(Message);\r
+               printf("\x1B[T");       // Scroll down 1 (free space below)\r
+               printf("[%s] %.*s\n", Source, avail, Message);\r
+               while( msglen > avail ) {\r
+                       msglen -= avail;\r
+                       Message += avail;\r
+                       printf("\x1B[T");\r
+                       SetCursorPos(giTerminal_Height-2, prefixlen);\r
+                       printf("%.*s\n", avail, Message);\r
                }\r
-               SetCursorPos(-1, pos);\r
+               printf("\x1b[u");\r
        }\r
        \r
        return ret;\r
@@ -519,11 +574,12 @@ void ParseServerLine(tServer *Server, char *Line)
         int    pos = 0;\r
        char    *ident, *cmd;\r
 \r
-       _SysDebug("Server %s: Line = %s", Server->Name, Line);  \r
+//     _SysDebug("[%s] %s", Server->Name, Line);       \r
        \r
        // Message?\r
        if( *Line == ':' )\r
        {\r
+               pos ++;\r
                ident = GetValue(Line, &pos);   // Ident (user or server)\r
                cmd = GetValue(Line, &pos);\r
                \r
@@ -547,20 +603,32 @@ void ParseServerLine(tServer *Server, char *Line)
                        \r
                        switch(num)\r
                        {\r
+                       case 332:       // Topic\r
+                               user = message; // Channel\r
+                               message = Line + pos + 1;       // Topic\r
+                               Message_AppendF(Server, MSG_TYPE_SERVER, user, user, "Topic: %s", message);\r
+                               break;\r
+                       case 333:       // Topic set by\r
+                               user = message; // Channel\r
+                               message = GetValue(Line, &pos); // User\r
+                               GetValue(Line, &pos);   // Timestamp\r
+                               Message_AppendF(Server, MSG_TYPE_SERVER, user, user, "Set by %s", message);\r
+                               break;\r
                        case 353:       // /NAMES list\r
                                // <user> = <channel> :list\r
-//                             GetValue(Line, &pos);   // '='\r
+                               // '=' was eaten in and set to message\r
                                user = GetValue(Line, &pos);    // Actually channel\r
                                message = Line + pos + 1;       // List\r
-                               Message_AppendF(Server, MSG_TYPE_SERVER, user, "", "Names: %s", message);\r
+                               Message_AppendF(Server, MSG_TYPE_SERVER, user, user, "Names: %s", message);\r
                                break;\r
                        case 366:       // end of /NAMES list\r
                                // <user> <channel> :msg\r
                                user = message;\r
                                message = Line + pos + 1;\r
-                               Message_Append(Server, MSG_TYPE_SERVER, user, "", message);\r
+                               Message_Append(Server, MSG_TYPE_SERVER, user, user, message);\r
                                break;\r
                        case 372:       // MOTD Data\r
+                       case 375:       // MOTD Start\r
                        case 376:       // MOTD End\r
                                \r
                        default:\r
@@ -584,6 +652,10 @@ void ParseServerLine(tServer *Server, char *Line)
                        }\r
                        \r
                        //printf("[%s] NOTICE %s: %s\n", Server->Name, ident, message);\r
+                       char *ident_bang = strchr(ident, '!');\r
+                       if( ident_bang ) {\r
+                               *ident_bang = '\0';\r
+                       }\r
                        Message_Append(Server, MSG_TYPE_NOTICE, ident, "", message);\r
                }\r
                else if( strcmp(cmd, "PRIVMSG") == 0 )\r
@@ -597,13 +669,20 @@ void ParseServerLine(tServer *Server, char *Line)
                        else {\r
                                message = GetValue(Line, &pos);\r
                        }\r
+\r
+                       // TODO: Catch when the privmsg is addressed to the user\r
+\r
 //                     Cmd_PRIVMSG(Server, dest, ident, message);\r
+                       char *ident_bang = strchr(ident, '!');\r
+                       if( ident_bang ) {\r
+                               *ident_bang = '\0';\r
+                       }\r
                        Message_Append(Server, MSG_TYPE_STANDARD, ident, dest, message);\r
                }\r
                else if( strcmp(cmd, "JOIN" ) == 0 )\r
                {\r
                        char    *channel;\r
-                       channel = GetValue(Line, &pos) + 1;\r
+                       channel = Line + pos + 1;\r
                        Window_Create(Server, channel);\r
                }\r
                else\r
@@ -638,11 +717,11 @@ int ProcessIncoming(tServer *Server)
        // ioctl#8 on a TCP client gets the number of bytes in the recieve buffer\r
        // - Used to avoid blocking\r
        #if NON_BLOCK_READ\r
-       while( (len = ioctl(Server->FD, 8, NULL)) > 0 )\r
+       while( (len = _SysIOCtl(Server->FD, 8, NULL)) > 0 )\r
        {\r
        #endif\r
                // Read data\r
-               len = read(Server->FD, &Server->InBuf[Server->ReadPos], BUFSIZ - Server->ReadPos);\r
+               len = _SysRead(Server->FD, &Server->InBuf[Server->ReadPos], BUFSIZ - Server->ReadPos);\r
                if( len == -1 ) {\r
                        return -1;\r
                }\r
@@ -687,7 +766,7 @@ int writef(int FD, const char *Format, ...)
         int    len;\r
        \r
        va_start(args, Format);\r
-       len = vsnprintf(NULL, 1000, Format, args);\r
+       len = vsnprintf(NULL, 0, Format, args);\r
        va_end(args);\r
        \r
        {\r
@@ -696,7 +775,7 @@ int writef(int FD, const char *Format, ...)
                vsnprintf(buf, len+1, Format, args);\r
                va_end(args);\r
                \r
-               return write(FD, buf, len);\r
+               return _SysWrite(FD, buf, len);\r
        }\r
 }\r
 \r
@@ -726,12 +805,12 @@ int OpenTCP(const char *AddressString, short PortNumber)
        \r
        // Set remote port and address\r
 //     printf("Setting port and remote address\n");\r
-       ioctl(fd, 5, &PortNumber);\r
-       ioctl(fd, 6, addrBuffer);\r
+       _SysIOCtl(fd, 5, &PortNumber);\r
+       _SysIOCtl(fd, 6, addrBuffer);\r
        \r
        // Connect\r
 //     printf("Initiating connection\n");\r
-       if( ioctl(fd, 7, NULL) == 0 ) {\r
+       if( _SysIOCtl(fd, 7, NULL) == 0 ) {\r
                // Shouldn't happen :(\r
                fprintf(stderr, "Unable to start connection\n");\r
                return -1;\r
@@ -769,16 +848,9 @@ char *GetValue(char *Src, int *Ofs)
        return ret;\r
 }\r
 \r
-int SetCursorPos(int Row, int Col)\r
+void SetCursorPos(int Row, int Col)\r
 {\r
-        int    rv;\r
-       if( Row == -1 ) {\r
-               Row = Col / giTerminal_Width;\r
-               Col = Col % giTerminal_Width;\r
-       }\r
-       rv = ioctl(1, 9, NULL); // Ugh, constants\r
        printf("\x1B[%i;%iH", Col, Row);\r
-       return rv;\r
 }\r
 \r
 static inline int isdigit(int ch)\r

UCC git Repository :: git.ucc.asn.au