#!/usr/bin/perl -w
#
#  rhrdlibs
#
#  Copyright (C) 2015-2016 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);

sub print_usage
{
  print STDERR "Usage: rhrd-user list\n" .
               "       rhrd-user (check|remove|groups|is-grids) <username>\n" .
               "       rhrd-user add <username> [ <fullname> ]\n" .
               "       rhrd-user set-grids <username> (1|0)\n";
}

sub list
{
  my ($ctx) = @_;

  my @users = RHRD::rddb::list_users($ctx);
  if(!defined $users[0] && defined $users[1]) {
    print STDERR "$users[1]: $users[2]";
    return 1;
  }
  for my $user (@users) {
    print $user . "\n";
  }
  return 0;
}

sub check
{
  my ($ctx, $username) = @_;

  my ($result, $status, $errorstring) = RHRD::rddb::check_user($ctx, $username);
  print STDERR "$result, $status: $errorstring\n";
  return $result;
}

sub add
{
  my ($ctx, $username, $fullname) = @_;

  my $token = mkpasswd(-length => 16, -minnum => 3, -minupper => 3, -minspecial => 0);
  my ($cnt, undef, $errorstring) = RHRD::rddb::add_user($ctx, $username, $token, $fullname);
  unless(defined $cnt) {
    print STDERR "$errorstring\n";
    return 1;
  }
  print int($cnt) . " rows affected\n";
  return 0;
}

sub remove
{
  my ($ctx, $username) = @_;

  my @results = RHRD::rddb::remove_user($ctx, $username);
  if(!defined $results[0] && defined $results[2]) {
    print STDERR "$results[2]\n";
    return 1;
  }
  for my $href (@results) {
    print int($href->{cnt}) . " " . $href->{name} . " deleted\n";
  }
  return 0;
}

sub groups
{
  my ($ctx, $username) = @_;

  my @groups = RHRD::rddb::get_user_groups($ctx, $username);
  if(!defined $groups[0] && defined $groups[1]) {
    print STDERR "$groups[1]: $groups[2]";
    return 1;
  }
  for my $user (@groups) {
    print $user . "\n";
  }
  return 0;
}

sub is_grids
{
  my ($ctx, $username) = @_;

  my ($authorized, $status, $errorstring) = RHRD::rddb::is_musicgrid_user($ctx, $username);
  if(!defined $authorized) {
    print STDERR "$status: $errorstring\n";
    return 1;
  }
  print $username . " is" . (($authorized) ? "" : " *not*") . " allowed to edit music grids\n";
  return (($authorized) ? 0 : 1);
}

sub set_grids
{
  my ($ctx, $username, $value) = @_;

  unless(defined($value) && ($value == '0' || $value == '1')) {
    print_usage();
    return 1;
  }

  my ($cnt, $status, $errorstring) = RHRD::rddb::set_musicgrid_user($ctx, $username, $value);
  if(!defined $cnt) {
    print STDERR "$status: $errorstring\n";
    return 1;
  }
  print int($cnt) . " rows affected\n";
  return 0;
}


my $num_args = $#ARGV + 1;
if($num_args < 1) {
  print_usage();
  exit(1);
}
my $cmd = $ARGV[0];
my $username = $ARGV[1];
my $ret = 0;

my ($ctx, undef, $errorstring) = RHRD::rddb::init();
if(defined $ctx) {
  if($cmd eq "list") {
    if($num_args != 1) {
      print_usage();
      $ret = 1;
    } else {
      $ret = list($ctx);
    }
  }
  elsif($cmd eq "check") {
    if($num_args != 2) {
      print_usage();
      $ret = 1;
    } else {
      $ret = check($ctx, $username);
    }
  } elsif($cmd eq "add") {
    if($num_args < 2 || $num_args > 3) {
      print_usage();
      $ret = 1;
    } else {
      $ret = add($ctx, $username, $ARGV[2]);
    }
  } elsif($cmd eq "remove") {
    if($num_args != 2) {
      print_usage();
      $ret = 1;
    } else {
      $ret = remove($ctx, $username);
    }
  } elsif($cmd eq "groups") {
    if($num_args != 2) {
      print_usage();
      $ret = 1;
    } else {
      $ret = groups($ctx, $username);
    }
  } elsif($cmd eq "is-grids") {
    if($num_args != 2) {
      print_usage();
      $ret = 1;
    } else {
      $ret = is_grids($ctx, $username);
    }
  } elsif($cmd eq "set-grids") {
    if($num_args != 3) {
      print_usage();
      $ret = 1;
    } else {
      $ret = set_grids($ctx, $username, $ARGV[2]);
    }
  } else {
    print_usage();
    $ret = 1;
  }

  RHRD::rddb::destroy($ctx);
} else {
  print STDERR "$errorstring\n";
  $ret = 1;
}

exit $ret;