|
|
|
|
@@ -2,7 +2,6 @@ package backends
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"encoding/json"
|
|
|
|
|
"errors"
|
|
|
|
|
"fmt"
|
|
|
|
|
"path/filepath"
|
|
|
|
|
"reflect"
|
|
|
|
|
@@ -11,327 +10,41 @@ import (
|
|
|
|
|
"strings"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// CommandParserConfig holds configuration for parsing command line arguments
|
|
|
|
|
type CommandParserConfig struct {
|
|
|
|
|
// ExecutableNames are the names of executables to detect (e.g., "llama-server", "mlx_lm.server")
|
|
|
|
|
ExecutableNames []string
|
|
|
|
|
// SubcommandNames are optional subcommands (e.g., "serve" for vllm)
|
|
|
|
|
SubcommandNames []string
|
|
|
|
|
// MultiValuedFlags are flags that can accept multiple values
|
|
|
|
|
MultiValuedFlags map[string]struct{}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// ParseCommand parses a command string using the provided configuration
|
|
|
|
|
func ParseCommand(command string, config CommandParserConfig, target any) error {
|
|
|
|
|
// 1. Normalize the command - handle multiline with backslashes
|
|
|
|
|
trimmed := normalizeMultilineCommand(command)
|
|
|
|
|
if trimmed == "" {
|
|
|
|
|
// ParseCommand parses a command string into a target struct
|
|
|
|
|
func ParseCommand(command string, executableNames []string, subcommandNames []string, multiValuedFlags map[string]bool, target any) error {
|
|
|
|
|
// Normalize multiline commands
|
|
|
|
|
command = normalizeCommand(command)
|
|
|
|
|
if command == "" {
|
|
|
|
|
return fmt.Errorf("command cannot be empty")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 2. Extract arguments from command
|
|
|
|
|
args, err := extractArgumentsFromCommand(trimmed, config)
|
|
|
|
|
// Extract arguments
|
|
|
|
|
args, err := extractArgs(command, executableNames, subcommandNames)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 3. Parse arguments into map
|
|
|
|
|
options := make(map[string]any)
|
|
|
|
|
|
|
|
|
|
i := 0
|
|
|
|
|
for i < len(args) {
|
|
|
|
|
arg := args[i]
|
|
|
|
|
|
|
|
|
|
if !strings.HasPrefix(arg, "-") { // skip positional / stray values
|
|
|
|
|
i++
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Reject malformed flags with more than two leading dashes (e.g. ---model) to surface user mistakes
|
|
|
|
|
if strings.HasPrefix(arg, "---") {
|
|
|
|
|
return fmt.Errorf("malformed flag: %s", arg)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Unified parsing for --flag=value vs --flag value
|
|
|
|
|
var rawFlag, rawValue string
|
|
|
|
|
hasEquals := false
|
|
|
|
|
if strings.Contains(arg, "=") {
|
|
|
|
|
parts := strings.SplitN(arg, "=", 2)
|
|
|
|
|
rawFlag = parts[0]
|
|
|
|
|
rawValue = parts[1] // may be empty string
|
|
|
|
|
hasEquals = true
|
|
|
|
|
} else {
|
|
|
|
|
rawFlag = arg
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
flagCore := strings.TrimPrefix(strings.TrimPrefix(rawFlag, "-"), "-")
|
|
|
|
|
flagName := strings.ReplaceAll(flagCore, "-", "_")
|
|
|
|
|
|
|
|
|
|
// Detect value if not in equals form
|
|
|
|
|
valueProvided := hasEquals
|
|
|
|
|
if !hasEquals {
|
|
|
|
|
if i+1 < len(args) && !isFlag(args[i+1]) { // next token is value
|
|
|
|
|
rawValue = args[i+1]
|
|
|
|
|
valueProvided = true
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Determine if multi-valued flag
|
|
|
|
|
_, isMulti := config.MultiValuedFlags[flagName]
|
|
|
|
|
|
|
|
|
|
// Normalization helper: ensure slice for multi-valued flags
|
|
|
|
|
appendValue := func(valStr string) {
|
|
|
|
|
if existing, ok := options[flagName]; ok {
|
|
|
|
|
// Existing value; ensure slice semantics for multi-valued flags or repeated occurrences
|
|
|
|
|
if slice, ok := existing.([]string); ok {
|
|
|
|
|
options[flagName] = append(slice, valStr)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
// Convert scalar to slice
|
|
|
|
|
options[flagName] = []string{fmt.Sprintf("%v", existing), valStr}
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
// First value
|
|
|
|
|
if isMulti {
|
|
|
|
|
options[flagName] = []string{valStr}
|
|
|
|
|
} else {
|
|
|
|
|
// We'll parse type below for single-valued flags
|
|
|
|
|
options[flagName] = valStr
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if valueProvided {
|
|
|
|
|
// Use raw token for multi-valued flags; else allow typed parsing
|
|
|
|
|
appendValue(rawValue)
|
|
|
|
|
if !isMulti { // convert to typed value if scalar
|
|
|
|
|
if strVal, ok := options[flagName].(string); ok { // still scalar
|
|
|
|
|
options[flagName] = parseValue(strVal)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
// Advance index: if we consumed a following token as value (non equals form), skip it
|
|
|
|
|
if !hasEquals && i+1 < len(args) && rawValue == args[i+1] {
|
|
|
|
|
i += 2
|
|
|
|
|
} else {
|
|
|
|
|
i++
|
|
|
|
|
}
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Boolean flag (no value)
|
|
|
|
|
options[flagName] = true
|
|
|
|
|
i++
|
|
|
|
|
// Parse flags into map
|
|
|
|
|
options, err := parseFlags(args, multiValuedFlags)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 4. Convert to target struct using JSON marshaling
|
|
|
|
|
// Convert to target struct via JSON
|
|
|
|
|
jsonData, err := json.Marshal(options)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return fmt.Errorf("failed to marshal parsed options: %w", err)
|
|
|
|
|
return fmt.Errorf("failed to marshal options: %w", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := json.Unmarshal(jsonData, target); err != nil {
|
|
|
|
|
return fmt.Errorf("failed to parse command options: %w", err)
|
|
|
|
|
return fmt.Errorf("failed to unmarshal to target: %w", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// parseValue attempts to parse a string value into the most appropriate type
|
|
|
|
|
func parseValue(value string) any {
|
|
|
|
|
// Surrounding matching quotes (single or double)
|
|
|
|
|
if l := len(value); l >= 2 {
|
|
|
|
|
if (value[0] == '"' && value[l-1] == '"') || (value[0] == '\'' && value[l-1] == '\'') {
|
|
|
|
|
value = value[1 : l-1]
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
lower := strings.ToLower(value)
|
|
|
|
|
if lower == "true" {
|
|
|
|
|
return true
|
|
|
|
|
}
|
|
|
|
|
if lower == "false" {
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if intVal, err := strconv.Atoi(value); err == nil {
|
|
|
|
|
return intVal
|
|
|
|
|
}
|
|
|
|
|
if floatVal, err := strconv.ParseFloat(value, 64); err == nil {
|
|
|
|
|
return floatVal
|
|
|
|
|
}
|
|
|
|
|
return value
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// normalizeMultilineCommand handles multiline commands with backslashes
|
|
|
|
|
func normalizeMultilineCommand(command string) string {
|
|
|
|
|
// Handle escaped newlines (backslash followed by newline)
|
|
|
|
|
re := regexp.MustCompile(`\\\s*\n\s*`)
|
|
|
|
|
normalized := re.ReplaceAllString(command, " ")
|
|
|
|
|
|
|
|
|
|
// Clean up extra whitespace
|
|
|
|
|
re = regexp.MustCompile(`\s+`)
|
|
|
|
|
normalized = re.ReplaceAllString(normalized, " ")
|
|
|
|
|
|
|
|
|
|
return strings.TrimSpace(normalized)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// extractArgumentsFromCommand extracts arguments from various command formats
|
|
|
|
|
func extractArgumentsFromCommand(command string, config CommandParserConfig) ([]string, error) {
|
|
|
|
|
// Split command into tokens respecting quotes
|
|
|
|
|
tokens, err := splitCommandTokens(command)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if len(tokens) == 0 {
|
|
|
|
|
return nil, fmt.Errorf("no command tokens found")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
firstToken := tokens[0]
|
|
|
|
|
|
|
|
|
|
// Check for full path executable
|
|
|
|
|
if strings.Contains(firstToken, string(filepath.Separator)) {
|
|
|
|
|
baseName := filepath.Base(firstToken)
|
|
|
|
|
for _, execName := range config.ExecutableNames {
|
|
|
|
|
if strings.HasSuffix(baseName, execName) {
|
|
|
|
|
return skipExecutableAndSubcommands(tokens[1:], config.SubcommandNames)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
// Unknown executable, assume it's still an executable
|
|
|
|
|
return skipExecutableAndSubcommands(tokens[1:], config.SubcommandNames)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check for simple executable names
|
|
|
|
|
lowerFirstToken := strings.ToLower(firstToken)
|
|
|
|
|
for _, execName := range config.ExecutableNames {
|
|
|
|
|
if lowerFirstToken == strings.ToLower(execName) {
|
|
|
|
|
return skipExecutableAndSubcommands(tokens[1:], config.SubcommandNames)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check for subcommands (like "serve" for vllm)
|
|
|
|
|
for _, subCmd := range config.SubcommandNames {
|
|
|
|
|
if lowerFirstToken == strings.ToLower(subCmd) {
|
|
|
|
|
return tokens[1:], nil // Return everything except the subcommand
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Arguments only (starts with a flag)
|
|
|
|
|
if strings.HasPrefix(firstToken, "-") {
|
|
|
|
|
return tokens, nil // Return all tokens as arguments
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Unknown format - might be a different executable name
|
|
|
|
|
return skipExecutableAndSubcommands(tokens[1:], config.SubcommandNames)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// skipExecutableAndSubcommands removes subcommands from the beginning of tokens
|
|
|
|
|
func skipExecutableAndSubcommands(tokens []string, subcommands []string) ([]string, error) {
|
|
|
|
|
if len(tokens) == 0 {
|
|
|
|
|
return tokens, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check if first token is a subcommand
|
|
|
|
|
if len(subcommands) > 0 && len(tokens) > 0 {
|
|
|
|
|
lowerFirstToken := strings.ToLower(tokens[0])
|
|
|
|
|
for _, subCmd := range subcommands {
|
|
|
|
|
if lowerFirstToken == strings.ToLower(subCmd) {
|
|
|
|
|
return tokens[1:], nil // Skip the subcommand
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return tokens, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// splitCommandTokens splits a command string into tokens, respecting quotes
|
|
|
|
|
func splitCommandTokens(command string) ([]string, error) {
|
|
|
|
|
var tokens []string
|
|
|
|
|
var current strings.Builder
|
|
|
|
|
inQuotes := false
|
|
|
|
|
quoteChar := byte(0)
|
|
|
|
|
escaped := false
|
|
|
|
|
|
|
|
|
|
for i := 0; i < len(command); i++ {
|
|
|
|
|
c := command[i]
|
|
|
|
|
|
|
|
|
|
if escaped {
|
|
|
|
|
current.WriteByte(c)
|
|
|
|
|
escaped = false
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if c == '\\' {
|
|
|
|
|
escaped = true
|
|
|
|
|
current.WriteByte(c)
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if !inQuotes && (c == '"' || c == '\'') {
|
|
|
|
|
inQuotes = true
|
|
|
|
|
quoteChar = c
|
|
|
|
|
current.WriteByte(c)
|
|
|
|
|
} else if inQuotes && c == quoteChar {
|
|
|
|
|
inQuotes = false
|
|
|
|
|
quoteChar = 0
|
|
|
|
|
current.WriteByte(c)
|
|
|
|
|
} else if !inQuotes && (c == ' ' || c == '\t' || c == '\n') {
|
|
|
|
|
if current.Len() > 0 {
|
|
|
|
|
tokens = append(tokens, current.String())
|
|
|
|
|
current.Reset()
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
current.WriteByte(c)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if inQuotes {
|
|
|
|
|
return nil, errors.New("unterminated quoted string")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if current.Len() > 0 {
|
|
|
|
|
tokens = append(tokens, current.String())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return tokens, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// isFlag determines if a string is a command line flag or a value
|
|
|
|
|
// Handles the special case where negative numbers should be treated as values, not flags
|
|
|
|
|
func isFlag(arg string) bool {
|
|
|
|
|
if !strings.HasPrefix(arg, "-") {
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Special case: if it's a negative number, treat it as a value
|
|
|
|
|
if _, err := strconv.ParseFloat(arg, 64); err == nil {
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return true
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// SliceHandling defines how []string fields should be handled when building command args
|
|
|
|
|
type SliceHandling int
|
|
|
|
|
|
|
|
|
|
const (
|
|
|
|
|
// SliceAsMultipleFlags creates multiple flags: --flag value1 --flag value2
|
|
|
|
|
SliceAsMultipleFlags SliceHandling = iota
|
|
|
|
|
// SliceAsCommaSeparated creates single flag with comma-separated values: --flag value1,value2
|
|
|
|
|
SliceAsCommaSeparated
|
|
|
|
|
// SliceAsMixed uses different strategies for different flags (requires configuration)
|
|
|
|
|
SliceAsMixed
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// ArgsBuilderConfig holds configuration for building command line arguments
|
|
|
|
|
type ArgsBuilderConfig struct {
|
|
|
|
|
// SliceHandling defines the default strategy for []string fields
|
|
|
|
|
SliceHandling SliceHandling
|
|
|
|
|
// MultipleFlags specifies which flags should use multiple instances when SliceHandling is SliceAsMixed
|
|
|
|
|
MultipleFlags map[string]struct{}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// BuildCommandArgs converts a struct to command line arguments using reflection
|
|
|
|
|
func BuildCommandArgs(options any, config ArgsBuilderConfig) []string {
|
|
|
|
|
// BuildCommandArgs converts a struct to command line arguments
|
|
|
|
|
func BuildCommandArgs(options any, multipleFlags map[string]bool) []string {
|
|
|
|
|
var args []string
|
|
|
|
|
|
|
|
|
|
v := reflect.ValueOf(options).Elem()
|
|
|
|
|
@@ -341,27 +54,19 @@ func BuildCommandArgs(options any, config ArgsBuilderConfig) []string {
|
|
|
|
|
field := v.Field(i)
|
|
|
|
|
fieldType := t.Field(i)
|
|
|
|
|
|
|
|
|
|
// Skip unexported fields
|
|
|
|
|
if !field.CanInterface() {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Get the JSON tag to determine the flag name
|
|
|
|
|
jsonTag := fieldType.Tag.Get("json")
|
|
|
|
|
if jsonTag == "" || jsonTag == "-" {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Remove ",omitempty" from the tag
|
|
|
|
|
flagName := jsonTag
|
|
|
|
|
if commaIndex := strings.Index(jsonTag, ","); commaIndex != -1 {
|
|
|
|
|
flagName = jsonTag[:commaIndex]
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Convert snake_case to kebab-case for CLI flags
|
|
|
|
|
// Get flag name from JSON tag
|
|
|
|
|
flagName := strings.Split(jsonTag, ",")[0]
|
|
|
|
|
flagName = strings.ReplaceAll(flagName, "_", "-")
|
|
|
|
|
|
|
|
|
|
// Add the appropriate arguments based on field type and value
|
|
|
|
|
switch field.Kind() {
|
|
|
|
|
case reflect.Bool:
|
|
|
|
|
if field.Bool() {
|
|
|
|
|
@@ -380,8 +85,20 @@ func BuildCommandArgs(options any, config ArgsBuilderConfig) []string {
|
|
|
|
|
args = append(args, "--"+flagName, field.String())
|
|
|
|
|
}
|
|
|
|
|
case reflect.Slice:
|
|
|
|
|
if field.Type().Elem().Kind() == reflect.String {
|
|
|
|
|
args = append(args, handleStringSlice(field, flagName, config)...)
|
|
|
|
|
if field.Type().Elem().Kind() == reflect.String && field.Len() > 0 {
|
|
|
|
|
if multipleFlags[flagName] {
|
|
|
|
|
// Multiple flags: --flag value1 --flag value2
|
|
|
|
|
for j := 0; j < field.Len(); j++ {
|
|
|
|
|
args = append(args, "--"+flagName, field.Index(j).String())
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
// Comma-separated: --flag value1,value2
|
|
|
|
|
var values []string
|
|
|
|
|
for j := 0; j < field.Len(); j++ {
|
|
|
|
|
values = append(values, field.Index(j).String())
|
|
|
|
|
}
|
|
|
|
|
args = append(args, "--"+flagName, strings.Join(values, ","))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
@@ -389,43 +106,155 @@ func BuildCommandArgs(options any, config ArgsBuilderConfig) []string {
|
|
|
|
|
return args
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// handleStringSlice handles []string fields based on the configuration
|
|
|
|
|
func handleStringSlice(field reflect.Value, flagName string, config ArgsBuilderConfig) []string {
|
|
|
|
|
if field.Len() == 0 {
|
|
|
|
|
return nil
|
|
|
|
|
// normalizeCommand handles multiline commands with backslashes
|
|
|
|
|
func normalizeCommand(command string) string {
|
|
|
|
|
re := regexp.MustCompile(`\\\s*\n\s*`)
|
|
|
|
|
normalized := re.ReplaceAllString(command, " ")
|
|
|
|
|
re = regexp.MustCompile(`\s+`)
|
|
|
|
|
return strings.TrimSpace(re.ReplaceAllString(normalized, " "))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// extractArgs extracts arguments from command, removing executable and subcommands
|
|
|
|
|
func extractArgs(command string, executableNames []string, subcommandNames []string) ([]string, error) {
|
|
|
|
|
// Check for unterminated quotes
|
|
|
|
|
if strings.Count(command, `"`)%2 != 0 || strings.Count(command, `'`)%2 != 0 {
|
|
|
|
|
return nil, fmt.Errorf("unterminated quoted string")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var args []string
|
|
|
|
|
tokens := strings.Fields(command)
|
|
|
|
|
if len(tokens) == 0 {
|
|
|
|
|
return nil, fmt.Errorf("no tokens found")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
switch config.SliceHandling {
|
|
|
|
|
case SliceAsMultipleFlags:
|
|
|
|
|
// Multiple flags: --flag value1 --flag value2
|
|
|
|
|
for j := 0; j < field.Len(); j++ {
|
|
|
|
|
args = append(args, "--"+flagName, field.Index(j).String())
|
|
|
|
|
// Skip executable
|
|
|
|
|
start := 0
|
|
|
|
|
firstToken := tokens[0]
|
|
|
|
|
|
|
|
|
|
// Check for executable name (with or without path)
|
|
|
|
|
if strings.Contains(firstToken, string(filepath.Separator)) {
|
|
|
|
|
baseName := filepath.Base(firstToken)
|
|
|
|
|
for _, execName := range executableNames {
|
|
|
|
|
if strings.HasSuffix(strings.ToLower(baseName), strings.ToLower(execName)) {
|
|
|
|
|
start = 1
|
|
|
|
|
break
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
case SliceAsCommaSeparated:
|
|
|
|
|
// Comma-separated: --flag value1,value2
|
|
|
|
|
var values []string
|
|
|
|
|
for j := 0; j < field.Len(); j++ {
|
|
|
|
|
values = append(values, field.Index(j).String())
|
|
|
|
|
} else {
|
|
|
|
|
for _, execName := range executableNames {
|
|
|
|
|
if strings.EqualFold(firstToken, execName) {
|
|
|
|
|
start = 1
|
|
|
|
|
break
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
args = append(args, "--"+flagName, strings.Join(values, ","))
|
|
|
|
|
case SliceAsMixed:
|
|
|
|
|
// Check if this specific flag should use multiple instances
|
|
|
|
|
if _, useMultiple := config.MultipleFlags[flagName]; useMultiple {
|
|
|
|
|
// Multiple flags
|
|
|
|
|
for j := 0; j < field.Len(); j++ {
|
|
|
|
|
args = append(args, "--"+flagName, field.Index(j).String())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Skip subcommand if present
|
|
|
|
|
if start < len(tokens) {
|
|
|
|
|
for _, subCmd := range subcommandNames {
|
|
|
|
|
if strings.EqualFold(tokens[start], subCmd) {
|
|
|
|
|
start++
|
|
|
|
|
break
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Handle case where command starts with subcommand (no executable)
|
|
|
|
|
if start == 0 {
|
|
|
|
|
for _, subCmd := range subcommandNames {
|
|
|
|
|
if strings.EqualFold(firstToken, subCmd) {
|
|
|
|
|
start = 1
|
|
|
|
|
break
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return tokens[start:], nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// parseFlags parses command line flags into a map
|
|
|
|
|
func parseFlags(args []string, multiValuedFlags map[string]bool) (map[string]any, error) {
|
|
|
|
|
options := make(map[string]any)
|
|
|
|
|
|
|
|
|
|
for i := 0; i < len(args); i++ {
|
|
|
|
|
arg := args[i]
|
|
|
|
|
|
|
|
|
|
if !strings.HasPrefix(arg, "-") {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check for malformed flags (more than two leading dashes)
|
|
|
|
|
if strings.HasPrefix(arg, "---") {
|
|
|
|
|
return nil, fmt.Errorf("malformed flag: %s", arg)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Get flag name and value
|
|
|
|
|
var flagName, value string
|
|
|
|
|
var hasValue bool
|
|
|
|
|
|
|
|
|
|
if strings.Contains(arg, "=") {
|
|
|
|
|
parts := strings.SplitN(arg, "=", 2)
|
|
|
|
|
flagName = strings.TrimLeft(parts[0], "-")
|
|
|
|
|
value = parts[1]
|
|
|
|
|
hasValue = true
|
|
|
|
|
} else {
|
|
|
|
|
flagName = strings.TrimLeft(arg, "-")
|
|
|
|
|
if i+1 < len(args) && !strings.HasPrefix(args[i+1], "-") {
|
|
|
|
|
value = args[i+1]
|
|
|
|
|
hasValue = true
|
|
|
|
|
i++ // Skip next arg since we consumed it
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Convert kebab-case to snake_case for JSON
|
|
|
|
|
flagName = strings.ReplaceAll(flagName, "-", "_")
|
|
|
|
|
|
|
|
|
|
if hasValue {
|
|
|
|
|
// Handle multi-valued flags
|
|
|
|
|
if multiValuedFlags[flagName] {
|
|
|
|
|
if existing, ok := options[flagName].([]string); ok {
|
|
|
|
|
options[flagName] = append(existing, value)
|
|
|
|
|
} else {
|
|
|
|
|
options[flagName] = []string{value}
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
options[flagName] = parseValue(value)
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
// Comma-separated
|
|
|
|
|
var values []string
|
|
|
|
|
for j := 0; j < field.Len(); j++ {
|
|
|
|
|
values = append(values, field.Index(j).String())
|
|
|
|
|
}
|
|
|
|
|
args = append(args, "--"+flagName, strings.Join(values, ","))
|
|
|
|
|
// Boolean flag
|
|
|
|
|
options[flagName] = true
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return args
|
|
|
|
|
return options, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// parseValue converts string to appropriate type
|
|
|
|
|
func parseValue(value string) any {
|
|
|
|
|
// Remove quotes
|
|
|
|
|
if len(value) >= 2 {
|
|
|
|
|
if (value[0] == '"' && value[len(value)-1] == '"') || (value[0] == '\'' && value[len(value)-1] == '\'') {
|
|
|
|
|
value = value[1 : len(value)-1]
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Try boolean
|
|
|
|
|
switch strings.ToLower(value) {
|
|
|
|
|
case "true":
|
|
|
|
|
return true
|
|
|
|
|
case "false":
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Try integer
|
|
|
|
|
if intVal, err := strconv.Atoi(value); err == nil {
|
|
|
|
|
return intVal
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Try float
|
|
|
|
|
if floatVal, err := strconv.ParseFloat(value, 64); err == nil {
|
|
|
|
|
return floatVal
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Return as string
|
|
|
|
|
return value
|
|
|
|
|
}
|
|
|
|
|
|