Fabric/internal/plugins/template/extension_manager.go

package template

import (
	"fmt"
	"os"
	"path/filepath"
	"time"

	"gopkg.in/yaml.v3"
)

// ExtensionManager handles the high-level operations of the extension system
type ExtensionManager struct {
	registry  *ExtensionRegistry
	executor  *ExtensionExecutor
	configDir string
}

// NewExtensionManager creates a new extension manager instance
func NewExtensionManager(configDir string) *ExtensionManager {
	registry := NewExtensionRegistry(configDir)
	return &ExtensionManager{
		registry:  registry,
		executor:  NewExtensionExecutor(registry),
		configDir: configDir,
	}
}

// ListExtensions handles the listextensions flag action
func (em *ExtensionManager) ListExtensions() error {
	if em.registry == nil || em.registry.registry.Extensions == nil {
		return fmt.Errorf("extension registry not initialized")
	}

	for name, entry := range em.registry.registry.Extensions {
		fmt.Printf("Extension: %s\n", name)

		// Try to load extension details
		ext, err := em.registry.GetExtension(name)
		if err != nil {
			fmt.Printf("  Status: DISABLED - Hash verification failed: %v\n", err)
			fmt.Printf("  Config Path: %s\n\n", entry.ConfigPath)
			continue
		}

		// Print extension details if verification succeeded
		fmt.Printf("  Status: ENABLED\n")
		fmt.Printf("  Executable: %s\n", ext.Executable)
		fmt.Printf("  Type: %s\n", ext.Type)
		fmt.Printf("  Timeout: %s\n", ext.Timeout)
		fmt.Printf("  Description: %s\n", ext.Description)
		fmt.Printf("  Version: %s\n", ext.Version)

		fmt.Printf("  Operations:\n")
		for opName, opConfig := range ext.Operations {
			fmt.Printf("    %s:\n", opName)
			fmt.Printf("      Command Template: %s\n", opConfig.CmdTemplate)
		}

		if fileConfig := ext.GetFileConfig(); fileConfig != nil {
			fmt.Printf("  File Configuration:\n")
			for k, v := range fileConfig {
				fmt.Printf("    %s: %v\n", k, v)
			}
		}
		fmt.Printf("\n")
	}

	return nil
}

// RegisterExtension handles the addextension flag action
func (em *ExtensionManager) RegisterExtension(configPath string) error {
	absPath, err := filepath.Abs(configPath)
	if err != nil {
		return fmt.Errorf("invalid config path: %w", err)
	}

	// Get extension name before registration for status message
	data, err := os.ReadFile(absPath)
	if err != nil {
		return fmt.Errorf("failed to read config file: %w", err)
	}

	var ext ExtensionDefinition
	if err := yaml.Unmarshal(data, &ext); err != nil {
		return fmt.Errorf("failed to parse config file: %w", err)
	}

	if err := em.registry.Register(absPath); err != nil {
		return fmt.Errorf("failed to register extension: %w", err)
	}

	if _, err := time.ParseDuration(ext.Timeout); err != nil {
		return fmt.Errorf("invalid timeout value '%s': must be a duration like '30s' or '1m': %w", ext.Timeout, err)
	}

	// Print success message with extension details
	fmt.Printf("Successfully registered extension:\n")
	fmt.Printf("Name: %s\n", ext.Name)
	fmt.Printf("  Executable: %s\n", ext.Executable)
	fmt.Printf("  Type: %s\n", ext.Type)
	fmt.Printf("  Timeout: %s\n", ext.Timeout)
	fmt.Printf("  Description: %s\n", ext.Description)
	fmt.Printf("  Version: %s\n", ext.Version)

	fmt.Printf("  Operations:\n")
	for opName, opConfig := range ext.Operations {
		fmt.Printf("    %s:\n", opName)
		fmt.Printf("      Command Template: %s\n", opConfig.CmdTemplate)
	}

	if fileConfig := ext.GetFileConfig(); fileConfig != nil {
		fmt.Printf("  File Configuration:\n")
		for k, v := range fileConfig {
			fmt.Printf("    %s: %v\n", k, v)
		}
	}

	return nil
}

// RemoveExtension handles the rmextension flag action
func (em *ExtensionManager) RemoveExtension(name string) error {
	if err := em.registry.Remove(name); err != nil {
		return fmt.Errorf("failed to remove extension: %w", err)
	}

	return nil
}

// ProcessExtension handles template processing for extension directives
func (em *ExtensionManager) ProcessExtension(name, operation, value string) (string, error) {
	return em.executor.Execute(name, operation, value)
}