#!/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 && $num_args !=3) {
  print STDERR "Usage: rd-user (check|add|remove) <username> [ fullname ]\n";
  exit 1;
}

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

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 => 0);
    (my $cnt, undef, $errorstring) = RHRD::rddb::add_user($dbh, $username, $token, $fullname);
    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";
      }
    }
  } else {
    print STDERR "unknown command\n";
    RHRD::rddb::closedb($dbh);
    exit 1;
  }

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

exit 0