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;