From cb285cbfddfc0b633d6b8cdb4dc0d2bd2b8b51ef Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Thu, 05 Jan 2012 17:34:05 -0500 Subject: [PATCH] Fixed bug in receive hook for repositories in subfolders --- src/com/gitblit/RpcServlet.java | 97 ++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 85 insertions(+), 12 deletions(-) diff --git a/src/com/gitblit/RpcServlet.java b/src/com/gitblit/RpcServlet.java index b068a39..2d51388 100644 --- a/src/com/gitblit/RpcServlet.java +++ b/src/com/gitblit/RpcServlet.java @@ -33,6 +33,7 @@ import com.gitblit.models.RefModel; import com.gitblit.models.RepositoryModel; import com.gitblit.models.ServerSettings; +import com.gitblit.models.TeamModel; import com.gitblit.models.UserModel; import com.gitblit.utils.HttpUtils; import com.gitblit.utils.JGitUtils; @@ -47,6 +48,8 @@ public class RpcServlet extends JsonServlet { private static final long serialVersionUID = 1L; + + public static final int PROTOCOL_VERSION = 2; public RpcServlet() { super(); @@ -68,12 +71,19 @@ 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(); + boolean allowManagement = user != null && user.canAdmin + && GitBlit.getBoolean(Keys.web.enableRpcManagement, false); + + boolean allowAdmin = user != null && user.canAdmin + && GitBlit.getBoolean(Keys.web.enableRpcAdministration, false); + Object result = null; - if (RpcRequest.LIST_REPOSITORIES.equals(reqType)) { + if (RpcRequest.GET_PROTOCOL.equals(reqType)) { + // Return the protocol version + result = PROTOCOL_VERSION; + } else if (RpcRequest.LIST_REPOSITORIES.equals(reqType)) { // Determine the Gitblit clone url String gitblitUrl = HttpUtils.getGitblitURL(request); StringBuilder sb = new StringBuilder(); @@ -91,28 +101,31 @@ } result = repositories; } else if (RpcRequest.LIST_BRANCHES.equals(reqType)) { - // list all branches in all repositories accessible to user - Map<String, List<String>> allBranches = new HashMap<String, List<String>>(); + // list all local branches in all repositories accessible to user + Map<String, List<String>> localBranches = new HashMap<String, List<String>>(); List<RepositoryModel> models = GitBlit.self().getRepositoryModels(user); for (RepositoryModel model : models) { if (!model.hasCommits) { // skip empty repository continue; } - // get branches + // get local branches Repository repository = GitBlit.self().getRepository(model.name); List<RefModel> refs = JGitUtils.getLocalBranches(repository, false, -1); - refs.addAll(JGitUtils.getRemoteBranches(repository, false, -1)); + if (model.showRemoteBranches) { + // add remote branches if repository displays them + refs.addAll(JGitUtils.getRemoteBranches(repository, false, -1)); + } if (refs.size() > 0) { List<String> branches = new ArrayList<String>(); for (RefModel ref : refs) { branches.add(ref.getName()); } - allBranches.put(model.name, branches); + localBranches.put(model.name, branches); } repository.close(); } - result = allBranches; + result = localBranches; } else if (RpcRequest.LIST_USERS.equals(reqType)) { // list users List<String> names = GitBlit.self().getAllUsernames(); @@ -121,6 +134,14 @@ users.add(GitBlit.self().getUserModel(name)); } result = users; + } else if (RpcRequest.LIST_TEAMS.equals(reqType)) { + // list teams + List<String> names = GitBlit.self().getAllTeamnames(); + List<TeamModel> teams = new ArrayList<TeamModel>(); + for (String name : names) { + teams.add(GitBlit.self().getTeamModel(name)); + } + result = teams; } else if (RpcRequest.CREATE_REPOSITORY.equals(reqType)) { // create repository RepositoryModel model = deserialize(request, response, RepositoryModel.class); @@ -173,6 +194,33 @@ if (!GitBlit.self().deleteUser(model.username)) { response.setStatus(failureCode); } + } else if (RpcRequest.CREATE_TEAM.equals(reqType)) { + // create team + TeamModel model = deserialize(request, response, TeamModel.class); + try { + GitBlit.self().updateTeamModel(model.name, model, true); + } catch (GitBlitException e) { + response.setStatus(failureCode); + } + } else if (RpcRequest.EDIT_TEAM.equals(reqType)) { + // edit team + TeamModel model = deserialize(request, response, TeamModel.class); + // name parameter specifies original team name in event of rename + String teamname = objectName; + if (teamname == null) { + teamname = model.name; + } + try { + GitBlit.self().updateTeamModel(teamname, model, false); + } catch (GitBlitException e) { + response.setStatus(failureCode); + } + } else if (RpcRequest.DELETE_TEAM.equals(reqType)) { + // delete team + TeamModel model = deserialize(request, response, TeamModel.class); + if (!GitBlit.self().deleteTeam(model.name)) { + response.setStatus(failureCode); + } } else if (RpcRequest.LIST_REPOSITORY_MEMBERS.equals(reqType)) { // get repository members RepositoryModel model = GitBlit.self().getRepositoryModel(objectName); @@ -183,6 +231,18 @@ 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_REPOSITORY_TEAMS.equals(reqType)) { + // get repository teams + RepositoryModel model = GitBlit.self().getRepositoryModel(objectName); + result = GitBlit.self().getRepositoryTeams(model); + } else if (RpcRequest.SET_REPOSITORY_TEAMS.equals(reqType)) { + // update repository team access list + RepositoryModel model = GitBlit.self().getRepositoryModel(objectName); + Collection<String> names = deserialize(request, response, RpcUtils.NAMES_TYPE); + List<String> teams = new ArrayList<String>(names); + if (!GitBlit.self().setRepositoryTeams(model, teams)) { response.setStatus(failureCode); } } else if (RpcRequest.LIST_FEDERATION_REGISTRATIONS.equals(reqType)) { @@ -221,13 +281,26 @@ // return all settings result = settings; } else { - // return management settings only - String[] keys = { Keys.realm.minPasswordLength, Keys.realm.passwordStorage, - Keys.federation.sets }; + // anonymous users get a few settings to allow browser launching + List<String> keys = new ArrayList<String>(); + keys.add(Keys.web.siteName); + keys.add(Keys.web.mountParameters); + keys.add(Keys.web.syndicationEntries); + + if (allowManagement) { + // keys necessary for repository and/or user management + keys.add(Keys.realm.minPasswordLength); + keys.add(Keys.realm.passwordStorage); + keys.add(Keys.federation.sets); + } + // build the settings ServerSettings managementSettings = new ServerSettings(); for (String key : keys) { managementSettings.add(settings.get(key)); } + if (allowManagement) { + managementSettings.pushScripts = settings.pushScripts; + } result = managementSettings; } } else if (RpcRequest.EDIT_SETTINGS.equals(reqType)) { -- Gitblit v1.9.1