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

package source

import (
	"context"
	"fmt"

	"git.dragse.it/anthrove/otter-space-sdk/pkg/models"
	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
	log "github.com/sirupsen/logrus"
)

// CreateSourceNode will always create a new node, it checks if a node with the same url exits and merges it.
func CreateSourceNode(ctx context.Context, driver neo4j.DriverWithContext, anthroveSource *models.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("graph: %w", err)
	}

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

	return nil
}

// DeleteSourceNode will delete a node with a given Domain
func DeleteSourceNode(ctx context.Context, driver neo4j.DriverWithContext, anthroveSource *models.AnthroveSource) error {
	query := `
		MERGE (sourceNode:Source {domain: $domain})
		DELETE sourceNode
		`
	params := map[string]any{
		"domain": anthroveSource.Domain,
	}

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

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

	return nil
}

func GetAllSourceNodes(ctx context.Context, driver neo4j.DriverWithContext) ([]models.AnthroveSource, error) {
	var sources []models.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, models.AnthroveSource{
			DisplayName: models.AnthroveSourceDisplayName(source.Props["display_name"].(string)),
			Domain:      models.AnthroveSourceDomain(source.Props["domain"].(string)),
			Icon:        models.AnthroveSourceIcon(source.Props["icon"].(string)),
		})

	}

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

	return sources, nil
}

// GetSourceNodesByURL returns the first models.AnthroveSource provided by the database
func GetSourceNodesByURL(ctx context.Context, driver neo4j.DriverWithContext, sourceUrl models.AnthroveSourceDomain) (*models.AnthroveSource, error) {
	var source models.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 = models.AnthroveSourceDisplayName(record.Props["display_name"].(string))
	source.Domain = models.AnthroveSourceDomain(record.Props["domain"].(string))
	source.Icon = models.AnthroveSourceIcon(record.Props["icon"].(string))

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

	return &source, nil
}