94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
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")
|
|
}
|