Fabrice Bacchella
2015-06-03 b57b9ec3edd2ca7e7e44190f85e3743cea822e3e
src/main/java/com/gitblit/wicket/GitBlitWebApp.java
@@ -25,11 +25,17 @@
import org.apache.wicket.Request;
import org.apache.wicket.Response;
import org.apache.wicket.Session;
import org.apache.wicket.application.IClassResolver;
import org.apache.wicket.markup.html.WebPage;
import org.apache.wicket.protocol.http.WebApplication;
import ro.fortsoft.pf4j.PluginState;
import ro.fortsoft.pf4j.PluginWrapper;
import com.gitblit.AvatarGenerator;
import com.gitblit.IStoredSettings;
import com.gitblit.Keys;
import com.gitblit.extensions.GitblitWicketPlugin;
import com.gitblit.manager.IAuthenticationManager;
import com.gitblit.manager.IFederationManager;
import com.gitblit.manager.IGitblit;
@@ -38,9 +44,11 @@
import com.gitblit.manager.IProjectManager;
import com.gitblit.manager.IRepositoryManager;
import com.gitblit.manager.IRuntimeManager;
import com.gitblit.manager.IServicesManager;
import com.gitblit.manager.IUserManager;
import com.gitblit.tickets.ITicketService;
import com.gitblit.transport.ssh.IPublicKeyManager;
import com.gitblit.utils.XssFilter;
import com.gitblit.wicket.pages.ActivityPage;
import com.gitblit.wicket.pages.BlamePage;
import com.gitblit.wicket.pages.BlobDiffPage;
@@ -52,6 +60,7 @@
import com.gitblit.wicket.pages.DocPage;
import com.gitblit.wicket.pages.DocsPage;
import com.gitblit.wicket.pages.EditMilestonePage;
import com.gitblit.wicket.pages.EditRepositoryPage;
import com.gitblit.wicket.pages.EditTicketPage;
import com.gitblit.wicket.pages.ExportTicketPage;
import com.gitblit.wicket.pages.FederationRegistrationPage;
@@ -66,6 +75,7 @@
import com.gitblit.wicket.pages.MyDashboardPage;
import com.gitblit.wicket.pages.MyTicketsPage;
import com.gitblit.wicket.pages.NewMilestonePage;
import com.gitblit.wicket.pages.NewRepositoryPage;
import com.gitblit.wicket.pages.NewTicketPage;
import com.gitblit.wicket.pages.OverviewPage;
import com.gitblit.wicket.pages.PatchPage;
@@ -77,18 +87,31 @@
import com.gitblit.wicket.pages.SummaryPage;
import com.gitblit.wicket.pages.TagPage;
import com.gitblit.wicket.pages.TagsPage;
import com.gitblit.wicket.pages.TeamsPage;
import com.gitblit.wicket.pages.TicketsPage;
import com.gitblit.wicket.pages.TreePage;
import com.gitblit.wicket.pages.UserPage;
import com.gitblit.wicket.pages.UsersPage;
import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.Singleton;
public class GitBlitWebApp extends WebApplication {
@Singleton
public class GitBlitWebApp extends WebApplication implements GitblitWicketApp {
   private final Class<? extends WebPage> homePageClass = MyDashboardPage.class;
   private final Class<? extends WebPage> newRepositoryPageClass = NewRepositoryPage.class;
   private final Map<String, CacheControl> cacheablePages = new HashMap<String, CacheControl>();
   private final Provider<IPublicKeyManager> publicKeyManagerProvider;
   private final Provider<ITicketService> ticketServiceProvider;
   private final IStoredSettings settings;
   private final XssFilter xssFilter;
   private final IRuntimeManager runtimeManager;
@@ -100,8 +123,6 @@
   private final IAuthenticationManager authenticationManager;
   private final IPublicKeyManager publicKeyManager;
   private final IRepositoryManager repositoryManager;
   private final IProjectManager projectManager;
@@ -110,30 +131,38 @@
   private final IGitblit gitblit;
   private final IServicesManager services;
   @Inject
   public GitBlitWebApp(
         Provider<IPublicKeyManager> publicKeyManagerProvider,
         Provider<ITicketService> ticketServiceProvider,
         IRuntimeManager runtimeManager,
         IPluginManager pluginManager,
         INotificationManager notificationManager,
         IUserManager userManager,
         IAuthenticationManager authenticationManager,
         IPublicKeyManager publicKeyManager,
         IRepositoryManager repositoryManager,
         IProjectManager projectManager,
         IFederationManager federationManager,
         IGitblit gitblit) {
         IGitblit gitblit,
         IServicesManager services) {
      super();
      this.publicKeyManagerProvider = publicKeyManagerProvider;
      this.ticketServiceProvider = ticketServiceProvider;
      this.settings = runtimeManager.getSettings();
      this.xssFilter = runtimeManager.getXssFilter();
      this.runtimeManager = runtimeManager;
      this.pluginManager = pluginManager;
      this.notificationManager = notificationManager;
      this.userManager = userManager;
      this.authenticationManager = authenticationManager;
      this.publicKeyManager = publicKeyManager;
      this.repositoryManager = repositoryManager;
      this.projectManager = projectManager;
      this.federationManager = federationManager;
      this.gitblit = gitblit;
      this.services = services;
   }
   @Override
@@ -161,9 +190,9 @@
      // setup the standard gitweb-ish urls
      mount("/repositories", RepositoriesPage.class);
      mount("/overview", OverviewPage.class, "r", "h");
      mount("/overview", OverviewPage.class, "r");
      mount("/summary", SummaryPage.class, "r");
      mount("/reflog", ReflogPage.class, "r", "h");
      mount("/reflog", ReflogPage.class, "r");
      mount("/commits", LogPage.class, "r", "h");
      mount("/log", LogPage.class, "r", "h");
      mount("/tags", TagsPage.class, "r");
@@ -181,6 +210,7 @@
      mount("/metrics", MetricsPage.class, "r");
      mount("/blame", BlamePage.class, "r", "h", "f");
      mount("/users", UsersPage.class);
      mount("/teams", TeamsPage.class);
      mount("/logout", LogoutPage.class);
      // setup ticket urls
@@ -193,13 +223,15 @@
      mount("/mytickets", MyTicketsPage.class, "r", "h");
      // setup the markup document urls
      mount("/docs", DocsPage.class, "r");
      mount("/docs", DocsPage.class, "r", "h");
      mount("/doc", DocPage.class, "r", "h", "f");
      // federation urls
      mount("/proposal", ReviewProposalPage.class, "t");
      mount("/registration", FederationRegistrationPage.class, "u", "n");
      mount("/new", NewRepositoryPage.class);
      mount("/edit", EditRepositoryPage.class, "r");
      mount("/activity", ActivityPage.class, "r", "h");
      mount("/lucene", LuceneSearchPage.class);
      mount("/project", ProjectPage.class, "p");
@@ -208,18 +240,40 @@
      mount("/forks", ForksPage.class, "r");
      mount("/fork", ForkPage.class, "r");
      // allow started Wicket plugins to initialize
      for (PluginWrapper pluginWrapper : pluginManager.getPlugins()) {
         if (PluginState.STARTED != pluginWrapper.getPluginState()) {
            continue;
         }
         if (pluginWrapper.getPlugin() instanceof GitblitWicketPlugin) {
            GitblitWicketPlugin wicketPlugin = (GitblitWicketPlugin) pluginWrapper.getPlugin();
            wicketPlugin.init(this);
         }
      }
      // customize the Wicket class resolver to load from plugins
      IClassResolver coreResolver = getApplicationSettings().getClassResolver();
      PluginClassResolver classResolver = new PluginClassResolver(coreResolver, pluginManager);
      getApplicationSettings().setClassResolver(classResolver);
      getMarkupSettings().setDefaultMarkupEncoding("UTF-8");
      super.init();
      buildAvatarGenerator();
   }
   private void mount(String location, Class<? extends WebPage> clazz, String... parameters) {
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#mount(java.lang.String, java.lang.Class, java.lang.String)
    */
   @Override
   public void mount(String location, Class<? extends WebPage> clazz, String... parameters) {
      if (parameters == null) {
         parameters = new String[] {};
      }
      if (!settings.getBoolean(Keys.web.mountParameters, true)) {
         parameters = new String[] {};
      }
      mount(new GitblitParamUrlCodingStrategy(settings, location, clazz, parameters));
      mount(new GitblitParamUrlCodingStrategy(settings, xssFilter, location, clazz, parameters));
      // map the mount point to the cache control definition
      if (clazz.isAnnotationPresent(CacheControl.class)) {
@@ -228,15 +282,30 @@
      }
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#getHomePage()
    */
   @Override
   public Class<? extends WebPage> getHomePage() {
      return homePageClass;
   }
   public Class<? extends WebPage> getNewRepositoryPage() {
      return newRepositoryPageClass;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#isCacheablePage(java.lang.String)
    */
   @Override
   public boolean isCacheablePage(String mountPoint) {
      return cacheablePages.containsKey(mountPoint);
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#getCacheControl(java.lang.String)
    */
   @Override
   public CacheControl getCacheControl(String mountPoint) {
      return cacheablePages.get(mountPoint);
   }
@@ -252,15 +321,26 @@
      return gitBlitWebSession;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#settings()
    */
   @Override
   public IStoredSettings settings() {
      return settings;
   }
   /**
    * Is Gitblit running in debug mode?
    *
    * @return true if Gitblit is running in debug mode
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#xssFilter()
    */
   @Override
   public XssFilter xssFilter() {
      return xssFilter;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#isDebugMode()
    */
   @Override
   public boolean isDebugMode() {
      return runtimeManager.isDebugMode();
   }
@@ -269,58 +349,122 @@
    * These methods look strange... and they are... but they are the first
    * step towards modularization across multiple commits.
    */
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#getBootDate()
    */
   @Override
   public Date getBootDate() {
      return runtimeManager.getBootDate();
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#getLastActivityDate()
    */
   @Override
   public Date getLastActivityDate() {
      return repositoryManager.getLastActivityDate();
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#runtime()
    */
   @Override
   public IRuntimeManager runtime() {
      return runtimeManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#plugins()
    */
   @Override
   public IPluginManager plugins() {
      return pluginManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#notifier()
    */
   @Override
   public INotificationManager notifier() {
      return notificationManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#users()
    */
   @Override
   public IUserManager users() {
      return userManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#authentication()
    */
   @Override
   public IAuthenticationManager authentication() {
      return authenticationManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#keys()
    */
   @Override
   public IPublicKeyManager keys() {
      return publicKeyManager;
      return publicKeyManagerProvider.get();
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#repositories()
    */
   @Override
   public IRepositoryManager repositories() {
      return repositoryManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#projects()
    */
   @Override
   public IProjectManager projects() {
      return projectManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#federation()
    */
   @Override
   public IFederationManager federation() {
      return federationManager;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#gitblit()
    */
   @Override
   public IGitblit gitblit() {
      return gitblit;
   }
   public ITicketService tickets() {
      return gitblit.getTicketService();
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#services()
    */
   @Override
   public IServicesManager services() {
      return services;
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#tickets()
    */
   @Override
   public ITicketService tickets() {
      return ticketServiceProvider.get();
   }
   /* (non-Javadoc)
    * @see com.gitblit.wicket.Webapp#getTimezone()
    */
   @Override
   public TimeZone getTimezone() {
      return runtimeManager.getTimezone();
   }
@@ -336,4 +480,30 @@
   public static GitBlitWebApp get() {
      return (GitBlitWebApp) WebApplication.get();
   }
   AvatarGenerator generator = null;
   @SuppressWarnings("unchecked")
   private void buildAvatarGenerator() {
      Class<AvatarGenerator> clazz;
      try {
         clazz = (Class<AvatarGenerator>) getClass().getClassLoader().loadClass(settings.getString(Keys.web.avatarClass, "com.gitblit.GravatarGenerator"));
         generator = clazz.newInstance();
         generator.configure(settings);
      } catch (ClassNotFoundException | SecurityException | InstantiationException | IllegalAccessException e) {
         throw new RuntimeException(e);
      }
   }
   public String buildAvatarUrl(String username, String emailaddress, String cssClass, int width, boolean identicon) {
      if (width <= 0) {
         width = 50;
      }
      if(generator != null) {
         return (String) generator.getURL(username, emailaddress, identicon, width);
      }
      return null;
   }
}