lufi/lib/Lufi/DB/File/Pg.pm

# vim:set sw=4 ts=4 sts=4 ft=perl expandtab:
package Lufi::DB::File::Pg;
use Mojo::Base 'Lufi::DB::File';
use Mojo::File;
use Mojo::Collection 'c';
use Lufi::DB::Slice;

has 'record' => 0;

sub new {
    my $c = shift;

    $c = $c->SUPER::new(@_);

    return $c;
}

sub write {
    my $c = shift;

    if ($c->record) {
        $c->app->pg->db->query('UPDATE files SET short = ?, deleted = ?, mediatype = ?, filename = ?, filesize = ?, counter = ?, delete_at_first_view = ?, delete_at_day = ?, created_at = ?, created_by = ?, last_access_at = ?, mod_token = ?, nbslices = ?, complete = ?, passwd = ? WHERE short = ?', $c->short, $c->deleted, $c->mediatype, $c->filename, $c->filesize, $c->counter, $c->delete_at_first_view, $c->delete_at_day, $c->created_at, $c->created_by, $c->last_access_at, $c->mod_token, $c->nbslices, $c->complete, $c->passwd, $c->short);
    } else {
        $c->app->pg->db->query('INSERT INTO files (short, deleted, mediatype, filename, filesize, counter, delete_at_first_view, delete_at_day, created_at, created_by, last_access_at, mod_token, nbslices, complete, passwd) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)', $c->short, $c->deleted, $c->mediatype, $c->filename, $c->filesize, $c->counter, $c->delete_at_first_view, $c->delete_at_day, $c->created_at, $c->created_by, $c->last_access_at, $c->mod_token, $c->nbslices, $c->complete, $c->passwd);
        $c->record(1);
    }

    return $c;
}

sub count_empty {
    my $c = shift;

    return $c->app->pg->db->query('SELECT count(short) FROM files WHERE created_at IS NULL')->hashes->first->{count};
}

sub already_exists {
    my $c     = shift;
    my $short = shift;

    return $c->app->pg->db->query('SELECT count(short) FROM files WHERE short = ?', $short)->hashes->first->{count};
}

sub get_empty {
    my $c     = shift;

    my $r = $c->app->pg->db->query('SELECT * FROM files WHERE created_at IS NULL')->hashes->shuffle->first;

    return $c->_slurp($r)->created_at(time)->write;
}

sub get_stats {
    my $c = shift;

    my $files   = $c->app->pg->db->query('SELECT count(short) FROM files WHERE created_at IS NOT null AND deleted = 0')->hashes->first->{count};
    my $deleted = $c->app->pg->db->query('SELECT count(short) FROM files WHERE created_at IS NOT null AND deleted = 1')->hashes->first->{count};
    my $empty   = $c->app->pg->db->query('SELECT count(short) FROM files WHERE created_at IS null')->hashes->first->{count};

    return { files => $files, deleted => $deleted, empty => $empty };
}

sub from_short {
    my $c     = shift;
    my $short = shift;

    my $r = $c->app->pg->db->query('SELECT * FROM files WHERE short = ?', $short)->hashes;

    if ($r->size) {
        return $c->_slurp($r->first)->record(1);
    } else {
        return undef;
    }
}

sub get_oldest_undeleted_files {
    my $c   = shift;
    my $num = shift;

    my @files;
    my $records = $c->app->pg->db->query('SELECT * FROM files WHERE deleted = 0 ORDER BY created_at ASC LIMIT ?', $num)->hashes;
    $records->each(
        sub {
            my ($e, $num) = @_;
            my $i = Lufi::DB::File->new(app => $c->app);

            push @files, $i->_slurp($e);
        }
    );

    return c(@files);
}

sub get_expired {
    my $c    = shift;
    my $time = shift;

    my @files;
    ## Select only files expired since two days, to be sure that nobody is still downloading it
    my $records = $c->app->pg->db->query('SELECT * FROM files WHERE deleted = 0 AND ((delete_at_day + 2) * 86400) < (? - created_at) AND delete_at_day != 0', $time)->hashes;
    $records->each(
        sub {
            my ($e, $num) = @_;
            my $i = Lufi::DB::File->new(app => $c->app);

            push @files, $i->_slurp($e);
        }
    );

    return c(@files);
}

sub get_no_longer_viewed {
    my $c    = shift;
    my $time = shift;

    my @files;
    my $records = $c->app->pg->db->query('SELECT * FROM files WHERE deleted = 0 AND last_access_at < ?', $time)->hashes;
    $records->each(
        sub {
            my ($e, $num) = @_;
            my $i = Lufi::DB::File->new(app => $c->app);

            push @files, $i->_slurp($e);
        }
    );

    return c(@files);
}

sub delete_creator_before {
    my $c          = shift;
    my $separation = shift;

    $c->app->pg->db->query('UPDATE files SET created_by = NULL WHERE created_by IS NOT NULL AND created_at < ?', $separation);
}

sub _slurp {
    my $c = shift;
    my $r = shift;

    my $file;
    if (defined $r) {
        $file = $r;
    } else {
        my $files = $c->app->pg->db->query('SELECT * FROM files WHERE short = ?', $c->short)->hashes;

        if ($files->size) {
            $file = $files->first;
        }
    }

    if ($file) {
        $c->short($file->{short});
        $c->deleted($file->{deleted});
        $c->mediatype($file->{mediatype});
        $c->filename($file->{filename});
        $c->filesize($file->{filesize});
        $c->counter($file->{counter});
        $c->delete_at_first_view($file->{delete_at_first_view});
        $c->delete_at_day($file->{delete_at_day});
        $c->created_at($file->{created_at});
        $c->created_by($file->{created_by});
        $c->last_access_at($file->{last_access_at});
        $c->mod_token($file->{mod_token});
        $c->nbslices($file->{nbslices});
        $c->complete($file->{complete});
        $c->passwd($file->{passwd});

        $c->record(1) unless $c->record;
    }

    $c->slices(Lufi::DB::Slice->new(app => $c->app)->get_slices_of_file($c->short));

    return $c;
}

1;