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 | 87 +++++++++++++++++++++++++++++++++---------- 1 files changed, 66 insertions(+), 21 deletions(-) diff --git a/src/com/gitblit/RpcServlet.java b/src/com/gitblit/RpcServlet.java index c366a18..53628a0 100644 --- a/src/com/gitblit/RpcServlet.java +++ b/src/com/gitblit/RpcServlet.java @@ -22,7 +22,6 @@ import java.util.HashMap; import java.util.List; import java.util.Map; -import java.util.Properties; import javax.servlet.ServletException; import javax.servlet.http.HttpServletRequest; @@ -30,6 +29,7 @@ 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; @@ -64,6 +64,8 @@ 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(); Object result = null; @@ -95,7 +97,11 @@ } else if (RpcRequest.CREATE_REPOSITORY.equals(reqType)) { // create repository RepositoryModel model = deserialize(request, response, RepositoryModel.class); - GitBlit.self().updateRepositoryModel(model.name, model, true); + 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); @@ -104,7 +110,11 @@ if (repoName == null) { repoName = model.name; } - GitBlit.self().updateRepositoryModel(repoName, model, false); + 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); @@ -112,7 +122,11 @@ } else if (RpcRequest.CREATE_USER.equals(reqType)) { // create user UserModel model = deserialize(request, response, UserModel.class); - GitBlit.self().updateUserModel(model.username, model, true); + 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); @@ -121,11 +135,17 @@ if (username == null) { username = model.username; } - GitBlit.self().updateUserModel(username, model, false); + 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); - GitBlit.self().deleteUser(model.username); + 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); @@ -136,44 +156,69 @@ 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); + response.setStatus(failureCode); } } else if (RpcRequest.LIST_FEDERATION_REGISTRATIONS.equals(reqType)) { // return the list of federation registrations - result = GitBlit.self().getFederationRegistrations(); + 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 (GitBlit.canFederate()) { + if (allowAdmin && GitBlit.canFederate()) { result = GitBlit.self().getFederationResultRegistrations(); } else { - response.sendError(HttpServletResponse.SC_FORBIDDEN); + response.sendError(notAllowedCode); } } else if (RpcRequest.LIST_FEDERATION_PROPOSALS.equals(reqType)) { // return the list of federation proposals - if (GitBlit.canFederate()) { + if (allowAdmin && GitBlit.canFederate()) { result = GitBlit.self().getPendingFederationProposals(); } else { - response.sendError(HttpServletResponse.SC_FORBIDDEN); + response.sendError(notAllowedCode); } } else if (RpcRequest.LIST_FEDERATION_SETS.equals(reqType)) { // return the list of federation sets - if (GitBlit.canFederate()) { + if (allowAdmin && GitBlit.canFederate()) { String gitblitUrl = HttpUtils.getGitblitURL(request); result = GitBlit.self().getFederationSets(gitblitUrl); } else { - response.sendError(HttpServletResponse.SC_FORBIDDEN); + response.sendError(notAllowedCode); } } else if (RpcRequest.LIST_SETTINGS.equals(reqType)) { // return the server's settings - Properties settings = new Properties(); - List<String> keys = GitBlit.getAllKeys(null); - for (String key:keys) { - String value = GitBlit.getString(key, null); - if (value != null) { - settings.put(key, value); + 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; } - result = settings; + } 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