yao/agent/output/adapters/openai/converter.go

package openai

import (
	"fmt"
	"time"

	"github.com/yaoapp/yao/agent/output/message"
)

// ConverterRegistry manages message type converters
type ConverterRegistry struct {
	converters map[string]ConverterFunc
}

// NewConverterRegistry creates a new converter registry with default converters
func NewConverterRegistry() *ConverterRegistry {
	return &ConverterRegistry{
		converters: map[string]ConverterFunc{
			message.TypeText:     convertText,
			message.TypeThinking: convertThinking,
			message.TypeLoading:  convertLoading,
			message.TypeToolCall: convertToolCall,
			message.TypeError:    convertError,
			message.TypeImage:    convertImage,
			message.TypeAudio:    convertToLink,
			message.TypeVideo:    convertToLink,
			message.TypeAction:   convertAction,
		},
	}
}

// Register registers a custom converter for a message type
func (r *ConverterRegistry) Register(msgType string, converter ConverterFunc) {
	r.converters[msgType] = converter
}

// GetConverter retrieves a converter for a given message type.
func (r *ConverterRegistry) GetConverter(msgType string) (ConverterFunc, bool) {
	converter, exists := r.converters[msgType]
	return converter, exists
}

// Convert converts a message using registered converters
// If no converter is found, converts to link format
func (r *ConverterRegistry) Convert(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	// Check for registered converter
	if converter, exists := r.converters[msg.Type]; exists {
		return converter(msg, config)
	}

	// Fallback: convert to link format
	return convertToLink(msg, config)
}

// convertText converts text messages to OpenAI format
func convertText(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	content := getStringProp(msg.Props, "content", "")

	return []interface{}{
		createOpenAIChunk(msg.ID, config.Model, map[string]interface{}{
			"content": content,
		}),
	}, nil
}

// convertThinking converts thinking messages to OpenAI reasoning format (o1 series)
func convertThinking(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	content := getStringProp(msg.Props, "content", "")

	return []interface{}{
		createOpenAIChunk(msg.ID, config.Model, map[string]interface{}{
			"reasoning_content": content,
		}),
	}, nil
}

// convertLoading converts loading messages to OpenAI reasoning format
// This makes loading messages visible in standard OpenAI clients as thinking process
func convertLoading(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	message := getStringProp(msg.Props, "message", "Processing...")

	// Convert loading to reasoning_content so it shows in OpenAI clients
	return []interface{}{
		createOpenAIChunk(msg.ID, config.Model, map[string]interface{}{
			"reasoning_content": message,
		}),
	}, nil
}

// convertToolCall converts tool_call messages to OpenAI format
func convertToolCall(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	// Tool call format varies, pass through the props
	toolCalls := []map[string]interface{}{}

	// If props contain tool call data, use it
	if id, ok := msg.Props["id"].(string); ok {
		toolCall := map[string]interface{}{
			"id":   id,
			"type": "function",
		}

		if function, ok := msg.Props["function"].(map[string]interface{}); ok {
			toolCall["function"] = function
		}

		toolCalls = append(toolCalls, toolCall)
	}

	return []interface{}{
		createOpenAIChunk(msg.ID, config.Model, map[string]interface{}{
			"tool_calls": toolCalls,
		}),
	}, nil
}

// convertError converts error messages to OpenAI error format
func convertError(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	message := getStringProp(msg.Props, "message", "An error occurred")
	code := getStringProp(msg.Props, "code", "server_error")

	return []interface{}{
		map[string]interface{}{
			"error": map[string]interface{}{
				"message": message,
				"type":    code,
				"code":    code,
			},
		},
	}, nil
}

// convertAction converts action messages to nothing (silent in OpenAI clients)
// Action messages are system-level commands (open panel, navigate, etc.)
// and should not be sent to standard chat clients
func convertAction(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	// Return empty slice - no output for action messages in OpenAI format
	return []interface{}{}, nil
}

// convertImage converts image messages to Markdown image format
// Uses ![alt](url) which displays inline in Markdown-supporting clients
func convertImage(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	// Get URL
	url, ok := msg.Props["url"].(string)
	if !ok || url == "" {
		return nil, fmt.Errorf("image message missing url")
	}

	// Transform URL if transformer is provided
	if config.LinkTransformer != nil {
		transformedURL, err := config.LinkTransformer(url, msg.Type, msg.ID)
		if err != nil {
			return nil, err
		}
		url = transformedURL
	}

	// Get alt text (default to "Image")
	alt := getStringProp(msg.Props, "alt", "Image")

	// Format as Markdown image: ![alt](url)
	template := getLinkTemplate(msg.Type, config)
	text := fmt.Sprintf(template, alt, url)

	return []interface{}{
		createOpenAIChunk(msg.ID, config.Model, map[string]interface{}{
			"content": text,
		}),
	}, nil
}

// convertToLink converts any message type to a Markdown link format
func convertToLink(msg *message.Message, config *AdapterConfig) ([]interface{}, error) {
	// Generate link
	link, err := generateViewLink(msg, config)
	if err != nil {
		return nil, err
	}

	// Get template
	template := getLinkTemplate(msg.Type, config)

	// Format text
	var text string
	if msg.Type == "button" {
		// Button is special: needs button text
		buttonText := getStringProp(msg.Props, "text", "Button")
		text = fmt.Sprintf(template, buttonText, link)
	} else {
		text = fmt.Sprintf(template, link)
	}

	return []interface{}{
		createOpenAIChunk(msg.ID, config.Model, map[string]interface{}{
			"content": text,
		}),
	}, nil
}

// generateViewLink generates a view link for a message
func generateViewLink(msg *message.Message, config *AdapterConfig) (string, error) {
	// If Props contains a URL, use it
	if url, ok := msg.Props["url"].(string); ok {
		// Transform URL if transformer is provided
		if config.LinkTransformer != nil {
			return config.LinkTransformer(url, msg.Type, msg.ID)
		}
		return url, nil
	}

	// Generate view link: {baseURL}/agent/view/{type}/{id}
	baseURL := config.BaseURL
	if baseURL == "" {
		baseURL = "" // TODO: Get from environment or context
	}

	viewURL := fmt.Sprintf("%s/agent/view/%s/%s", baseURL, msg.Type, msg.ID)

	// Transform URL if transformer is provided
	if config.LinkTransformer != nil {
		return config.LinkTransformer(viewURL, msg.Type, msg.ID)
	}

	return viewURL, nil
}

// getLinkTemplate gets the link template for a message type
func getLinkTemplate(msgType string, config *AdapterConfig) string {
	if template, exists := config.LinkTemplates[msgType]; exists {
		return template
	}

	// Default fallback template
	return "📎 [View %s](" + msgType + ")"
}

// createOpenAIChunk creates an OpenAI chat completion chunk
func createOpenAIChunk(id string, model string, delta map[string]interface{}) map[string]interface{} {
	return map[string]interface{}{
		"id":      id,
		"object":  "chat.completion.chunk",
		"created": time.Now().Unix(),
		"model":   model,
		"choices": []map[string]interface{}{
			{
				"index":         0,
				"delta":         delta,
				"finish_reason": nil,
			},
		},
	}
}

// getStringProp safely gets a string property from props
func getStringProp(props map[string]interface{}, key string, defaultValue string) string {
	if val, ok := props[key].(string); ok {
		return val
	}
	return defaultValue
}