Gitblit devel
view | history | commit | commitdiff
James Moger
2011-10-02 f762b160efd5cafd919a6fd7f9587f578eceb454
commit | author | age
93f0b1 1 /*
JM 2  * Copyright 2011 gitblit.com.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.gitblit;
17
ca9d0f 18 import java.io.IOException;
93f0b1 19 import java.text.MessageFormat;
JM 20 import java.util.ArrayList;
31abc2 21 import java.util.Collection;
93f0b1 22 import java.util.HashMap;
JM 23 import java.util.List;
24 import java.util.Map;
25
ca9d0f 26 import javax.servlet.ServletException;
JM 27 import javax.servlet.http.HttpServletRequest;
93f0b1 28 import javax.servlet.http.HttpServletResponse;
JM 29
30 import com.gitblit.Constants.RpcRequest;
31 import com.gitblit.models.RepositoryModel;
32 import com.gitblit.models.UserModel;
33 import com.gitblit.utils.HttpUtils;
31abc2 34 import com.gitblit.utils.RpcUtils;
93f0b1 35
JM 36 /**
37  * Handles remote procedure calls.
38  * 
39  * @author James Moger
40  * 
41  */
42 public class RpcServlet extends JsonServlet {
43
44     private static final long serialVersionUID = 1L;
45
46     public RpcServlet() {
47         super();
48     }
49
50     /**
51      * Processes an rpc request.
52      * 
53      * @param request
54      * @param response
55      * @throws javax.servlet.ServletException
56      * @throws java.io.IOException
57      */
58     @Override
ca9d0f 59     protected void processRequest(HttpServletRequest request, HttpServletResponse response)
JM 60             throws ServletException, IOException {
93f0b1 61         RpcRequest reqType = RpcRequest.fromName(request.getParameter("req"));
31abc2 62         String objectName = request.getParameter("name");
93f0b1 63         logger.info(MessageFormat.format("Rpc {0} request from {1}", reqType,
JM 64                 request.getRemoteAddr()));
65
ca9d0f 66         UserModel user = (UserModel) request.getUserPrincipal();
93f0b1 67
JM 68         Object result = null;
69         if (RpcRequest.LIST_REPOSITORIES.equals(reqType)) {
70             // Determine the Gitblit clone url
71             String gitblitUrl = HttpUtils.getGitblitURL(request);
72             StringBuilder sb = new StringBuilder();
73             sb.append(gitblitUrl);
74             sb.append(Constants.GIT_PATH);
75             sb.append("{0}");
76             String cloneUrl = sb.toString();
77
ca9d0f 78             // list repositories
93f0b1 79             List<RepositoryModel> list = GitBlit.self().getRepositoryModels(user);
JM 80             Map<String, RepositoryModel> repositories = new HashMap<String, RepositoryModel>();
81             for (RepositoryModel model : list) {
82                 String url = MessageFormat.format(cloneUrl, model.name);
83                 repositories.put(url, model);
84             }
85             result = repositories;
86         } else if (RpcRequest.LIST_USERS.equals(reqType)) {
87             // list users
88             List<String> names = GitBlit.self().getAllUsernames();
89             List<UserModel> users = new ArrayList<UserModel>();
90             for (String name : names) {
91                 users.add(GitBlit.self().getUserModel(name));
92             }
93             result = users;
31abc2 94         } else if (RpcRequest.CREATE_REPOSITORY.equals(reqType)) {
JM 95             // create repository
96             RepositoryModel model = deserialize(request, response, RepositoryModel.class);
97             GitBlit.self().updateRepositoryModel(model.name, model, true);
98         } else if (RpcRequest.EDIT_REPOSITORY.equals(reqType)) {
99             // edit repository
100             RepositoryModel model = deserialize(request, response, RepositoryModel.class);
101             // name parameter specifies original repository name in event of
102             // rename
103             String repoName = objectName;
104             if (repoName == null) {
105                 repoName = model.name;
106             }
107             GitBlit.self().updateRepositoryModel(repoName, model, false);
108         } else if (RpcRequest.DELETE_REPOSITORY.equals(reqType)) {
109             // delete repository
110             RepositoryModel model = deserialize(request, response, RepositoryModel.class);
111             GitBlit.self().deleteRepositoryModel(model);
112         } else if (RpcRequest.CREATE_USER.equals(reqType)) {
113             // create user
114             UserModel model = deserialize(request, response, UserModel.class);
115             GitBlit.self().updateUserModel(model.username, model, true);
116         } else if (RpcRequest.EDIT_USER.equals(reqType)) {
117             // edit user
118             UserModel model = deserialize(request, response, UserModel.class);
119             // name parameter specifies original user name in event of rename
120             String username = objectName;
121             if (username == null) {
122                 username = model.username;
123             }
124             GitBlit.self().updateUserModel(username, model, false);
125         } else if (RpcRequest.DELETE_USER.equals(reqType)) {
126             // delete user
127             UserModel model = deserialize(request, response, UserModel.class);
128             GitBlit.self().deleteUser(model.username);
129         } else if (RpcRequest.LIST_REPOSITORY_MEMBERS.equals(reqType)) {
130             // get repository members
131             RepositoryModel model = GitBlit.self().getRepositoryModel(objectName);
132             result = GitBlit.self().getRepositoryUsers(model);
133         } else if (RpcRequest.SET_REPOSITORY_MEMBERS.equals(reqType)) {
134             // update repository access list
135             RepositoryModel model = GitBlit.self().getRepositoryModel(objectName);
136             Collection<String> names = deserialize(request, response, RpcUtils.NAMES_TYPE);
137             List<String> users = new ArrayList<String>(names);
138             if (!GitBlit.self().setRepositoryUsers(model, users)) {
139                 response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
140             }
141         } else if (RpcRequest.LIST_FEDERATION_REGISTRATIONS.equals(reqType)) {
142             // return the list of federation registrations
143             result = GitBlit.self().getFederationRegistrations();
144         } else if (RpcRequest.LIST_FEDERATION_RESULTS.equals(reqType)) {
145             // return the list of federation result registrations
146             if (GitBlit.canFederate()) {
147                 result = GitBlit.self().getFederationResultRegistrations();
148             } else {
149                 response.sendError(HttpServletResponse.SC_FORBIDDEN);
150             }
151         } else if (RpcRequest.LIST_FEDERATION_PROPOSALS.equals(reqType)) {
152             // return the list of federation proposals
153             if (GitBlit.canFederate()) {
154                 result = GitBlit.self().getPendingFederationProposals();
155             } else {
156                 response.sendError(HttpServletResponse.SC_FORBIDDEN);
157             }
158         } else if (RpcRequest.LIST_FEDERATION_SETS.equals(reqType)) {
159             // return the list of federation sets
160             if (GitBlit.canFederate()) {
161                 String gitblitUrl = HttpUtils.getGitblitURL(request);
162                 result = GitBlit.self().getFederationSets(gitblitUrl);
163             } else {
164                 response.sendError(HttpServletResponse.SC_FORBIDDEN);
165             }
93f0b1 166         }
JM 167
168         // send the result of the request
169         serialize(response, result);
170     }
171 }