kit/pkg/kit/template_bridge.go

package kit

import (
	"regexp"
	"strings"
	"sync"

	"github.com/mark3labs/kit/internal/extensions"
	"github.com/mark3labs/kit/internal/models"
	"github.com/mark3labs/kit/internal/prompts"
	"github.com/mark3labs/kit/internal/skills"
)

// ---------------------------------------------------------------------------
// Template Parsing Bridge for Extensions (Phase 3)
// ---------------------------------------------------------------------------

// ParseTemplate extracts {{variables}} from template content. The template
// grammar is shared with skill prompt templates, so a template parses
// identically regardless of which API loads it.
func ParseTemplate(name, content string) extensions.PromptTemplate {
	tpl := skills.NewPromptTemplate(name, content)
	vars := tpl.Variables
	if vars == nil {
		vars = []string{}
	}
	return extensions.PromptTemplate{
		Name:      tpl.Name,
		Content:   tpl.Content,
		Variables: vars,
	}
}

// RenderTemplate substitutes variables into template content.
// Handles {{name}} and {{ name }} (any whitespace) placeholders; missing
// variables are left as-is.
func RenderTemplate(tpl extensions.PromptTemplate, vars map[string]string) string {
	t := skills.PromptTemplate{Content: tpl.Content}
	return t.Expand(vars)
}

// ParseArguments parses command-line style arguments.
func ParseArguments(input string, pattern extensions.ArgumentPattern) extensions.ParseResult {
	result := extensions.ParseResult{
		Vars:  make(map[string]string),
		Flags: make(map[string]string),
	}

	fields := parseFields(input)
	if len(fields) == 0 {
		return result
	}

	// First field is the command itself (if present); skip it.
	startIdx := 0
	if len(fields) > 0 && !strings.HasPrefix(fields[0], "-") {
		startIdx = 1
	}

	// Parse flags
	i := startIdx
	for i < len(fields) {
		field := fields[i]

		// Check for flags
		if strings.HasPrefix(field, "--") {
			flagName := field[2:]
			if varName, ok := pattern.Flags["--"+flagName]; ok {
				// Flag with value
				if i+1 < len(fields) && !strings.HasPrefix(fields[i+1], "-") {
					result.Flags["--"+flagName] = fields[i+1]
					result.Vars[varName] = fields[i+1]
					i += 2
					continue
				}
				// Boolean flag
				result.Flags["--"+flagName] = "true"
				result.Vars[varName] = "true"
			}
			i++
			continue
		}

		if strings.HasPrefix(field, "-") && len(field) > 1 {
			flagName := field[1:]
			if varName, ok := pattern.Flags["-"+flagName]; ok {
				// Flag with value
				if i+1 < len(fields) && !strings.HasPrefix(fields[i+1], "-") {
					result.Flags["-"+flagName] = fields[i+1]
					result.Vars[varName] = fields[i+1]
					i += 2
					continue
				}
				// Boolean flag
				result.Flags["-"+flagName] = "true"
				result.Vars[varName] = "true"
			}
			i++
			continue
		}

		i++
	}

	// Collect remaining as positional args and "rest"
	positional := make([]string, 0)
	i = startIdx
	for i < len(fields) {
		field := fields[i]
		if !strings.HasPrefix(field, "-") {
			// Check if this was consumed as a flag value
			consumed := false
			for _, v := range result.Vars {
				if v == field {
					// Might be consumed, check previous field
					if i > 0 {
						prev := fields[i-1]
						if strings.HasPrefix(prev, "-") {
							consumed = true
							break
						}
					}
				}
			}
			if !consumed {
				positional = append(positional, field)
			}
		}
		i++
	}

	// Map positional args
	for i, name := range pattern.Positional {
		if i < len(positional) {
			result.Vars[name] = positional[i]
		}
	}

	// Set rest
	if pattern.Rest != "" && len(positional) > len(pattern.Positional) {
		restStart := len(pattern.Positional)
		if restStart < len(positional) {
			result.Vars[pattern.Rest] = strings.Join(positional[restStart:], " ")
		}
	}

	result.Rest = strings.Join(fields, " ")
	return result
}

// SimpleParseArguments parses $1, $2, $@ style arguments.
// Returns slice where [0]=full input, [1]=$1, [2]=$2, ... [n]=$@
func SimpleParseArguments(input string, count int) []string {
	fields := parseFields(input)
	result := make([]string, 0, count+2)
	result = append(result, input) // [0] = full input

	// [1]..[count] = positional args
	for i := range count {
		if i < len(fields) {
			result = append(result, fields[i])
		} else {
			result = append(result, "")
		}
	}

	// [n] = $@ (all remaining)
	if len(fields) > count {
		result = append(result, strings.Join(fields[count:], " "))
	} else {
		result = append(result, "")
	}

	return result
}

// parseFields splits input into arguments respecting quoted strings and
// backslash escaping. It delegates to the canonical tokenizer in
// internal/prompts so extension argument parsing and builtin prompt-template
// parsing agree on grammar.
func parseFields(input string) []string {
	return prompts.ParseCommandArgs(input)
}

// EvaluateModelConditional checks if condition matches current model.
// Condition supports wildcards: * matches any, ? matches single char.
func EvaluateModelConditional(currentModel, condition string) bool {
	// Handle comma-separated conditions (OR logic)
	for c := range strings.SplitSeq(condition, ",") {
		c = strings.TrimSpace(c)
		if matchModelPattern(currentModel, c) {
			return true
		}
	}
	return false
}

// modelPatternCache caches compiled regexps for model glob patterns.
var modelPatternCache sync.Map

// matchModelPattern matches a model against a pattern with wildcards.
// Compiled regexps are cached to avoid recompilation on hot paths.
func matchModelPattern(model, pattern string) bool {
	rePattern := "^" + strings.ReplaceAll(strings.ReplaceAll(pattern, "*", ".*"), "?", ".") + "$"
	var re *regexp.Regexp
	if v, ok := modelPatternCache.Load(rePattern); ok {
		re = v.(*regexp.Regexp)
	} else {
		compiled, err := regexp.Compile(rePattern)
		if err != nil {
			// Fallback: exact match
			return model == pattern
		}
		modelPatternCache.Store(rePattern, compiled)
		re = compiled
	}
	return re.MatchString(model)
}

// RenderWithModelConditionals processes <if-model> blocks in content.
func RenderWithModelConditionals(content, currentModel string) string {
	// Simple regex-based processor for <if-model> blocks
	// Supports: <if-model is="pattern">content</if-model>
	// And: <if-model is="pattern">content<else>other</if-model>

	result := content

	// Pattern for if-model blocks
	ifModelRegex := regexp.MustCompile(`(?s)<if-model\s+is="([^"]+)">(.*?)(?:<else>(.*?))?</if-model>`)

	for {
		match := ifModelRegex.FindStringSubmatchIndex(result)
		if match == nil {
			break
		}

		condition := result[match[2]:match[3]]
		ifContent := result[match[4]:match[5]]
		elseContent := ""
		if match[6] >= 0 && match[7] >= 0 {
			elseContent = result[match[6]:match[7]]
		}

		var replacement string
		if EvaluateModelConditional(currentModel, condition) {
			replacement = ifContent
		} else {
			replacement = elseContent
		}

		result = result[:match[0]] + replacement + result[match[1]:]
	}

	return result
}

// ---------------------------------------------------------------------------
// Model Resolution Bridge for Extensions (Phase 4)
// ---------------------------------------------------------------------------

// ResolveModelChain attempts each model in order until one is available.
func ResolveModelChain(preferences []string) extensions.ModelResolutionResult {
	result := extensions.ModelResolutionResult{
		Attempted: make([]string, 0, len(preferences)),
	}

	registry := models.GetGlobalRegistry()

	for _, pref := range preferences {
		pref = strings.TrimSpace(pref)
		result.Attempted = append(result.Attempted, pref)

		// Parse model string
		provider, modelID, err := models.ParseModelString(pref)
		if err != nil {
			continue
		}

		// Check if provider exists
		if registry.GetProviderInfo(provider) == nil {
			continue
		}

		// Check if model exists in registry
		modelInfo := registry.LookupModel(provider, modelID)
		if modelInfo == nil {
			// Try with just the model as bare name
			continue
		}

		// Found available model
		result.Model = provider + "/" + modelID
		result.Capabilities = extensions.ModelCapabilities{
			Provider:     provider,
			ModelID:      modelID,
			ContextLimit: modelInfo.Limit.Context,
			OutputLimit:  modelInfo.Limit.Output,
			Reasoning:    modelInfo.Reasoning,
			Streaming:    true, // Assume streaming support
		}
		return result
	}

	result.Error = "no models in chain are available"
	return result
}

// GetModelCapabilities returns capabilities for a specific model.
// If model is empty, returns zero capabilities.
func GetModelCapabilities(model string) (extensions.ModelCapabilities, string) {
	if model == "" {
		return extensions.ModelCapabilities{}, "no model specified"
	}

	provider, modelID, err := models.ParseModelString(model)
	if err != nil {
		return extensions.ModelCapabilities{}, err.Error()
	}

	registry := models.GetGlobalRegistry()
	modelInfo := registry.LookupModel(provider, modelID)
	if modelInfo == nil {
		return extensions.ModelCapabilities{}, "model not found in registry"
	}

	return extensions.ModelCapabilities{
		Provider:     provider,
		ModelID:      modelID,
		ContextLimit: modelInfo.Limit.Context,
		OutputLimit:  modelInfo.Limit.Output,
		Reasoning:    modelInfo.Reasoning,
		Streaming:    true,
	}, ""
}

// CheckModelAvailable verifies if a model string is valid and provider exists.
func CheckModelAvailable(model string) bool {
	provider, _, err := models.ParseModelString(model)
	if err != nil {
		return false
	}

	registry := models.GetGlobalRegistry()
	if registry.GetProviderInfo(provider) == nil {
		return false
	}

	// Model doesn't need to be in registry - could be dynamic/Ollama
	return true
}

// GetCurrentProvider extracts provider from model string.
func GetCurrentProvider(model string) string {
	provider, _, _ := models.ParseModelString(model)
	return provider
}

// GetCurrentModelID extracts model ID from model string.
func GetCurrentModelID(model string) string {
	_, modelID, _ := models.ParseModelString(model)
	return modelID
}

// JoinModel combines provider and model ID into a model string.
func JoinModel(provider, modelID string) string {
	if provider == "" {
		return modelID
	}
	return provider + "/" + modelID
}

// MatchModelGlob matches a model against a glob pattern.
// Pattern can contain * (match any) and ? (match single).
func MatchModelGlob(model, pattern string) bool {
	return matchModelPattern(model, pattern)
}

// ExtractProviderFromPath extracts provider from a path-like model string.
//
// Deprecated: Use GetCurrentProvider instead.
func ExtractProviderFromPath(model string) string {
	return GetCurrentProvider(model)
}

// ExtractModelFromPath extracts model ID from a path-like model string.
//
// Deprecated: Use RemoveProviderFromModel instead, which correctly handles
// model IDs containing "/" (e.g. "openrouter/meta/llama").
func ExtractModelFromPath(model string) string {
	return RemoveProviderFromModel(model)
}

// IsBareModelID checks if a string is a bare model ID (no provider).
func IsBareModelID(model string) bool {
	return !strings.Contains(model, "/")
}

// AddProviderToModel adds a provider prefix to a bare model ID.
func AddProviderToModel(provider, model string) string {
	if strings.Contains(model, "/") {
		return model // Already has provider
	}
	return provider + "/" + model
}

// RemoveProviderFromModel removes the provider prefix from a model string.
func RemoveProviderFromModel(model string) string {
	parts := strings.SplitN(model, "/", 2)
	if len(parts) == 2 {
		return parts[1]
	}
	return model
}