src/cmd_funcmds.c

   1 /* cmd_funcmds.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_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
  27 static const struct default_language_entry msgtab[] = {
  28     {"FUN_DICE", "$b%s$b: A $b%d$b shows on the %d-sided die."}, /* {ARGS: "TestUser", 5, 6} */
  29     {"FUN_DICE_NUM", "I do not understand $b%s$b. Please use a single number above 1."}, /* {ARGS: "bla"} */
  30     {NULL, NULL}
  31 };
  32
  33 void init_funcmds() {
  34     register_default_language_table(msgtab);
  35 }
  36
  37 struct current_funcmd_header {
  38     struct ClientSocket *client;
  39     struct UserNode *user;
  40     struct ChanNode *chan;
  41     char send_notice;
  42 };
  43
  44 static struct current_funcmd_header current_funcmd;
  45
  46 #define FUNCMD_HEADER \
  47 current_funcmd.client = getTextBot(); \
  48 current_funcmd.user = user; \
  49 current_funcmd.chan = chan; \
  50 {\
  51     MYSQL_RES *res; \
  52     MYSQL_ROW row; \
  53     printf_mysql_query("SELECT `channel_toys` FROM `channels` WHERE `channel_name` = '%s'", escape_string(chan->name)); \
  54     res = mysql_use(); \
  55     row = mysql_fetch_row(res); \
  56     if(!row || !strcmp(row[0], "0")) { \
  57         reply(getTextBot(), user, "NS_FUN_DISABLED", chan->name); \
  58         return; \
  59     } else if(!strcmp(row[0], "1")) \
  60         current_funcmd.send_notice = 1; \
  61     else \
  62         current_funcmd.send_notice = 0; \
  63 }
  64
  65 static void funcmd_reply(const char *text, ...) {
  66     const char *reply_format = get_language_string(current_funcmd.user, text);
  67     if(reply_format)
  68         text = reply_format;
  69     char formatBuf[MAXLEN];
  70     if(current_funcmd.send_notice)
  71         sprintf(formatBuf, "NOTICE %s :%s", current_funcmd.user->nick, text);
  72     else
  73         sprintf(formatBuf, "PRIVMSG %s :%s", current_funcmd.chan->name, text);
  74     va_list arg_list;
  75     char sendBuf[MAXLEN];
  76     int pos;
  77     if (!(current_funcmd.client->flags & SOCKET_FLAG_CONNECTED)) return;
  78     sendBuf[0] = '\0';
  79     va_start(arg_list, text);
  80     pos = vsnprintf(sendBuf, MAXLEN - 2, formatBuf, arg_list);
  81     va_end(arg_list);
  82     if (pos < 0 || pos > (MAXLEN - 2)) pos = MAXLEN - 2;
  83     sendBuf[pos] = '\n';
  84     sendBuf[pos+1] = '\0';
  85     write_socket(current_funcmd.client, sendBuf, pos+1);
  86 }
  87
  88 CMD_BIND(funcmd_ping) {
  89     FUNCMD_HEADER;
  90     funcmd_reply("\002%s\002: Pong!", user->nick);
  91 }
  92
  93 CMD_BIND(funcmd_pong) {
  94     FUNCMD_HEADER;
  95     funcmd_reply("\002%s\002: Ping!", user->nick);
  96 }
  97
  98 CMD_BIND(funcmd_dice) {
  99     FUNCMD_HEADER;
 100     int max = atoi(argv[0]);
 101     if(max > 1) {
 102         int val = (rand() % max) + 1;
 103         funcmd_reply("FUN_DICE", user->nick, val, max);
 104     } else
 105         funcmd_reply("FUN_DICE_NUM", argv[0]);
 106 }