|
Luc Didry |
40deab |
package Lufi::Command::cron::watch;
|
|
Luc Didry |
40deab |
use Mojo::Base 'Mojolicious::Command';
|
|
Luc Didry |
40deab |
use Filesys::DiskUsage qw/du/;
|
|
Luc Didry |
d909b8 |
use Lufi::DB::File;
|
|
Luc Didry |
40deab |
use Switch;
|
|
Luc Didry |
40deab |
use FindBin qw($Bin);
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
has description => 'Watch the files directory and take action when over quota';
|
|
Luc Didry |
40deab |
has usage => sub { shift->extract_usage };
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
sub run {
|
|
Luc Didry |
40deab |
my $c = shift;
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
d909b8 |
my $cfile = Mojo::File->new($Bin, '..' , 'lufi.conf');
|
|
Luc Didry |
d909b8 |
if (defined $ENV{MOJO_CONFIG}) {
|
|
Luc Didry |
d909b8 |
$cfile = Mojo::File->new($ENV{MOJO_CONFIG});
|
|
Luc Didry |
d909b8 |
unless (-e $cfile->to_abs) {
|
|
Luc Didry |
d909b8 |
$cfile = Mojo::File->new($Bin, '..', $ENV{MOJO_CONFIG});
|
|
Luc Didry |
d909b8 |
}
|
|
Luc Didry |
d909b8 |
}
|
|
Luc Didry |
40deab |
my $config = $c->app->plugin('Config', {
|
|
Luc Didry |
d909b8 |
file => $cfile,
|
|
Luc Didry |
40deab |
default => {
|
|
Luc Didry |
d909b8 |
dbtype => 'sqlite',
|
|
Luc Didry |
40deab |
policy_when_full => 'warn'
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
});
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
if (defined($config->{max_total_size})) {
|
|
Luc Didry |
40deab |
my $total = du(qw/files/);
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
if ($total > $config->{max_total_size}) {
|
|
Luc Didry |
40deab |
say "[Lufi cron job watch] Files directory is over quota ($total > ".$config->{max_total_size}.")";
|
|
Luc Didry |
40deab |
switch ($config->{policy_when_full}) {
|
|
Luc Didry |
40deab |
case 'warn' {
|
|
Luc Didry |
40deab |
say "[Lufi cron job watch] Please, delete some files or increase quota (".$config->{max_total_size}.")";
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
case 'stop-upload' {
|
|
Luc Didry |
40deab |
open (my $fh, '>', 'stop-upload') or die ("Couldn't open stop-upload: $!");
|
|
Luc Didry |
40deab |
close($fh);
|
|
Luc Didry |
40deab |
say '[Lufi cron job watch] Uploads are stopped. Delete some images and the stop-upload file to reallow uploads.';
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
case 'delete' {
|
|
Luc Didry |
40deab |
say '[Lufi cron job watch] Older files are being deleted';
|
|
Luc Didry |
d909b8 |
my $ldfile = Lufi::DB::File->new(app => $c->app);
|
|
Luc Didry |
40deab |
do {
|
|
Luc Didry |
d909b8 |
$ldfile->get_oldest_undeleted_files(50)->each(
|
|
Luc Didry |
d909b8 |
sub {
|
|
Luc Didry |
d909b8 |
my ($f, $num) = @_;
|
|
Luc Didry |
d909b8 |
$f->delete;
|
|
Luc Didry |
d909b8 |
}
|
|
Luc Didry |
d909b8 |
);
|
|
Luc Didry |
40deab |
} while (du(qw/files/) > $config->{max_total_size});
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
else {
|
|
Luc Didry |
40deab |
say '[Lufi cron job watch] Unrecognized policy_when_full option: '.$config->{policy_when_full}.'. Aborting.';
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
} else {
|
|
Luc Didry |
40deab |
unlink 'stop-upload' if (-f 'stop-upload');
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
} else {
|
|
Luc Didry |
40deab |
say "[Lufi cron job watch] No max_total_size found in the configuration file. Aborting.";
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
}
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
=encoding utf8
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
=head1 NAME
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
Lufi::Command::cron::watch - Watch the files directory and take action when over quota
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
=head1 SYNOPSIS
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
Usage: script/lufi cron watch
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
=cut
|
|
Luc Didry |
40deab |
|
|
Luc Didry |
40deab |
1;
|