File Coverage

blib/lib/Kwiki/Scode.pm

Criterion	Covered	Total	%
statement	1	3	33.3
branch			n/a
condition			n/a
subroutine	1	1	100.0
pod			n/a
total	2	4	50.0

package Kwiki::Scode;
use Kwiki::Plugin -Base;
use mixin 'Kwiki::Installer';
use GD;
our $VERSION = '0.03';

const class_id => 'scode';
const class_title => 'Scode prevents wiki spam';
const cgi_class => 'Kwiki::Scode::CGI';

field 'captcha_code';

my $tmpdir = "/tmp/";
my $scode_length = 6;
my $scode_maxtmp = 50;

sub register {
    my $reg = shift;
    $reg->add(action => 'captcha');
    $reg->add(hook => 'edit:edit', post => 'generate_scode');
    $reg->add(hook => 'edit:save', pre => 'check_scode');
}

sub generate_scode {
    my $hook = pop;
    my $scode = $self->hub->load_class('scode');
    srand int (time/10)+$$;
    my $code = int rand($scode->scode_tmp());
    $code++;
    $scode->scode_create($code);
    $scode->captcha_code($code);
    my ($html) = $hook->returned;
    my $img_html = $scode->template_process('scode_image.html', captcha_code => $code);
    $html =~ s{
    return $html;
}

sub check_scode {
    my $hook = pop;
    my $scode = $self->hub->load_class('scode');
    unless($scode->scode_granted) {
        $hook->code(undef);
        return $self->redirect($self->redirect($self->pages->current->uri));
    }
}

sub scode_granted {
    my $answer = $self->scode_get($self->cgi->code);
    $answer eq $self->cgi->captcha;
}

sub captcha {
    my $code = $self->cgi->code;

    # Calculate code
    my $scode = $self->scode_get($code);

    # lets define the image
    my $im_length = ($self->scode_len()+1)*10;
    my $im = new GD::Image($im_length,25);

    # define the color we going to use
    my $c_background = $im->colorAllocate(224,224,224);
    my $c_border = $im->colorAllocate(0,0,0);
    my $c_line = $im->colorAllocate(192,192,192);
    my $c_code = $im->colorAllocate(128,128,128);

    # Fill in the background
    $im->fill(50,50,$c_background);

    # Draw the borders lines
    for (my $i=0;$i<$im_length;$i+=5) {
        $im->line($i,0,$i,24,$c_line);
    }

    for (my $i=0;$i<25;$i+=5) {
        $im->line(0,$i,$im_length-1,$i,$c_line);
    }

    $im->rectangle(0,0,$im_length-1,24,$c_border);

    # Write the code
    $im->string(gdGiantFont,8,5,$scode,$c_code);

    binmode STDOUT;
    for(qw(png jpeg gif)) {
        my $img = $im->$_;
        if($img) {
            $self->hub->headers->content_type("image/$_");
            $self->hub->headers->print;
            print $img;
            last;
        }
    }
    return;
}


## Following code comes from MT::Scode plugin ##########

sub scode_len {
    return $scode_length;
}

sub scode_tmp {
    return $scode_maxtmp;
}

sub scode_generate {
    return int rand( (10**($scode_length)) - (10**($scode_length-1)) ) +
                      10**($scode_length-1);
}

sub scode_create {
    my $code = shift;
    $tmpdir = $self->plugin_directory . '/';

    return if (-e $tmpdir.$code);

    if ($code>0 && $code<=$scode_maxtmp) {
    	my $scode = scode_generate();
        open(OUTFILE,">${tmpdir}${code}");
        print OUTFILE $scode;
        close(OUTFILE);
    }
}

sub scode_delete {
    my $code = shift;

    if ($code>0 && $code<=$scode_maxtmp) {
        unlink $tmpdir.$code;
    }
}

sub scode_get {
    my $code = shift;
    $tmpdir = $self->plugin_directory . '/';

    srand time;

    # Random number back...if have not initialized
    if ($code<=0 || $code>$scode_maxtmp || !-e $tmpdir.$code ) {
        return $self->scode_generate();
    }

    open(INFILE, $tmpdir.$code);
    my $scode = ;
    close(INFILE);

    $scode =~ s/\D//g;
    return $scode;
}

package Kwiki::Scode::CGI;
use base 'Kwiki::CGI';

cgi 'code';
cgi 'captcha';

package Kwiki::Scode;

__DATA__