lib/SQLSession/CurrentUser.pm

use strict;
use warnings;

package SQLSession::CurrentUser;
use base qw/Jifty::CurrentUser/;

=head2 _init

This function will load user data from database and turn on admin mode in
Jifty if user has C<admin> role.

=cut

sub _init {
        my $self = shift;
        my %args = (@_);

        if ( delete $args{'_bootstrap'} ) {
                $self->is_bootstrap_user(1);
        } elsif( keys %args ) {
                $self->user_object( SQLSession::Model::User->new( current_user => $self ) );
                $self->user_object->load_by_cols( %args );

        }

        $self->SUPER::_init(%args);

        if ( $self->current_user->id ) {
                my $r = SQLSession::Model::Role->new();
                $r->load_by_cols(
                        user_id => $self->current_user->id,
                        role => 'admin',
                );
                if ($r->id) {
                        Jifty->log->debug("turn on AdminMode for ", $self->current_user->user_object->email);
                        $self->is_superuser(1);
                }
        }

        # honor AdminMode from config.yml
        $self->is_superuser(1) if Jifty->config->framework('AdminMode');
}

=head2 role

Test if user has role

  Jifty->web->current_user->role( 'edit' );

=cut

sub role {
        my $self = shift;
        my $role = shift;

        return 0 unless ($self->current_user->id);

        return 1 if ($role eq 'admin' && $self->current_user->is_superuser);

        return 1 if ($role eq 'edit' && $self->current_user->user_object->email_confirmed);

        my $r = SQLSession::Model::Role->new();
        $r->load_by_cols(
                user_id => $self->current_user->id,
                role => $role
        );

        Jifty->log->debug("role is '$role' for ",
                $self->current_user->user_object->email, " ",
                $r->id ? 'OK' : 'DENY'
        );

        return $r->id;
}

1;
1	dpavlin	45	use strict;
2			use warnings;
3
4			package SQLSession::CurrentUser;
5			use base qw/Jifty::CurrentUser/;
6
7	dpavlin	58	=head2 _init
8
9			This function will load user data from database and turn on admin mode in
10			Jifty if user has C<admin> role.
11
12			=cut
13
14	dpavlin	45	sub _init {
15			my $self = shift;
16			my %args = (@_);
17
18			if ( delete $args{'_bootstrap'} ) {
19			$self->is_bootstrap_user(1);
20			} elsif( keys %args ) {
21			$self->user_object( SQLSession::Model::User->new( current_user => $self ) );
22			$self->user_object->load_by_cols( %args );
23	dpavlin	58
24	dpavlin	45	}
25
26			$self->SUPER::_init(%args);
27	dpavlin	58
28			if ( $self->current_user->id ) {
29			my $r = SQLSession::Model::Role->new();
30			$r->load_by_cols(
31			user_id => $self->current_user->id,
32			role => 'admin',
33			);
34			if ($r->id) {
35	dpavlin	88	Jifty->log->debug("turn on AdminMode for ", $self->current_user->user_object->email);
36	dpavlin	58	$self->is_superuser(1);
37			}
38			}
39	dpavlin	60
40			# honor AdminMode from config.yml
41			$self->is_superuser(1) if Jifty->config->framework('AdminMode');
42	dpavlin	45	}
43
44	dpavlin	47	=head2 role
45
46			Test if user has role
47
48			Jifty->web->current_user->role( 'edit' );
49
50			=cut
51
52	dpavlin	45	sub role {
53			my $self = shift;
54			my $role = shift;
55
56			return 0 unless ($self->current_user->id);
57
58	dpavlin	58	return 1 if ($role eq 'admin' && $self->current_user->is_superuser);
59
60	dpavlin	69	return 1 if ($role eq 'edit' && $self->current_user->user_object->email_confirmed);
61
62	dpavlin	47	my $r = SQLSession::Model::Role->new();
63			$r->load_by_cols(
64			user_id => $self->current_user->id,
65			role => $role
66			);
67	dpavlin	45
68	dpavlin	88	Jifty->log->debug("role is '$role' for ",
69			$self->current_user->user_object->email, " ",
70			$r->id ? 'OK' : 'DENY'
71			);
72	dpavlin	45
73	dpavlin	47	return $r->id;
74	dpavlin	45	}
75
76	dpavlin	59	1;