File Coverage

blib/lib/Apache/AxKit/Plugin/Upload.pm

Criterion	Covered	Total	%
statement	4	6	66.6
branch			n/a
condition			n/a
subroutine	2	2	100.0
pod			n/a
total	6	8	75.0

package Apache::AxKit::Plugin::Upload;
use strict;

use Apache::Constants qw(:common);
use Number::Format;
use Fcntl qw(LOCK_EX LOCK_NB O_RDWR O_CREAT);
use IO::Handle;

BEGIN {
    our $VERSION = 0.2;
}

eval {
  require Time::HiRes;
};
if (!$@) {
  use Time::HiRes;
}

my ($start, $sizetotal, $file, $location, $nf, $custom, $running, $lock);

sub print_html {
    my ($now, $sizedone) = @_;
    my $elapsed = $now-$start;

    my $todo = 0;
    $todo = ($sizetotal - $sizedone) * $elapsed / $sizedone if $sizedone > 0;

    my $percent = 0;
    $percent = int($sizedone*100/$sizetotal) if $sizetotal > 0;
    my $showpercent = ($percent?$percent:1);
    #warn("printing: $percent");

    my $speed = 0;
    $speed = $nf->format_bytes($sizedone / $elapsed,1) if $elapsed > 0;

    $sizedone = $nf->format_bytes($sizedone,1);
    my $sizetotal = $nf->format_bytes($sizetotal,1);
    $elapsed = sprintf("%i:%02i",int($elapsed/60),$elapsed%60);
    $todo = sprintf("%i:%02i",int($todo/60),$todo%60);

    local(*FH);
    open(FH, ">$file.tmp") || warn("could not open $file.tmp: $!");

    if (!$custom && $percent == 100) {
        print FH << "EOF";


    Upload Status


     




53




 





 





 





 





 





 



        




54




 





 





 





 





 





 



         $percent\% 




55




 





 





 





 





 





 



          




56




 





 





 





 





 





 



            




57




 





 





 





 





 





 



              




58




 





 





 





 





 





 



                 




59




 





 





 





 





 





 



              




60




 





 





 





 





 





 



            




61




 





 





 





 





 





 



          




62




 





 





 





 





 





 



        




63




 





 





 





 





 





 



        




64




 





 





 





 





 





 



           




65




 





 





 





 





 





 



         $sizedone / $sizetotal, ${speed}b/s, $todo 




66




 





 





 





 





 





 



        




67















    window.close()


EOF
    } else {
        print FH << "EOF";


    Upload Status



     




81




 





 





 





 





 





 



        




82




 





 





 





 





 





 



         $percent\% 




83




 





 





 





 





 





 



          




84




 





 





 





 





 





 



            




85




 





 





 





 





 





 



              




86




 





 





 





 





 





 



                 




87




 





 





 





 





 





 



              




88




 





 





 





 





 





 



            




89




 





 





 





 





 





 



          




90




 





 





 





 





 





 



        




91




 





 





 





 





 





 



        




92




 





 





 





 





 





 



           




93




 





 





 





 





 





 



         $sizedone / $sizetotal, ${speed}b/s, $todo 




94




 





 





 





 





 





 



        




95

















EOF
    }
    rename("$file.tmp",$file);
}

sub init {
    my ($r,$upload_id) = @_;
    my $destdir = $r->dir_config('AxUploadStatusDir') || return 0;
    $destdir = $r->document_root.'/'.$destdir if substr($destdir,0,1) ne '/';
    my $destloc = $r->dir_config('AxUploadStatusLocation') || return 0;
    my $format = lc($r->dir_config('AxUploadFormat')) || 'html';
    $location = "$destloc/$upload_id.$format";
    $file = "$destdir/$upload_id.$format";
    local(*FH);
    open(FH, ">$file.tmp") || warn("could not open $file.tmp: $!");
    print FH << "EOF";


    Upload Status






EOF
    close(FH);
    rename("$file.tmp",$file);
}

sub progress {
    my ($done, $total, $message) = @_;
    my $now = time();
    my $elapsed = $now-$start;

    my $todo = 0;
    $todo = ($total - $done) * $elapsed / $done if $done > 0;

    my $percent = 0;
    $percent = int($done*100/$total) if $total > 0;
    my $showpercent = ($percent?$percent:1);

    $todo = sprintf("%i:%02i",int($todo/60),$todo%60);

    local(*FH);
    open(FH, ">$file.tmp") || warn("could not open $file.tmp: $!");
    if ($percent == 100) {
        print FH << "EOF";


    Upload Status


     




152




 





 





 





 





 





 



        




153




 





 





 





 





 





 



         $percent\% 




154




 





 





 





 





 





 



          




155




 





 





 





 





 





 



            




156




 





 





 





 





 





 



              




157




 





 





 





 





 





 



                 




158




 





 





 





 





 





 



              




159




 





 





 





 





 





 



            




160




 





 





 





 





 





 



          




161




 





 





 





 





 





 



        




162




 





 





 





 





 





 



        




163




 





 





 





 





 





 



           




164




 





 





 





 





 





 



         $message 




165




 





 





 





 





 





 



        




166















    window.close()


EOF
        close($lock);
    } else {
        print FH << "EOF";


    Upload Status



     




181




 





 





 





 





 





 



        




182




 





 





 





 





 





 



         $percent\% 




183




 





 





 





 





 





 



          




184




 





 





 





 





 





 



            




185




 





 





 





 





 





 



              




186




 





 





 





 





 





 



                 




187




 





 





 





 





 





 



              




188




 





 





 





 





 





 



            




189




 





 





 





 





 





 



          




190




 





 





 





 





 





 



        




191




 





 





 





 





 





 



        




192




 





 





 





 





 





 



           




193




 





 





 





 





 





 



         $message 




194




 





 





 





 





 





 



        




195

















EOF
    }
    close(FH);
    rename("$file.tmp",$file);
}

sub is_running {
    my ($r,$upload_id) = @_;
    return 0 if ($running eq $upload_id);
    my $destdir = $r->dir_config('AxUploadStatusDir') || return 0;
    $destdir = $r->document_root.'/'.$destdir if substr($destdir,0,1) ne '/';
    my $format = lc($r->dir_config('AxUploadFormat')) || 'html';
    $file = $destdir."/".$upload_id.".".$format;
    local (*FH);
    sysopen(FH,$file.".lck",O_RDWR) || return undef;
    my $unlocked = flock(FH,LOCK_EX|LOCK_NB);
    close(FH);
    return !$unlocked;
}

sub upload_handler_html {
    my ($upload, $buf, $len, $hook_data) = @_;
    my ($sizedone, $lasttime) = @$hook_data;
    my $now = time();

    $sizedone += $len;
    $$hook_data[0] = $sizedone;

    #warn("now: $now, done: $sizedone, last: $lasttime");
    return if (int($now) == $lasttime);
    $$hook_data[1] = int($now);

    print_html($now, $sizedone);
}

sub handler {
    my ($r) = @_;

    my %args = $r->args;
    my $upload_id = $args{'axkit_upload_id'};
    $running = '';

    my $destdir = $r->dir_config('AxUploadStatusDir') || return OK;
    $destdir = $r->document_root.'/'.$destdir if substr($destdir,0,1) ne '/';
    my $destloc = $r->dir_config('AxUploadStatusLocation') || return OK;
    my $format = lc($r->dir_config('AxUploadFormat')) || 'html';
    $custom = $r->dir_config('AxUploadCustom');
    $file = "$destdir/$upload_id.$format";
    $location = "$destloc/$upload_id.$format";
    return OK unless $upload_id;
    return OK unless $r->method eq 'POST';
    $lock = new IO::Handle;
    sysopen($lock,$file.".lck",O_RDWR|O_CREAT) || return OK;
    my $unlocked = flock($lock,LOCK_EX|LOCK_NB);
    return OK if !$unlocked;

    $running = $upload_id;
    # from Apache::RequestNotes
    my $maxsize   = $r->dir_config('MaxPostSize') || 1024;
    my $uploads   = $r->dir_config('DisableUploads') =~ m/Off/i ? 0 : 1;

    $nf = new Number::Format(split(/ /,$r->dir_config('AxUploadNumberFormat')));

    $sizetotal = $r->header_in('Content-Length');
    $start = time();
    AxKit::Debug(3,"[Upload] managing upload: $sizetotal bytes, status in $destdir/$upload_id.$format");

    print_html($start, 0);

    my $apr = Apache::Request->instance($r,
        POST_MAX => $maxsize,
        DISABLE_UPLOADS => $uploads,
        HOOK_DATA => [ $file, $location, $nf, $sizetotal, 0, $start, -1 ],
        UPLOAD_HOOK => \&upload_handler_html,
    );
    $apr->parse;

    print_html(time(), $sizetotal);

    $start = time();
    return OK;
}

1;
__END__