package gocmd

import (
	"os"
	"os/exec"

	"github.com/spf13/cobra"
)

var (
	fmtFix   bool
	fmtDiff  bool
	fmtCheck bool
)

func addGoFmtCommand(parent *cobra.Command) {
	fmtCmd := &cobra.Command{
		Use:   "fmt",
		Short: "Format Go code",
		Long: "Format Go code using gofmt or goimports.\n\n" +
			"Examples:\n" +
			"  core go fmt              # Check formatting\n" +
			"  core go fmt --fix        # Fix formatting\n" +
			"  core go fmt --diff       # Show diff",
		RunE: func(cmd *cobra.Command, args []string) error {
			fmtArgs := []string{}
			if fmtFix {
				fmtArgs = append(fmtArgs, "-w")
			}
			if fmtDiff {
				fmtArgs = append(fmtArgs, "-d")
			}
			if !fmtFix && !fmtDiff {
				fmtArgs = append(fmtArgs, "-l")
			}
			fmtArgs = append(fmtArgs, ".")

			// Try goimports first, fall back to gofmt
			var execCmd *exec.Cmd
			if _, err := exec.LookPath("goimports"); err == nil {
				execCmd = exec.Command("goimports", fmtArgs...)
			} else {
				execCmd = exec.Command("gofmt", fmtArgs...)
			}

			execCmd.Stdout = os.Stdout
			execCmd.Stderr = os.Stderr
			return execCmd.Run()
		},
	}

	fmtCmd.Flags().BoolVar(&fmtFix, "fix", false, "Fix formatting in place")
	fmtCmd.Flags().BoolVar(&fmtDiff, "diff", false, "Show diff of changes")
	fmtCmd.Flags().BoolVar(&fmtCheck, "check", false, "Check only, exit 1 if not formatted")

	parent.AddCommand(fmtCmd)
}

var lintFix bool

func addGoLintCommand(parent *cobra.Command) {
	lintCmd := &cobra.Command{
		Use:   "lint",
		Short: "Run golangci-lint",
		Long: "Run golangci-lint on the codebase.\n\n" +
			"Examples:\n" +
			"  core go lint\n" +
			"  core go lint --fix",
		RunE: func(cmd *cobra.Command, args []string) error {
			lintArgs := []string{"run"}
			if lintFix {
				lintArgs = append(lintArgs, "--fix")
			}

			execCmd := exec.Command("golangci-lint", lintArgs...)
			execCmd.Stdout = os.Stdout
			execCmd.Stderr = os.Stderr
			return execCmd.Run()
		},
	}

	lintCmd.Flags().BoolVar(&lintFix, "fix", false, "Fix issues automatically")

	parent.AddCommand(lintCmd)
}