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 = (@_);

        warn "_init";

        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) {
                        warn "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
        );

        warn "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			warn "_init";
19
20			if ( delete $args{'_bootstrap'} ) {
21			$self->is_bootstrap_user(1);
22			} elsif( keys %args ) {
23			$self->user_object( SQLSession::Model::User->new( current_user => $self ) );
24			$self->user_object->load_by_cols( %args );
25	dpavlin	58
26	dpavlin	45	}
27
28			$self->SUPER::_init(%args);
29	dpavlin	58
30			if ( $self->current_user->id ) {
31			my $r = SQLSession::Model::Role->new();
32			$r->load_by_cols(
33			user_id => $self->current_user->id,
34			role => 'admin',
35			);
36			if ($r->id) {
37			warn "turn on AdminMode for ", $self->current_user->user_object->email, $/;
38			$self->is_superuser(1);
39			}
40			}
41	dpavlin	60
42			# honor AdminMode from config.yml
43			$self->is_superuser(1) if Jifty->config->framework('AdminMode');
44	dpavlin	45	}
45
46	dpavlin	47	=head2 role
47
48			Test if user has role
49
50			Jifty->web->current_user->role( 'edit' );
51
52			=cut
53
54	dpavlin	45	sub role {
55			my $self = shift;
56			my $role = shift;
57
58			return 0 unless ($self->current_user->id);
59
60	dpavlin	58	return 1 if ($role eq 'admin' && $self->current_user->is_superuser);
61
62	dpavlin	69	return 1 if ($role eq 'edit' && $self->current_user->user_object->email_confirmed);
63
64	dpavlin	47	my $r = SQLSession::Model::Role->new();
65			$r->load_by_cols(
66			user_id => $self->current_user->id,
67			role => $role
68			);
69	dpavlin	45
70	dpavlin	47	warn "role is '$role' for ", $self->current_user->user_object->email, " ",
71			$r->id ? 'OK' : 'DENY', $/;
72	dpavlin	45
73	dpavlin	47	return $r->id;
74	dpavlin	45	}
75
76	dpavlin	59	1;