2 #include "cmd_neonserv.h"
6 * argv[0/1] - nick / *auth
8 static AUTHLOOKUP_CALLBACK(neonserv_cmd_register_auth_lookup);
9 static USERAUTH_CALLBACK(neonserv_cmd_register_nick_lookup);
10 static void neonserv_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth);
12 struct neonserv_cmd_register_cache {
13 struct ClientSocket *client, *textclient;
14 struct UserNode *user;
15 struct ChanNode *chan;
21 CMD_BIND(neonserv_cmd_register) {
24 char *channel = argv[0];
25 if(!is_valid_chan(channel)) {
26 reply(getTextBot(), user, "NS_INVALID_CHANNEL_NAME", argv[0]);
29 printf_mysql_query("SELECT `botid` 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' AND `botclass` = '%d'", escape_string(channel), client->botid);
31 if ((row = mysql_fetch_row(res)) != NULL) {
32 reply(getTextBot(), user, "NS_REGISTER_ALREADY", argv[0], client->user->nick);
36 if(argv[1][0] == '*') {
39 printf_mysql_query("SELECT `user_user` FROM `users` WHERE `user_user` = '%s'", escape_string(argv[1]));
41 if ((row = mysql_fetch_row(res)) != NULL) {
42 neonserv_cmd_register_async1(client, getTextBot(), user, chan, event, channel, row[0]);
44 //we need to create a new user...
45 //but first lookup the auth to check if it really exists
46 struct neonserv_cmd_register_cache *cache = malloc(sizeof(*cache));
48 perror("malloc() failed");
51 cache->client = client;
52 cache->textclient = getTextBot();
56 cache->nick = strdup(argv[1]);
57 cache->channel = strdup(channel);
58 lookup_authname(argv[1], neonserv_cmd_register_auth_lookup, cache);
61 struct UserNode *cuser = getUserByNick(argv[1]);
63 cuser = createTempUser(argv[1]);
64 cuser->flags |= USERFLAG_ISTMPUSER;
66 if(cuser->flags & USERFLAG_ISAUTHED) {
67 neonserv_cmd_register_async1(client, getTextBot(), user, chan, event, channel, cuser->auth);
69 struct neonserv_cmd_register_cache *cache = malloc(sizeof(*cache));
71 perror("malloc() failed");
74 cache->client = client;
75 cache->textclient = getTextBot();
79 cache->nick = strdup(argv[1]);
80 cache->channel = strdup(channel);
81 get_userauth(cuser, neonserv_cmd_register_nick_lookup, cache);
86 static AUTHLOOKUP_CALLBACK(neonserv_cmd_register_auth_lookup) {
87 struct neonserv_cmd_register_cache *cache = data;
90 reply(cache->textclient, cache->user, "NS_AUTH_UNKNOWN", cache->nick);
92 neonserv_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, auth);
98 static USERAUTH_CALLBACK(neonserv_cmd_register_nick_lookup) {
99 struct neonserv_cmd_register_cache *cache = data;
101 //USER_DOES_NOT_EXIST
102 reply(cache->textclient, cache->user, "NS_USER_UNKNOWN", cache->nick);
104 else if(!(user->flags & USERFLAG_ISAUTHED)) {
106 reply(cache->textclient, cache->user, "NS_USER_NEED_AUTH", cache->nick);
109 neonserv_cmd_register_async1(cache->client, cache->textclient, cache->user, cache->chan, cache->event, cache->channel, user->auth);
110 free(cache->channel);
115 static void neonserv_cmd_register_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, struct Event *event, char *channel, char *auth) {
116 //we've got a valid auth now...
120 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
122 if ((row = mysql_fetch_row(res)) != NULL)
123 adminid = atoi(row[0]);
126 printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth));
128 if ((row = mysql_fetch_row(res)) != NULL) {
129 userid = atoi(row[0]);
131 printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth));
132 userid = (int) mysql_insert_id(mysql_conn);
134 printf_mysql_query("SELECT `id`, `max_channels`, `defaulttrigger` FROM `bots` WHERE `botclass` = '%d' ORDER BY `register_priority` DESC", client->botid);
138 while ((row = mysql_fetch_row(res)) != NULL) {
139 //check channel count
140 printf_mysql_query("SELECT COUNT(*) FROM `bot_channels` WHERE `botid` = '%s'", row[0]);
141 row2 = mysql_fetch_row(mysql_use());
142 if(atoi(row2[0]) < atoi(row[1])) {
143 botid = atoi(row[0]);
149 reply(textclient, user, "NS_REGISTER_FULL");
153 printf_mysql_query("SELECT `channel_id` FROM `channels` WHERE `channel_name` = '%s'", escape_string(channel));
155 if ((row = mysql_fetch_row(res)) != NULL) {
156 chanid = atoi(row[0]);
157 printf_mysql_query("UPDATE `channels` SET `channel_registered` = UNIX_TIMESTAMP(), `channel_registrator` = '%d' WHERE `channel_id` = '%d'", adminid, chanid);
159 printf_mysql_query("INSERT INTO `channels` (`channel_name`, `channel_registered`, `channel_registrator`) VALUES ('%s', UNIX_TIMESTAMP(), '%d')", escape_string(channel), adminid);
160 chanid = (int) mysql_insert_id(mysql_conn);
162 struct ClientSocket *bot;
163 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
164 if(bot->clientid == botid)
168 putsock(bot, "JOIN %s", channel);
170 reply(textclient, user, "NS_REGISTER_DISCONNECTED");
171 printf_mysql_query("INSERT INTO `bot_channels` (`botid`, `chanid`, `trigger`) VALUES ('%d', '%d', '%s')", botid, chanid, bottrigger);
172 printf_mysql_query("DELETE FROM `chanusers` WHERE `chanuser_cid` = '%d'", chanid);
173 printf_mysql_query("INSERT INTO `chanusers` (`chanuser_cid`, `chanuser_uid`, `chanuser_access`) VALUES ('%d', '%d', '%d')", chanid, userid, 500);
174 reply(textclient, user, "NS_REGISTER_DONE", channel, auth);