From a70b43cde76b4baab82b4ce0d9ff82883f80b8df Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Wed, 26 Oct 2011 17:54:07 -0400
Subject: [PATCH] Added status icon

---
 src/com/gitblit/RpcServlet.java |  131 +++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 131 insertions(+), 0 deletions(-)

diff --git a/src/com/gitblit/RpcServlet.java b/src/com/gitblit/RpcServlet.java
index 9d26ee0..53628a0 100644
--- a/src/com/gitblit/RpcServlet.java
+++ b/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.ServerSettings;
 import com.gitblit.models.UserModel;
 import com.gitblit.utils.HttpUtils;
+import com.gitblit.utils.RpcUtils;
 
 /**
  * Handles remote procedure calls.
@@ -57,8 +60,11 @@
 	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()));
+
+		boolean allowAdmin = GitBlit.getBoolean(Keys.web.enableRpcAdministration, false);
 
 		UserModel user = (UserModel) request.getUserPrincipal();
 
@@ -88,6 +94,131 @@
 				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
+			if (allowAdmin) {
+				result = GitBlit.self().getFederationRegistrations();
+			} else {
+				response.sendError(notAllowedCode);
+			}
+		} else if (RpcRequest.LIST_FEDERATION_RESULTS.equals(reqType)) {
+			// return the list of federation result registrations
+			if (allowAdmin && 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 (allowAdmin && 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 (allowAdmin && 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
+			ServerSettings settings = GitBlit.self().getSettingsModel();
+			if (allowAdmin) {
+				// return all settings
+				result = settings;
+			} else {
+				// return management settings only
+				String[] keys = { Keys.realm.minPasswordLength, Keys.realm.passwordStorage,
+						Keys.federation.sets };
+				ServerSettings managementSettings = new ServerSettings();
+				for (String key : keys) {
+					managementSettings.add(settings.get(key));
+				}
+				result = managementSettings;
+			}
+		} else if (RpcRequest.EDIT_SETTINGS.equals(reqType)) {
+			// update settings on the server
+			if (allowAdmin) {
+				Map<String, String> 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
+			if (allowAdmin) {
+				result = GitBlit.self().getStatus();
+			} else {
+				response.sendError(notAllowedCode);
+			}
 		}
 
 		// send the result of the request

--
Gitblit v1.9.1