James Moger
2011-10-02 31abc26dd0354bc2dafe27c011c2e54934a89486
src/com/gitblit/RpcServlet.java
@@ -18,6 +18,7 @@
import java.io.IOException;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -30,6 +31,7 @@
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.HttpUtils;
import com.gitblit.utils.RpcUtils;
/**
 * Handles remote procedure calls.
@@ -57,6 +59,7 @@
   protected void processRequest(HttpServletRequest request, HttpServletResponse response)
         throws ServletException, IOException {
      RpcRequest reqType = RpcRequest.fromName(request.getParameter("req"));
      String objectName = request.getParameter("name");
      logger.info(MessageFormat.format("Rpc {0} request from {1}", reqType,
            request.getRemoteAddr()));
@@ -88,6 +91,78 @@
            users.add(GitBlit.self().getUserModel(name));
         }
         result = users;
      } else if (RpcRequest.CREATE_REPOSITORY.equals(reqType)) {
         // create repository
         RepositoryModel model = deserialize(request, response, RepositoryModel.class);
         GitBlit.self().updateRepositoryModel(model.name, model, true);
      } else if (RpcRequest.EDIT_REPOSITORY.equals(reqType)) {
         // edit repository
         RepositoryModel model = deserialize(request, response, RepositoryModel.class);
         // name parameter specifies original repository name in event of
         // rename
         String repoName = objectName;
         if (repoName == null) {
            repoName = model.name;
         }
         GitBlit.self().updateRepositoryModel(repoName, model, false);
      } else if (RpcRequest.DELETE_REPOSITORY.equals(reqType)) {
         // delete repository
         RepositoryModel model = deserialize(request, response, RepositoryModel.class);
         GitBlit.self().deleteRepositoryModel(model);
      } else if (RpcRequest.CREATE_USER.equals(reqType)) {
         // create user
         UserModel model = deserialize(request, response, UserModel.class);
         GitBlit.self().updateUserModel(model.username, model, true);
      } else if (RpcRequest.EDIT_USER.equals(reqType)) {
         // edit user
         UserModel model = deserialize(request, response, UserModel.class);
         // name parameter specifies original user name in event of rename
         String username = objectName;
         if (username == null) {
            username = model.username;
         }
         GitBlit.self().updateUserModel(username, model, false);
      } else if (RpcRequest.DELETE_USER.equals(reqType)) {
         // delete user
         UserModel model = deserialize(request, response, UserModel.class);
         GitBlit.self().deleteUser(model.username);
      } else if (RpcRequest.LIST_REPOSITORY_MEMBERS.equals(reqType)) {
         // get repository members
         RepositoryModel model = GitBlit.self().getRepositoryModel(objectName);
         result = GitBlit.self().getRepositoryUsers(model);
      } else if (RpcRequest.SET_REPOSITORY_MEMBERS.equals(reqType)) {
         // update repository access list
         RepositoryModel model = GitBlit.self().getRepositoryModel(objectName);
         Collection<String> names = deserialize(request, response, RpcUtils.NAMES_TYPE);
         List<String> users = new ArrayList<String>(names);
         if (!GitBlit.self().setRepositoryUsers(model, users)) {
            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
         }
      } else if (RpcRequest.LIST_FEDERATION_REGISTRATIONS.equals(reqType)) {
         // return the list of federation registrations
         result = GitBlit.self().getFederationRegistrations();
      } else if (RpcRequest.LIST_FEDERATION_RESULTS.equals(reqType)) {
         // return the list of federation result registrations
         if (GitBlit.canFederate()) {
            result = GitBlit.self().getFederationResultRegistrations();
         } else {
            response.sendError(HttpServletResponse.SC_FORBIDDEN);
         }
      } else if (RpcRequest.LIST_FEDERATION_PROPOSALS.equals(reqType)) {
         // return the list of federation proposals
         if (GitBlit.canFederate()) {
            result = GitBlit.self().getPendingFederationProposals();
         } else {
            response.sendError(HttpServletResponse.SC_FORBIDDEN);
         }
      } else if (RpcRequest.LIST_FEDERATION_SETS.equals(reqType)) {
         // return the list of federation sets
         if (GitBlit.canFederate()) {
            String gitblitUrl = HttpUtils.getGitblitURL(request);
            result = GitBlit.self().getFederationSets(gitblitUrl);
         } else {
            response.sendError(HttpServletResponse.SC_FORBIDDEN);
         }
      }
      // send the result of the request