com.beijunyi.parallelgit.filesystem.utils.GitUriBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of parallelgit-filesystem Show documentation
There is a newer version: 2.0.0
Show newest version
package com.beijunyi.parallelgit.filesystem.utils;

import java.io.File;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.LinkedHashMap;
import java.util.Map;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import com.beijunyi.parallelgit.filesystem.GitFileSystem;
import com.beijunyi.parallelgit.filesystem.GitFileSystemProvider;
import com.beijunyi.parallelgit.filesystem.GitPath;
import org.eclipse.jgit.lib.Repository;

public class GitUriBuilder {

  private String repository;
  private String file;
  private final Map params = new LinkedHashMap<>();

  @Nonnull
  public static GitUriBuilder prepare() {
    return new GitUriBuilder();
  }

  @Nonnull
  public static GitUriBuilder fromFileSystem(@Nonnull GitFileSystem gfs) {
    return prepare()
             .repository(gfs.getRepository())
             .sid(gfs.getSessionId());
  }

  @Nonnull
  public GitUriBuilder sid(@Nullable String session) {
    if(session != null)
      params.put(GitUriUtils.SID_KEY, session);
    else
      params.remove(GitUriUtils.SID_KEY);
    return this;
  }

  @Nonnull
  public GitUriBuilder repository(@Nullable String repoDirPath) {
    this.repository = repoDirPath;
    return this;
  }

  @Nonnull
  public GitUriBuilder repository(@Nullable File repoDir) {
    return repository(repoDir != null ? repoDir.toURI().getPath() : null);
  }

  @Nonnull
  public GitUriBuilder repository(@Nullable Repository repository) {
    return repository(repository != null
                        ? (repository.isBare() ? repository.getDirectory() : repository.getWorkTree())
                        : null);
  }

  @Nonnull
  public GitUriBuilder file(@Nullable String filePathStr) {
    this.file = filePathStr;
    return this;
  }

  @Nonnull
  public GitUriBuilder file(@Nullable GitPath filePath) {
    return file(filePath != null ? filePath.toRealPath().toString() : null);
  }

  @Nonnull
  private String buildPath() {
    if(repository == null)
      throw new IllegalArgumentException("Missing repository");
    if(!repository.startsWith("/"))
      throw new IllegalArgumentException("Repository location must be an absolute path");
    return repository;
  }

  @Nullable
  private String buildQuery() {
    if(params.isEmpty())
      return null;
    String query = "";
    for(Map.Entry param : params.entrySet()) {
      if(!query.isEmpty()) query += "&";
      query += param.getKey() + "=" + param.getValue();
    }
    return query;
  }

  @Nullable
  private String buildFragment() {
    if(file == null || file.isEmpty() || file.equals("/"))
      return null;
    String fragment = "";
    if(!file.startsWith("/"))
      fragment += "/";
    fragment += file;
    return fragment;
  }

  @Nonnull
  public URI build() {
    try {
      return new URI(GitFileSystemProvider.GIT_FS_SCHEME, null, buildPath(), buildQuery(), buildFragment());
    } catch(URISyntaxException e) {
      throw new IllegalStateException(e);
    }
  }
}