tools.go

package cobrax

import (
	"encoding/json"
	"fmt"
	"log/slog"
	"os"

	"github.com/spf13/cobra"
)

// toolCommandFlags holds flags for the tools command.
type toolCommandFlags struct {
	logLevel string
}

// toolCommand creates the 'mcp tools' command to export tool definitions.
func toolCommand(config *Config) *cobra.Command {
	toolFlags := &toolCommandFlags{}
	cmd := &cobra.Command{
		Use:   "tools",
		Short: "Export tools as JSON",
		Long:  `Export available MCP tools to mcp-tools.json for inspection`,
		RunE: func(cmd *cobra.Command, _ []string) error {
			if config == nil {
				config = &Config{}
			}

			if toolFlags.logLevel != "" {
				if config.SloggerOptions == nil {
					config.SloggerOptions = &slog.HandlerOptions{}
				}

				config.SloggerOptions.Level = parseLogLevel(toolFlags.logLevel)
			}

			file, err := os.OpenFile("mcp-tools.json", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0o644)
			if err != nil {
				return fmt.Errorf("failed to create or open mcp-tools.json file: %w", err)
			}
			defer func() {
				if closeErr := file.Close(); closeErr != nil {
					cmd.Printf("Warning: failed to close file: %v\n", closeErr)
				}
			}()

			encoder := json.NewEncoder(file)
			encoder.SetIndent("", "  ")
			config.registerTools(cmd)
			err = encoder.Encode(config.tools)
			if err != nil {
				return fmt.Errorf("failed to encode MCP tools to JSON: %w", err)
			}

			cmd.Printf("Successfully exported %d tools to mcp-tools.json\n", len(config.tools))
			return nil
		},
	}

	// Add flags
	flags := cmd.Flags()
	flags.StringVar(&toolFlags.logLevel, "log-level", "", "Log level (debug, info, warn, error)")
	return cmd
}