James Moger
2011-10-23 fe326255202dcfac8b0991ca9d28e3cf4bcc4fe6
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;
@@ -28,8 +29,10 @@
import com.gitblit.Constants.RpcRequest;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.SettingModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.HttpUtils;
import com.gitblit.utils.RpcUtils;
/**
 * Handles remote procedure calls.
@@ -57,6 +60,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 +92,114 @@
            users.add(GitBlit.self().getUserModel(name));
         }
         result = users;
      } else if (RpcRequest.CREATE_REPOSITORY.equals(reqType)) {
         // create repository
         RepositoryModel model = deserialize(request, response, RepositoryModel.class);
         try {
            GitBlit.self().updateRepositoryModel(model.name, model, true);
         } catch (GitBlitException e) {
            response.setStatus(failureCode);
         }
      } else if (RpcRequest.EDIT_REPOSITORY.equals(reqType)) {
         // edit repository
         RepositoryModel model = deserialize(request, response, RepositoryModel.class);
         // name specifies original repository name in event of rename
         String repoName = objectName;
         if (repoName == null) {
            repoName = model.name;
         }
         try {
            GitBlit.self().updateRepositoryModel(repoName, model, false);
         } catch (GitBlitException e) {
            response.setStatus(failureCode);
         }
      } 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);
         try {
            GitBlit.self().updateUserModel(model.username, model, true);
         } catch (GitBlitException e) {
            response.setStatus(failureCode);
         }
      } 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;
         }
         try {
            GitBlit.self().updateUserModel(username, model, false);
         } catch (GitBlitException e) {
            response.setStatus(failureCode);
         }
      } else if (RpcRequest.DELETE_USER.equals(reqType)) {
         // delete user
         UserModel model = deserialize(request, response, UserModel.class);
         if (!GitBlit.self().deleteUser(model.username)) {
            response.setStatus(failureCode);
         }
      } 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(failureCode);
         }
      } 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(notAllowedCode);
         }
      } else if (RpcRequest.LIST_FEDERATION_PROPOSALS.equals(reqType)) {
         // return the list of federation proposals
         if (GitBlit.canFederate()) {
            result = GitBlit.self().getPendingFederationProposals();
         } else {
            response.sendError(notAllowedCode);
         }
      } 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(notAllowedCode);
         }
      } else if (RpcRequest.LIST_SETTINGS.equals(reqType)) {
         // return the server's settings
         if (GitBlit.getBoolean(Keys.web.enableRpcAdministration, false)) {
            result = GitBlit.self().getSettingsModel();
         } else {
            response.sendError(notAllowedCode);
         }
      } else if (RpcRequest.EDIT_SETTINGS.equals(reqType)) {
         // update settings on the server
         if (GitBlit.getBoolean(Keys.web.enableRpcAdministration, false)) {
            Collection<SettingModel> settings = deserialize(request, response,
                  RpcUtils.SETTINGS_TYPE);
            GitBlit.self().updateSettings(settings);
         } else {
            response.sendError(notAllowedCode);
         }
      } else if (RpcRequest.LIST_STATUS.equals(reqType)) {
         // return the server's status information
         result = GitBlit.self().getStatus();
      }
      // send the result of the request