package commands

import (
	"bytes"
	"context"

	"github.com/LazyBachelor/LazyPM/internal/service"
	"github.com/charmbracelet/fang"

	"github.com/spf13/cobra"
)

type contextKey string

const appKey contextKey = "app"

// app is a package-level variable used during command setup
var app *service.App

// Flags struct to hold command-line flag values for issues.
type Flags struct {
	interactive bool
	limit       int

	title       string
	description string
	status      string
	issueType   string
	priority    int
}

// rootCmd is the base command for the CLI application.
var rootCmd = &cobra.Command{
	Short: "Project Management CLI",
	Long:  `Project Management CLI for managing issues and tasks.`,
	PersistentPreRun: func(cmd *cobra.Command, args []string) {
		// Inject app into context for all commands
		if app != nil {
			cmd.SetContext(context.WithValue(cmd.Context(), appKey, app))
		}
	},
}

// SetApp sets the app variable for use in command execution.
// Must be called before executing any commands to ensure services are available.
func SetApp(application *service.App) {
	app = application
	rootCmd.Use = app.Config.RootCmd
}

// AppFromContext retrieves the App from the command context
func AppFromContext(ctx context.Context) *service.App {
	if a, ok := ctx.Value(appKey).(*service.App); ok {
		return a
	}
	// Fallback to package-level app (for testing or edge cases)
	return app
}

// Execute executes the root command using the fang library.
func Execute() error {
	return fang.Execute(context.Background(), rootCmd,
		fang.WithColorSchemeFunc(fang.AnsiColorScheme))
}

// ExecuteArgs executes the command with the given arguments using the fang library.
func ExecuteArgs(args []string) error {
	rootCmd.SetArgs(args)
	return fang.Execute(context.Background(), rootCmd,
		fang.WithColorSchemeFunc(fang.AnsiColorScheme))
}

// ExecuteArgsString executes the command with the given arguments and returns the output as a string.
// This is useful for testing command outputs and used in the REPL
func ExecuteArgsString(args []string) (string, error) {
	buf := new(bytes.Buffer)

	rootCmd.SetOut(buf)
	rootCmd.SetErr(buf)
	rootCmd.SetArgs(args)

	err := rootCmd.Execute()

	return buf.String(), err
}

// init function to set up the command hierarchy and options.
func init() {
	rootCmd.CompletionOptions.DisableDefaultCmd = false
	rootCmd.AddGroup(&cobra.Group{ID: "help", Title: "Helping Commands"})
	rootCmd.SetCompletionCommandGroupID("help")
	rootCmd.SetHelpCommandGroupID("help")
}