initial commit of file from CVS for smeserver-manager on Fri Mar 22 14:54:28 AEDT 2024

root/usr/share/smanager/lib/SrvMngr/Controller/Quota.pm

@@ -0,0 +1,193 @@
+package SrvMngr::Controller::Quota;
+
+#----------------------------------------------------------------------
+# heading     : User management
+# description : Quotas
+# navigation  : 2000 300
+#----------------------------------------------------------------------
+#
+# routes : end
+#----------------------------------------------------------------------
+use strict;
+use warnings;
+use Mojo::Base 'Mojolicious::Controller';
+
+use esmith::FormMagick::Panel::quota;
+
+#use esmith::TestUtils;
+use Scalar::Util qw(looks_like_number);
+
+use Locale::gettext;
+use SrvMngr::I18N;
+
+use SrvMngr qw(theme_list init_session);
+
+#our $db = esmith::ConfigDB->open || die "Couldn't open config db";
+our $adb = esmith::AccountsDB->open || die "Couldn't open accounts db";
+
+
+sub main {
+
+    my $c = shift;
+    $c->app->log->info($c->log_req);
+
+    my %quo_datas = ();
+    my $title = $c->l('quo_FORM_TITLE');
+
+    $quo_datas{'trt'} = 'LIST';
+
+    my @userAccounts;
+    if ($adb) {
+        @userAccounts = $adb->users();
+    }
+
+    $c->stash( title => $title, quo_datas => \%quo_datas, userAccounts => \@userAccounts );
+    $c->render(template => 'quota');
+
+};
+
+
+sub do_display {
+
+    my $c = shift;
+
+    my $rt = $c->current_route;
+    my $trt = ($c->param('trt') || 'LIST');
+    my $user = $c->param('user') || '';
+
+    $trt = 'UPD' if ( $user );
+
+    my %quo_datas = ();
+    my $title = $c->l('quo_FORM_TITLE');
+
+    $quo_datas{'trt'} = $trt;
+
+        if ( $trt eq 'UPD' ) {
+
+	    my $rec = $adb->get($user);
+	    if ($rec and $rec->prop('type') eq 'user') {
+    		$quo_datas{user} = $user;
+    		$quo_datas{userRec} = $rec;
+		my $max = esmith::FormMagick::Panel::quota->toBestUnit($rec->prop('MaxBlocks'));
+    		$quo_datas{hardlim} = $max;
+		$max = esmith::FormMagick::Panel::quota->toBestUnit($rec->prop('MaxBlocksSoftLim'));
+    		$quo_datas{softlim} = $max;
+	    }
+
+        }
+
+    $c->stash( title => $title, quo_datas => \%quo_datas );
+    $c->render( template => 'quota' );
+
+};
+
+
+sub do_update {
+
+    my $c = shift;
+    $c->app->log->info($c->log_req);
+
+    my $title = $c->l('quo_FORM_TITLE');
+    my %quo_datas = ();
+
+    my $rt = $c->current_route;
+    my $trt = ($c->param('trt') || 'LIST');
+
+    $quo_datas{trt} = $trt;
+    my $result = '';
+    my $res;
+
+    if ( $trt eq 'UPD' ) {
+
+	$quo_datas{user} = ($c->param('user') || '');
+	$quo_datas{softlim} = ($c->param('Soft') || '');
+        $quo_datas{hardlim} = ($c->param('Hard') || '');
+
+	# controls
+	$res = validate_quota( $c, $quo_datas{user}, $quo_datas{softlim}, $quo_datas{hardlim} );
+	$result .= $res unless $res eq 'OK';
+
+	if ( ! $result ) {
+	    $result = $c->l('quo_SUCCESSFULLY_MODIFIED') . ' ' . $quo_datas{user};
+	} else {
+	    $quo_datas{userRec} = $adb->get($quo_datas{user}) || undef;
+	}
+    }
+
+    # common parts
+
+    if ($res ne 'OK') {
+	$c->stash( error => $result );
+	$c->stash( title => $title, quo_datas => \%quo_datas );
+	return $c->render('quota');
+    }
+
+    my $message = "'Quota' updates ($trt) DONE";
+    $c->app->log->info($message);
+    $c->flash( success => $result );
+
+    $c->redirect_to('/quota');
+
+};
+
+
+sub validate_quota {
+    my ($c, $acct, $softlim, $hardlim ) = @_;
+    my $msg;
+
+    my $rec = $adb->get($acct);
+    return $c->l('quo_ERR_NO_SUCH_ACCT') . ' : ' . $acct  unless (defined $rec);
+
+    my $type = $rec->prop('type');
+    unless ($type eq "user") {
+	$msg = $c->l('quo_ERR_NOT_A_USER_ACCT').$acct.$c->l('quo_ACCOUNT_IS_TYPE').$type;
+	return $msg;
+    }
+    my $uid = getpwnam($acct);
+    return $c->l('COULD_NOT_GET_UID').$acct unless ($uid);
+
+    if (($softlim !~ /^(.+?)\s*([KMGT])?$/ ) || (!looks_like_number ($1)))  {
+	return $c->l('quo_SOFT_VAL_MUST_BE_NUMBER');
+    }
+    
+   my $exponent = 1; # Entries with no suffix are assumed to be in megabytes.
+   if (defined ($2)) { 
+   	$exponent = index("KMGT",$2);
+   }
+   $softlim = ($1 * 1024 ** $exponent);
+
+    if (($hardlim !~ /^(.+?)\s*([KMGT])?$/ ) || (!looks_like_number ($1))) {
+	return $c->l('quo_HARD_VAL_MUST_BE_NUMBER');
+    }
+   $exponent = 1; # Entries with no suffix are assumed to be in megabytes.
+   if (defined ($2)) 
+   { 
+   	$exponent = index("KMGT",$2);
+   }   
+   $hardlim = ($1 * 1024 ** $exponent);
+
+    #------------------------------------------------------------
+    # Make sure that soft limit is less than hard limit.
+    #------------------------------------------------------------
+
+    unless ($hardlim == 0 or $hardlim > $softlim) {
+	return $c->l('quo_ERR_HARD_LT_SOFT');
+    }
+
+    #------------------------------------------------------------
+    # Update accounts database and signal the user-modify event.
+    #------------------------------------------------------------
+
+    $rec->set_prop('MaxBlocks', $hardlim);
+    $rec->set_prop('MaxBlocksSoftLim', $softlim);
+
+    # Untaint $acct before using in system().
+    $acct =~ /^(\w[\-\w_\.]*)$/; $acct = $1;
+    system ("/sbin/e-smith/signal-event", "user-modify", "$acct") == 0
+        or die ($c->l('quo_ERR_MODIFYING')."\n");
+
+    return 'OK';
+}
+
+
+1