// Package eino 封装基于 CloudWeGo Eino 的 Agent 图编排引擎。
package eino

import (
	"context"
	"errors"
	"io"
	"log"
	"time"

	"github.com/cloudwego/eino/compose"
	"github.com/cloudwego/eino/schema"

	"github.com/sundynix/sundynix-dispatcher/internal/harness"
	"github.com/sundynix/sundynix-dispatcher/internal/llm"
	"github.com/sundynix/sundynix-shared/contract"
)

// TokenSink 是 Token 流回流出口（由 NATS bus 实现）。
type TokenSink interface {
	PublishToken(taskID string, token []byte) error
	CompleteStream(taskID string) error
}

// ToolCaller 经 NATS 调起第 5 层 MCP 工具（由 NATS bus 实现）。
type ToolCaller interface {
	CallTool(ctx context.Context, subject string, call *contract.ToolCall) (*contract.ToolResult, error)
}

// 工具调用超时；超时即降级（不带工具上下文继续推理）。
const toolCallTimeout = 3 * time.Second

// Orchestrator 把 DSL 任务交给编译好的 Eino 图执行（记忆召回 → 注入 → 流式）。
type Orchestrator struct {
	breaker *harness.CircuitBreaker
	sink    TokenSink
	tools   ToolCaller
	run     compose.Runnable[*contract.Task, *schema.Message]
}

// NewOrchestrator 构建并编译记忆增强图。
func NewOrchestrator(pool *llm.Pool, breaker *harness.CircuitBreaker, sink TokenSink, tools ToolCaller) (*Orchestrator, error) {
	o := &Orchestrator{breaker: breaker, sink: sink, tools: tools}
	run, err := buildGraph(context.Background(), pool, o.fetchMemory)
	if err != nil {
		return nil, err
	}
	o.run = run
	return o, nil
}

// Handle 消费一个任务：执行 Eino 图，把 Token 流回流到 sundynix.streams.<id>。
func (o *Orchestrator) Handle(ctx context.Context, t *contract.Task) error {
	if !o.breaker.Allow() {
		log.Printf("[eino] circuit open, drop task %s", t.ID)
		return nil
	}
	log.Printf("[eino] task %s received (graph=%d bytes), running graph...", t.ID, len(t.Graph))

	stream, err := o.run.Stream(ctx, t)
	if err != nil {
		log.Printf("[eino] task %s graph error: %v", t.ID, err)
		_ = o.sink.CompleteStream(t.ID)
		o.breaker.Report(false)
		return err
	}
	defer stream.Close()

	n := 0
	for {
		chunk, rerr := stream.Recv()
		if errors.Is(rerr, io.EOF) {
			break
		}
		if rerr != nil {
			log.Printf("[eino] task %s stream recv error: %v", t.ID, rerr)
			break
		}
		if chunk == nil || chunk.Content == "" {
			continue
		}
		if perr := o.sink.PublishToken(t.ID, []byte(chunk.Content)); perr != nil {
			log.Printf("[eino] publish token failed: %v", perr)
			break
		}
		n++
	}

	if cerr := o.sink.CompleteStream(t.ID); cerr != nil {
		log.Printf("[eino] complete stream failed: %v", cerr)
	}
	log.Printf("[eino] task %s done, %d tokens streamed", t.ID, n)
	o.breaker.Report(true)

	// 写回阶段：流已排空（= 模型生成结束），此处离开热路径、异步抽取记忆。
	// 注：流式节点用 OnEndWithStreamOutput 而非 OnEndFn，故不走回调而在此触发。
	go o.memorize(t)
	return nil
}

// fetchMemory 经 MCP memory_get 工具召回用户常驻画像。
// 工具不可用/超时/无 user_id 时返回空串，降级为无记忆推理（不阻断主流程）。
func (o *Orchestrator) fetchMemory(ctx context.Context, userID, _ string) string {
	if o.tools == nil || userID == "" {
		return ""
	}
	cctx, cancel := context.WithTimeout(ctx, toolCallTimeout)
	defer cancel()
	res, err := o.tools.CallTool(cctx, contract.ToolSubjectGo("memory_get"), &contract.ToolCall{
		Tool: "memory_get",
		Args: map[string]any{"user_id": userID},
	})
	if err != nil {
		log.Printf("[eino] memory_get unavailable for %s, degrade: %v", userID, err)
		return ""
	}
	if !res.OK {
		log.Printf("[eino] memory_get error for %s: %s", userID, res.Error)
		return ""
	}
	log.Printf("[eino] memory_get ok for %s: %s", userID, res.Content)
	return res.Content
}

// memorize 写回阶段：从本轮对话抽取并更新偏好记忆。
// 目前发占位日志；真实实现应跑抽取 LLM → 去重/更新 → memory_upsert（异步，离开热路径）。
func (o *Orchestrator) memorize(t *contract.Task) {
	uid, _ := t.Meta[contract.MetaUserID].(string)
	if uid == "" {
		return
	}
	log.Printf("[eino] (writeback) task %s 完成，待抽取 user=%s 的新偏好记忆", t.ID, uid)
	// TODO: 发 sundynix.memory.extract 事件 → memory worker 抽取 → memory_upsert
}