Converted the .flags command

[NDIRC.git] / Commands / Usermgm.pm

diff --git a/Commands/Usermgm.pm b/Commands/Usermgm.pm
index 5ec8a2727c8eba3ff8cb286353283da4d2082bf5..8a87f8e1fae85147459336c9e4bcf654420c6f29 100644 (file)
--- a/Commands/Usermgm.pm
+++ b/Commands/Usermgm.pm
@@ -86,4 +86,98 @@ UPDATE users SET hostmask = ?, password = '' WHERE uid = ?
        $f->finish;
 }
 
+sub chattr
+       : Help(syntax: .chattr username [-]flags | % can be used for wildcards \%arro% will match barrow, if a - is given then flags will be removed, otherwise added)
+       : ACL(irc_chattr)
+{
+       my ($self,$c,$msg) = @_;
+
+       my ($nick, $flags) = $msg =~ /^(\S+) ((\+|-)?\w+)$/ or die 'ARGS';
+       my $dbh = $c->model;
+
+       my $f = $dbh->prepare(q{SELECT uid,username FROM users WHERE username ILIKE ?});
+       $f->execute($nick);
+       my $user = $f->fetchrow_hashref;
+       if ($f->rows == 1){
+               my $update;
+               if ($flags =~ /^(-)/){
+                       $update = $dbh->prepare(q{
+DELETE FROM groupmembers WHERE uid = $1 AND
+       gid IN (SELECT gid FROM groups WHERE flag = ANY($2))
+                       });
+               }else{
+                       $update = $dbh->prepare(q{
+INSERT INTO groupmembers (uid,gid) (SELECT $1,gid FROM groups
+       WHERE flag = ANY($2) AND gid NOT IN (SELECT gid FROM groupmembers WHERE uid = $1))
+                       });
+               }
+               my @flags = split /\W*/,$flags;
+               $update->execute($user->{uid},\@flags);
+               $update = $dbh->prepare(q{
+SELECT concat(flag)
+FROM (SELECT uid,flag FROM groupmembers NATURAL JOIN groups ORDER BY uid,flag ) g
+WHERE uid = ?
+               });
+               $flags = $dbh->selectrow_array($update,undef,$user->{uid});
+               $c->reply("Flags for <b>$user->{username}</b> are now: $flags");
+       }elsif ($f->rows == 0){
+               $c->reply("No hit, maybe spelling mistake, or add % as wildcard");
+       }else{
+               $c->reply("More than 1 user matched, please refine the search");
+       }
+       $f->finish;
+}
+
+sub whois
+       : Help(syntax: .whois [username] | % can be used for wildcards \%arro% will match barrow. If no username is given then all flags will be listed)
+       : Alias(flags)
+       : ACL(irc_whois)
+{
+       my ($self,$c,$msg) = @_;
+       my ($nick) = $msg =~ /^(\S+)?$/ or die 'ARGS';
+       my $dbh = $c->model;
+
+       unless ($nick){
+               my ($flags) = $dbh->selectrow_array(q{
+SELECT TRIM(', ' FROM concat(flag||':'||groupname||', ')) FROM groups
+               });
+               $c->reply("Current flags: $flags");
+               return;
+       }
+
+       my $f = $dbh->prepare(q{
+SELECT username, pnick, hostmask, concat(flag) AS flags
+FROM users u
+       LEFT OUTER JOIN (SELECT uid,flag FROM groupmembers
+               NATURAL JOIN groups ORDER BY uid,flag ) g USING (uid)
+WHERE username ILIKE ?
+GROUP BY username,pnick,hostmask LIMIT 5
+               });
+       $f->execute($nick);
+       while (my $user = $f->fetchrow_hashref){
+               $c->reply("<b>$user->{username} (/$user->{pnick})</b> flags: ($user->{flags}) host: $user->{hostmask}");
+       }
+       if ($f->rows == 0){
+               $c->reply("No hit, maybe spelling mistake, or add % as wildcard");
+       }
+}
+
+sub flag
+       : Help(syntax: .flag flag | Lists all users with the given flag.)
+       : ACL(irc_flag)
+{
+       my ($self,$c,$msg) = @_;
+       my ($flag) = $msg =~ /^(\w)$/ or die 'ARGS';
+
+       my $f = $c->model->prepare(q{
+SELECT TRIM(', ' FROM concat(username||', ')),count(username)
+FROM (SELECT uid, username FROM users ORDER BY username) u
+       JOIN groupmembers gm USING (uid)
+       JOIN groups g USING (gid)
+WHERE flag = $1
+               });
+       my ($users,$count) = $c->model->selectrow_array($f,undef,$flag);
+       $c->reply("<b>$count</b> Users with flag <b>$flag</b>: $users");
+}
+
 1;