git_reset_hard.go

package main

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

	"github.com/hexops/cmder"
	"github.com/hexops/wrench/internal/wrench/scripts"
)

func init() {
	const usage = `
Examples:

  Git reset hard all repositories:

    $ wrench git resethard

`

	// Parse flags for our subcommand.
	flagSet := flag.NewFlagSet("resethard", flag.ExitOnError)
	excludeSet := flagSet.String("exclude", "", "comma separated list of repositories to exclude")
	accept := flagSet.Bool("accept", false, "Actually run the commands (do not run in dry-run mode)")

	// Handles calls to our subcommand.
	handler := func(args []string) error {
		_ = flagSet.Parse(args)

		excludedRepos := map[string]struct{}{}
		for _, excluded := range strings.FieldsFunc(*excludeSet, func(r rune) bool {
			return r == ','
		}) {
			excludedRepos[excluded] = struct{}{}
		}

		for _, repo := range scripts.AllRepos {
			repoName := strings.Split(repo.Name, "/")[1]
			if _, err := os.Stat(repoName); err != nil {
				continue
			}
			if _, excluded := excludedRepos[repoName]; excluded {
				continue
			}
			if !*accept {
				fmt.Printf("$ cd %s/ && git reset --hard origin/main\n", repoName)
			} else {
				_ = scripts.ExecArgs("git", []string{"reset", "--hard", "origin/main"}, scripts.WorkDir(repoName))(os.Stderr)
			}
		}
		return nil
	}

	// Register the command.
	gitCommands = append(gitCommands, &cmder.Command{
		FlagSet: flagSet,
		Aliases: []string{},
		Handler: handler,
		UsageFunc: func() {
			fmt.Fprintf(flag.CommandLine.Output(), "Usage of 'wrench git status %s':\n", flagSet.Name())
			flagSet.PrintDefaults()
			fmt.Printf("%s", usage)
		},
	})
}