Usermgm.pm

   1 #**************************************************************************
   2 #   Copyright (C) 2006 by Michael Andreen <harvATruinDOTnu>               *
   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 2 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, write to the                         *
  16 #   Free Software Foundation, Inc.,                                       *
  17 #   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  18 #**************************************************************************/
  19 package ND::IRC::Usermgm;
  20 use strict;
  21 use warnings;
  22 use ND::DB;
  23 use ND::IRC::Access;
  24 require Exporter;
  25
  26 our @ISA = qw/Exporter/;
  27
  28 our @EXPORT = qw/addUser whois flags flag laston addPoints chattrG setHost setPNick deactivateUser getShips/;
  29
  30 sub addUser {
  31         my ($nick,$pnick) = @_;
  32         DB();
  33         if (hc()){
  34                 $pnick = $nick unless $pnick;
  35                 my $host = "$pnick.users.netgamers.org";
  36                 my ($username,$hostname,$p_nick) = $ND::DBH->selectrow_array(q{SELECT username, hostmask,pnick
  37                         FROM users WHERE username ILIKE ? OR hostmask ILIKE ? OR pnick ILIKE ?}
  38                         ,undef,$nick,$host,$pnick);
  39                 if ((not defined $username) && $ND::DBH->do("INSERT INTO users (username,hostmask,pnick,password) VALUES(?,?,?,'')"
  40                                 ,undef,$nick,$host,$pnick)){
  41                         $ND::server->command("msg $ND::target Added $ND::B$nick(/$pnick)$ND::B with host: $ND::B$host$ND::B");
  42                 }elsif(defined $username){
  43                         $ND::server->command("msg $ND::target $ND::B$username ($p_nick)$ND::B already exists with host: $ND::B$hostname$ND::B.");
  44
  45                 }else{
  46                         $ND::server->command("msg $ND::target Something went wrong when trying to add $ND::B$nick ($pnick)$ND::B with host: $ND::B$host$ND::B, ".$ND::DBH->errstr);
  47                 }
  48         }else{
  49                 $ND::server->command("msg $ND::target Only HCs are allowed to add users");
  50         }
  51 }
  52 sub whois {
  53         my ($nick) = @_;
  54         DB();
  55         if (officer()){
  56                 my $f = $ND::DBH->prepare("SELECT username, hostmask, concat(flag) FROM users u LEFT OUTER JOIN (SELECT uid,flag FROM groupmembers NATURAL JOIN groups ORDER BY uid,flag ) g ON g.uid = u.uid  WHERE username ILIKE ? GROUP BY username,hostmask");
  57                 $f->execute($nick);
  58                 while (my @row = $f->fetchrow()){
  59                         $ND::server->command("msg $ND::target $row[0] flags: ($row[2]) host: $row[1]");
  60                 }
  61                 if ($f->rows == 0){
  62                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
  63                 }
  64         }else{
  65                 $ND::server->command("msg $ND::target Only officers are allowed to check that");
  66         }
  67 }
  68
  69 sub flags {
  70         my ($nick) = @_;
  71         DB();
  72         unless ($1){
  73                 my ($flags) = $ND::DBH->selectrow_array("SELECT TRIM(', ' FROM concat(flag||':'||groupname||', ')) FROM groups");
  74                 $ND::server->command("msg $ND::target $flags");
  75         }elsif (hc()){
  76                 my $f = $ND::DBH->prepare("SELECT username, concat(flag), TRIM(', ' FROM concat(groupname||', ')) FROM users u LEFT OUTER JOIN (SELECT uid,flag,groupname FROM groupmembers NATURAL JOIN groups ORDER BY uid,flag ) g ON g.uid = u.uid  WHERE username ILIKE ? GROUP BY username,hostmask");
  77                 $f->execute($nick);
  78                 while (my @row = $f->fetchrow()){
  79                         $ND::server->command("msg $ND::target Flags for $row[0] on: $ND::target: $row[1]| (Global: $row[2])");
  80                 }
  81                 if ($f->rows == 0){
  82                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
  83                 }
  84         }else{
  85                 $ND::server->command("msg $ND::target Only HCs are allowed to check that");
  86         }
  87 }
  88
  89 sub flag {
  90         my ($flag) = @_;
  91
  92         if (officer()|| ($ND::target eq $ND::scanchan && $flag eq 'S')){
  93                 my $f = $ND::DBH->prepare(qq{
  94 SELECT TRIM(', ' FROM concat(username||', ')),count(username) FROM
  95         (SELECT uid, username FROM users ORDER BY username) u NATURAL JOIN groupmembers gm
  96         JOIN groups g ON g.gid = gm.gid
  97 WHERE flag = ?;
  98                         });
  99                 if (my ($users,$count) = $ND::DBH->selectrow_array($f,undef,$flag)){
 100                         $ND::server->command("msg $ND::target $ND::B$count$ND::B Users with flag $ND::B$flag$ND::B: $users");
 101                 }
 102         }else{
 103                 $ND::server->command("msg $ND::target Only officers are allowed to check that");
 104         }
 105 }
 106
 107 sub laston {
 108         my ($flag,$min) = @_;
 109
 110         if (officer()){
 111                 my $f = $ND::DBH->prepare(qq{SELECT username,last
 112                         FROM (SELECT uid,username, date_part('day',now() - laston)::int AS last,laston FROM users) u NATURAL JOIN groupmembers NATURAL JOIN groups WHERE flag = ? AND (last >= ? OR last IS NULL) ORDER BY laston
 113                         });
 114                 $min = 0 unless defined $min;
 115                 $f->execute($flag,$min);
 116                 my $text;
 117                 my $i = 0;
 118                 while (my $user = $f->fetchrow_hashref){
 119                         $user->{last} = '?' unless defined $user->{last};
 120                         $text .= "$user->{username}($user->{last}) ";
 121                         $i++;
 122                 }
 123                 $ND::server->command("msg $ND::target $ND::B$i$ND::B Users(days) with flag $ND::B$flag$ND::B: $text");
 124         }else{
 125                 $ND::server->command("msg $ND::target Only officers are allowed to check that");
 126         }
 127 }
 128
 129
 130 sub getShips {
 131         my ($ship) = @_;
 132
 133         if (officer() || dc()){
 134                 my $f = $ND::DBH->prepare(qq{SELECT username,amount
 135                         FROM users u JOIN fleets f USING (uid) JOIN fleet_ships fs ON f.id = fs.fleet WHERE f.fleet = 0 AND ship ILIKE ? ORDER BY amount DESC
 136                         });
 137                 $f->execute($ship);
 138                 my $text;
 139                 my $i = 0;
 140                 my $total = 0;
 141                 while (my $user = $f->fetchrow_hashref){
 142                         $user->{last} = '?' unless defined $user->{last};
 143                         $text .= "$user->{username}: $user->{amount} ";
 144                         $i++;
 145                         $total += $user->{amount};
 146                 }
 147                 if ($text){
 148                         $ND::server->command("notice $ND::nick $ND::B$i$ND::B Users with $ND::B$total $ship$ND::B: $text");
 149                 }else{
 150                         $ND::server->command("msg $ND::target $ND::B$i$ND::B Couldn't find any user with $ND::B$ship$ND::B:");
 151                 }
 152         }
 153 }
 154
 155 sub addPoints {
 156         my ($t,$nick,$p) = @_;
 157         DB();
 158         if (   ($t eq "d" && dc())
 159                 || ($t eq "a" && bc())
 160                 || ($t eq "h" && officer())
 161                 || ($t eq "s" && scanner())){
 162                 my $points = 1;
 163                 if ($p){
 164                         $points = $p;
 165                 }
 166                 if ($points*$points > 400){
 167                         $ND::server->command("msg $ND::target Values between -20 and 20 please");
 168                         return;
 169                 }
 170                 my $f = $ND::DBH->prepare("SELECT uid,username FROM users WHERE username ILIKE ?");
 171                 $f->execute($nick);
 172                 my @row = $f->fetchrow();
 173                 if ($f->rows == 1){
 174                         my $type = "defense";
 175                         $type = "attack" if $t eq "a";
 176                         $type = "humor" if $t eq "h";
 177                         $type = "scan" if $t eq "s";
 178                         my ($fleets) = $ND::DBH->selectrow_array('SELECT count(*) FROM raids r JOIN raid_targets rt ON r.id = rt.raid JOIN raid_claims rc ON rt.id = rc.target WHERE not launched AND uid = ? AND tick + 24 > tick();',undef,$row[0]);
 179                         if ($t eq 'a' && $fleets > 0 && $points > 0){
 180                                 $ND::server->command("msg $ND::target $row[1] has $fleets claimed waves the last 24 ticks that aren't marked as launched, so no points.");
 181                                 return;
 182                         }
 183                         $type .= "_points";
 184                         $ND::DBH->do("UPDATE users SET $type = $type + ? WHERE uid = ?",undef,$points,$row[0]);
 185                         $ND::server->command("msg $ND::target $row[1] has been given $points $type");
 186                 }elsif ($f->rows == 0){
 187                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
 188                 }else{
 189                         $ND::server->command("msg $ND::target More than 1 user matched, please refine the search");
 190                 }
 191                 $f->finish;
 192
 193         }else{
 194                 $ND::server->command("msg $ND::target You don't have access for that");
 195         }
 196 }
 197
 198 sub chattrG {
 199         my ($nick, $flags) = @_;
 200         DB();
 201         if (hc() || ($flags =~ /^(\+|-)?x$/ && $ND::address eq 'Assassin.users.netgamers.org')){
 202                 my $f = $ND::DBH->prepare("SELECT uid,username FROM users WHERE username ILIKE ?");
 203                 $f->execute($nick);
 204                 my @user = $f->fetchrow();
 205                 if ($f->rows == 1){
 206                         my $add = 1;
 207                         $flags =~ /^(-)/;
 208                         my $update;
 209                         if ($1 eq "-"){
 210                                 $update = $ND::DBH->prepare("DELETE FROM groupmembers WHERE uid = ? AND gid = (SELECT gid FROM groups WHERE flag = ?)");
 211                         }else{
 212                                 $update = $ND::DBH->prepare("INSERT INTO groupmembers (uid,gid) VALUES(?,(SELECT gid FROM groups WHERE flag = ?))");
 213                         }
 214                         while ($flags =~ m/(\w)/g){
 215                                 $update->execute($user[0],$1);
 216                         }
 217                         $update = $ND::DBH->prepare("SELECT concat(flag) FROM (SELECT uid,flag FROM groupmembers NATURAL JOIN groups ORDER BY uid,flag ) g WHERE uid = ? ");
 218                         my @flags = $ND::DBH->selectrow_array($update,undef,$user[0]);
 219                         $ND::server->command("msg $ND::target Global flags for $user[1] are now: $flags[0]");
 220                 }elsif ($f->rows == 0){
 221                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
 222                 }else{
 223                         $ND::server->command("msg $ND::target More than 1 user matched, please refine the search");
 224                 }
 225                 $f->finish;
 226         }
 227 }
 228
 229 sub setHost {
 230         my ($nick, $host) = @_;
 231         DB();
 232         if (hc()){
 233                 my $f = $ND::DBH->prepare("SELECT uid,username FROM users WHERE username ILIKE ?");
 234                 $f->execute($nick);
 235                 my ($uid,$nick) = $f->fetchrow();
 236                 if ($f->rows == 1){
 237                         my ($username,$hostname) = $ND::DBH->selectrow_array("SELECT username, hostmask FROM users WHERE hostmask ILIKE ? AND NOT (username ILIKE ?)",undef,$host,$nick);
 238                         if ((not defined $username) && $ND::DBH->do("UPDATE users SET hostmask = ? WHERE uid = ?",undef,$host,$uid) > 0){
 239                                 $ND::server->command("msg $ND::target Updated $ND::B$nick${ND::B}'s host to: $ND::B$host$ND::B");
 240                         }elsif(defined $username){
 241                                 $ND::server->command("msg $ND::target $ND::B$username$ND::B already exists with host: $ND::B$hostname$ND::B.");
 242                         }else{
 243                                 $ND::server->command("msg $ND::target Couldn't update $ND::B$username${ND::B}'s host");
 244                         }
 245                 }elsif ($f->rows == 0){
 246                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
 247                 }else{
 248                         $ND::server->command("msg $ND::target More than 1 user matched, please refine the search");
 249                 }
 250                 $f->finish;
 251         }
 252 }
 253
 254 sub setPNick {
 255         my ($nick, $pnick) = @_;
 256         DB();
 257         if (hc()){
 258                 my $f = $ND::DBH->prepare("SELECT uid,username FROM users WHERE username ILIKE ?");
 259                 $f->execute($nick);
 260                 my ($uid,$nick) = $f->fetchrow();
 261                 if ($f->rows == 1){
 262                         my ($username,$p_nick) = $ND::DBH->selectrow_array("SELECT username, pnick FROM users WHERE pnick ILIKE ? AND NOT (username ILIKE ?)",undef,$pnick,$nick);
 263                         if ((not defined $username) && $ND::DBH->do("UPDATE users SET pnick = ? WHERE uid = ?",undef,$pnick,$uid) > 0){
 264                                 $ND::server->command("msg $ND::target Updated $ND::B$nick${ND::B}'s pnick to: $ND::B$pnick$ND::B");
 265                         }elsif(defined $username){
 266                                 $ND::server->command("msg $ND::target $ND::B$username$ND::B already exists with pnick $ND::B$p_nick$ND::B.");
 267                         }else{
 268                                 $ND::server->command("msg $ND::target Couldn't update $ND::B$username${ND::B}'s host");
 269                         }
 270                 }elsif ($f->rows == 0){
 271                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
 272                 }else{
 273                         $ND::server->command("msg $ND::target More than 1 user matched, please refine the search");
 274                 }
 275                 $f->finish;
 276         }
 277 }
 278
 279 sub deactivateUser {
 280         my ($nick) = @_;
 281         DB();
 282         if (hc()){
 283                 my $f = $ND::DBH->prepare("SELECT uid,username FROM users WHERE username ILIKE ?");
 284                 $f->execute($nick);
 285                 my ($uid,$username) = $f->fetchrow();
 286                 if ($f->rows == 1){
 287                         my $updated = $ND::DBH->do("UPDATE users SET hostmask = ?, password = '' WHERE uid = ?",undef,$username,$uid);
 288                         if ($updated > 0){
 289                                 my $groups = $ND::DBH->do("DELETE FROM groupmembers WHERE uid = ?",undef,$uid);
 290                                 $ND::server->command("msg $ND::target $ND::B$username$ND::B has been deactivated.");
 291                         }else{
 292                                 $ND::server->command("msg $ND::target Something went wrong when trying to modify $ND::B$username$ND::B");
 293                         }
 294                 }elsif ($f->rows == 0){
 295                         $ND::server->command("msg $ND::target No hit, maybe spelling mistake, or add % as wildcard");
 296                 }else{
 297                         $ND::server->command("msg $ND::target More than 1 user matched, please refine the search");
 298                 }
 299                 $f->finish;
 300         }
 301 }
 302
 303 1;