@@ -9,6 +9,7 @@ use DBIx::Custom;

 use Validator::Custom;

 use Encode qw/encode decode/;

 use Mojo::JSON;

+use Gitprep::API;

 has 'git';

 has 'dbi';

@@ -167,6 +168,9 @@ EOS

   # Validator

   my $validator = Validator::Custom->new;

   $self->validator($validator);

+  

+  # Helper

+  $self->helper(gitprep_api => sub { Gitprep::API->new(shift) });

 }

 1;

@@ -51,5 +51,22 @@ sub users {

   return $users;

 }

+sub logined {

+  my $self = shift;

+  

+  my $c = $self->cntl;

+  

+  my $dbi = $c->app->dbi;

+  

+  my $id = $c->session('user_id');

+  my $password = $c->session('user_password');

+  

+  my $row = $dbi->model('user')->select('config', id => $id)->one;

+  return unless $row;

+  my $config = $self->json($row->{config});

+  

+  return $password eq $config->{password};

+}

+

 1;

@@ -1,9 +1,9 @@

 <%

-  my $user_id = session('user_id');

+  my $api = gitprep_api;

 %>

 <div class="container">

-  % if (defined $user_id) {

+  % if ($api->logined) {

     <div>

       <ul>

         <li><a href="<%= url_for('/') %>">GitPrep</a></li>

@@ -13,6 +13,7 @@

   my $dbi = $self->app->dbi;

   if ($op eq 'login') {

+    sleep 3;

     $state = 'login';

     my $params = {

@@ -55,11 +56,14 @@

     if ($vresult->is_ok) {

       my $valid_params = $vresult->data;

       my $id = $valid_params->{id};

+      my $password = $valid_params->{password};

+      my $password_md5 = md5_sum $password;

       my $config_json = $self->app->dbi->model('user')->select('config', id => $id)->value;

       my $config = $api->json($config_json);

       my $admin = $config->{admin};

       session(user_id => $id);

+      session(user_password => $password_md5);

       if ($admin) {

         $self->redirect_to('/_admin');