1 /* cmd_global_register.c - NeonServ v5.2
2 * Copyright (C) 2011 Philipp Kreil (pk910)
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.
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.
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/>.
18 #include "cmd_global.h"
22 * argv[0/1] - nick / *auth
24 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup);
25 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup);
26 static void global_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth, int multibot);
28 struct global_cmd_register_cache {
29 struct ClientSocket *client, *textclient;
30 struct UserNode *user;
31 struct ChanNode *chan;
38 CMD_BIND(global_cmd_register) {
41 char *channel = argv[0];
43 if(!is_valid_chan(channel)) {
44 reply(getTextBot(), user, "NS_INVALID_CHANNEL_NAME", argv[0]);
47 printf_mysql_query("SELECT `botid`, `botclass` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` LEFT JOIN `channels` ON `bot_channels`.`chanid` = `channels`.`channel_id` WHERE `channel_name` = '%s'", escape_string(channel));
49 while ((row = mysql_fetch_row(res)) != NULL) {
50 if(atoi(row[1]) == client->botid) {
51 reply(getTextBot(), user, "NS_REGISTER_ALREADY", argv[0], client->user->nick);
56 //if theres already another bot in the channel we don't need a owner parameter...
57 if(multibot && argc < 2) {
58 //skip all these owner check lines
60 global_cmd_register_async1(client, getTextBot(), user, chan, event, channel, NULL, multibot);
63 reply(getTextBot(), user, "MODCMD_LESS_PARAM_COUNT");
67 if(argv[1][0] == '*') {
70 printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[1]));
72 if ((row = mysql_fetch_row(res)) != NULL) {
73 global_cmd_register_async1(client, getTextBot(), user, chan, event, channel, row[0], multibot);
75 //we need to create a new user...
76 //but first lookup the auth to check if it really exists
77 struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
79 perror("malloc() failed");
82 cache->client = client;
83 cache->textclient = getTextBot();
87 cache->nick = strdup(argv[1]);
88 cache->channel = strdup(channel);
89 cache->multibot = multibot;
90 lookup_authname(argv[1], global_cmd_register_auth_lookup, cache);
93 struct UserNode *cuser = getUserByNick(argv[1]);
95 cuser = createTempUser(argv[1]);
96 cuser->flags |= USERFLAG_ISTMPUSER;
98 if(cuser->flags & USERFLAG_ISAUTHED) {
99 global_cmd_register_async1(client, getTextBot(), user, chan, event, channel, cuser->auth, multibot);
101 struct global_cmd_register_cache *cache = malloc(sizeof(*cache));
103 perror("malloc() failed");
106 cache->client = client;
107 cache->textclient = getTextBot();
110 cache->event = event;
111 cache->nick = strdup(argv[1]);
112 cache->channel = strdup(channel);
113 cache->multibot = multibot;
114 get_userauth(cuser, global_cmd_register_nick_lookup, cache);
119 static AUTHLOOKUP_CALLBACK(global_cmd_register_auth_lookup) {
120 struct global_cmd_register_cache *cache = data;
122 //AUTH_DOES_NOT_EXIST
123 reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
125 global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, auth, cache->multibot);
126 free(cache->channel);
131 static USERAUTH_CALLBACK(global_cmd_register_nick_lookup) {
132 struct global_cmd_register_cache *cache = data;
134 //USER_DOES_NOT_EXIST
135 reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
137 else if(!(user->flags & USERFLAG_ISAUTHED)) {
139 reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
142 global_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, user->auth, cache->multibot);
143 free(cache->channel);
148 static void global_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth, int multibot) {
149 //we've got a valid auth now...
152 int userid = 0, adminid;
153 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
155 if ((row = mysql_fetch_row(res)) != NULL)
156 adminid = atoi(row[0]);
160 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
162 if ((row = mysql_fetch_row(res)) != NULL) {
163 userid = atoi(row[0]);
165 printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
166 userid = (int) mysql_insert_id(mysql_conn);
169 printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger` FROM `bots` WHERE `botclass` = '%d' ORDER BY `register_priority` DESC", client->botid);
173 while ((row = mysql_fetch_row(res)) != NULL) {
174 //check channel count
175 printf_mysql_query("SELECT COUNT(*) FROM `bot_channels` WHERE `botid` = '%s'", row[0]);
176 row2 = mysql_fetch_row(mysql_use());
177 if(atoi(row2[0]) < atoi(row[1])) {
178 botid = atoi(row[0]);
184 reply(textclient, user, "NS_REGISTER_FULL");
188 printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
190 if ((row = mysql_fetch_row(res)) != NULL) {
191 chanid = atoi(row[0]);
192 printf_mysql_query("UPDATE `channels` SET `channel_registered` = UNIX_TIMESTAMP(), `channel_registrator` = '%d' WHERE `channel_id` = '%d'", adminid, chanid);
194 printf_mysql_query("INSERT INTO `channels` (`channel_name`, `channel_registered`, `channel_registrator`) VALUES ('%s', UNIX_TIMESTAMP(), '%d')", escape_string(channel), adminid);
195 chanid = (int) mysql_insert_id(mysql_conn);
197 struct ClientSocket *bot;
198 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
199 if(bot->clientid == botid)
203 putsock(bot, "JOIN %s", channel);
205 reply(textclient, user, "NS_REGISTER_DISCONNECTED");
206 printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', '%s')", botid, chanid, bottrigger);
209 printf_mysql_query("UPDATE `chanusers` SET `chanuser_access` = 499 WHERE `chanuser_cid` = '%d' AND `chanuser_access` = '500'", chanid);
210 printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d' AND `chanuser_uid` = '%d'", chanid, userid);
212 printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d'", chanid);
213 printf_mysql_query("INSERT INTO `chanusers` (`chanuser_cid`, `chanuser_uid`, `chanuser_access`) VALUES ('%d', '%d', '%d')", chanid, userid, 500);
214 reply(textclient, user, "NS_REGISTER_DONE", channel, auth);
216 reply(textclient, user, "NS_REGISTER_DONE_NOAUTH", channel);