gogitlabber/cmd/gogitlabber/input.go

package main

import (
	"flag"
	"fmt"
	"os"
	"strconv"
	"strings"

	"github.com/scornet256/go-logger"
)

// set default values and override values from environment variables
func setDefaultsFromEnv() {

	// set default values
	debug = false
	concurrency = 15
	gitHost = "gitlab.com"
	gitToken = ""
	includeArchived = "excluded"
	repoDestinationPre = "$HOME/Documents"

	// override with environment variables if present
	if envDebug := os.Getenv("GOGITLABBER_DEBUG"); envDebug != "" {
		if debugVal, err := strconv.ParseBool(envDebug); err == nil {
			debug = debugVal
		} else {
			logger.Print("Warning: Invalid debug value in environment, using default", nil)
		}
	}

	if envBackend := os.Getenv("GOGITLABBER_BACKEND"); envBackend != "" {
		gitBackend = envBackend
	}

	if envToken := os.Getenv("GIT_API_TOKEN"); envToken != "" {
		gitToken = envToken
	}

	if envHost := os.Getenv("GIT_URL"); envHost != "" {
		gitHost = envHost
	}

	if envRepoDest := os.Getenv("GOGITLABBER_DESTINATION"); envRepoDest != "" {
		repoDestinationPre = envRepoDest
	}

	if envConcurrency := os.Getenv("GOGITLABBER_CONCURRENCY"); envConcurrency != "" {
		if concurrencyVal, err := strconv.Atoi(envConcurrency); err == nil {
			concurrency = concurrencyVal
		} else {
			logger.Print("Warning: Invalid concurrency value in environment, using default", nil)
		}
	}

	if envArchived := os.Getenv("GOGITLABBER_ARCHIVED"); envArchived != "" {
		switch envArchived {
		case "any", "exclusive", "excluded":
			includeArchived = envArchived
		default:
			logger.Print("Warning: Invalid archived value in environment, using default", nil)
		}
	}
}

func manageArguments() {

	// set defaults from environment variables
	setDefaultsFromEnv()

	// define flags (which will override environment variables)
	var archivedFlag = flag.String(
		"archived",
		includeArchived,
		"To include archived repositories (any|excluded|exclusive)\n  example: -archived=any\nenv = GOGITLABBER_ARCHIVED\n")

	var concurrencyFlag = flag.Int(
		"concurrency",
		concurrency,
		"Specify repository concurrency\n  example: -concurrency=15\nenv = GOGITLABBER_CONCURRENCY\n")

	var destinationFlag = flag.String(
		"destination",
		repoDestinationPre,
		"Specify where to check the repositories out\n  example: -destination=$HOME/repos\nenv = GOGITLABBER_DESTINATION\n")

	var backendFlag = flag.String(
		"backend",
		gitBackend,
		"Specify git backend\n  example -backend=gitlab\nenv = GOGITLABBER_BACKEND\n")

	var hostFlag = flag.String(
		"git-url",
		gitHost,
		"Specify GitLab/Gitea host\n  example: -git-url=gitlab.com\nenv = GIT_URL\n")

	var tokenFlag = flag.String(
		"git-api-token",
		gitToken,
		"Specify GitLab/Gitea API token\n  example: -git-api=glpat-xxxx\nenv = GIT_API_TOKEN\n")

	var debugFlag = flag.Bool(
		"debug",
		debug,
		"Toggle debug mode\n  example: -debug=true\nenv = GOGITLABBER_DEBUG\n")

	versionFlag := flag.Bool("version", false, "Print the version and exit")

	flag.Parse()

	// print version
	if *versionFlag {
		fmt.Println(version)
		os.Exit(0)
	}

	// override with flag values (higher precedence)
	concurrency = *concurrencyFlag
	debug = *debugFlag
	gitHost = *hostFlag
	gitToken = *tokenFlag
	gitBackend = *backendFlag
	includeArchived = *archivedFlag
	repoDestinationPre = *destinationFlag

	// add slash 🎩🎸 if not provided
	if !strings.HasSuffix(repoDestinationPre, "/") {
		repoDestinationPre += "/"
	}

	// validate required parameters
	if gitToken == "" {
		flag.Usage()
		logger.Fatal("Configuration: API Token not found", nil)
	}

	// validate archived option
	switch includeArchived {
	case "any", "exclusive", "excluded":
	default:
		flag.Usage()
		logger.Fatal("Configuration: Invalid archive option: "+includeArchived, nil)
	}

	// log configuration
	logger.Print("Configuration: Using host: "+gitHost, nil)
	logger.Print("Configuration: Using destination: "+repoDestinationPre, nil)
	logger.Print("Configuration: Using concurrency: "+strconv.Itoa(concurrency), nil)
	logger.Print("Configuration: Using archived option: "+includeArchived, nil)
	if debug {
		logger.Print("Configuration: Debug mode enabled", nil)
	}
}