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);
                }
        }
}

=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);

        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	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
43	=head2 role
44
45	Test if user has role
46
47	Jifty->web->current_user->role( 'edit' );
48
49	=cut
50
51	sub role {
52	my $self = shift;
53	my $role = shift;
54
55	return 0 unless ($self->current_user->id);
56
57	return 1 if ($role eq 'admin' && $self->current_user->is_superuser);
58
59	my $r = SQLSession::Model::Role->new();
60	$r->load_by_cols(
61	user_id => $self->current_user->id,
62	role => $role
63	);
64
65	warn "role is '$role' for ", $self->current_user->user_object->email, " ",
66	$r->id ? 'OK' : 'DENY', $/;
67
68	return $r->id;
69	}
70
71