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	use strict;
2	use warnings;
3
4	package SQLSession::CurrentUser;
5	use base qw/Jifty::CurrentUser/;
6
7	=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	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
26	}
27
28	$self->SUPER::_init(%args);
29
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
42	# honor AdminMode from config.yml
43	$self->is_superuser(1) if Jifty->config->framework('AdminMode');
44	}
45
46	=head2 role
47
48	Test if user has role
49
50	Jifty->web->current_user->role( 'edit' );
51
52	=cut
53
54	sub role {
55	my $self = shift;
56	my $role = shift;
57
58	return 0 unless ($self->current_user->id);
59
60	return 1 if ($role eq 'admin' && $self->current_user->is_superuser);
61
62	return 1 if ($role eq 'edit' && $self->current_user->user_object->email_confirmed);
63
64	my $r = SQLSession::Model::Role->new();
65	$r->load_by_cols(
66	user_id => $self->current_user->id,
67	role => $role
68	);
69
70	warn "role is '$role' for ", $self->current_user->user_object->email, " ",
71	$r->id ? 'OK' : 'DENY', $/;
72
73	return $r->id;
74	}
75
76	1;