iris-context/commands/dev/git.go

package dev

import (
	"context"
	"os/exec"
	"strings"
	"time"

	"github.com/versenilvis/iris/commands/core"
)

func GitRemoteGenerator(tokens []string, _ string, _ string) []core.Suggestion {
	return getGitResults(tokens, "remote")
}

func GitStashGenerator(tokens []string, _ string, _ string) []core.Suggestion {
	return getGitResults(tokens, "stash", "list", "--format=%gd: %gs")
}

func getGitResults(tokens []string, args ...string) []core.Suggestion {
	return getGitResultsFiltered(tokens, false, args...)
}

func getGitResultsFiltered(tokens []string, localOnly bool, args ...string) []core.Suggestion {
	cwd := core.GetCWD()

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	cmd := exec.CommandContext(ctx, "git", args...)
	cmd.Dir = cwd
	out, err := cmd.Output()
	if err != nil {
		return nil
	}

	activeBranch := ""
	if args[0] == "branch" {
		activeCmd := exec.CommandContext(ctx, "git", "rev-parse", "--abbrev-ref", "HEAD")
		activeCmd.Dir = cwd
		if activeOut, err := activeCmd.Output(); err == nil {
			activeBranch = strings.TrimSpace(string(activeOut))
		}
	}

	seen := make(map[string]bool)
	lines := strings.Split(string(out), "\n")
	var results []core.Suggestion

	// more robust subcommand detection: find the first non-flag after "git"
	subcommand := ""
	for i := 1; i < len(tokens); i++ {
		if !strings.HasPrefix(tokens[i], "-") {
			subcommand = tokens[i]
			break
		}
	}

	for _, line := range lines {
		line = strings.TrimSpace(line)
		if line == "" {
			continue
		}

		isRemote := strings.HasPrefix(line, "remotes/")

		// skip remote tracking branches if localOnly mode
		if localOnly && isRemote {
			continue
		}

		// strip remotes/ prefix to get the usable form: origin/main
		if isRemote {
			line = strings.TrimPrefix(line, "remotes/")
		}

		// only skip active branch for checkout/switch commands
		if subcommand == "checkout" || subcommand == "switch" {
			if line == activeBranch {
				continue
			}
			// also skip any remote branch that tracks the active branch (e.g. origin/main)
			if idx := strings.Index(line, "/"); isRemote && idx != -1 {
				if line[idx+1:] == activeBranch {
					continue
				}
			}
		}

		// dedup: origin/dev and dev would both appear with -a; skip if already seen the short name
		shortName := line
		if idx := strings.Index(line, "/"); isRemote && idx != -1 {
			shortName = line[idx+1:] // "origin/dev" -> "dev"
		}
		if seen[shortName] {
			continue
		}
		seen[shortName] = true

		suggestionCmd := line
		suggestionDesc := args[0]

		if args[0] == "stash" {
			parts := strings.SplitN(line, ": ", 2)
			if len(parts) == 2 {
				suggestionCmd = parts[0]
				suggestionDesc = parts[1]
			}
		}

		results = append(results, core.Suggestion{
			Cmd:  suggestionCmd,
			Desc: suggestionDesc,
		})
	}

	if activeBranch != "" {
		for i, r := range results {
			if r.Cmd == activeBranch {
				// move active branch to front if found
				newResults := make([]core.Suggestion, 0, len(results))
				newResults = append(newResults, r)
				newResults = append(newResults, results[:i]...)
				newResults = append(newResults, results[i+1:]...)
				results = newResults
				break
			}
		}
	}

	return results
}

// GitBranchGenerator suggests git branches (local + remote, deduped)
func GitBranchGenerator(tokens []string, _ string, _ string) []core.Suggestion {
	// check if we are in "create" mode (-b or -B or -c)
	isCreateMode := false
	for _, t := range tokens {
		if t == "-b" || t == "-B" || t == "-c" || t == "-C" {
			isCreateMode = true
			break
		}
	}

	if isCreateMode {
		return nil
	}

	return getGitResults(tokens, "branch", "-a", "--format=%(refname:short)")
}

// gitLocalBranchGenerator is like GitBranchGenerator but only local branches
// used for push/pull where remote tracking branches cause duplicates
func gitLocalBranchGenerator(tokens []string, _ string, _ string) []core.Suggestion {
	isCreateMode := false
	for _, t := range tokens {
		if t == "-b" || t == "-B" || t == "-c" || t == "-C" {
			isCreateMode = true
			break
		}
	}
	if isCreateMode {
		return nil
	}
	return getGitResultsFiltered(tokens, true, "branch", "-a", "--format=%(refname:short)")
}

func GitPushPullGenerator(tokens []string, prefix string, partial string) []core.Suggestion {
	// count completed positional args (not flags, not the partial being typed)
	// tokens[0] = "git", tokens[1] = "push"/"pull", so start at 2
	// exclude tokens[len-1] because that's the partial being typed
	pArgs := []string{}
	for i := 2; i < len(tokens)-1; i++ {
		t := tokens[i]
		if t == "" || strings.HasPrefix(t, "-") {
			continue
		}
		pArgs = append(pArgs, t)
	}

	// no remote confirmed yet, suggest remotes
	if len(pArgs) == 0 {
		return GitRemoteGenerator(tokens, prefix, partial)
	}

	// remote is set, suggest local branches only (no duplicates with origin/xxx)
	return gitLocalBranchGenerator(tokens, prefix, partial)
}

func init() {
	core.Register(&core.Spec{
		Name:        "git",
		Description: "version control",
		Options: []core.Option{
			{Name: "--version", Description: "print version"},
			{Name: "--help", Description: "show help"},
		},
		Subcommands: []core.Subcommand{
			{
				Name:        "init",
				Description: "create empty repo",
				Options: []core.Option{
					{Name: "--bare", Description: "create bare repo"},
					{Name: "-b", Description: "initial branch name"},
				},
			},
			{
				Name:        "clone",
				Description: "clone a repository",
				Options: []core.Option{
					{Name: "--depth", Description: "shallow clone depth"},
					{Name: "--branch", Description: "specific branch"},
					{Name: "--bare", Description: "clone as bare"},
				},
			},
			{
				Name:        "status",
				Description: "show working tree",
				Options: []core.Option{
					{Name: "-s", Description: "short format"},
					{Name: "--porcelain", Description: "machine format"},
				},
			},
			{
				Name:        "add",
				Description: "stage changes",
				Generator:   core.FileGenerator(),
				Options: []core.Option{
					{Name: "-A", Description: "add all files"},
					{Name: "-p", Description: "interactive patch"},
					{Name: ".", Description: "add current dir"},
				},
			},
			{
				Name:        "commit",
				Description: "record changes",
				Options: []core.Option{
					{Name: "-m", Description: "commit message"},
					{Name: "-a", Description: "auto stage tracked"},
					{Name: "--amend", Description: "amend last commit"},
					{Name: "--no-edit", Description: "keep message"},
				},
			},
			{
				Name:        "push",
				Description: "update remote refs",
				Generator:   GitPushPullGenerator,
				Options: []core.Option{
					{Name: "-u", Description: "set upstream"},
					{Name: "--force", Description: "force push"},
					{Name: "--tags", Description: "push tags"},
				},
			},
			{
				Name:        "pull",
				Description: "fetch and merge",
				Generator:   GitPushPullGenerator,
				Options: []core.Option{
					{Name: "--rebase", Description: "rebase on pull"},
				},
			},
			{
				Name:        "fetch",
				Description: "download objects",
				Generator:   GitRemoteGenerator,
				Options: []core.Option{
					{Name: "--all", Description: "fetch all remotes"},
					{Name: "--prune", Description: "remove stale refs"},
				},
			},
			{
				Name:        "checkout",
				Description: "switch branches",
				Generator: func(tokens []string, prefix string, partial string) []core.Suggestion {
					for _, t := range tokens {
						if t == "-b" || t == "-B" {
							return nil
						}
					}
					branches := GitBranchGenerator(tokens, prefix, partial)
					files := core.FileGenerator()(tokens, prefix, partial)
					return append(branches, files...)
				},
				Options: []core.Option{
					{Name: "-b", Description: "create new branch"},
				},
			},
			{
				Name:        "switch",
				Description: "switch branches",
				Generator:   GitBranchGenerator,
				Options: []core.Option{
					{Name: "-c", Description: "create and switch"},
				},
			},
			{
				Name:        "branch",
				Description: "manage branches",
				Generator:   GitBranchGenerator,
				Options: []core.Option{
					{Name: "-d", Description: "delete branch"},
					{Name: "-D", Description: "force delete"},
					{Name: "-a", Description: "list all"},
					{Name: "-m", Description: "rename branch"},
				},
			},
			{
				Name:        "merge",
				Description: "join branches",
				Generator:   GitBranchGenerator,
				Options: []core.Option{
					{Name: "--no-ff", Description: "no fast forward"},
					{Name: "--squash", Description: "squash commits"},
					{Name: "--abort", Description: "abort merge"},
				},
			},
			{
				Name:        "rebase",
				Description: "reapply commits",
				Generator:   GitBranchGenerator,
				Options: []core.Option{
					{Name: "-i", Description: "interactive"},
					{Name: "--onto", Description: "rebase onto"},
					{Name: "--abort", Description: "abort rebase"},
					{Name: "--continue", Description: "continue rebase"},
				},
			},
			{
				Name:        "log",
				Description: "show commit log",
				Generator:   core.FileGenerator(),
				Options: []core.Option{
					{Name: "--oneline", Description: "compact format"},
					{Name: "--graph", Description: "show graph"},
					{Name: "-n", Description: "limit count"},
				},
			},
			{
				Name:        "diff",
				Description: "show changes",
				Generator:   core.FileGenerator(),
				Options: []core.Option{
					{Name: "--staged", Description: "staged changes"},
					{Name: "--stat", Description: "diffstat only"},
					{Name: "--", Description: "separate paths"},
				},
			},
			{
				Name:        "tag",
				Description: "manage tags",
				Generator: func(tokens []string, prefix string, partial string) []core.Suggestion {
					return getGitResults(tokens, "tag", "-l")
				},
				Options: []core.Option{
					{Name: "-a", Description: "annotated tag"},
					{Name: "-d", Description: "delete tag"},
					{Name: "-l", Description: "list tags"},
					{Name: "--delete", Description: "delete tag"},
					{Name: "-m", Description: "tag message"},
				},
			},
			{
				Name:        "reset",
				Description: "reset HEAD",
				Generator: func(tokens []string, prefix string, partial string) []core.Suggestion {
					branches := GitBranchGenerator(tokens, prefix, partial)
					files := core.FileGenerator()(tokens, prefix, partial)
					return append(branches, files...)
				},
				Options: []core.Option{
					{Name: "--hard", Description: "discard changes"},
					{Name: "--soft", Description: "keep staged"},
					{Name: "--mixed", Description: "unstage changes"},
				},
			},
			{
				Name:        "restore",
				Description: "restore working tree files",
				Generator:   core.FileGenerator(),
				Options: []core.Option{
					{Name: "-s", Description: "source tree"},
					{Name: "-W", Description: "working tree"},
				},
			},
			{
				Name:        "rm",
				Description: "remove files",
				Generator:   core.FileGenerator(),
				Options: []core.Option{
					{Name: "-f", Description: "force"},
					{Name: "-r", Description: "recursive"},
					{Name: "--cached", Description: "unstage only"},
				},
			},
			{
				Name:        "stash",
				Description: "stash changes",
				Subcommands: []core.Subcommand{
					{Name: "pop", Description: "apply and drop", Generator: GitStashGenerator, Options: []core.Option{{Name: "--index", Description: "try to reinstate index"}}},
					{Name: "apply", Description: "apply stash", Generator: GitStashGenerator},
					{Name: "drop", Description: "remove stash", Generator: GitStashGenerator},
					{Name: "list", Description: "list stashes"},
					{Name: "show", Description: "show stash diff", Generator: GitStashGenerator},
					{Name: "push", Description: "push to stash"},
					{Name: "branch", Description: "create branch from stash", Generator: GitBranchGenerator},
				},
			},
			{
				Name:        "remote",
				Description: "manage remotes",
				Subcommands: []core.Subcommand{
					{Name: "add", Description: "add remote", Options: []core.Option{{Name: "-f", Description: "fetch immediately"}}},
					{Name: "remove", Description: "remove remote", Generator: GitRemoteGenerator},
					{Name: "rename", Description: "rename remote", Generator: GitRemoteGenerator},
					{Name: "set-url", Description: "change remote url", Generator: GitRemoteGenerator},
					{Name: "-v", Description: "verbose list"},
				},
			},
			{
				Name:        "cherry-pick",
				Description: "apply commit",
				Options: []core.Option{
					{Name: "--no-commit", Description: "no auto commit"},
					{Name: "--abort", Description: "abort pick"},
				},
			},
			{
				Name:        "bisect",
				Description: "binary search bug",
				Subcommands: []core.Subcommand{
					{Name: "start", Description: "start bisect"},
					{Name: "good", Description: "mark good"},
					{Name: "bad", Description: "mark bad"},
					{Name: "reset", Description: "end bisect"},
				},
			},
		},
	})
}