James Moger
2011-10-23 fe326255202dcfac8b0991ca9d28e3cf4bcc4fe6
src/com/gitblit/RpcServlet.java
@@ -15,18 +15,24 @@
 */
package com.gitblit;
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;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
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.
@@ -51,26 +57,17 @@
    * @throws java.io.IOException
    */
   @Override
   protected void processRequest(javax.servlet.http.HttpServletRequest request,
         javax.servlet.http.HttpServletResponse response) throws javax.servlet.ServletException,
         java.io.IOException {
   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()));
      if (!GitBlit.getBoolean(Keys.web.enableRpcServlet, false)) {
         logger.warn(Keys.web.enableRpcServlet + " must be set TRUE for rpc requests.");
         response.sendError(HttpServletResponse.SC_FORBIDDEN);
         return;
      }
      // TODO user authentication and authorization
      UserModel user = null;
      UserModel user = (UserModel) request.getUserPrincipal();
      Object result = null;
      if (RpcRequest.LIST_REPOSITORIES.equals(reqType)) {
         // list repositories
         // Determine the Gitblit clone url
         String gitblitUrl = HttpUtils.getGitblitURL(request);
         StringBuilder sb = new StringBuilder();
@@ -79,6 +76,7 @@
         sb.append("{0}");
         String cloneUrl = sb.toString();
         // list repositories
         List<RepositoryModel> list = GitBlit.self().getRepositoryModels(user);
         Map<String, RepositoryModel> repositories = new HashMap<String, RepositoryModel>();
         for (RepositoryModel model : list) {
@@ -88,17 +86,120 @@
         result = repositories;
      } else if (RpcRequest.LIST_USERS.equals(reqType)) {
         // list users
         if (user == null || !user.canAdmin) {
            response.sendError(HttpServletResponse.SC_FORBIDDEN);
            return;
         }
         // user is authorized to retrieve all accounts
         List<String> names = GitBlit.self().getAllUsernames();
         List<UserModel> users = new ArrayList<UserModel>();
         for (String name : names) {
            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