James Moger
2012-09-10 fabe060d3a435f116128851f828e35c2af5fde67
src/com/gitblit/client/GitblitClient.java
@@ -21,23 +21,31 @@
import java.util.Collections;
import java.util.Date;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import com.gitblit.Constants;
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.AuthorizationControl;
import com.gitblit.GitBlitException.ForbiddenException;
import com.gitblit.GitBlitException.NotAllowedException;
import com.gitblit.GitBlitException.UnauthorizedException;
import com.gitblit.GitBlitException.UnknownRequestException;
import com.gitblit.Keys;
import com.gitblit.models.FederationModel;
import com.gitblit.models.FeedEntryModel;
import com.gitblit.models.FeedModel;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.ServerSettings;
import com.gitblit.models.ServerStatus;
import com.gitblit.models.SyndicatedEntryModel;
import com.gitblit.models.TeamModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.ArrayUtils;
import com.gitblit.utils.RpcUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.utils.SyndicationUtils;
/**
@@ -51,6 +59,8 @@
   private static final long serialVersionUID = 1L;
   private static final Date NEVER = new Date(0);
   protected final GitblitRegistration reg;
   public final String url;
@@ -58,6 +68,8 @@
   public final String account;
   private final char[] password;
   private volatile int protocolVersion;
   private volatile boolean allowManagement;
@@ -69,11 +81,13 @@
   private final List<UserModel> allUsers;
   private final List<TeamModel> allTeams;
   private final List<FederationModel> federationRegistrations;
   private final List<FeedModel> availableFeeds;
   private final List<SyndicatedEntryModel> syndicatedEntries;
   private final List<FeedEntryModel> syndicatedEntries;
   private final Set<String> subscribedRepositories;
@@ -86,29 +100,28 @@
      this.password = reg.password;
      this.allUsers = new ArrayList<UserModel>();
      this.allTeams = new ArrayList<TeamModel>();
      this.allRepositories = new ArrayList<RepositoryModel>();
      this.federationRegistrations = new ArrayList<FederationModel>();
      this.availableFeeds = new ArrayList<FeedModel>();
      this.syndicatedEntries = new ArrayList<SyndicatedEntryModel>();
      this.syndicatedEntries = new ArrayList<FeedEntryModel>();
      this.subscribedRepositories = new HashSet<String>();
   }
   public void login() throws IOException {
      protocolVersion = RpcUtils.getProtocolVersion(url, account, password);
      refreshSettings();
      refreshAvailableFeeds();
      refreshRepositories();
      try {
         // RSS feeds may be disabled by server
         refreshSubscribedFeeds();
      } catch (IOException e) {
         e.printStackTrace();
      }
      refreshSubscribedFeeds(0);
      try {
         // credentials may not have administrator access
         // or server may have disabled rpc management
         refreshUsers();
         refreshSettings();
         if (protocolVersion > 1) {
            refreshTeams();
         }
         allowManagement = true;
      } catch (UnauthorizedException e) {
      } catch (ForbiddenException e) {
@@ -130,7 +143,10 @@
      } catch (IOException e) {
         e.printStackTrace();
      }
   }
   public int getProtocolVersion() {
      return protocolVersion;
   }
   public boolean allowManagement() {
@@ -143,6 +159,156 @@
   public boolean isOwner(RepositoryModel model) {
      return account != null && account.equalsIgnoreCase(model.owner);
   }
   public String getURL(String action, String repository, String objectId) {
      boolean mounted = settings.get(Keys.web.mountParameters).getBoolean(true);
      StringBuilder sb = new StringBuilder();
      sb.append(url);
      sb.append('/');
      sb.append(action);
      sb.append('/');
      if (mounted) {
         // mounted url/action/repository/objectId
         sb.append(StringUtils.encodeURL(repository));
         if (!StringUtils.isEmpty(objectId)) {
            sb.append('/');
            sb.append(objectId);
         }
         return sb.toString();
      } else {
         // parameterized url/action/&r=repository&h=objectId
         sb.append("?r=");
         sb.append(repository);
         if (!StringUtils.isEmpty(objectId)) {
            sb.append("&h=");
            sb.append(objectId);
         }
         return sb.toString();
      }
   }
   public AccessRestrictionType getDefaultAccessRestriction() {
      String restriction = null;
      if (settings.hasKey(Keys.git.defaultAccessRestriction)) {
         restriction = settings.get(Keys.git.defaultAccessRestriction).currentValue;
      }
      return AccessRestrictionType.fromName(restriction);
   }
   public AuthorizationControl getDefaultAuthorizationControl() {
      String authorization = null;
      if (settings.hasKey(Keys.git.defaultAuthorizationControl)) {
         authorization = settings.get(Keys.git.defaultAuthorizationControl).currentValue;
      }
      return AuthorizationControl.fromName(authorization);
   }
   /**
    * Returns the list of pre-receive scripts the repository inherited from the
    * global settings and team affiliations.
    *
    * @param repository
    *            if null only the globally specified scripts are returned
    * @return a list of scripts
    */
   public List<String> getPreReceiveScriptsInherited(RepositoryModel repository) {
      Set<String> scripts = new LinkedHashSet<String>();
      // Globals
      for (String script : settings.get(Keys.groovy.preReceiveScripts).getStrings()) {
         if (script.endsWith(".groovy")) {
            scripts.add(script.substring(0, script.lastIndexOf('.')));
         } else {
            scripts.add(script);
         }
      }
      // Team Scripts
      if (repository != null) {
         for (String teamname : getPermittedTeamnames(repository)) {
            TeamModel team = getTeamModel(teamname);
            if (!ArrayUtils.isEmpty(team.preReceiveScripts)) {
               scripts.addAll(team.preReceiveScripts);
            }
         }
      }
      return new ArrayList<String>(scripts);
   }
   /**
    * Returns the list of all available Groovy pre-receive push hook scripts
    * that are not already inherited by the repository. Script files must have
    * .groovy extension
    *
    * @param repository
    *            optional parameter
    * @return list of available hook scripts
    */
   public List<String> getPreReceiveScriptsUnused(RepositoryModel repository) {
      Set<String> inherited = new TreeSet<String>(getPreReceiveScriptsInherited(repository));
      // create list of available scripts by excluding inherited scripts
      List<String> scripts = new ArrayList<String>();
      for (String script : settings.pushScripts) {
         if (!inherited.contains(script)) {
            scripts.add(script);
         }
      }
      return scripts;
   }
   /**
    * Returns the list of post-receive scripts the repository inherited from
    * the global settings and team affiliations.
    *
    * @param repository
    *            if null only the globally specified scripts are returned
    * @return a list of scripts
    */
   public List<String> getPostReceiveScriptsInherited(RepositoryModel repository) {
      Set<String> scripts = new LinkedHashSet<String>();
      // Global Scripts
      for (String script : settings.get(Keys.groovy.postReceiveScripts).getStrings()) {
         if (script.endsWith(".groovy")) {
            scripts.add(script.substring(0, script.lastIndexOf('.')));
         } else {
            scripts.add(script);
         }
      }
      // Team Scripts
      if (repository != null) {
         for (String teamname : getPermittedTeamnames(repository)) {
            TeamModel team = getTeamModel(teamname);
            if (!ArrayUtils.isEmpty(team.postReceiveScripts)) {
               scripts.addAll(team.postReceiveScripts);
            }
         }
      }
      return new ArrayList<String>(scripts);
   }
   /**
    * Returns the list of unused Groovy post-receive push hook scripts that are
    * not already inherited by the repository. Script files must have .groovy
    * extension
    *
    * @param repository
    *            optional parameter
    * @return list of available hook scripts
    */
   public List<String> getPostReceiveScriptsUnused(RepositoryModel repository) {
      Set<String> inherited = new TreeSet<String>(getPostReceiveScriptsInherited(repository));
      // create list of available scripts by excluding inherited scripts
      List<String> scripts = new ArrayList<String>();
      if (!ArrayUtils.isEmpty(settings.pushScripts)) {
         for (String script : settings.pushScripts) {
            if (!inherited.contains(script)) {
               scripts.add(script);
            }
         }
      }
      return scripts;
   }
   public ServerSettings getSettings() {
@@ -174,6 +340,13 @@
      return allUsers;
   }
   public List<TeamModel> refreshTeams() throws IOException {
      List<TeamModel> teams = RpcUtils.getTeams(url, account, password);
      allTeams.clear();
      allTeams.addAll(teams);
      return allTeams;
   }
   public ServerSettings refreshSettings() throws IOException {
      settings = RpcUtils.getSettings(url, account, password);
      return settings;
@@ -184,17 +357,31 @@
      return status;
   }
   public List<String> getBranches(String repository) {
      List<FeedModel> feeds = getAvailableFeeds(repository);
      List<String> branches = new ArrayList<String>();
      for (FeedModel feed : feeds) {
         branches.add(feed.branch);
      }
      Collections.sort(branches);
      return branches;
   }
   public List<FeedModel> getAvailableFeeds() {
      return availableFeeds;
   }
   public List<FeedModel> getAvailableFeeds(RepositoryModel repository) {
      return getAvailableFeeds(repository.name);
   }
   public List<FeedModel> getAvailableFeeds(String repository) {
      List<FeedModel> repositoryFeeds = new ArrayList<FeedModel>();
      if (repository == null) {
         return repositoryFeeds;
      }
      for (FeedModel feed : availableFeeds) {
         if (feed.repository.equalsIgnoreCase(repository.name)) {
         if (feed.repository.equalsIgnoreCase(repository)) {
            repositoryFeeds.add(feed);
         }
      }
@@ -209,16 +396,18 @@
      return availableFeeds;
   }
   public List<SyndicatedEntryModel> refreshSubscribedFeeds() throws IOException {
      Set<SyndicatedEntryModel> allEntries = new HashSet<SyndicatedEntryModel>();
   public List<FeedEntryModel> refreshSubscribedFeeds(int page) throws IOException {
      Set<FeedEntryModel> allEntries = new HashSet<FeedEntryModel>();
      if (reg.feeds.size() > 0) {
         for (FeedModel feed : reg.feeds) {
            feed.lastRefresh = new Date();
            List<SyndicatedEntryModel> entries = SyndicationUtils.readFeed(url,
                  feed.repository, feed.branch, feed.maxRetrieval, account, password);
            feed.lastRefreshDate = feed.currentRefreshDate;
            feed.currentRefreshDate = new Date();
            List<FeedEntryModel> entries = SyndicationUtils.readFeed(url, feed.repository,
                  feed.branch, -1, page, account, password);
            allEntries.addAll(entries);
         }
      }
      reg.cacheFeeds();
      syndicatedEntries.clear();
      syndicatedEntries.addAll(allEntries);
      Collections.sort(syndicatedEntries);
@@ -241,12 +430,36 @@
      }
   }
   public Date getLastFeedRefresh(String repository, String branch) {
      FeedModel feed = new FeedModel();
      feed.repository = repository;
      feed.branch = branch;
      if (reg.feeds.contains(feed)) {
         int idx = reg.feeds.indexOf(feed);
         feed = reg.feeds.get(idx);
         return feed.lastRefreshDate;
      }
      return NEVER;
   }
   public boolean isSubscribed(RepositoryModel repository) {
      return subscribedRepositories.contains(repository.name.toLowerCase());
   }
   public List<SyndicatedEntryModel> getSyndicatedEntries() {
   public List<FeedEntryModel> getSyndicatedEntries() {
      return syndicatedEntries;
   }
   public List<FeedEntryModel> log(String repository, String branch, int numberOfEntries, int page)
         throws IOException {
      return SyndicationUtils.readFeed(url, repository, branch, numberOfEntries, page, account,
            password);
   }
   public List<FeedEntryModel> search(String repository, String branch, String fragment,
         Constants.SearchType type, int numberOfEntries, int page) throws IOException {
      return SyndicationUtils.readSearchFeed(url, repository, branch, fragment, type,
            numberOfEntries, page, account, password);
   }
   public List<FederationModel> refreshFederationRegistrations() throws IOException {
@@ -279,6 +492,38 @@
      return usernames;
   }
   public List<TeamModel> getTeams() {
      return allTeams;
   }
   public List<String> getTeamnames() {
      List<String> teamnames = new ArrayList<String>();
      for (TeamModel team : this.allTeams) {
         teamnames.add(team.name);
      }
      Collections.sort(teamnames);
      return teamnames;
   }
   public List<String> getPermittedTeamnames(RepositoryModel repository) {
      List<String> teamnames = new ArrayList<String>();
      for (TeamModel team : this.allTeams) {
         if (team.repositories.contains(repository.name)) {
            teamnames.add(team.name);
         }
      }
      return teamnames;
   }
   public TeamModel getTeamModel(String name) {
      for (TeamModel team : allTeams) {
         if (team.name.equalsIgnoreCase(name)) {
            return team;
         }
      }
      return null;
   }
   public List<String> getFederationSets() {
      return settings.get(Keys.federation.sets).getStrings();
   }
@@ -289,11 +534,21 @@
   public boolean createRepository(RepositoryModel repository, List<String> permittedUsers)
         throws IOException {
      return createRepository(repository, permittedUsers, null);
   }
   public boolean createRepository(RepositoryModel repository, List<String> permittedUsers,
         List<String> permittedTeams) throws IOException {
      boolean success = true;
      success &= RpcUtils.createRepository(repository, url, account, password);
      if (permittedUsers.size() > 0) {
      success &= RpcUtils.createRepository(repository, url, account, password);
      if (permittedUsers != null && permittedUsers.size() > 0) {
         // if new repository has named members, set them
         success &= RpcUtils.setRepositoryMembers(repository, permittedUsers, url, account,
               password);
      }
      if (permittedTeams != null && permittedTeams.size() > 0) {
         // if new repository has named teams, set them
         success &= RpcUtils.setRepositoryTeams(repository, permittedTeams, url, account,
               password);
      }
      return success;
@@ -301,16 +556,31 @@
   public boolean updateRepository(String name, RepositoryModel repository,
         List<String> permittedUsers) throws IOException {
      return updateRepository(name, repository, permittedUsers, null);
   }
   public boolean updateRepository(String name, RepositoryModel repository,
         List<String> permittedUsers, List<String> permittedTeams) throws IOException {
      boolean success = true;
      success &= RpcUtils.updateRepository(name, repository, url, account, password);
      // always set the repository members
      success &= RpcUtils
            .setRepositoryMembers(repository, permittedUsers, url, account, password);
      // set the repository members
      if (permittedUsers != null) {
         success &= RpcUtils.setRepositoryMembers(repository, permittedUsers, url, account,
               password);
      }
      if (permittedTeams != null) {
         success &= RpcUtils.setRepositoryTeams(repository, permittedTeams, url, account,
               password);
      }
      return success;
   }
   public boolean deleteRepository(RepositoryModel repository) throws IOException {
      return RpcUtils.deleteRepository(repository, url, account, password);
   }
   public boolean clearRepositoryCache() throws IOException {
      return RpcUtils.clearRepositoryCache(url, account, password);
   }
   public boolean createUser(UserModel user) throws IOException {
@@ -325,6 +595,18 @@
      return RpcUtils.deleteUser(user, url, account, password);
   }
   public boolean createTeam(TeamModel team) throws IOException {
      return RpcUtils.createTeam(team, url, account, password);
   }
   public boolean updateTeam(String name, TeamModel team) throws IOException {
      return RpcUtils.updateTeam(name, team, url, account, password);
   }
   public boolean deleteTeam(TeamModel team) throws IOException {
      return RpcUtils.deleteTeam(team, url, account, password);
   }
   public boolean updateSettings(Map<String, String> newSettings) throws IOException {
      return RpcUtils.updateSettings(newSettings, url, account, password);
   }