Jason Pyeron
2012-08-19 d1231c63669f4bc3643985b9032de7f998612e08
src/com/gitblit/wicket/pages/BasePage.java
@@ -17,23 +17,38 @@
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.ResourceBundle;
import java.util.TimeZone;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import org.apache.wicket.Application;
import org.apache.wicket.MarkupContainer;
import org.apache.wicket.PageParameters;
import org.apache.wicket.RestartResponseAtInterceptPageException;
import org.apache.wicket.RestartResponseException;
import org.apache.wicket.markup.html.CSSPackageResource;
import org.apache.wicket.markup.html.WebPage;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.markup.html.link.BookmarkablePageLink;
import org.apache.wicket.markup.html.link.ExternalLink;
import org.apache.wicket.markup.html.panel.FeedbackPanel;
import org.apache.wicket.markup.html.panel.Fragment;
import org.apache.wicket.protocol.http.WebRequest;
import org.apache.wicket.protocol.http.WebResponse;
import org.apache.wicket.protocol.http.servlet.ServletWebRequest;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.gitblit.Constants;
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.FederationStrategy;
import com.gitblit.GitBlit;
import com.gitblit.Keys;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.TimeUtils;
import com.gitblit.wicket.GitBlitWebSession;
import com.gitblit.wicket.WicketUtils;
import com.gitblit.wicket.panels.LinkPanel;
@@ -41,15 +56,89 @@
public abstract class BasePage extends WebPage {
   private final Logger logger;
   private transient TimeUtils timeUtils;
   public BasePage() {
      super();
      logger = LoggerFactory.getLogger(getClass());
      customizeHeader();
      loginByCookie();
   }
   public BasePage(PageParameters params) {
      super(params);
      logger = LoggerFactory.getLogger(getClass());
      customizeHeader();
      loginByCookie();
   }
   private void customizeHeader() {
      if (GitBlit.getBoolean(Keys.web.useResponsiveLayout, true)) {
         add(CSSPackageResource.getHeaderContribution("bootstrap/css/bootstrap-responsive.css"));
      }
   }
   protected String getLanguageCode() {
      return GitBlitWebSession.get().getLocale().getLanguage();
   }
   protected TimeUtils getTimeUtils() {
      if (timeUtils == null) {
         ResourceBundle bundle;
         try {
            bundle = ResourceBundle.getBundle("com.gitblit.wicket.GitBlitWebApp", GitBlitWebSession.get().getLocale());
         } catch (Throwable t) {
            bundle = ResourceBundle.getBundle("com.gitblit.wicket.GitBlitWebApp");
         }
         timeUtils = new TimeUtils(bundle);
      }
      return timeUtils;
   }
   @Override
   protected void onBeforeRender() {
      if (GitBlit.isDebugMode()) {
         // strip Wicket tags in debug mode for jQuery DOM traversal
         Application.get().getMarkupSettings().setStripWicketTags(true);
      }
      super.onBeforeRender();
   }
   @Override
   protected void onAfterRender() {
      if (GitBlit.isDebugMode()) {
         // restore Wicket debug tags
         Application.get().getMarkupSettings().setStripWicketTags(false);
      }
      super.onAfterRender();
   }
   private void loginByCookie() {
      if (!GitBlit.getBoolean(Keys.web.allowCookieAuthentication, false)) {
         return;
      }
      UserModel 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
      if (user != null) {
         // Set the user into the session
         GitBlitWebSession session = GitBlitWebSession.get();
         // issue 62: fix session fixation vulnerability
         session.replaceSession();
         session.setUser(user);
         // Set Cookie
         WebResponse response = (WebResponse) getRequestCycle().getResponse();
         GitBlit.self().setCookie(response, user);
         continueToOriginalDestination();
      }
   }
   protected void setupPage(String repositoryName, String pageName) {
@@ -58,15 +147,10 @@
      } else {
         add(new Label("title", getServerName()));
      }
      // header
      String siteName = GitBlit.getString(Keys.web.siteName, Constants.NAME);
      if (siteName == null || siteName.trim().length() == 0) {
         siteName = Constants.NAME;
      }
      add(new Label("siteName", siteName));
      add(new LinkPanel("repositoryName", null, repositoryName, SummaryPage.class,
            WicketUtils.newRepositoryParameter(repositoryName)));
      add(new Label("pageName", pageName));
      ExternalLink rootLink = new ExternalLink("rootLink", urlFor(RepositoriesPage.class, null).toString());
      WicketUtils.setHtmlTooltip(rootLink, GitBlit.getString(Keys.web.siteName, Constants.NAME));
      add(rootLink);
      // Feedback panel for info, warning, and non-fatal error messages
      add(new FeedbackPanel("feedback"));
@@ -74,17 +158,12 @@
      // footer
      if (GitBlit.getBoolean(Keys.web.authenticateViewPages, true)
            || GitBlit.getBoolean(Keys.web.authenticateAdminPages, true)) {
         if (GitBlitWebSession.get().isLoggedIn()) {
            // logout
            add(new LinkPanel("userPanel", null, getString("gb.logout") + " "
                  + GitBlitWebSession.get().getUser().toString(), LogoutPage.class));
         } else {
            // login
            add(new LinkPanel("userPanel", null, getString("gb.login"), LoginPage.class));
         }
         UserFragment userFragment = new UserFragment("userPanel", "userFragment", BasePage.this);
         add(userFragment);
      } else {
         add(new Label("userPanel", ""));
      }
      add(new Label("gbVersion", "v" + Constants.VERSION));
      if (GitBlit.getBoolean(Keys.web.aggressiveHeapManagement, false)) {
         System.gc();
@@ -111,10 +190,28 @@
      }
      return map;
   }
   protected Map<FederationStrategy, String> getFederationTypes() {
      Map<FederationStrategy, String> map = new LinkedHashMap<FederationStrategy, String>();
      for (FederationStrategy type : FederationStrategy.values()) {
         switch (type) {
         case EXCLUDE:
            map.put(type, getString("gb.excludeFromFederation"));
            break;
         case FEDERATE_THIS:
            map.put(type, getString("gb.federateThis"));
            break;
         case FEDERATE_ORIGIN:
            map.put(type, getString("gb.federateOrigin"));
            break;
         }
      }
      return map;
   }
   protected TimeZone getTimeZone() {
      return GitBlit.getBoolean(Keys.web.useClientTimezone, false) ? GitBlitWebSession.get()
            .getTimezone() : TimeZone.getDefault();
            .getTimezone() : GitBlit.getTimezone();
   }
   protected String getServerName() {
@@ -122,12 +219,31 @@
      HttpServletRequest req = servletWebRequest.getHttpServletRequest();
      return req.getServerName();
   }
   protected String getRepositoryUrl(RepositoryModel repository) {
      StringBuilder sb = new StringBuilder();
      sb.append(WicketUtils.getGitblitURL(getRequestCycle().getRequest()));
      sb.append(Constants.GIT_PATH);
      sb.append(repository.name);
      // inject username into repository url if authentication is required
      if (repository.accessRestriction.exceeds(AccessRestrictionType.NONE)
            && GitBlitWebSession.get().isLoggedIn()) {
         String username = GitBlitWebSession.get().getUser().username;
         sb.insert(sb.indexOf("://") + 3, username + "@");
      }
      return sb.toString();
   }
   public void warn(String message, Throwable t) {
      logger.warn(message, t);
   }
   public void error(String message, boolean redirect) {
      logger.error(message);
      if (redirect) {
         GitBlitWebSession.get().cacheErrorMessage(message);
         throw new RestartResponseAtInterceptPageException(getApplication().getHomePage());
         throw new RestartResponseException(getApplication().getHomePage());
      } else {
         super.error(message);
      }
@@ -137,9 +253,49 @@
      logger.error(message, t);
      if (redirect) {
         GitBlitWebSession.get().cacheErrorMessage(message);
         throw new RestartResponseAtInterceptPageException(getApplication().getHomePage());
         throw new RestartResponseException(getApplication().getHomePage());
      } else {
         super.error(message);
      }
   }
   public void authenticationError(String message) {
      logger.error(message);
      if (GitBlitWebSession.get().isLoggedIn()) {
         error(message, true);
      } else {
         throw new RestartResponseAtInterceptPageException(RepositoriesPage.class);
      }
   }
   /**
    * Panel fragment for displaying login or logout/change_password links.
    *
    */
   static class UserFragment extends Fragment {
      private static final long serialVersionUID = 1L;
      public UserFragment(String id, String markupId, MarkupContainer markupProvider) {
         super(id, markupId, markupProvider);
         if (GitBlitWebSession.get().isLoggedIn()) {
            // username, logout, and change password
            add(new Label("username", GitBlitWebSession.get().getUser().getDisplayName() + ":"));
            add(new LinkPanel("loginLink", null, markupProvider.getString("gb.logout"),
                  LogoutPage.class));
            boolean editCredentials = GitBlit.self().supportsCredentialChanges();
            // quick and dirty hack for showing a separator
            add(new Label("separator", "|").setVisible(editCredentials));
            add(new BookmarkablePageLink<Void>("changePasswordLink",
                  ChangePasswordPage.class).setVisible(editCredentials));
         } else {
            // login
            add(new Label("username").setVisible(false));
            add(new Label("loginLink").setVisible(false));
            add(new Label("separator").setVisible(false));
            add(new Label("changePasswordLink").setVisible(false));
         }
      }
   }
}