Renamed User to UserModel.

James Moger

2011-05-11 511554c6bdd7daa080506fb62400bf6d295ab9e2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package com.gitblit.wicket; 
 
import org.apache.wicket.Component; 
import org.apache.wicket.RestartResponseAtInterceptPageException; 
import org.apache.wicket.authorization.IUnauthorizedComponentInstantiationListener; 
import org.apache.wicket.authorization.strategies.page.AbstractPageAuthorizationStrategy; 
 
import com.gitblit.GitBlit; 
import com.gitblit.Keys; 
import com.gitblit.wicket.models.UserModel; 
import com.gitblit.wicket.pages.RepositoriesPage; 
 
public class AuthorizationStrategy extends AbstractPageAuthorizationStrategy implements IUnauthorizedComponentInstantiationListener { 
 
    public AuthorizationStrategy() { 
    } 
 
    @SuppressWarnings({ "unchecked", "rawtypes" }) 
    @Override 
    protected boolean isPageAuthorized(Class pageClass) { 
        if (BasePage.class.isAssignableFrom(pageClass)) { 
            boolean authenticateView = GitBlit.self().settings().getBoolean(Keys.web.authenticateViewPages, true); 
            boolean authenticateAdmin = GitBlit.self().settings().getBoolean(Keys.web.authenticateAdminPages, true); 
            boolean allowAdmin = GitBlit.self().settings().getBoolean(Keys.web.allowAdministration, true); 
             
            GitBlitWebSession session = GitBlitWebSession.get();             
            if (authenticateView && !session.isLoggedIn()) { 
                // authentication required 
                return false; 
            } 
             
            UserModel user = session.getUser(); 
            if (pageClass.isAnnotationPresent(AdminPage.class)) { 
                // admin page 
                if (allowAdmin) { 
                    if (authenticateAdmin) { 
                        // authenticate admin 
                        if (user != null) { 
                            return user.canAdmin(); 
                        } 
                        return false; 
                    } else { 
                        // no admin authentication required 
                        return true; 
                    } 
                } else { 
                    //admin prohibited 
                    return false; 
                } 
            } 
        } 
        return true; 
    } 
 
    @Override 
    public void onUnauthorizedInstantiation(Component component) { 
        if (component instanceof BasePage) { 
            GitBlitWebSession session = GitBlitWebSession.get(); 
            if (!session.isLoggedIn()) 
                throw new RestartResponseAtInterceptPageException(LoginPage.class); 
            else 
                throw new RestartResponseAtInterceptPageException(RepositoriesPage.class); 
        } 
    } 
}