package monitor

import (
	"context"
	"time"

	"eino-project/internal/conf"

	cozeloop "github.com/coze-dev/cozeloop-go"
	"github.com/go-kratos/kratos/v2/log"
)

// cozeLoopMonitor 是一个包装器：在调用本地监控逻辑的同时，将关键指标以 Trace 形式上报到 Coze Loop
type cozeLoopMonitor struct {
	base   Monitor
	client cozeloop.Client
	log    *log.Helper
}

// NewCozeLoopMonitor 根据配置初始化 Coze Loop 客户端并返回包装后的监控器
func NewCozeLoopMonitor(base Monitor, cfg *conf.Monitoring_CozeLoop, logger log.Logger) Monitor {
	helper := log.NewHelper(logger)

	ctx := context.Background()

	// 按官方 simple demo 风格，直接从环境变量初始化客户端
	client, err := cozeloop.NewClient()
	if err != nil {
		helper.WithContext(ctx).Warnf("init CozeLoop client failed: %v, fallback to base monitor", err)
		return base
	}

	m := &cozeLoopMonitor{
		base:   base,
		client: client,
		log:    helper,
	}

	return m
}

// RecordRequest 记录请求并上报简化 Trace
func (m *cozeLoopMonitor) RecordRequest(ctx context.Context, requestType string, duration time.Duration, success bool) error {
	if err := m.base.RecordRequest(ctx, requestType, duration, success); err != nil {
		return err
	}
	// 简化：不生成trace
	return nil
}

// RecordAIRequest 记录 AI 请求并上报简化 Trace
func (m *cozeLoopMonitor) RecordAIRequest(ctx context.Context, duration time.Duration, success bool) error {
	if err := m.base.RecordAIRequest(ctx, duration, success); err != nil {
		return err
	}
	// 简化：不生成trace
	return nil
}

// RecordVectorOperation 记录向量操作并上报简化 Trace
func (m *cozeLoopMonitor) RecordVectorOperation(ctx context.Context, operation string, success bool) error {
	if err := m.base.RecordVectorOperation(ctx, operation, success); err != nil {
		return err
	}
	// 简化：不生成trace
	return nil
}

// RecordSessionOperation 记录会话操作并上报简化 Trace
func (m *cozeLoopMonitor) RecordSessionOperation(ctx context.Context, operation string) error {
	if err := m.base.RecordSessionOperation(ctx, operation); err != nil {
		return err
	}
	// 简化：不生成trace
	return nil
}

// RecordKnowledgeOperation 记录知识库操作并上报简化 Trace
func (m *cozeLoopMonitor) RecordKnowledgeOperation(ctx context.Context, operation string) error {
	if err := m.base.RecordKnowledgeOperation(ctx, operation); err != nil {
		return err
	}
	// 简化：不生成trace
	return nil
}

// 透传LLM使用事件（不生成trace，不做周期性上报）
func (m *cozeLoopMonitor) RecordLLMUsage(ctx context.Context, usage *LLMUsage) error {
	return m.base.RecordLLMUsage(ctx, usage)
}

// GetMetrics 透传本地指标（不直接从 Coze Loop 读取）
func (m *cozeLoopMonitor) GetMetrics(ctx context.Context) (*Metrics, error) {
	return m.base.GetMetrics(ctx)
}

// CreateAlert 创建告警（不生成trace，保持简化）
func (m *cozeLoopMonitor) CreateAlert(ctx context.Context, alertType, level, title, message string, metadata map[string]interface{}) error {
	if err := m.base.CreateAlert(ctx, alertType, level, title, message, metadata); err != nil {
		return err
	}
	return nil
}

// GetAlerts 透传本地告警
func (m *cozeLoopMonitor) GetAlerts(ctx context.Context, resolved bool) ([]*Alert, error) {
	return m.base.GetAlerts(ctx, resolved)
}

// ResolveAlert 透传本地告警处理
func (m *cozeLoopMonitor) ResolveAlert(ctx context.Context, alertID string) error {
	return m.base.ResolveAlert(ctx, alertID)
}

// HealthCheck 透传本地健康检查
func (m *cozeLoopMonitor) HealthCheck(ctx context.Context) error {
	return m.base.HealthCheck(ctx)
}

// ResetMetrics 重置本地指标
func (m *cozeLoopMonitor) ResetMetrics(ctx context.Context) error {
	return m.base.ResetMetrics(ctx)
}