File Coverage

blib/lib/Kwiki/UserName/Auth.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::UserName::Auth;
use Kwiki::UserName -Base;
use mixin 'Kwiki::Installer';

our $VERSION = '0.10';

const cgi_class => 'Kwiki::UserName::Auth::CGI';
const css_file  => 'user_name_auth.css';

# Forward to this URL after process form submittion.
field 'forward';
field 'display_msg';

sub register {
    my $registry = shift;
    $registry->add(preload => 'user_name');
    $registry->add(action  => 'user_name');
}

## Override Plugin.pm's render_screen()
sub render_screen {
    if(@_) {
	$self->template_process($self->screen_template, @_);
    } else {
	super;
    }
}

sub user_name {
    if(my $mode = $self->cgi->run_mode) {
	$self->$mode;
    } else {
	$self->setup;
    }
    $self->render_screen;
}

## 5 run modes: setup,create,login,logout,mail_password
sub setup {
    $self->display_msg("Please login or register");
}

sub login {
    my ($email,$password) = (@_,$self->cgi->email,$self->cgi->password);
    if(my $user = $self->db_verify($email,$password)) {
	my $session = $self->hub->session->load;
	$session->param("users_auth",{name => $user, email => $email });
	if($self->cgi->forward) {
	    return $self->redirect($self->cgi->forward);
	}
	$self->display_msg("Login success");
    } else {
	$self->display_msg("Login falied: Wrong user name or password.");
    }
}

sub logout {
    $self->hub->session->load->param("users_auth", {});
    $self->display_msg("Logout success");
}

sub create {
    if($self->cgi->password) {
	if($self->cgi->password_verify eq $self->cgi->password) {
	    if($self->db_add($self->cgi->all)) {
		$self->display_msg("Registered");
	    } else {
		$self->display_msg("Duplicated username");
	    }
	} else {
	    $self->display_msg("Password does not match");
	}
    }
}

sub mail_password {
}

# DB subs

field dbi => -init => '$self->hub->dbi';

sub dbinit {
    my $db = shift;
    my $dbh = $self->dbi->connect("dbi:SQLite:dbname=$db","","",
				 { RaiseError => 1, AutoCommit => 1 });
    $dbh->do('CREATE TABLE user_name (displayname NOT NULL,email NOT NULL,password NOT NULL,regdate DEFAULT CURRENT_TIMESTAMP)');
    $dbh->disconnect;
}

sub dbpath {
    my $path = $self->plugin_directory;
    my $filename =  io->catfile($path,"user_name.sqlt")->name;
    $self->dbinit($filename) unless -f $filename;
    return $filename;
}

sub db_connect {
    my $db  = $self->dbpath;
    $self->dbi->connect("dbi:SQLite:dbname=$db","","",
		       { RaiseError => 1, AutoCommit => 1 });
}

sub db_verify {
    my ($email,$password) = @_;
    my $dbh = $self->db_connect;
    my $user = $dbh->selectrow_hashref('SELECT displayname,email from user_name WHERE email=? AND password=?',undef,$email,$password);
    $dbh->disconnect;
    if($user) {
        return $user->{displayname} unless $user->{displayname} =~ /^\s*$/;
        $email =~ s/(.+?)@.*$/$1/;
        return $email;
    }
    return;
}

sub db_retrieve {
    my $email = shift;
    my $dbh   = $self->db_connect;
    my $user  = $dbh->selectrow_hashref('SELECT displayname FROM user_name WHERE email = ?',undef,$email);
    return $self->utf8_decode($user->{displayname}) if $user;
}

sub db_add {
    my %vars = @_;
    return 0 if $self->db_exists($vars{email});
    my $dbh = $self->db_connect;
    my $sth = $dbh->prepare('INSERT INTO user_name (displayname, email, password) values(?,?,?)');
    $sth->execute(@vars{qw(display_name email password)}, );
    $sth->finish;
    $dbh->disconnect;
    return 1;
}

sub db_exists {
    my $email = shift;
    my $dbh = $self->db_connect;
    my $r = $dbh->selectrow_arrayref('SELECT email FROM user_name WHERE email=?',undef,$email);
    $dbh->disconnect;
    return $r;
}

package Kwiki::UserName::Auth::CGI;
use base 'Kwiki::CGI';

# Forward to this URL after process form submittion.
cgi 'forward';

cgi 'run_mode';

cgi 'display_name';
cgi 'email';
cgi 'password';
cgi 'password_verify';

package Kwiki::UserName::Auth;

__DATA__