com.capitalone.dashboard.model.RepoBranch Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core Show documentation
Core package shared by API layer and Microservices
There is a newer version: 4.0.2
package com.capitalone.dashboard.model;

import javax.validation.constraints.NotNull;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URL;
import java.util.Locale;

public class RepoBranch {
    private String url = "";
    private String branch = "";
    private RepoType type = RepoType.Unknown;
    private static final String GIT_EXTN = ".git";
    private static final String GIT_SCHEME = "git@";
    private static final String DEFAULT_SCHEME = "https://";
    private static final String GIT_SEPERATOR = ":";
    private static final String DEFAULT_SEPERATOR = "/";

    public enum RepoType {
        SVN,
        GIT,
        Unknown;

        public static com.capitalone.dashboard.model.RepoBranch.RepoType fromString(String value) {
            if (value ==  null) return RepoType.Unknown;
            for (com.capitalone.dashboard.model.RepoBranch.RepoType repoType : values()) {
                if (repoType.toString().equalsIgnoreCase(value)) {
                    return repoType;
                }
            }
            throw new IllegalArgumentException(value + " is not a valid RepoType.");
        }
    }

    public RepoBranch(String url, String branch, RepoType repoType) {
        this.url = url;
        this.branch = branch;
        this.type = repoType;
    }

    public RepoBranch() {
    }

    public String getUrl() {
        switch (this.type) {
            case GIT:
                return getGITNormalizedURL(this.url);
            case SVN:
                return this.url;
            default:
                return this.url;
        }
    }

    public void setUrl(String url) {
        this.url = url.toLowerCase(Locale.US);
    }

    public String getBranch() {
        switch (this.getType()) {
            case GIT: return getGITNormalizedBranch(branch);
            case SVN: return branch;
            default: return branch;
        }
    }

    public void setBranch(String branch) {
        switch (this.getType()) {
            case GIT: this.branch = getGITNormalizedBranch(branch); break;
            case SVN: this.branch = branch; break;
            default: this.branch = branch; break;
        }
    }

    public RepoType getType() {
        return type;
    }

    public void setType(RepoType type) {
        this.type = type;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || !(o instanceof RepoBranch)) return false;

        RepoBranch that = (RepoBranch) o;

        return getRepoName().equals(that.getRepoName()) && getBranch().equals(that.getBranch());
    }

    @Override
    public int hashCode() {
        int result = this.getUrl().hashCode();
        result = 31 * result + getBranch().hashCode();
        return result;
    }

    private String getRepoName() {
        try {
            URL temp = new URL(getUrl());
            return temp.getHost() + temp.getPath();
        } catch (MalformedURLException e) {
            return getUrl();
        }

    }

    private String getGITNormalizedBranch (@NotNull String branch) {
        String[] tokens = branch.split("/", -1);
        return tokens[tokens.length-1];
    }

    private String getGITNormalizedURL(@NotNull String url) {
        // check for http or https or ssh or git
        if (url.endsWith(GIT_EXTN)) {
            url = url.substring(0, url.lastIndexOf(GIT_EXTN));
        }

        if (url.contains(GIT_SCHEME)) {
            url = url.replace(GIT_SEPERATOR, DEFAULT_SEPERATOR);
            url = url.replace(GIT_SCHEME, DEFAULT_SCHEME);
        }

        String host;
        String path;
        try {
            URI processedURI = URI.create(url.replaceAll(" ", "%20"));
            host = processedURI.getHost();
            path = processedURI.getPath();
        } catch (IllegalArgumentException e) {
            return url;
        }

        /*
         * Force the urls to use https just as a means of Normalization.
         * */
        return DEFAULT_SCHEME + host + path;
    }

}