go-clone/main.go

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"go.wit.com/dev/alexflint/arg"
	"go.wit.com/gui"
	"go.wit.com/lib/gui/repolist"
	"go.wit.com/lib/gui/repostatus"
	"go.wit.com/lib/gui/shell"
	"go.wit.com/log"
)

// sent via -ldflags
var VERSION string
var BUILDTIME string

var rv *repolist.RepoList

func main() {
	pp := arg.MustParse(&argv)

	// for very new users or users unfamilar with the command line, this may help them
	if argv.Repo == "version" || argv.Repo == "help" || argv.Repo == "?" {
		pp.WriteHelp(os.Stdout)
		os.Exit(0)
	}

	// figures out where you're go.work file is
	wdir, err := findWorkFile()
	if err != nil {
		log.Info(err)
		os.Exit(-1)
	}
	if wdir != "" {
		os.Setenv("REPO_WORK_PATH", wdir)

	}

	fullgitdir := filepath.Join(wdir, argv.Repo, ".git")
	if shell.IsDir(fullgitdir) {
		if ! argv.Recursive {
			log.Info("repo already cloned", filepath.Join(wdir, argv.Repo))
			os.Exit(0)
		}
	}

	// readControlFile()

	b := gui.RawBox()
	rv = repolist.AutotypistView(b)

	if argv.Pull {
		count := scanForRepos(wdir)
		log.Info("Total repositories:", count)
		log.Info("Going to run git pull in each one")
		log.Sleep(1)
		pull := []string{"git", "pull"}

		var trycount, errcount int
		loop := rv.ReposSortByName()
		for loop.Scan() {
			repo := loop.Repo()
			if argv.DryRun {
				log.Info("git pull --dry-run", repo.Status.Path())
			} else {
				trycount += 1
				log.Info("actually run: git pull:", repo.Status.Path())
				if result := shell.PathRunRealtime(repo.Status.Path(), pull); result.Error != nil {
					log.Info("git pull error:", result.Error)
					errcount += 1
				}
			}
		}
		log.Info("Total repositories:", count, "Total attempted:", trycount, "Errors:", errcount)
		os.Exit(0)
	}

	// if the user didn't provide a repo, stop here unless --git-pull
	if argv.Repo == "" || argv.Pull {
		pp.WriteHelp(os.Stdout)
		os.Exit(0)
	}

	os.Setenv("REPO_AUTO_CLONE", "true")
	newr, err := rv.NewRepo(argv.Repo)
	if err != nil {
		log.Info("could not download:", err)
		os.Exit(-1)
	}
	newr.Status.MakeRedomod()

	fullgitdir = filepath.Join(wdir, argv.Repo, ".git")
	if ! shell.IsDir(fullgitdir) {
		log.Info("repo cloned failed", filepath.Join(wdir, argv.Repo))
		os.Exit(-1)
	}

	log.Info("scanning for repo in:", filepath.Join(wdir, argv.Repo))

	// rv.NewRepo("go.wit.com/apps/helloworld")
	for _, path := range repostatus.ScanGitDirectories(wdir) {
		gopath := strings.TrimPrefix(path, wdir)
		gopath = strings.Trim(gopath, "/")
		// log.Info("Also should add:", gopath)
		rv.NewRepo(gopath)
	}

	godep := newr.Status.GetGoDeps()
	if argv.Recursive {
		for gopath, version := range godep {
			repo, err := rv.NewRepo(gopath)
			if err != nil {
				log.Info("git clone failed for", gopath, version)
				continue
			}
			repo.Status.MakeRedomod()
		}
	}

	var count int
	loop := rv.ReposSortByName()
	for loop.Scan() {
		repo := loop.Repo()
		count += 1
		if !repo.Status.Exists("go.mod") {
			repo.Status.MakeRedomod()
		}
	}

	log.Info("Total repositories:", count)
	log.Info("Finished go-clone for", argv.Repo)
	if argv.AutoWork {
		log.Info("About to re-create", wdir+"/go.work")
		log.Info("Sleep 3. original go.work saved as go.work.last (hit ctrl-c to cancel)")
		log.Sleep(3)
		shell.PathRun(wdir, []string{"mv", "go.work", "go.work.last"})
		rv.MakeGoWork()
		shell.PathRun(wdir, []string{"go", "work", "use"})
		log.Info("")
		log.Info("original go.work file saved as go.work.last")
		log.Info("")
	}
}

// look for a go.work file
// otherwise use ~/go/src
func findWorkFile() (string, error) {
	pwd, err := os.Getwd()
	if err == nil {
		// Check for go.work in the current directory and then move up until root
		if pwd, err := digup(pwd); err == nil {
			log.Info("using go.work file in directory", pwd)
			// found an existing go.work file
			os.Chdir(pwd)
			return pwd, nil
		}
	}

	// there are no go.work files, resume the ~/go/src behavior from prior to golang 1.22
	pwd, err = useGoSrc()
	log.Info("using ~/go/src directory", pwd)
	return pwd, err
}

// this is the 'old way" and works fine for me. I use it because I like the ~/go/src directory
// because I know exactly what is in it: GO stuff & nothing else
func useGoSrc() (string, error) {
	homeDir, err := os.UserHomeDir()
	if err != nil {
		return "", err
	}
	pwd := filepath.Join(homeDir, "go/src")
	shell.Mkdir(pwd)
	os.Chdir(pwd)
	return pwd, nil
}

func digup(path string) (string, error) {
	for {
		workFilePath := filepath.Join(path, "go.work")
		if _, err := os.Stat(workFilePath); err == nil {
			return path, nil // Found the go.work file
		} else if !os.IsNotExist(err) {
			return "", err // An error other than not existing
		}

		parentPath := filepath.Dir(path)
		if parentPath == path {
			break // Reached the filesystem root
		}
		path = parentPath
	}

	return "", fmt.Errorf("no go.work file found")
}

func scanForRepos(wdir string) int {
	var count int
	log.Info("scanning for repo in:", filepath.Join(wdir, argv.Repo))

	// rv.NewRepo("go.wit.com/apps/helloworld")
	for _, path := range repostatus.ScanGitDirectories(wdir) {
		count += 1
		gopath := strings.TrimPrefix(path, wdir)
		gopath = strings.Trim(gopath, "/")
		// log.Info("Also should add:", gopath)
		rv.NewRepo(gopath)
	}
	return count
}