src/modules/NeonServ.mod/cmd_neonserv_search.c

   1 /* cmd_neonserv_search.c - NeonServ v5.4
   2  * Copyright (C) 2011-2012  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_neonserv.h"
  19
  20 #define CMD_SEARCH_FLAG_HAS_NODELETE 0x001
  21 #define CMD_SEARCH_FLAG_NOT_NODELETE 0x002
  22 #define CMD_SEARCH_FLAG_HAS_SUSPENDED 0x004
  23 #define CMD_SEARCH_FLAG_NOT_SUSPENDED 0x008
  24 #define CMD_SEARCH_FLAG_IS_JOINED  0x010
  25 #define CMD_SEARCH_FLAG_NOT_JOINED 0x020
  26 #define CMD_SEARCH_FLAG_IS_OPPED   0x040
  27 #define CMD_SEARCH_FLAG_NOT_OPPED  0x080
  28 #define CMD_SEARCH_FLAG_IS_VOICED  0x100
  29 #define CMD_SEARCH_FLAG_NOT_VOICED 0x200
  30 #define CMD_SEARCH_FLAG_STATES     0x3f0
  31
  32 struct neonserv_cmd_search_criteria {
  33     char *name;
  34     char *registrar;
  35     char *onchan;
  36     unsigned int flags : 16;
  37     unsigned int unvisited;
  38     unsigned int registered;
  39     unsigned int limit : 16;
  40 };
  41
  42 CMD_BIND(neonserv_cmd_search) {
  43     //ok parse the criterias
  44     struct neonserv_cmd_search_criteria criteria;
  45     memset(&criteria, 0, sizeof(criteria));
  46     criteria.limit = 50;
  47     int i, show_chans = 0, positive;
  48     if(!stricmp(argv[0], "print")) {
  49         show_chans = 1;
  50     }
  51     for(i = 1; i < argc; i += 2) {
  52         if(argc <= i+1) {
  53             reply(getTextBot(), user, "MODCMD_LESS_PARAM_COUNT");
  54             return;
  55         }
  56         if(!stricmp(argv[i], "name")) criteria.name = argv[i+1];
  57         else if(!stricmp(argv[i], "registrar")) criteria.registrar = argv[i+1];
  58         else if(!stricmp(argv[i], "onchan")) criteria.onchan = argv[i+1];
  59         else if(!stricmp(argv[i], "unvisited")) criteria.unvisited = strToTime(user, argv[i+1]);
  60         else if(!stricmp(argv[i], "registered")) criteria.registered = strToTime(user, argv[i+1]);
  61         else if(!stricmp(argv[i], "flags")) {
  62             if(argv[i+1][0] == '+') {
  63                 positive = 1;
  64                 argv[i+1]++;
  65             } else if(argv[i+1][0] == '-') {
  66                 positive = 0;
  67                 argv[i+1]++;
  68             } else
  69                 positive = 1;
  70             if(!stricmp(argv[i+1], "nodelete")) {
  71                 if(positive)
  72                     criteria.flags |= CMD_SEARCH_FLAG_HAS_NODELETE;
  73                 else
  74                     criteria.flags |= CMD_SEARCH_FLAG_NOT_NODELETE;
  75             } else if(!stricmp(argv[i+1], "suspended")) {
  76                 if(positive)
  77                     criteria.flags |= CMD_SEARCH_FLAG_HAS_SUSPENDED;
  78                 else
  79                     criteria.flags |= CMD_SEARCH_FLAG_NOT_SUSPENDED;
  80             }
  81         }
  82         else if(!stricmp(argv[i], "state")) {
  83             if(argv[i+1][0] == '+') {
  84                 positive = 1;
  85                 argv[i+1]++;
  86             } else if(argv[i+1][0] == '-') {
  87                 positive = 0;
  88                 argv[i+1]++;
  89             } else
  90                 positive = 1;
  91             if(!stricmp(argv[i+1], "joined")) {
  92                 if(positive)
  93                     criteria.flags |= CMD_SEARCH_FLAG_IS_JOINED;
  94                 else
  95                     criteria.flags |= CMD_SEARCH_FLAG_NOT_JOINED;
  96             } else if(!stricmp(argv[i+1], "opped")) {
  97                 if(positive)
  98                     criteria.flags |= CMD_SEARCH_FLAG_IS_OPPED;
  99                 else
 100                     criteria.flags |= CMD_SEARCH_FLAG_NOT_OPPED;
 101             } else if(!stricmp(argv[i+1], "voiced")) {
 102                 if(positive)
 103                     criteria.flags |= CMD_SEARCH_FLAG_IS_VOICED;
 104                 else
 105                     criteria.flags |= CMD_SEARCH_FLAG_NOT_VOICED;
 106             }
 107         }
 108         else if(!stricmp(argv[i], "limit")) {
 109             criteria.limit = atoi(argv[i+1]);
 110         }
 111     }
 112     int matches = 0;
 113     reply(getTextBot(), user, "NS_SEARCH_HEADER");
 114     MYSQL_RES *res, *res2;
 115     MYSQL_ROW row, row2;
 116     printf_mysql_query("SELECT `channel_name`, `user_user`, `channel_registered`, `channel_nodelete`, `suspended`, `channel_id` FROM `bot_channels` LEFT JOIN `bots` ON `bots`.`id` = `botid` LEFT JOIN `channels` ON `chanid` = `channel_id` LEFT JOIN `users` ON `channel_registrator` = `user_id` WHERE `botclass` = '%d' AND `active` = '1'", client->botid);
 117     res = mysql_use();
 118     while ((row = mysql_fetch_row(res)) != NULL) {
 119         if(show_chans && matches == criteria.limit) {
 120             //too many
 121             break;
 122         }
 123         if(criteria.name && match(criteria.name, row[0])) continue;
 124         if(criteria.registrar && row[1] && match(criteria.registrar, row[1])) continue;
 125         if(criteria.unvisited) {
 126             printf_mysql_query("SELECT `chanuser_seen` FROM `chanusers` WHERE `chanuser_cid` = '%s' ORDER BY `chanuser_seen` DESC LIMIT 1", row[5]);
 127             res2 = mysql_use();
 128             row2 = mysql_fetch_row(res2);
 129             if(!row2) continue;
 130             if((time(0) - atoi(row2[0])) < criteria.unvisited) continue;
 131         }
 132         if(criteria.registered && (time(0) - atoi(row[2])) < criteria.registered) continue;
 133
 134         if((criteria.flags & CMD_SEARCH_FLAG_HAS_NODELETE) && strcmp(row[3], "1")) continue;
 135         if((criteria.flags & CMD_SEARCH_FLAG_NOT_NODELETE) && strcmp(row[3], "0")) continue;
 136         if((criteria.flags & CMD_SEARCH_FLAG_HAS_SUSPENDED) && strcmp(row[4], "1")) continue;
 137         if((criteria.flags & CMD_SEARCH_FLAG_NOT_SUSPENDED) && strcmp(row[4], "0")) continue;
 138         if((criteria.flags & CMD_SEARCH_FLAG_STATES) || criteria.onchan) {
 139             struct ChanNode *channel = getChanByName(row[0]);
 140             if(criteria.onchan) {
 141                 if(!channel) continue;
 142                 struct ChanUser *chanuser = NULL;
 143                 for(chanuser = getChannelUsers(channel, NULL); chanuser; chanuser = getChannelUsers(channel, chanuser)) {
 144                     if(!match(criteria.onchan, chanuser->user->nick)) break;
 145                 }
 146                 if(!chanuser) continue;
 147             }
 148             if((criteria.flags & CMD_SEARCH_FLAG_IS_JOINED) && !channel) continue;
 149             if((criteria.flags & CMD_SEARCH_FLAG_NOT_JOINED) && channel) continue;
 150             if(channel && (criteria.flags & (CMD_SEARCH_FLAG_IS_OPPED | CMD_SEARCH_FLAG_NOT_OPPED | CMD_SEARCH_FLAG_IS_VOICED | CMD_SEARCH_FLAG_NOT_VOICED))) {
 151                 int flags = 0;
 152                 struct ClientSocket *bot;
 153                 for(bot = getBots(SOCKET_FLAG_READY, NULL); bot; bot = getBots(SOCKET_FLAG_READY, bot)) {
 154                     if(bot->botid == client->botid) {
 155                         struct ChanUser *chanuser = getChanUser(bot->user, channel);
 156                         if(chanuser) {
 157                             flags = chanuser->flags;
 158                             break;
 159                         }
 160                     }
 161                 }
 162                 if((criteria.flags & CMD_SEARCH_FLAG_IS_OPPED) && !(flags & CHANUSERFLAG_OPPED)) continue;
 163                 if((criteria.flags & CMD_SEARCH_FLAG_NOT_OPPED) && (flags & CHANUSERFLAG_OPPED)) continue;
 164                 if((criteria.flags & CMD_SEARCH_FLAG_IS_VOICED) && !(flags & CHANUSERFLAG_VOICED)) continue;
 165                 if((criteria.flags & CMD_SEARCH_FLAG_NOT_VOICED) && (flags & CHANUSERFLAG_VOICED)) continue;
 166             }
 167         }
 168         matches++;
 169         //output
 170         if(show_chans) {
 171             reply(getTextBot(), user, "%s", row[0]);
 172         }
 173     }
 174     reply(getTextBot(), user, "NS_TABLE_COUNT", matches);
 175 }