1 /* cmd_funcmds.c - NeonServ v5.4
2 * Copyright (C) 2011-2012 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/>.
17 #include "../module.h"
18 #include "cmd_funcmds.h"
19 #include "../../modcmd.h"
20 #include "../../mysqlConn.h"
21 #include "../../IRCParser.h"
22 #include "../../ClientSocket.h"
23 #include "../../UserNode.h"
24 #include "../../ChanNode.h"
25 #include "../../lang.h"
26 #include "../../tools.h"
27 #include "../../DBHelper.h"
29 static const struct default_language_entry msgtab[] = {
30 {"FUN_DICE", "$b%s$b: A $b%d$b shows on the %d-sided die."}, /* {ARGS: "TestUser", 5, 6} */
31 {"FUN_DICE_NUM", "I do not understand $b%s$b. Please use a single number above 1."}, /* {ARGS: "bla"} */
32 {"FUN_8BALL", "$b%s$b: %s"}, /* {ARGS: "TestUser", "Not a chance."} */
33 {"FUN_8BALL_REPLIES", "Not a chance.|In your dreams.|Absolutely!|Could be, could be.|No!"},
34 {"FUN_COOKIE", "gives %1$s a very big chocolate cookie. %1$s has got %2$d cookies until now (%3$d in this channel)."}, /* {ARGS: "TestUser", 20, 50} */
39 register_default_language_table(msgtab);
43 void register_commands() {
45 register_command_alias(3, "FunCMD");
46 #define USER_COMMAND(NAME,FUNCTION,PARAMCOUNT,FLAGS) register_command(3, NAME, module_id, FUNCTION, PARAMCOUNT, NULL, 0, FLAGS)
47 // NAME FUNCTION PARAMS FLAGS
48 //USER_COMMAND("extscript", neonserv_cmd_extscript, 0, CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_EMPTY_ARGS | CMDFLAG_CHAN_PARAM | CMDFLAG_FUNCMD);
49 USER_COMMAND("ping", funcmd_ping, 0, CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_FUNCMD);
50 USER_COMMAND("pong", funcmd_pong, 0, CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_FUNCMD);
51 USER_COMMAND("dice", funcmd_dice, 1, CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_FUNCMD);
52 USER_COMMAND("8ball", funcmd_8ball, 1, CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_FUNCMD);
53 USER_COMMAND("cookie", funcmd_cookie, 0, CMDFLAG_REQUIRE_CHAN | CMDFLAG_REGISTERED_CHAN | CMDFLAG_FUNCMD);
57 struct current_funcmd_header {
58 struct ClientSocket *client;
59 struct UserNode *user;
60 struct ChanNode *chan;
64 static struct current_funcmd_header current_funcmd;
66 #define FUNCMD_HEADER \
67 current_funcmd.client = getTextBot(); \
68 current_funcmd.user = user; \
69 current_funcmd.chan = chan; \
73 printf_mysql_query("SELECT `channel_toys` FROM `channels` WHERE `channel_name` = '%s'", escape_string(chan->name)); \
75 row = mysql_fetch_row(res); \
76 if(!row || !strcmp(row[0], "0")) { \
77 reply(getTextBot(), user, "NS_FUN_DISABLED", chan->name); \
79 } else if(!strcmp(row[0], "1")) \
80 current_funcmd.send_notice = 1; \
82 current_funcmd.send_notice = 0; \
85 #define REPLYTYPE_NORMAL 0
86 #define REPLYTYPE_ACTION 1
87 static void funcmd_reply(const char *text, int type, ...) {
88 if (!(current_funcmd.client->flags & SOCKET_FLAG_CONNECTED)) return;
89 const char *reply_format = get_language_string(current_funcmd.user, text);
92 char formatBuf[MAXLEN];
93 if(current_funcmd.send_notice) {
94 if(type == REPLYTYPE_ACTION)
95 sprintf(formatBuf, "NOTICE %s :%s %s", current_funcmd.user->nick, current_funcmd.client->user->nick, text);
97 sprintf(formatBuf, "NOTICE %s :%s", current_funcmd.user->nick, text);
99 if(type == REPLYTYPE_ACTION)
100 sprintf(formatBuf, "PRIVMSG %s :\001ACTION %s\001", current_funcmd.chan->name, text);
102 sprintf(formatBuf, "PRIVMSG %s :%s", current_funcmd.chan->name, text);
105 char sendBuf[MAXLEN];
108 va_start(arg_list, type);
109 pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
111 if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
113 sendBuf[pos+1] = '\0';
114 write_socket(current_funcmd.client, sendBuf, pos+1);
117 static char* getSetting(struct UserNode *user, struct ChanNode *chan, const char *setting) {
123 uname = ((user->flags & USERFLAG_ISAUTHED) ? user->auth : "*");
126 loadChannelSettings(chan);
127 if(chan->flags & CHANFLAG_CHAN_REGISTERED)
128 cid = chan->channel_id;
130 printf_mysql_query("SELECT `value` FROM `fundata` WHERE `user` = '%s' AND `cid` = '%d' AND `name` = '%s'", escape_string(uname), cid, escape_string(setting));
132 if ((row = mysql_fetch_row(res)) != NULL) {
138 static void setSetting(struct UserNode *user, struct ChanNode *chan, const char *setting, const char *value) {
144 uname = ((user->flags & USERFLAG_ISAUTHED) ? user->auth : "*");
147 loadChannelSettings(chan);
148 if(chan->flags & CHANFLAG_CHAN_REGISTERED)
149 cid = chan->channel_id;
151 printf_mysql_query("SELECT `id`, `value` FROM `fundata` WHERE `user` = '%s' AND `cid` = '%d' AND `name` = '%s'", escape_string(uname), cid, escape_string(setting));
153 if ((row = mysql_fetch_row(res)) != NULL) {
154 if(strcmp(row[1], value))
155 printf_mysql_query("UPDATE `fundata` SET `value` = '%s' WHERE `id` = '%s'", escape_string(value), row[0]);
157 printf_mysql_query("INSERT INTO `fundata` (`user`, `cid`, `name`, `value`) VALUES ('%s', '%d', '%s', '%s')", escape_string(uname), cid, escape_string(setting), escape_string(value));
160 CMD_BIND(funcmd_ping) {
162 funcmd_reply("\002%s\002: Pong!", REPLYTYPE_NORMAL, user->nick);
165 CMD_BIND(funcmd_pong) {
167 funcmd_reply("\002%s\002: Ping!", REPLYTYPE_NORMAL, user->nick);
170 CMD_BIND(funcmd_dice) {
172 int max = atoi(argv[0]);
174 int val = (rand() % max) + 1;
175 funcmd_reply("FUN_DICE", REPLYTYPE_NORMAL, user->nick, val, max);
177 funcmd_reply("FUN_DICE_NUM", REPLYTYPE_NORMAL, argv[0]);
180 CMD_BIND(funcmd_8ball) {
182 char *message = merge_argv(argv, 0, argc);
183 const char *const_replies = get_language_string(current_funcmd.user, "FUN_8BALL_REPLIES");
184 char replies[MAXLEN];
185 int i, reply_count = 1;
186 for(i = 0; const_replies[i]; i++) {
187 if(const_replies[i] == '|')
189 replies[i] = const_replies[i];
192 unsigned int crc32_val = (crc32(message)) % reply_count;
193 char *creply = (crc32_val == 0 ? replies : NULL);
195 for(i = 0; replies[i]; i++) {
196 if(replies[i] == '|') {
202 if(reply_count == crc32_val) {
203 creply = &replies[i+1];
209 funcmd_reply("FUN_8BALL", REPLYTYPE_NORMAL, user->nick, creply);
213 CMD_BIND(funcmd_cookie) {
216 if(!(user = getUserByNick(argv[0]))) {
217 reply(current_funcmd.client, current_funcmd.user, "NS_USER_UNKNOWN", argv[0]);
222 int user_count = ((tmp = getSetting(user, chan, "cookies")) ? atoi(tmp) : 0);
223 int total_count = ((tmp = getSetting(user, NULL, "cookies")) ? atoi(tmp) : 0);
227 sprintf(buf, "%d", user_count);
228 setSetting(user, chan, "cookies", buf);
229 sprintf(buf, "%d", total_count);
230 setSetting(user, NULL, "cookies", buf);
231 funcmd_reply("FUN_COOKIE", REPLYTYPE_ACTION, user->nick, total_count, user_count);