Revise, complete, and fully test the FORK rpc

4 files changed, 49 insertions, 76 deletions

diff --git a/src/main/java/com/gitblit/Constants.java b/src/main/java/com/gitblit/Constants.java
index cbbc0cac..c8ce83c1 100644
--- a/src/main/java/com/gitblit/Constants.java
+++ b/src/main/java/com/gitblit/Constants.java
@@ -350,10 +350,11 @@ public class Constants {
 	 * a client.

 	 */

 	public static enum RpcRequest {

-		// Order is important here.  anything above LIST_SETTINGS requires

+		// Order is important here.  anything after LIST_SETTINGS requires

 		// administrator privileges and web.allowRpcManagement.

-		CLEAR_REPOSITORY_CACHE, REINDEX_TICKETS, GET_PROTOCOL, LIST_REPOSITORIES, LIST_BRANCHES, GET_USER, LIST_SETTINGS,

-		CREATE_REPOSITORY, FORK_REPOSITORY, EDIT_REPOSITORY, DELETE_REPOSITORY,

+		CLEAR_REPOSITORY_CACHE, REINDEX_TICKETS, GET_PROTOCOL, LIST_REPOSITORIES, LIST_BRANCHES, GET_USER,

+		FORK_REPOSITORY, LIST_SETTINGS,

+		CREATE_REPOSITORY, EDIT_REPOSITORY, DELETE_REPOSITORY,

 		LIST_USERS, CREATE_USER, EDIT_USER, DELETE_USER,

 		LIST_TEAMS, CREATE_TEAM, EDIT_TEAM, DELETE_TEAM,

 		LIST_REPOSITORY_MEMBERS, SET_REPOSITORY_MEMBERS, LIST_REPOSITORY_TEAMS, SET_REPOSITORY_TEAMS,

diff --git a/src/main/java/com/gitblit/models/UserRepositoryCompositeModel.java b/src/main/java/com/gitblit/models/UserRepositoryCompositeModel.java
deleted file mode 100644
index 122bb3ba..00000000
--- a/src/main/java/com/gitblit/models/UserRepositoryCompositeModel.java
+++ /dev/null
@@ -1,47 +0,0 @@
-package com.gitblit.models;
-
-/*
- * Copyright 2011 gitblit.com.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-
-import com.gitblit.utils.StringUtils;
-
-import java.io.Serializable;
-
-public class UserRepositoryCompositeModel implements Serializable {
-
-    private static final long serialVersionUID = 1L;
-
-    public UserModel userModel;
-    public RepositoryModel repositoryModel;
-
-    public UserModel getUserModel() {
-        return userModel;
-    }
-
-    public void setUserModel(UserModel userModel) {
-        this.userModel = userModel;
-    }
-
-    public RepositoryModel getRepositoryModel() {
-        return repositoryModel;
-    }
-
-    public void setRepositoryModel(RepositoryModel repositoryModel) {
-        this.repositoryModel = repositoryModel;
-    }
-
-}
\ No newline at end of file
diff --git a/src/main/java/com/gitblit/servlet/RpcServlet.java b/src/main/java/com/gitblit/servlet/RpcServlet.java
index 80c50714..b3016ad6 100644
--- a/src/main/java/com/gitblit/servlet/RpcServlet.java
+++ b/src/main/java/com/gitblit/servlet/RpcServlet.java
@@ -27,7 +27,6 @@ import javax.servlet.ServletException;
 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 

-import com.gitblit.models.*;

 import org.eclipse.jgit.lib.Repository;

 

 import com.gitblit.Constants;

@@ -36,6 +35,12 @@ import com.gitblit.GitBlitException;
 import com.gitblit.IStoredSettings;

 import com.gitblit.Keys;

 import com.gitblit.manager.IGitblit;

+import com.gitblit.models.RefModel;

+import com.gitblit.models.RegistrantAccessPermission;

+import com.gitblit.models.RepositoryModel;

+import com.gitblit.models.ServerSettings;

+import com.gitblit.models.TeamModel;

+import com.gitblit.models.UserModel;

 import com.gitblit.utils.DeepCopier;

 import com.gitblit.utils.HttpUtils;

 import com.gitblit.utils.JGitUtils;

@@ -53,7 +58,7 @@ public class RpcServlet extends JsonServlet {
 

     private static final long serialVersionUID = 1L;

 

-    public static final int PROTOCOL_VERSION = 7;

+    public static final int PROTOCOL_VERSION = 8;

 

     private IStoredSettings settings;

 

@@ -191,22 +196,31 @@ public class RpcServlet extends JsonServlet {
                 response.setStatus(failureCode);

             }

         } else if (RpcRequest.FORK_REPOSITORY.equals(reqType)) {

-            // fork repository

-            UserRepositoryCompositeModel userRepositoryCompositeModel = deserialize(request, response,

-                    UserRepositoryCompositeModel.class);

-            RepositoryModel repoModel = userRepositoryCompositeModel.getRepositoryModel();

-            UserModel userModel = userRepositoryCompositeModel.getUserModel();

-            try {

-                if (repoModel != null && userModel != null) {

-                    gitblit.fork(repoModel, userModel);

-                } else {

-                    System.out.println("Non existing user model or repo model");

-                    response.setStatus(failureCode);

-                }

-

-            } catch (GitBlitException e) {

-                response.setStatus(failureCode);

-            }

+        	// fork repository

+        	RepositoryModel origin = gitblit.getRepositoryModel(objectName);

+        	if (origin == null) {

+        		// failed to find repository, error is logged by the repository manager

+        		response.setStatus(failureCode);

+        	} else {

+        		if (user == null || !user.canFork(origin)) {

+        			logger.error("User {} is not permitted to fork '{}'!",

+        					user == null ? "anonymous" : user.username, objectName);

+        			response.setStatus(failureCode);

+        		} else {

+        			try {

+        				// fork the origin

+        				RepositoryModel fork = gitblit.fork(origin, user);

+        				if (fork == null) {

+        					logger.error("Failed to fork repository '{}'!", objectName);

+        					response.setStatus(failureCode);

+        				} else {

+        					logger.info("User {} has forked '{}'!", user.username, objectName);

+        				}

+        			} catch (GitBlitException e) {

+        				response.setStatus(failureCode);

+        			}

+        		}

+        	}

         } else if (RpcRequest.EDIT_REPOSITORY.equals(reqType)) {

             // edit repository

             RepositoryModel model = deserialize(request, response, RepositoryModel.class);

diff --git a/src/main/java/com/gitblit/utils/RpcUtils.java b/src/main/java/com/gitblit/utils/RpcUtils.java
index 904461c8..82202154 100644
--- a/src/main/java/com/gitblit/utils/RpcUtils.java
+++ b/src/main/java/com/gitblit/utils/RpcUtils.java
@@ -25,7 +25,16 @@ import java.util.Map;
 import com.gitblit.Constants;

 import com.gitblit.Constants.RpcRequest;

 import com.gitblit.GitBlitException.UnknownRequestException;

-import com.gitblit.models.*;

+import com.gitblit.models.FederationModel;

+import com.gitblit.models.FederationProposal;

+import com.gitblit.models.FederationSet;

+import com.gitblit.models.FeedModel;

+import com.gitblit.models.RegistrantAccessPermission;

+import com.gitblit.models.RepositoryModel;

+import com.gitblit.models.ServerSettings;

+import com.gitblit.models.ServerStatus;

+import com.gitblit.models.TeamModel;

+import com.gitblit.models.UserModel;

 import com.google.gson.reflect.TypeToken;

 

 /**

@@ -195,20 +204,16 @@ public class RpcUtils {
 	}

 

     /**

-     * Create a fork of an already existing repo

+     * Create a fork of a repository.

      *

      * @param repository

-     * @param user

 

      * @return true if the action succeeded

      * @throws IOException

      */

-    public static boolean forkRpository(RepositoryModel repository, UserModel user, String serverUrl,

+    public static boolean forkRepository(RepositoryModel repository, String serverUrl,

                                         String account, char[] password) throws IOException {

-        UserRepositoryCompositeModel userRepositoryCompositeModel = new UserRepositoryCompositeModel();

-        userRepositoryCompositeModel.setRepositoryModel(repository);

-        userRepositoryCompositeModel.setUserModel(user);

-        return doAction(RpcRequest.FORK_REPOSITORY, null, userRepositoryCompositeModel, serverUrl, account, password);

+        return doAction(RpcRequest.FORK_REPOSITORY, repository.name, null, serverUrl, account, password);

     }