summaryrefslogtreecommitdiff
path: root/utils/rd-user
blob: 520a6538b915470bec5b9201dd3f0997e7e198aa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#!/usr/bin/perl -w
#
#  rhrdlibs
#
#  Copyright (C) 2015 Christian Pointner <equinox@helsinki.at>
#
#  This file is part of rhrdlibs.
#
#  rhrdlibs is free software: you can redistribute it and/or modify
#  it under the terms of the GNU Affero General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  any later version.
#
#  rhrdlibs is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU Affero General Public License for more details.
#
#  You should have received a copy of the GNU Affero General Public License
#  along with rhrdlibs. If not, see <http://www.gnu.org/licenses/>.
#

use strict;
use RHRD::rddb;
use String::MkPasswd qw(mkpasswd);

# this is ridiculous but makes it a little harder to create/remove users...
if ($> != 0 ) {
  print STDERR "this must be run as root!\n";
  exit 1;
}

my $num_args = $#ARGV + 1;
if ($num_args != 2) {
  print STDERR "Usage: rd-user (check|add|remove) <username>\n";
  exit 1;
}

my $cmd = $ARGV[0];
my $username = $ARGV[1];

my ($dbh, undef, $errorstring) = RHRD::rddb::opendb();
if(defined $dbh) {
  if($cmd eq "check") {
    (my $result, my $status, $errorstring) = RHRD::rddb::check_user($dbh, $username);
    print "$result, $status: $errorstring\n";
    RHRD::rddb::closedb($dbh);
    exit $result;
  } elsif($cmd eq "add") {
    my $token = mkpasswd(-length => 16, -minnum => 3, -minupper => 3, -minspecial => 2);
    (my $cnt, undef, $errorstring) = RHRD::rddb::add_user($dbh, $username, $token);
    unless(defined $cnt) {
      print "$errorstring\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print int($cnt) . " rows affected\n";
  } elsif($cmd eq "remove") {
    my @results = RHRD::rddb::remove_user($dbh, $username);
    if(!defined $results[0] && defined $results[2]) {
      print "$results[2]\n";
    } else {
      for my $href (@results) {
        print int($href->{cnt}) . " " . $href->{name} . " deleted\n";
      }
    }
  }

  RHRD::rddb::closedb($dbh);
} else {
  print STDERR "$errorstring\n";
  exit 1;
}

exit 0