James Moger
2011-04-14 155bf78e3377910d29b2c912f58c0f496cb428e8
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package com.gitblit.wicket;
 
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
 
import org.apache.wicket.PageParameters;
import org.apache.wicket.markup.html.WebPage;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.markup.html.form.Form;
import org.apache.wicket.markup.html.form.PasswordTextField;
import org.apache.wicket.markup.html.form.TextField;
import org.apache.wicket.markup.html.image.ContextImage;
import org.apache.wicket.markup.html.panel.FeedbackPanel;
import org.apache.wicket.model.IModel;
import org.apache.wicket.model.Model;
import org.apache.wicket.protocol.http.WebRequest;
import org.apache.wicket.protocol.http.WebResponse;
import org.apache.wicket.protocol.http.servlet.ServletWebRequest;
 
import com.gitblit.Constants;
import com.gitblit.GitBlit;
 
public class LoginPage extends WebPage {
 
    IModel<String> username = new Model<String>("");
    IModel<String> password = new Model<String>("");
 
    public LoginPage(PageParameters params) {
        super(params);
 
        tryAutomaticLogin();
 
        add(new Label("title", getServerName()));
        add(new ContextImage("logo", "gitblt2.png"));
        add(new Label("name", Constants.NAME));
 
        Form<Void> loginForm = new LoginForm("loginForm");
        loginForm.add(new TextField<String>("username", username));
        loginForm.add(new PasswordTextField("password", password));
        loginForm.add(new FeedbackPanel("feedback"));
        add(loginForm);
    }
 
    protected String getServerName() {
        ServletWebRequest servletWebRequest = (ServletWebRequest) getRequest();
        HttpServletRequest req = servletWebRequest.getHttpServletRequest();
        return req.getServerName();
    }
 
    class LoginForm extends Form<Void> {
        private static final long serialVersionUID = 1L;
 
        public LoginForm(String id) {
            super(id);
        }
 
        @Override
        public void onSubmit() {
            String username = LoginPage.this.username.getObject();
            char[] password = LoginPage.this.password.getObject().toCharArray();
 
            User user = GitBlit.self().authenticate(username, password);
            if (user == null)
                error("Invalid username or password!");
            else
                loginUser(user);
        }
    }
 
    private void tryAutomaticLogin() {
        User user = null;
 
        // Grab cookie from Browser Session
        Cookie[] cookies = ((WebRequest) getRequestCycle().getRequest()).getCookies();
        if (cookies != null && cookies.length > 0) {
            user = GitBlit.self().authenticate(cookies);
        }
 
        // Login the user
        loginUser(user);
    }
 
    private void loginUser(User user) {
        if (user != null) {
            GitBlitWebSession session = GitBlitWebSession.get();
 
            // Set Cookie
            WebResponse response = (WebResponse) getRequestCycle().getResponse();
            GitBlit.self().setCookie(response, user);
 
            // track user object so that we do not have to continue
            // re-authenticating on each request.
            session.setUser(user);
 
            // Redirect to original page OR to first available tab
            if (!continueToOriginalDestination()) {
                // Redirect to home page
                setResponsePage(session.getApplication().getHomePage());
            }
        }
    }
}