config.go

package main

import (
	"errors"
	giturls "github.com/whilp/git-urls"
	"net/url"
	"os"
	"os/user"
	"regexp"
	"strings"
)

type ConfigInput struct {
	src     string
	dst     string
	target  string
	branch  string
	cdir    string
	key     string
	proto   string
	git     bool
	quiet   bool
	verbose bool
}

type Config struct {
	src     []string
	dst     []string
	url     url.URL
	branch  string
	cdir    string
	key     string
	proto   string
	git     bool
	quiet   bool
	verbose bool
}

func NewConfig(ci ConfigInput) (Config, error) {
	// validate src & dst must not be empty
	if ci.src == "" {
		ci.src = "."
	}

	// parse src & dst strings into slices
	src := strings.Split(ci.src, ",")
	dst := strings.Split(ci.dst, ",")

	// validate clone directory
	if ci.cdir == "" {
		return Config{}, errors.New("config: cdir value must not be empty")
	}
	if _, err := os.Stat(ci.cdir); os.IsNotExist(err) {
		return Config{}, errors.New("config: cdir must be a valid path")
	}

	// validate proto is auto, https, or ssh
	if ci.proto == "" {
		return Config{}, errors.New("config: proto must not be empty")
	} else if ci.proto != "auto" && ci.proto != "https" && ci.proto != "ssh" {
		return Config{}, errors.New("config: invalid proto")
	}

	// validate url directory
	if ci.target == "" {
		return Config{}, errors.New("config: url must not be empty")
	}
	// convert short target into url
	gitUrl, err := giturls.Parse(ci.target)
	if err != nil {
		return Config{}, err
	}

	// convert short format to actual url
	filePath := regexp.MustCompile(`^file://@[a-z0-9-]{0,38}/`)
	if filePath.MatchString(gitUrl.String()) {
		configURL := ""
		if ci.proto == "ssh" || ci.proto == "auto" {
			// TODO: Only set auto to ssh if the repo is private
			ci.proto = "ssh"
			configURL = "git@github.com:" + ci.target
		} else {
			configURL = "https://github.com/" + ci.target + ".git"
		}
		gitUrl, _ = giturls.Parse(configURL)
	}

	// validate ssh key exists if proto is ssh and repo is private
	if ci.proto == "ssh" && ci.key == "" {
		usr, _ := user.Current()
		ci.key = usr.HomeDir + "/.ssh/id_rsa"
	}

	return Config{
		url:     *gitUrl,
		src:     src,
		dst:     dst,
		branch:  ci.branch,
		cdir:    ci.cdir,
		key:     ci.key,
		proto:   ci.proto,
		git:     ci.git,
		quiet:   ci.quiet,
		verbose: ci.verbose,
	}, nil
}