#!/usr/bin/perl
# Show disk usage for one virtual server, broken down by directory

require './virtual-server-lib.pl';
&ReadParse();
$d = &get_domain($in{'dom'});
&can_edit_domain($d) || &error($text{'newbw_ecannot'});

$subh = &domain_in($d);
@subs = &get_domain_by("parent", $d->{'id'});
&ui_print_header($subh, $text{'usage_title'}, "", "usage");

# Compute usage by database
foreach $sd ($d, @subs) {
	foreach $db (&domain_databases($sd)) {
		($dbu, undef, $dbq, $dbc) = &get_one_database_usage($sd, $db);
		push(@dbusage, [ $db->{'name'}, &show_domain_name($sd),
				 $dbu, $dbc ]);
		}
	}

# Compute usage by users
$homesize = &quota_bsize("home");
$mailsize = &quota_bsize("mail");
foreach $sd ($d, &get_domain_by("parent", $d->{'id'})) {
	@users = &list_domain_users($sd, 0, 1, 0, 1);
	foreach my $u (@users) {
		next if ($u->{'webowner'});
		my ($uusage, $ucount);
		if ($u->{'domainowner'}) {
			# Only show mail for domain owner
			($uusage, undef, undef, $ucount) = &mail_file_size($u);
			}
		elsif (&has_home_quotas()) {
			$uusage = $u->{'uquota'}*$homesize +
				  $u->{'umquota'}*$mailsize;
			$ucount = $u->{'ufquota'} + $u->{'umfquota'};
			}
		else {
			($uusage, undef, $ucount) =
				&recursive_disk_usage_mtime($u->{'home'});
			if (!&mail_under_home()) {
				($umail) = &mail_file_size($u);
				$uusage += $umail;
				}
			}
		push(@userusage,
		     [ &remove_userdom($u->{'user'}, $sd),
		       &show_domain_name($sd),
		       $uusage,
		       $ucount, ]);
		}
	}

# First work out what tabs we have
@tabs = ( );
$prog = "usage.cgi?dom=$in{'dom'}&mode=";
if (&has_home_quotas()) {
	push(@tabs, [ "summary", $text{'usage_tabsummary'}, $prog."summary" ]);
	}
push(@tabs, [ "homes", $text{'usage_tabhomes'}, $prog."homes" ]);
if (@userusage) {
	push(@tabs, [ "users", $text{'usage_tabusers'}, $prog."users" ]);
	}
if (@subs) {
	push(@tabs, [ "subs", $text{'usage_tabsubs'}, $prog."subs" ]);
	}
if (@dbusage) {
	push(@tabs, [ "dbs", $text{'usage_tabdbs'}, $prog."dbs" ]);
	}
print &ui_tabs_start(\@tabs, "mode", $in{'mode'} || $tabs[0]->[0], 1);

# Show quota usage summary
if (&has_home_quotas()) {
	print &ui_tabs_start_tab("mode", "summary");
	print $text{'usage_summaryheader'},"<p>\n";
	print &ui_table_start(undef, undef, 4);

	# Total virtual server quota
	print &ui_table_row($text{'usage_squota'},
			    &quota_show($d->{'quota'}));

	# Quota used
	($home, $mail, $db) = &get_domain_quota($d, 1);
	$usage = $home*$homesize + $mail*$mailsize;
	print &ui_table_row($text{'usage_susage'},
			    &nice_size($usage));

	# Space used by databases
	print &ui_table_row($text{'usage_sdb'},
			    &nice_size($db));

	# Percentage used
	if ($d->{'quota'}) {
		$pc = 100*($usage + $db) / ($d->{'quota'}*$homesize);
		$pc = int($pc)."%";
		print &ui_table_row($text{'usage_spercent'},
			   $pc >= 100 ? "<font color=#ff0000>$pc</font>" : $pc);
		}

	print &ui_table_end();
	print &ui_tabs_end_tab();
	}

# Show usage by each sub-directory under home, and sub-dirs under public_html
opendir(DIR, $d->{'home'});
@dirs = grep { $_ ne ".." } readdir(DIR);
closedir(DIR);
$phd = &public_html_dir($d, 1);
if (-r "$d->{'home'}/$phd") {
	opendir(DIR, "$d->{'home'}/$phd");
	push(@dirs, map { "$phd/$_" }
			grep { $_ ne ".." && $_ ne "." } readdir(DIR));
	closedir(DIR);
	}
@dirs = sort { $a cmp $b } @dirs;
$others = 0;
$othersc = 0;
foreach $dir (@dirs) {
	my $path = "$d->{'home'}/$dir";
	my $levels = $dir eq "domains" || $dir eq "homes" ||
			$dir eq "." || $dir eq $phd ? 0 : undef;
	my ($dirusage, undef, $dircount) =
		&recursive_disk_usage_mtime($path, undef, $levels);
	if (-d $path && $dir ne ".") {
		push(@dirusage, [ $dir, $dirusage, $dircount ]);
		}
	else {
		$others += $dirusage;
		$othersc += $dircount;
		}
	}
push(@dirusage, [ "<i>$text{'usage_others'}</i>", $others, $othersc ]);
closedir(DIR);

# Add an extra directories outside the home
foreach my $edir (split(/\t+/, $config{'quota_dirs'})) {
	my $path = &substitute_domain_template($edir, $d);
	my ($dirgid, undef, $dircount) =
		&recursive_disk_usage_mtime($path, $d->{'gid'}, undef);
	push(@dirusage, [ $path, $dirgid, $dircount ]);
	}

print &ui_tabs_start_tab("mode", "homes");
my $msg = $config{'quota_dirs'} ? $text{'usage_dirheader2'}
			        : $text{'usage_dirheader'};
&usage_table(\@dirusage, $text{'usage_dir'}, 0, $msg);
print &ui_tabs_end_tab();

# Show usage by top 10 mail users, in all domains
if (@userusage) {
	print &ui_tabs_start_tab("mode", "users");
	&usage_table(\@userusage, $text{'usage_user'}, $in{'all'} ? 0 : 10,
		     $text{'usage_userheader'}, $text{'usage_dom'});
	print &ui_tabs_end_tab();
	}

# Show usage by sub-servers, if there are any
if (@subs) {
	foreach $sd (@subs) {
		next if (!$sd->{'dir'});
		($susage, undef, $scount) =
			&recursive_disk_usage_mtime($sd->{'home'});
		push(@subusage, [ &show_domain_name($sd), $susage, $scount ]);
		}
	print &ui_tabs_start_tab("mode", "subs");
	&usage_table(\@subusage, $text{'usage_sub'}, $in{'all'} ? 0 : 10,
		     $text{'usage_subheader'});
	print &ui_tabs_end_tab();
	}

# Show usage by databases
if (@dbusage) {
	print &ui_tabs_start_tab("mode", "dbs");
	&usage_table(\@dbusage, $text{'usage_db'}, $in{'all'} ? 0 : 10,
		     $text{'usage_dbheader'}, $text{'usage_dom'});
	print &ui_tabs_end_tab();
	}

print &ui_tabs_end(1);

&ui_print_footer(&domain_footer_link($d));

sub usage_table
{
my ($list, $type, $max, $title, $type2) = @_;
my @table;

# Make the data
my $i = 0;
my $total = 0;
my $ctotal = 0;
foreach my $l (sort { $b->[@$b-2] <=> $a->[@$a-2] } @$list) {
	my @rest = @$l;
	my $ct = pop(@rest);
	my $sz = pop(@rest);
	push(@table, [ @rest, &nice_size($sz), $ct ]);
	$i++;
	last if ($max && $i > $max);
	}
foreach my $l (@$list) {
	$total += $l->[@$l-2];
	$ctotal += $l->[@$l-1];
	}
push(@table, [ "<b>$text{'usage_total'}</b>",
		$type2 ? ( "" ) : ( ),
		"<b>".&nice_size($total)."</b>",
		"<b>".$ctotal."</b>" ]);

# Show the table
print $title,"<p>\n";
print &ui_columns_table(
	[ $type, $type2 ? ( $type2 ) : ( ),
	  $text{'usage_size'}, $text{'usage_count'} ],
	undef,
	\@table,
	undef,
	0,
	undef,
	$text{'usage_none'},
	);
if ($max && @$list > $max) {
	print "<i>",&text('usage_max', $max)," ",
	      "<a href='usage.cgi?dom=$in{'dom'}&all=1&mode=$in{'mode'}'>",
	      $text{'usage_showall'},"</a></i><br>\n";
	}
}