package internal

import (
	"context"

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

func EstablishAnthrovePostToSourceLink(ctx context.Context, driver neo4j.DriverWithContext, anthrovePostID models.AnthrovePostID, anthroveSourceDomain string, anthrovePostRelationship *models.AnthrovePostRelationship) error {
	query := `
		MATCH (sourceNode:Source {domain: $source_url})
		MATCH (postNode:AnthrovePost {post_id: $anthrove_post_id})
		MERGE (sourceNode)-[:REFERENCE {url: $source_post_url, source_post_id: $source_post_id}]->(postNode) 
		`

	params := map[string]any{
		"source_url":       anthroveSourceDomain,
		"anthrove_post_id": anthrovePostID,
		"source_post_url":  anthrovePostRelationship.Url,
		"source_post_id":   anthrovePostRelationship.PostID,
	}

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

	log.WithFields(log.Fields{
		"source_url":       anthroveSourceDomain,
		"anthrove_post_id": anthrovePostID,
		"source_post_url":  anthrovePostRelationship.Url,
		"source_post_id":   anthrovePostRelationship.PostID,
	}).Trace("graph: creating anthrove post to source link")

	return nil
}