package heartwood

import (
	"git.wisehodl.dev/jay/go-heartwood/graph"
	roots "git.wisehodl.dev/jay/go-roots/events"
)

func EventToSubgraph(e roots.Event, exp ExpanderRegistry) *graph.Subgraph {
	subgraph := graph.NewSubgraph()

	// Create Event node
	eventNode := graph.NewEventNode(e.ID)
	eventNode.Props["created_at"] = e.CreatedAt
	eventNode.Props["kind"] = e.Kind
	eventNode.Props["content"] = e.Content

	// Create User node
	userNode := graph.NewUserNode(e.PubKey)

	// Create SIGNED rel
	signedRel := graph.NewSignedRel(userNode, eventNode, nil)

	// Create Tag nodes
	tagNodes := []*graph.Node{}
	for _, tag := range e.Tags {
		if !isValidTag(tag) {
			continue
		}
		tagNodes = append(tagNodes, graph.NewTagNode(tag[0], tag[1]))
	}

	// Create Tag rels
	tagRels := []*graph.Relationship{}
	for _, tagNode := range tagNodes {
		tagRels = append(tagRels, graph.NewTaggedRel(eventNode, tagNode, nil))
	}

	// Populate subgraph
	subgraph.AddNode(eventNode)
	subgraph.AddNode(userNode)
	subgraph.AddRel(signedRel)
	for _, node := range tagNodes {
		subgraph.AddNode(node)
	}
	for _, rel := range tagRels {
		subgraph.AddRel(rel)
	}

	// Run expanders
	for _, expander := range exp {
		expander(e, subgraph)
	}

	return subgraph
}

func isValidTag(t roots.Tag) bool {
	if len(t) < 2 {
		// Skip tags that do not have name and value fields
		return false
	}
	if len(t[0])+len(t[1]) > 8192 {
		// Skip tags that are too large for the neo4j indexer
		return false
	}
	return true
}