Method to generate fork network graph

diff --git a/src/com/gitblit/GitBlit.java b/src/com/gitblit/GitBlit.java
index f86c66a3e16d23e1f57741a686b9a1e4ad3fe408..51c44f662bd394506fa81c8c0c602a8ab1052a3f 100644 (file)
--- a/src/com/gitblit/GitBlit.java
+++ b/src/com/gitblit/GitBlit.java
@@ -82,6 +82,7 @@ import com.gitblit.Constants.FederationToken;
 import com.gitblit.models.FederationModel;\r
 import com.gitblit.models.FederationProposal;\r
 import com.gitblit.models.FederationSet;\r
+import com.gitblit.models.ForkModel;\r
 import com.gitblit.models.Metric;\r
 import com.gitblit.models.ProjectModel;\r
 import com.gitblit.models.RepositoryModel;\r
@@ -1401,6 +1402,38 @@ public class GitBlit implements ServletContextListener {
                // user does not have a fork\r
                return null;\r
        }\r
+       \r
+       /**\r
+        * Returns the fork network for a repository by traversing up the fork graph\r
+        * to discover the root and then down through all children of the root node.\r
+        * \r
+        * @param repository\r
+        * @return a ForkModel\r
+        */\r
+       public ForkModel getForkNetwork(String repository) {\r
+               if (settings.getBoolean(Keys.git.cacheRepositoryList, true)) {\r
+                       // find the root\r
+                       RepositoryModel model = repositoryListCache.get(repository);\r
+                       while (model.originRepository != null) {\r
+                               model = repositoryListCache.get(model.originRepository);\r
+                       }\r
+                       ForkModel root = getForkModel(model.name);\r
+                       return root;\r
+               }\r
+               return null;\r
+       }\r
+       \r
+       private ForkModel getForkModel(String repository) {\r
+               RepositoryModel model = repositoryListCache.get(repository);\r
+               ForkModel fork = new ForkModel(model.originRepository, model.name);\r
+               if (!ArrayUtils.isEmpty(model.forks)) {\r
+                       for (String aFork : model.forks) {\r
+                               ForkModel fm = getForkModel(aFork);\r
+                               fork.forks.add(fm);\r
+                       }\r
+               }\r
+               return fork;\r
+       }\r
 \r
        /**\r
         * Returns the size in bytes of the repository. Gitblit caches the\r

diff --git a/src/com/gitblit/models/ForkModel.java b/src/com/gitblit/models/ForkModel.java
new file mode 100644 (file)
index 0000000..b394396
--- /dev/null
+++ b/src/com/gitblit/models/ForkModel.java
@@ -0,0 +1,80 @@
+/*
+ * Copyright 2012 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.
+ */
+package com.gitblit.models;
+
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+import com.gitblit.utils.ArrayUtils;
+import com.gitblit.utils.StringUtils;
+
+/**
+ * A ForkModel represents a repository, its direct descendants, and its origin.
+ * 
+ * @author James Moger
+ *
+ */
+public class ForkModel implements Serializable {
+
+       private static final long serialVersionUID = 1L;
+       
+       public final String originRepository;
+       
+       public final String repository;
+       
+       public final List<ForkModel> forks;
+       
+       public ForkModel(String origin, String repository) {
+               this.originRepository = origin;
+               this.repository = repository;
+               this.forks = new ArrayList<ForkModel>();
+       }
+       
+       public boolean isRoot() {
+               return StringUtils.isEmpty(originRepository);
+       }
+       
+       public boolean isNode() {
+               return !ArrayUtils.isEmpty(forks);
+       }
+       
+       public boolean isLeaf() {
+               return ArrayUtils.isEmpty(forks);
+       }
+       
+       public boolean isPersonalRepository() {
+               return repository.charAt(0) == '~';
+       }
+       
+       @Override
+       public int hashCode() {
+               return repository.hashCode();
+       }
+       
+       @Override
+       public boolean equals(Object o) {
+               if (o instanceof ForkModel) {
+                       return repository.equals(((ForkModel) o).repository);
+               }
+               return false;
+       }
+       
+       @Override
+       public String toString() {
+               return repository;
+       }
+}