Wrote write goroutine functions.

Refactored subpackages back to root package.

schema.go

@@ -0,0 +1,140 @@
+package heartwood
+
+import (
+	"context"
+	"fmt"
+	"github.com/neo4j/neo4j-go-driver/v6/neo4j"
+)
+
+// ========================================
+// Schema Match Keys
+// ========================================
+
+func NewSimpleMatchKeys() *SimpleMatchKeys {
+	return &SimpleMatchKeys{
+		Keys: map[string][]string{
+			"User":  {"pubkey"},
+			"Relay": {"url"},
+			"Event": {"id"},
+			"Tag":   {"name", "value"},
+		},
+	}
+}
+
+// ========================================
+// Node Constructors
+// ========================================
+
+func NewUserNode(pubkey string) *Node {
+	return NewNode("User", Properties{"pubkey": pubkey})
+}
+
+func NewRelayNode(url string) *Node {
+	return NewNode("Relay", Properties{"url": url})
+}
+
+func NewEventNode(id string) *Node {
+	return NewNode("Event", Properties{"id": id})
+}
+
+func NewTagNode(name string, value string) *Node {
+	return NewNode("Tag", Properties{
+		"name":  name,
+		"value": value})
+}
+
+// ========================================
+// Relationship Constructors
+// ========================================
+
+func NewSignedRel(
+	start *Node, end *Node, props Properties) *Relationship {
+	return NewRelationshipWithValidation(
+		"SIGNED", "User", "Event", start, end, props)
+
+}
+
+func NewTaggedRel(
+	start *Node, end *Node, props Properties) *Relationship {
+	return NewRelationshipWithValidation(
+		"TAGGED", "Event", "Tag", start, end, props)
+}
+
+func NewReferencesEventRel(
+	start *Node, end *Node, props Properties) *Relationship {
+	return NewRelationshipWithValidation(
+		"REFERENCES", "Tag", "Event", start, end, props)
+}
+
+func NewReferencesUserRel(
+	start *Node, end *Node, props Properties) *Relationship {
+	return NewRelationshipWithValidation(
+		"REFERENCES", "Tag", "User", start, end, props)
+}
+
+// ========================================
+// Relationship Constructor Helpers
+// ========================================
+
+func validateNodeLabel(node *Node, role string, expectedLabel string) {
+	if !node.Labels.Contains(expectedLabel) {
+		panic(fmt.Errorf(
+			"expected %s node to have label %q. got %v",
+			role, expectedLabel, node.Labels.ToArray(),
+		))
+	}
+}
+
+func NewRelationshipWithValidation(
+	rtype string,
+	startLabel string,
+	endLabel string,
+	start *Node,
+	end *Node,
+	props Properties) *Relationship {
+
+	validateNodeLabel(start, "start", startLabel)
+	validateNodeLabel(end, "end", endLabel)
+
+	return NewRelationship(rtype, start, end, props)
+}
+
+// ========================================
+// Schema Indexes and Constraints
+// ========================================
+
+// SetNeo4jSchema ensures that the necessary indexes and constraints exist in
+// the database
+func SetNeo4jSchema(ctx context.Context, driver neo4j.Driver) error {
+	schemaQueries := []string{
+		`CREATE CONSTRAINT user_pubkey IF NOT EXISTS
+		 FOR (n:User) REQUIRE n.pubkey IS UNIQUE`,
+
+		`CREATE INDEX user_pubkey IF NOT EXISTS
+		 FOR (n:User) ON (n.pubkey)`,
+
+		`CREATE INDEX event_id IF NOT EXISTS
+		 FOR (n:Event) ON (n.id)`,
+
+		`CREATE INDEX event_kind IF NOT EXISTS
+		 FOR (n:Event) ON (n.kind)`,
+
+		`CREATE INDEX tag_name_value IF NOT EXISTS
+		 FOR (n:Tag) ON (n.name, n.value)`,
+	}
+
+	// Create indexes and constraints
+	for _, query := range schemaQueries {
+		_, err := neo4j.ExecuteQuery(ctx, driver,
+			query,
+			nil,
+			neo4j.EagerResultTransformer,
+			neo4j.ExecuteQueryWithDatabase("neo4j"))
+
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}