completed cmd_adduser and added CMDFLAG_REGISTERED_CHAN flag
[NeonServV5.git] / modcmd.c
index 092408e47c634069fa7eae076006d8cc5fb9098f..9273f0265b999b010435b3bc64da26c8824266dd 100644 (file)
--- a/modcmd.c
+++ b/modcmd.c
@@ -1,10 +1,13 @@
 
 #include "modcmd.h"
 #include "IRCEvents.h"
+#include "IRCParser.h"
 #include "ClientSocket.h"
 #include "UserNode.h"
 #include "ChanNode.h"
 #include "ChanUser.h"
+#include "WHOHandler.h"
+#include "lang.h"
 
 struct trigger_callback {
     int botid;
@@ -13,9 +16,30 @@ struct trigger_callback {
     struct trigger_callback *next;
 };
 
+struct command_check_user_cache {
+    struct ClientSocket *client, *textclient;
+    struct UserNode *user;
+    struct ChanNode *chan;
+    char **argv;
+    int argc;
+    char *message;
+    struct cmd_binding *cbind;
+};
+
 static struct cmd_binding **cmd_binds;
 static struct cmd_function *cmd_functions = NULL;
 static struct trigger_callback *trigger_callbacks = NULL;
+static struct ClientSocket *tmp_text_client;
+
+static const struct default_language_entry msgtab[] = {
+    {"MODCMD_LESS_PARAM_COUNT", "This command requires more parameters."},
+    {"MODCMD_CHAN_REQUIRED",    "You must provide the name of a channel that exists."},
+    {"MODCMD_AUTH_REQUIRED",    "You need to be authenticated with AuthServ to use this command."},
+    {"MODCMD_PRIVILEGED",       "§b%s§b is a privileged command."},
+    
+    
+    {NULL, NULL}
+};
 
 static int get_binds_index(char first_char) {
     if(tolower(first_char) >= 'a' && tolower(first_char) <= 'z') {
@@ -61,6 +85,25 @@ static char* get_channel_trigger(int botid, struct ChanNode *chan) {
     return trigger->trigger;
 }
 
+static USERAUTH_CALLBACK(command_checked_auth) {
+    struct command_check_user_cache *cache = data;
+    int execute_cmd = 1;
+    tmp_text_client = cache->textclient;
+    if((cache->cbind->func->flags & CMDFLAG_REQUIRE_AUTH) && !(user->flags & USERFLAG_ISAUTHED)) {
+        reply(tmp_text_client, user, "MODCMD_AUTH_REQUIRED");
+        execute_cmd = 0;
+    }
+    else if((cache->cbind->func->flags & CMDFLAG_REQUIRE_GOD) && !isGodMode(user)) {
+        reply(tmp_text_client, user, "MODCMD_PRIVILEGED", cache->cbind->cmd);
+        execute_cmd = 0;
+    }
+    if(execute_cmd) {
+        cache->cbind->func->func(cache->client, user, cache->chan, cache->argv, cache->argc);
+    }
+    free(cache->message);
+    free(cache);
+}
+
 static void handle_command(struct ClientSocket *client, struct UserNode *user, struct ChanNode *chan, char *message) {
     if(message[0] == '#') {
         char *chanName = message;
@@ -72,6 +115,7 @@ static void handle_command(struct ClientSocket *client, struct UserNode *user, s
         if(chan2)
             chan = chan2;
     }
+    message = strdup(message);
     int bind_index = get_binds_index(message[0]);
     char *args = strstr(message, " ");
     if(args) {
@@ -81,20 +125,23 @@ static void handle_command(struct ClientSocket *client, struct UserNode *user, s
     struct cmd_binding *cbind;
     for(cbind = cmd_binds[bind_index]; cbind; cbind = cbind->next) {
         if(cbind->botid == client->botid && strcmp(cbind->cmd, message) == 0) {
-            struct cmd_function *cmdfunc = cbind->func;
+            //get a text bot
+            tmp_text_client = get_prefered_bot(client->botid);
             //parse the arguments...
             char *arga[MAXNUMPARAMS];
             char **argv;
             int argc = 0;
-            while(*args) {
-                //skip leading spaces
-                while (*args == ' ')
-                    *args++ = 0;
-                arga[argc++] = args;
-                if (argc >= MAXNUMPARAMS)
-                    break;
-                while (*args != ' ' && *args)
-                    args++;
+            if(args) {
+                while(*args) {
+                    //skip leading spaces
+                    while (*args == ' ')
+                        *args++ = 0;
+                    arga[argc++] = args;
+                    if (argc >= MAXNUMPARAMS)
+                        break;
+                    while (*args != ' ' && *args)
+                        args++;
+                }
             }
             argv = arga;
             if(argc != 0 && argv[0][0] == '#') {
@@ -105,10 +152,101 @@ static void handle_command(struct ClientSocket *client, struct UserNode *user, s
                     chan = chan2;
                 }
             }
-            cmdfunc->func(client, user, chan, argv, argc);
-            return;
+            if(cbind->parameters) {
+                //userdefined parameters...
+                char *uarga[MAXNUMPARAMS];
+                char params[strlen(cbind->parameters)+1];
+                strcpy(params, cbind->parameters);
+                int uargpos = 0, argi, allargs = 0;
+                char *ppos = params;
+                char *prev_ppos = params;
+                while((ppos = strstr(ppos, " "))) {
+                    *ppos = '\0';
+                    if(prev_ppos[0] == '%') {
+                        prev_ppos++;
+                        if(prev_ppos[strlen(prev_ppos)-1] == '-') {
+                            allargs = 1;
+                            prev_ppos[strlen(prev_ppos)-1] = '\0';
+                        } else
+                            allargs = 0;
+                        argi = atoi(prev_ppos);
+                        if(argi > 0) {
+                            if(argi <= argc) continue;
+                            uarga[uargpos++] = argv[argi-1];
+                            if(allargs) {
+                                for(;argi < argc; argi++)
+                                    uarga[uargpos++] = argv[argi-1];
+                            }
+                        } else if(!strcmp(prev_ppos, "c"))
+                            uarga[uargpos++] = (chan ? chan->name : NULL);
+                        else if(!strcmp(prev_ppos, "n")) 
+                            uarga[uargpos++] = user->nick;
+                    } else {
+                        uarga[uargpos++] = prev_ppos;
+                    }
+                    ppos++;
+                    prev_ppos = ppos;
+                }
+                argv = uarga;
+                argc = uargpos;
+            }
+            if(argc < cbind->func->paramcount) {
+                reply(tmp_text_client, user, "MODCMD_LESS_PARAM_COUNT");
+                break;
+            }
+            if((cbind->func->flags & CMDFLAG_REQUIRE_CHAN) && !chan) {
+                reply(tmp_text_client, user, "MODCMD_CHAN_REQUIRED");
+                break;
+            }
+            if((cbind->func->flags & CMDFLAG_REGISTERED_CHAN)) {
+                load_channel_settings(chan);
+                if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
+                    reply(tmp_text_client, user, "MODCMD_CHAN_REQUIRED");
+                    break;
+                }
+                check_mysql();
+                MYSQL_RES *res;
+                MYSQL_ROW row;
+                printf_mysql_query("SELECT `botid` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '%d'", chan->channel_id, client->botid);
+                res = mysql_use();
+                if ((row = mysql_fetch_row(res)) == NULL) {
+                    reply(tmp_text_client, user, "MODCMD_CHAN_REQUIRED");
+                    break;
+                }
+            }
+            if((cbind->func->flags & CMDFLAG_CHECK_AUTH) && !(user->flags & USERFLAG_ISAUTHED)) {
+                //check auth...
+                struct command_check_user_cache *data = malloc(sizeof(*data));
+                char **temp_argv = malloc(argc*sizeof(*temp_argv));
+                if (!data || !temp_argv) {
+                    perror("malloc() failed");
+                    break;
+                }
+                memcpy(temp_argv, argv, argc*sizeof(*temp_argv));
+                data->argv = temp_argv;
+                data->argc = argc;
+                data->client = client;
+                data->user = user;
+                data->chan = chan;
+                data->message = message;
+                data->cbind = cbind;
+                data->textclient = tmp_text_client;
+                get_userauth(user, command_checked_auth, data);
+                return;
+            }
+            if((cbind->func->flags & CMDFLAG_REQUIRE_AUTH) && !(user->flags & USERFLAG_ISAUTHED)) {
+                reply(tmp_text_client, user, "MODCMD_AUTH_REQUIRED");
+                break;
+            }
+            if((cbind->func->flags & CMDFLAG_REQUIRE_GOD) && !isGodMode(user)) {
+                reply(tmp_text_client, user, "MODCMD_PRIVILEGED", cbind->cmd);
+                break;
+            }
+            cbind->func->func(client, user, chan, argv, argc);
+            break;
         }
     }
+    free(message);
 }
 
 static void got_chanmsg(struct UserNode *user, struct ChanNode *chan, char *message) {
@@ -145,7 +283,7 @@ static void got_privmsg(struct UserNode *user, struct UserNode *target, char *me
     }
 }
 
-int register_command(int botid, char *name, cmd_bind_t *func) {
+int register_command(int botid, char *name, cmd_bind_t *func, int paramcount, unsigned int flags) {
     struct cmd_function *cmdfunc;
     for(cmdfunc = cmd_functions; cmdfunc; cmdfunc = cmdfunc->next) {
         if(cmdfunc->botid == botid && strcmp(cmdfunc->name, name) == 0)
@@ -159,6 +297,8 @@ int register_command(int botid, char *name, cmd_bind_t *func) {
     cmdfunc->botid = botid;
     cmdfunc->name = strdup(name);
     cmdfunc->func = func;
+    cmdfunc->flags = 0;
+    cmdfunc->paramcount = paramcount;
     cmdfunc->next = cmd_functions;
     cmd_functions = cmdfunc;
     return 1;
@@ -211,6 +351,9 @@ int bind_cmd_to_function(int botid, char *cmd, struct cmd_function *func) {
     cbind->botid = botid;
     cbind->cmd = strdup(cmd);
     cbind->func = func;
+    cbind->parameters = NULL;
+    cbind->gaccess = 0;
+    cbind->flags = 0;
     cbind->next = cmd_binds[bind_index];
     cmd_binds[bind_index] = cbind;
     return 1;
@@ -238,6 +381,9 @@ int bind_cmd_to_command(int botid, char *cmd, char *func) {
     cbind->cmd = strdup(cmd);
     cbind->func = cmdfunc;
     cbind->next = cmd_binds[bind_index];
+    cbind->parameters = NULL;
+    cbind->gaccess = 0;
+    cbind->flags = 0;
     cmd_binds[bind_index] = cbind;
     return 1;
 }
@@ -252,6 +398,8 @@ int unbind_cmd(int botid, char *cmd) {
             else
                 cmd_binds[bind_index] = cbind->next;
             free(cbind->cmd);
+            if(cbind->parameters)
+                free(cbind->parameters);
             free(cbind);
             return 1;
         } else
@@ -260,10 +408,15 @@ int unbind_cmd(int botid, char *cmd) {
     return 0;
 }
 
+struct ClientSocket *getTextBot() {
+    return tmp_text_client;
+}
+
 void init_modcmd() {
     cmd_binds = calloc(27, sizeof(*cmd_binds));
     bind_chanmsg(got_chanmsg);
     bind_privmsg(got_privmsg);
+    register_default_language_table(msgtab);
 }
 
 void free_modcmd() {
@@ -273,6 +426,8 @@ void free_modcmd() {
         for(cbind = cmd_binds[i]; cbind; cbind = next) {
             next = cbind->next;
             free(cbind->cmd);
+            if(cbind->parameters)
+                free(cbind->parameters);
             free(cbind);
         }
     }
@@ -292,3 +447,11 @@ void free_modcmd() {
     trigger_callbacks = NULL;
 }
 
+void bind_set_parameters(int botid, char *cmd, char *parameters) {
+    
+}
+
+void bind_set_gaccess(int botid, char *cmd, int gaccess) {
+    
+}
+