added channel rejoin faker (session recover)

[TransparentIRC.git] / src / IRCClient.c

diff --git a/src/IRCClient.c b/src/IRCClient.c
index eb91bafdbf684f3e6671269bb03c2dcbdfd6c61f..eee86b1f8162d78e6f4fc81cc0416dfed58b17ef 100644 (file)
--- a/src/IRCClient.c
+++ b/src/IRCClient.c
@@ -72,6 +72,35 @@ static void ircclient_handshake(struct IRCClient *client) {
     session->realname = NULL;
 }
 
+static struct IRCUser ircclient_parse_user(char *from) {
+    struct IRCUser user;
+    char *a = from, *b = from;
+    while(*b != '!') {
+        b++;
+    }
+    user.nick = a;
+    if(!*b) {
+        user.ident = "";
+        user.host = "";
+        return user;
+    }
+    *b = '\0';
+    b++;
+    a = b;
+    while(*b && *b != '@') {
+        b++;
+    }
+    user.ident = a;
+    if(!*b) {
+        user.host = "";
+        return user;
+    }
+    *b = '\0';
+    b++;
+    user.host = b;
+    return user;
+}
+
 static void ircclient_recv(struct IRCClient *client, char *line) {
     struct UserSession *session = client->session;
     char *argv[MAXNUMPARAMS];
@@ -91,6 +120,11 @@ static void ircclient_recv(struct IRCClient *client, char *line) {
             pass_to_client = 0;
         }
     } else {
+        if(!stricmp(argv[1], "001")) {
+            free(session->nick);
+            session->nick = strdup(argv[2]);
+            client->fully_connected = 1;
+        }
         if(!stricmp(argv[1], "001") ||
             !stricmp(argv[1], "002") ||
             !stricmp(argv[1], "003") ||
@@ -113,6 +147,61 @@ static void ircclient_recv(struct IRCClient *client, char *line) {
                 else
                     client->recover_header = new_line;
             }
+        } else if(!stricmp(argv[1], "JOIN")) {
+            struct IRCUser from = ircclient_parse_user(argv[0]);
+            if(!stricmp(from.nick, session->nick)) {
+                struct IRCChannel *lchannel = NULL, *channel = malloc(sizeof(*channel));
+                if(client->channel)
+                    for(lchannel = client->channel; lchannel->next; lchannel = lchannel->next) {}
+                channel->name = strdup(argv[2]);
+                channel->next = NULL;
+                channel->prev = lchannel;
+                if(lchannel)
+                    lchannel->next = channel;
+                else
+                    client->channel = channel;
+            }
+        } else if(!stricmp(argv[1], "PART")) {
+            struct IRCUser from = ircclient_parse_user(argv[0]);
+            if(!stricmp(from.nick, session->nick)) {
+                struct IRCChannel *channel;
+                for(channel = client->channel; channel; channel = channel->next) {
+                    if(!stricmp(channel->name, argv[2])) {
+                        if(channel->prev)
+                            channel->prev->next = channel->next;
+                        else
+                            client->channel = channel->next;
+                        if(channel->next)
+                            channel->next->prev = channel->prev;
+                        free(channel->name);
+                        free(channel);
+                        break;
+                    }
+                }
+            }
+        } else if(!stricmp(argv[1], "KICK")) {
+            if(!stricmp(argv[3], session->nick)) {
+                struct IRCChannel *channel;
+                for(channel = client->channel; channel; channel = channel->next) {
+                    if(!stricmp(channel->name, argv[2])) {
+                        if(channel->prev)
+                            channel->prev->next = channel->next;
+                        else
+                            client->channel = channel->next;
+                        if(channel->next)
+                            channel->next->prev = channel->prev;
+                        free(channel->name);
+                        free(channel);
+                        break;
+                    }
+                }
+            }
+        } else if(!stricmp(argv[1], "NICK")) {
+            struct IRCUser from = ircclient_parse_user(argv[0]);
+            if(!stricmp(from.nick, session->nick)) {
+                free(session->nick);
+                session->nick = strdup(argv[2]);
+            }
         }
     }
     if(pass_to_client)
@@ -150,5 +239,13 @@ void ircclient_recover_session(struct UserSession *session) {
     for(recover_line = client->recover_header; recover_line; recover_line = recover_line->next) {
         usersession_client_raw(session, recover_line->line);
     }
-    
+    struct IRCChannel *channel;
+    char raw[LINELEN];
+    for(channel = client->channel; channel; channel = channel->next) {
+        sprintf(raw, ":%s!%s@TransparentIRC.session.recover JOIN %s", client->session->nick, client->session->username, channel->name);
+        usersession_client_raw(session, raw);
+        
+        sprintf(raw, "NAMES %s", channel->name);
+        ircclient_send(client, raw);
+    }
 }