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