1 /* bot_NeonBackup.c - NeonServ v5.6
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"
20 #include "bot_NeonBackup.h"
21 #include "../../modcmd.h"
22 #include "cmd_neonbackup.h"
23 #include "../../lang.h"
24 #include "../../mysqlConn.h"
25 #include "../../ClientSocket.h"
26 #include "../../UserNode.h"
27 #include "../../ChanNode.h"
28 #include "../../ChanUser.h"
29 #include "../../IRCEvents.h"
30 #include "../../IRCParser.h"
31 #include "../../bots.h"
32 #include "../../DBHelper.h"
33 #include "../../WHOHandler.h"
35 #define BOTID NEONBACKUP_BOTID
36 #define BOTALIAS "NeonBackup"
38 static const struct default_language_entry msgtab[] = {
42 static void neonbackup_bot_ready(struct ClientSocket *client) {
43 if(client->botid != BOTID)
48 printf_mysql_query("SELECT `automodes`, `oper_user`, `oper_pass` FROM `bots` WHERE `id` = '%d'", client->clientid);
50 if ((row = mysql_fetch_row(res)) != NULL) {
51 if(row[1] && row[2]) {
52 putsock(client, "OPER %s %s", row[1], row[2]);
54 putsock(client, "MODE %s +%s", client->user->nick, row[0]);
57 printf_mysql_query("SELECT `channel_name`, `channel_key` FROM `bot_channels` LEFT JOIN `channels` ON `chanid` = `channel_id` WHERE `botid` = '%d' AND `suspended` = '0'", client->clientid);
60 while ((row = mysql_fetch_row(res)) != NULL) {
61 putsock(client, "JOIN %s %s", row[0], row[1]);
65 static void neonbackup_trigger_callback(int clientid, struct ChanNode *chan, char *trigger) {
68 loadChannelSettings(chan);
69 if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
73 printf_mysql_query("SELECT `trigger`, `defaulttrigger` FROM `bot_channels` LEFT JOIN `bots` ON `botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '%d'", chan->channel_id, BOTID);
75 if(!(row = mysql_fetch_row(res))) {
80 strcpy(trigger, row[0]);
82 strcpy(trigger, ((row[1] && *row[1]) ? row[1] : "+"));
85 static void start_bots(int type) {
86 struct ClientSocket *client;
87 MYSQL_RES *res, *res2;
90 if(type == MODSTATE_STARTSTOP) {
91 printf_mysql_query("SELECT `nick`, `ident`, `realname`, `server`, `port`, `pass`, `textbot`, `id`, `queue`, `ssl`, `bind`, `secret` FROM `bots` WHERE `botclass` = '%d' AND `active` = '1'", BOTID);
94 while ((row = mysql_fetch_row(res)) != NULL) {
95 client = create_socket(row[3], atoi(row[4]), row[10], row[5], row[0], row[1], row[2]);
96 client->flags |= (strcmp(row[6], "0") ? SOCKET_FLAG_PREFERRED : 0);
97 client->flags |= (strcmp(row[8], "0") ? SOCKET_FLAG_USE_QUEUE : 0);
98 client->flags |= (strcmp(row[9], "0") ? SOCKET_FLAG_SSL : 0);
99 client->flags |= (strcmp(row[11], "0") ? SOCKET_FLAG_SECRET_BOT : 0);
100 client->flags |= SOCKET_FLAG_REQUEST_INVITE | SOCKET_FLAG_REQUEST_OP;
101 client->botid = BOTID;
102 client->clientid = atoi(row[7]);
103 connect_socket(client);
107 printf_mysql_query("SELECT `command`, `function`, `parameters`, `global_access`, `chan_access`, `flags` FROM `bot_binds` WHERE `botclass` = '%d'", BOTID);
109 while ((row = mysql_fetch_row(res2)) != NULL) {
110 if(bind_cmd_to_command(BOTID, row[0], row[1])) {
111 if(row[2] && strcmp(row[2], "")) {
112 bind_set_parameters(BOTID, row[0], row[2]);
115 bind_set_global_access(BOTID, row[0], atoi(row[3]));
118 bind_set_channel_access(BOTID, row[0], row[4]);
120 if(strcmp(row[5], "0"))
121 bind_set_bind_flags(BOTID, row[0], atoi(row[5]));
124 bind_unbound_required_functions(BOTID);
127 void neonbackup_recover_chan(struct ChanNode *chan) {
128 struct ClientSocket *bot = getChannelBot(chan, BOTID); // prefer backup bot ;)
129 struct ChanUser *chanuser = (bot ? getChanUser(bot->user, chan) : NULL);
130 if(!chanuser || !(chanuser->flags & CHANUSERFLAG_OPPED)) {
131 //search an opped bot
132 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
133 if((chanuser = getChanUser(bot->user, chan)) && (chanuser->flags & CHANUSERFLAG_OPPED))
137 if(!bot) //no opped bots present... channel can't be recovered
139 struct ClientSocket *target;
140 struct ModeBuffer *modeBuf;
141 modeBuf = initModeBuffer(bot, chan);
142 for(target = getBots(SOCKET_FLAG_READY, NULL); target; target = getBots(SOCKET_FLAG_READY, target)) {
143 if((target->flags & SOCKET_FLAG_REQUEST_OP) && (chanuser = getChanUser(target->user, chan)) && !(chanuser->flags & CHANUSERFLAG_OPPED)) {
144 modeBufferOp(modeBuf, target->user->nick);
147 freeModeBuffer(modeBuf);
150 static void neonbackup_event_join(struct ChanUser *chanuser) {
151 struct ClientSocket *client = getChannelBot(chanuser->chan, BOTID);
152 if(!client) return; //we can't "see" this event
153 if(chanuser->user == client->user) {
154 requestOp(client->user, chanuser->chan);
155 neonbackup_recover_chan(chanuser->chan);
159 static void neonbackup_event_invite(struct ClientSocket *client, struct UserNode *user, char *channel) {
160 if(client->botid != BOTID)
164 printf_mysql_query("SELECT `botid`, `bot_channels`.`id`, `suspended` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` LEFT JOIN `channels` ON `chanid` = `channel_id` WHERE `channel_name` = '%s' AND `botclass` = '%d'", escape_string(channel), client->botid);
166 if ((row = mysql_fetch_row(res)) == NULL) {
167 reply(client, user, "NS_INVITE_FAIL", channel, client->user->nick);
170 if(!strcmp(row[2], "1")) {
171 reply(client, user, "MODCMD_CHAN_SUSPENDED");
174 int botid = atoi(row[0]);
175 struct ClientSocket *bot;
176 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
177 if(bot->clientid == botid)
181 struct ChanNode *chan = getChanByName(channel);
182 if(chan && isUserOnChan(bot->user, chan)) {
183 reply(client, user, "NS_INVITE_ON_CHAN", bot->user->nick, chan->name);
185 putsock(bot, "JOIN %s", channel);
189 void init_NeonBackup(int type) {
190 set_bot_alias(BOTID, BOTALIAS);
193 if(type == MODSTATE_REBIND) return;
196 bind_bot_ready(neonbackup_bot_ready, module_id);
197 bind_join(neonbackup_event_join, module_id);
198 bind_invite(neonbackup_event_invite, module_id);
200 set_trigger_callback(BOTID, module_id, neonbackup_trigger_callback);
202 register_default_language_table(msgtab);
205 void free_NeonBackup(int type) {
206 unbind_allcmd(BOTID);
207 if(type == MODSTATE_STARTSTOP) {
208 //disconnect all our bots
209 struct ClientSocket *client;
210 for(client = getBots(0, NULL); client; client = getBots(0, client)) {
211 if(client->botid == BOTID) {
212 unbind_botwise_allcmd(0, client->clientid);
213 close_socket(client);