main.go

package main

import (
	"os"
	"fmt"
	"errors"
	"runtime"

	"github.com/mgutz/ansi"
	"github.com/spf13/cobra"
	"github.com/abdfnx/gomo/tools"
	"github.com/abdfnx/gomo/cmd/gomo"
	"github.com/abdfnx/gomo/cmd/factory"
	"github.com/abdfnx/gomo/internal/checker"
	"github.com/AlecAivazis/survey/v2/terminal"
	surveyCore "github.com/AlecAivazis/survey/v2/core"
)

var (
	version string
	buildDate string
)

type exitCode int

const (
	exitOK     exitCode = 0
	exitError  exitCode = 1
	exitCancel exitCode = 2
)

func main() {
	code := mainRun()
	os.Exit(int(code))
}

func mainRun() exitCode {
	runtime.LockOSThread()

	cmdFactory := factory.New()
	hasDebug := os.Getenv("DEBUG") != ""
	stderr := cmdFactory.IOStreams.ErrOut

	if !cmdFactory.IOStreams.ColorEnabled() {
		surveyCore.DisableColor = true
	} else {
		surveyCore.TemplateFuncsWithColor["color"] = func(style string) string {
			switch style {
				case "white":
					if cmdFactory.IOStreams.ColorSupport256() {
						return fmt.Sprintf("\x1b[%d;5;%dm", 38, 242)
					}

					return ansi.ColorCode("default")

				default:
					return ansi.ColorCode(style)
			}
		}
	}

	if len(os.Args) > 1 && os.Args[1] != "" {
		cobra.MousetrapHelpText = ""
	}

	RootCmd := gomo.Execute(cmdFactory, version, buildDate)

	if cmd, err := RootCmd.ExecuteC(); err != nil {
		if err == tools.SilentError {
			return exitError
		} else if tools.IsUserCancellation(err) {
			if errors.Is(err, terminal.InterruptErr) {
				fmt.Fprint(stderr, "\n")
			}

			return exitCancel
		}

		tools.PrintError(stderr, err, cmd, hasDebug)

		return exitError
	}

	if gomo.HasFailed() {
		return exitError
	}

	checker.Check(version)

	return exitOK
}