otter-space-sdk/internal/graph/source.go

package graph

import (
	"context"
	"fmt"
	"git.dragse.it/anthrove/otter-space-sdk/pkg/models/graphModels"

	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
	log "github.com/sirupsen/logrus"
)

func CreateSourceNode(ctx context.Context, driver neo4j.DriverWithContext, anthroveSource *graphModels.AnthroveSource) error {
	query := `
		MERGE (sourceNode:Source {domain: $source_url})
			ON CREATE SET sourceNode.domain = $source_url, sourceNode.display_name = $source_display_name, sourceNode.icon = $source_icon
		`
	params := map[string]any{
		"source_url":          anthroveSource.Domain,
		"source_display_name": anthroveSource.DisplayName,
		"source_icon":         anthroveSource.Icon,
	}

	_, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
	if err != nil {
		return fmt.Errorf("database: %w", err)
	}

	log.WithFields(log.Fields{
		"node_source_url":         anthroveSource.Domain,
		"node_source_displayName": anthroveSource.DisplayName,
		"node_source_icon":        anthroveSource.Icon,
	}).Trace("database: created source node")

	return nil
}

func GetAllSourceNodes(ctx context.Context, driver neo4j.DriverWithContext) ([]graphModels.AnthroveSource, error) {
	var sources []graphModels.AnthroveSource

	query := `
		MATCH (s:Source)
		RETURN s as source
		`
	params := map[string]any{}

	result, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
	if err != nil {
		return nil, err
	}

	if len(result.Records) == 0 {
		return nil, nil
	}

	for i := range result.Records {
		record := result.Records[i]

		source, _, err := neo4j.GetRecordValue[neo4j.Node](record, "source")
		if err != nil {
			return nil, err
		}

		sources = append(sources, graphModels.AnthroveSource{
			DisplayName: source.Props["display_name"].(string),
			Domain:      source.Props["domain"].(string),
			Icon:        source.Props["icon"].(string),
		})

	}

	log.WithFields(log.Fields{
		"tag_amount": len(sources),
	}).Trace("database: created tag node")

	return sources, nil
}

func GetSourceNodesByURL(ctx context.Context, driver neo4j.DriverWithContext, sourceUrl string) (*graphModels.AnthroveSource, error) {

	var source graphModels.AnthroveSource

	query := `
		MATCH (s:Source {domain: $source_url})
		RETURN s as source
		`
	params := map[string]any{
		"source_url": sourceUrl,
	}

	result, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
	if err != nil {
		return nil, err
	}

	if len(result.Records) == 0 {
		return nil, fmt.Errorf("source not found")
	}

	record, _, err := neo4j.GetRecordValue[neo4j.Node](result.Records[0], "source")
	if err != nil {
		return nil, err
	}

	source.DisplayName = record.Props["display_name"].(string)
	source.Domain = record.Props["domain"].(string)
	source.Icon = record.Props["icon"].(string)

	log.WithFields(log.Fields{
		"source_url": sourceUrl,
	}).Trace("database: got source node")

	return &source, nil
}