src/cmd_neonserv_assignrank.c

   1 /* cmd_neonserv_assignrank.c - NeonServ v5.2
   2  * Copyright (C) 2011  Philipp Kreil (pk910)
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include "cmd_neonserv.h"
  19
  20 /*
  21 * argv[0] - nick / *auth
  22 * argv[1-*] - rank name
  23 */
  24 static AUTHLOOKUP_CALLBACK(neonserv_cmd_assignrank_auth_lookup);
  25 static USERAUTH_CALLBACK(neonserv_cmd_assignrank_nick_lookup);
  26 static void neonserv_cmd_assignrank_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct Event *event, char *nick, char *auth, int rank_id);
  27
  28 struct neonserv_cmd_assignrank_cache {
  29     struct ClientSocket *client, *textclient;
  30     struct UserNode *user;
  31     struct Event *event;
  32     int rank_id;
  33     char *nick;
  34 };
  35
  36 CMD_BIND(neonserv_cmd_assignrank) {
  37     MYSQL_RES *res;
  38     MYSQL_ROW row;
  39     char *name = merge_argv(argv, 1, argc);
  40     int rank_id = 0;
  41     if(stricmp(name, "*") && stricmp(name, "user") && stricmp(name, "none")) {
  42         printf_mysql_query("SELECT `rank_id`, `rank_name` FROM `support_ranks` WHERE `rank_name` = '%s'", escape_string(name));
  43         res = mysql_use();
  44         if ((row = mysql_fetch_row(res)) == NULL) {
  45             reply(getTextBot(), user, "NS_DELRANK_NOT_FOUND", name);
  46             return;
  47         }
  48         rank_id = atoi(row[0]);
  49     }
  50     if(argv[0][0] == '*') {
  51         //we've got an auth
  52         argv[0]++;
  53         printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[0]));
  54         res = mysql_use();
  55         if ((row = mysql_fetch_row(res)) != NULL) {
  56             neonserv_cmd_assignrank_async1(client, getTextBot(), user, event, argv[0], row[0], rank_id);
  57         } else {
  58             //we need to create a new user...
  59             //but first lookup the auth to check if it really exists
  60             struct neonserv_cmd_assignrank_cache *cache = malloc(sizeof(*cache));
  61             if (!cache) {
  62                 perror("malloc() failed");
  63                 return;
  64             }
  65             cache->client = client;
  66             cache->textclient = getTextBot();
  67             cache->user = user;
  68             cache->event = event;
  69             cache->rank_id = rank_id;
  70             cache->nick = strdup(argv[0]);
  71             lookup_authname(argv[0], neonserv_cmd_assignrank_auth_lookup, cache);
  72         }
  73     } else {
  74         struct UserNode *cuser = getUserByNick(argv[0]);
  75         if(!cuser) {
  76             cuser = createTempUser(argv[0]);
  77             cuser->flags |= USERFLAG_ISTMPUSER;
  78         }
  79         if(cuser->flags & USERFLAG_ISAUTHED) {
  80             neonserv_cmd_assignrank_async1(client, getTextBot(), user, event, argv[0], cuser->auth, rank_id);
  81         } else {
  82             struct neonserv_cmd_assignrank_cache *cache = malloc(sizeof(*cache));
  83             if (!cache) {
  84                 perror("malloc() failed");
  85                 return;
  86             }
  87             cache->client = client;
  88             cache->textclient = getTextBot();
  89             cache->user = user;
  90             cache->event = event;
  91             cache->rank_id = rank_id;
  92             cache->nick = strdup(argv[0]);
  93             get_userauth(cuser, neonserv_cmd_assignrank_nick_lookup, cache);
  94         }
  95     }
  96 }
  97
  98 static AUTHLOOKUP_CALLBACK(neonserv_cmd_assignrank_auth_lookup) {
  99     struct neonserv_cmd_assignrank_cache *cache = data;
 100     if(!exists) {
 101         //AUTH_DOES_NOT_EXIST
 102         reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
 103     } else
 104         neonserv_cmd_assignrank_async1(cache->client, cache->textclient, cache->user, cache->event, cache->nick, auth, cache->rank_id);
 105     free(cache->nick);
 106     free(cache);
 107 }
 108
 109 static USERAUTH_CALLBACK(neonserv_cmd_assignrank_nick_lookup) {
 110     struct neonserv_cmd_assignrank_cache *cache = data;
 111     if(!user) {
 112         //USER_DOES_NOT_EXIST
 113         reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
 114     }
 115     else if(!(user->flags & USERFLAG_ISAUTHED)) {
 116         //USER_NOT_AUTHED
 117         reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
 118     }
 119     else
 120         neonserv_cmd_assignrank_async1(cache->client, cache->textclient, cache->user, cache->event, user->nick, user->auth, cache->rank_id);
 121     free(cache->nick);
 122     free(cache);
 123 }
 124
 125 static void neonserv_cmd_assignrank_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct Event *event, char *nick, char *auth, int rank_id) {
 126     //we've got a valid auth now...
 127     MYSQL_RES *res;
 128     MYSQL_ROW row, rank = NULL;
 129     int caccess = 0;
 130     if(rank_id) {
 131         printf_mysql_query("SELECT `rank_access`, `rank_name` FROM `support_ranks` WHERE `rank_id` = '%d'", rank_id);
 132         res = mysql_use();
 133         rank = mysql_fetch_row(res);
 134         caccess = atoi(rank[0]);
 135     }
 136     printf_mysql_query("SELECT `user_id`, `user_rank` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
 137     res = mysql_use();
 138     if ((row = mysql_fetch_row(res)) != NULL) {
 139         if(atoi(row[1]) != rank_id) {
 140             printf_mysql_query("UPDATE `users` SET `user_access` = '%d', `user_rank` = '%d' WHERE `user_id` = '%s'", caccess, rank_id, row[0]);
 141         }
 142     } else if(rank_id) {
 143         printf_mysql_query("INSERT INTO `users` (`user_user`, `user_access`, `user_rank`) VALUES ('%s', '%d', '%d')", escape_string(auth), caccess, rank_id);
 144     }
 145     reply(textclient, user, "NS_ASSIGNRANK_DONE", auth, (rank_id ? rank[1] : "*"));
 146     logEvent(event);
 147 }