#!/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;

# 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-group ( (check|remove|get-members|get-carts|get-reports) <groupname>            |\n" .
               "                   add <groupname> [ <description> ]                                      |\n" .
               "                   (add-member|remove-member|is-member) <groupname> <user>                |\n" .
               "                   set-carts <groupname> [ <low> [ <high> [ <type> [ <enforce range> ]]]  |\n" .
               "                   set-reports <groupname> [ <nownext> [ <traffic> [ <music> ]]]  )\n";
  exit 1;
}

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

my ($dbh, undef, $errorstring) = RHRD::rddb::opendb();
if(defined $dbh) {
  if($cmd eq "check") {
    (my $result, my $status, $errorstring) = RHRD::rddb::check_group($dbh, $groupname);
    print "$result, $status: $errorstring\n";
    RHRD::rddb::closedb($dbh);
    exit $result;
  } elsif($cmd eq "add") {
    (my $cnt, undef, $errorstring) = RHRD::rddb::add_group($dbh, $groupname, $ARGV[2]);
    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_group($dbh, $groupname);
    if(!defined $results[0] && defined $results[2]) {
      print "$results[2]\n";
    } else {
      for my $href (@results) {
        print int($href->{cnt}) . " " . $href->{name} . " deleted\n";
      }
    }
  } elsif($cmd eq "get-members") {
    my @users = RHRD::rddb::get_group_members($dbh, $groupname);
    if(!defined $users[0] && defined $users[1]) {
      print "$users[2]\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    for my $user (sort @users) {
      print "$user\n";
    }
  } elsif($cmd eq "add-member") {
    (my $cnt, undef, $errorstring) = RHRD::rddb::add_group_member($dbh, $groupname, $ARGV[2]);
    unless(defined $cnt) {
      print "$errorstring\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print int($cnt) . " rows affected\n";
  } elsif($cmd eq "remove-member") {
    (my $cnt, undef, $errorstring) = RHRD::rddb::remove_group_member($dbh, $groupname, $ARGV[2]);
    unless(defined $cnt) {
      print "$errorstring\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print int($cnt) . " rows affected\n";
  } elsif($cmd eq "is-member") {
    (my $cnt, undef, $errorstring) = RHRD::rddb::is_group_member($dbh, $groupname, $ARGV[2]);
    RHRD::rddb::closedb($dbh);
    unless(defined $cnt) {
      print "$errorstring\n";
      exit 1;
    }
    print $ARGV[2] . " is" . (($cnt) ? "" : " not") . " a member\n";
    exit (($cnt) ? 0 : 1);
  } elsif($cmd eq "get-carts") {
    my ($low, $high, $type, $enforce_range) = RHRD::rddb::get_group_carts($dbh, $groupname);
    unless(defined $low) {
      print "$type\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print "Range: " . int($low) . " - " . int($high) . ", Type: " . int($type) . ", Enforce Range: " . $enforce_range . "\n";
  } elsif($cmd eq "set-carts") {
    (my $cnt, undef, $errorstring) = RHRD::rddb::set_group_carts($dbh, $groupname, $ARGV[2], $ARGV[3], $ARGV[4], $ARGV[5]);
    unless(defined $cnt) {
      print "$errorstring\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print int($cnt) . " rows affected\n";
  } elsif($cmd eq "get-reports") {
    my ($nownext, $traffic, $music) = RHRD::rddb::get_group_reports($dbh, $groupname);
    unless(defined $nownext) {
      print "$music\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print "Now-Next: " . $nownext . ", Traffic: " . $traffic . ", Music: " . $music . "\n";
  } elsif($cmd eq "set-reports") {
    (my $cnt, undef, $errorstring) = RHRD::rddb::set_group_reports($dbh, $groupname, $ARGV[2], $ARGV[3], $ARGV[4]);
    unless(defined $cnt) {
      print "$errorstring\n";
      RHRD::rddb::closedb($dbh);
      exit 1;
    }
    print int($cnt) . " rows affected\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