package test

import (
	stdContext "context"

	"github.com/yaoapp/yao/agent/context"
	"github.com/yaoapp/yao/agent/output/message"
	"github.com/yaoapp/yao/openapi/oauth/types"
)

// NewTestContext creates a new context for testing
// This is similar to newAgentNextTestContext in agent_next_test.go
// but configurable via Environment
func NewTestContext(chatID, assistantID string, env *Environment) *context.Context {
	// Build authorized info from environment
	authorized := &types.AuthorizedInfo{
		Subject:  env.UserID,
		UserID:   env.UserID,
		TenantID: env.TeamID,
	}

	// Create context with standard initialization
	ctx := context.New(stdContext.Background(), authorized, chatID)
	ctx.ID = chatID
	ctx.AssistantID = assistantID
	ctx.Locale = env.Locale
	ctx.Client = context.Client{
		Type:      env.ClientType,
		UserAgent: "yao-agent-test/1.0",
		IP:        env.ClientIP,
	}
	ctx.Referer = env.Referer
	ctx.Accept = context.AcceptStandard
	ctx.IDGenerator = message.NewIDGenerator()
	ctx.Metadata = make(map[string]interface{})

	// Initialize interrupt controller
	ctx.Interrupt = context.NewInterruptController()

	// Close the default logger created by context.New() and use noop logger
	// to suppress LLM debug output during tests
	if ctx.Logger != nil {
		ctx.Logger.Close()
	}
	ctx.Logger = context.Noop()

	return ctx
}

// NewTestContextFromOptions creates a test context from test options and test case
func NewTestContextFromOptions(chatID, assistantID string, opts *Options, tc *Case) *context.Context {
	// Get environment from test case (with options override)
	env := tc.GetEnvironment(opts)
	return NewTestContext(chatID, assistantID, env)
}

// GenerateChatID generates a unique chat ID for testing
func GenerateChatID(testID string, runNumber int) string {
	if runNumber > 1 {
		return "test-" + testID + "-run" + string(rune('0'+runNumber))
	}
	return "test-" + testID
}