Method to generate fork network graph

James Moger

2012-10-01 22b181154c11cb250a2f7ad9eb7820a462cf3079

Method to generate fork network graph

 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


 src/com/gitblit/models/ForkModel.java

New file
@@ -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;
    }
}