From 22b181154c11cb250a2f7ad9eb7820a462cf3079 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Mon, 01 Oct 2012 22:29:42 -0400
Subject: [PATCH] Method to generate fork network graph
---
src/com/gitblit/GitBlit.java | 33 ++++++++++++++++
src/com/gitblit/models/ForkModel.java | 80 ++++++++++++++++++++++++++++++++++++++++
2 files changed, 113 insertions(+), 0 deletions(-)
diff --git a/src/com/gitblit/GitBlit.java b/src/com/gitblit/GitBlit.java
index f86c66a..51c44f6 100644
--- a/src/com/gitblit/GitBlit.java
+++ b/src/com/gitblit/GitBlit.java
@@ -82,6 +82,7 @@
import com.gitblit.models.FederationModel;
import com.gitblit.models.FederationProposal;
import com.gitblit.models.FederationSet;
+import com.gitblit.models.ForkModel;
import com.gitblit.models.Metric;
import com.gitblit.models.ProjectModel;
import com.gitblit.models.RepositoryModel;
@@ -1401,6 +1402,38 @@
// user does not have a fork
return null;
}
+
+ /**
+ * Returns the fork network for a repository by traversing up the fork graph
+ * to discover the root and then down through all children of the root node.
+ *
+ * @param repository
+ * @return a ForkModel
+ */
+ public ForkModel getForkNetwork(String repository) {
+ if (settings.getBoolean(Keys.git.cacheRepositoryList, true)) {
+ // find the root
+ RepositoryModel model = repositoryListCache.get(repository);
+ while (model.originRepository != null) {
+ model = repositoryListCache.get(model.originRepository);
+ }
+ ForkModel root = getForkModel(model.name);
+ return root;
+ }
+ return null;
+ }
+
+ private ForkModel getForkModel(String repository) {
+ RepositoryModel model = repositoryListCache.get(repository);
+ ForkModel fork = new ForkModel(model.originRepository, model.name);
+ if (!ArrayUtils.isEmpty(model.forks)) {
+ for (String aFork : model.forks) {
+ ForkModel fm = getForkModel(aFork);
+ fork.forks.add(fm);
+ }
+ }
+ return fork;
+ }
/**
* Returns the size in bytes of the repository. Gitblit caches the
diff --git a/src/com/gitblit/models/ForkModel.java b/src/com/gitblit/models/ForkModel.java
new file mode 100644
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;
+ }
+}
--
Gitblit v1.9.1