package copy_mission

import (
	kgusecase "haixun-backend/internal/model/knowledge_graph/domain/usecase"
	"haixun-backend/internal/types"
)

func toKnowledgeGraphData(graph *kgusecase.GraphSummary) types.KnowledgeGraphData {
	if graph == nil {
		return types.KnowledgeGraphData{}
	}
	nodes := make([]types.KnowledgeGraphNodeData, 0, len(graph.Nodes))
	for _, node := range graph.Nodes {
		evidence := make([]types.KnowledgeGraphEvidenceData, 0, len(node.Evidence))
		for _, item := range node.Evidence {
			evidence = append(evidence, types.KnowledgeGraphEvidenceData{
				URL:     item.URL,
				Snippet: item.Snippet,
				Query:   item.Query,
			})
		}
		nodes = append(nodes, types.KnowledgeGraphNodeData{
			ID:              node.ID,
			Label:           node.Label,
			NodeKind:        node.NodeKind,
			Type:            node.Type,
			Layer:           node.Layer,
			Relation:        node.Relation,
			PlacementValue:  node.PlacementValue,
			ProductFitScore: node.ProductFitScore,
			SelectedForScan: node.SelectedForScan,
			RelevanceTags:   append([]string{}, node.DerivedTags.Relevance...),
			RecencyTags:     append([]string{}, node.DerivedTags.Recency...),
			Evidence:        evidence,
		})
	}
	edges := make([]types.KnowledgeGraphEdgeData, 0, len(graph.Edges))
	for _, edge := range graph.Edges {
		edges = append(edges, types.KnowledgeGraphEdgeData{
			From:     edge.From,
			To:       edge.To,
			Relation: edge.Relation,
		})
	}
	sources := make([]types.BraveSourceData, 0, len(graph.BraveSources))
	for _, src := range graph.BraveSources {
		sources = append(sources, types.BraveSourceData{
			Query:   src.Query,
			Title:   src.Title,
			URL:     src.URL,
			Snippet: src.Snippet,
		})
	}
	return types.KnowledgeGraphData{
		ID:             graph.ID,
		BrandID:        graph.BrandID,
		Seed:           graph.Seed,
		Nodes:          nodes,
		Edges:          edges,
		BraveSources:   sources,
		ExpandStrategy: graph.ExpandStrategy,
		PainTagCount:   graph.PainTagCount,
		GeneratedAt:    graph.GeneratedAt,
		CreateAt:       graph.CreateAt,
		UpdateAt:       graph.UpdateAt,
	}
}