101 lines
3.0 KiB
Go
101 lines
3.0 KiB
Go
package internal
|
|
|
|
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"
|
|
)
|
|
|
|
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
|
|
}
|
|
|
|
func EstablishUserToPostLink(ctx context.Context, driver neo4j.DriverWithContext, anthroveUser *models.AnthroveUser, anthrovePost *models.AnthrovePost) error {
|
|
|
|
query := `
|
|
MATCH (user:User {user_id: $anthrove_user_id})
|
|
MATCH (anthrovePost:AnthrovePost {post_id: $anthrove_post_id})
|
|
MERGE (user)-[:FAV]->(anthrovePost)
|
|
`
|
|
|
|
params := map[string]any{
|
|
"anthrove_post_id": anthrovePost.PostID,
|
|
"anthrove_user_id": anthroveUser.UserID,
|
|
}
|
|
|
|
_, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"anthrove_post_id": anthrovePost.PostID,
|
|
"anthrove_user_id": anthroveUser.UserID,
|
|
}).Trace("graph: created user to post link")
|
|
|
|
return nil
|
|
}
|
|
|
|
func CheckUserToPostLink(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID, sourcePostID string, sourceUrl string) (bool, error) {
|
|
query := `
|
|
OPTIONAL MATCH (:User {user_id: $anthrove_user_id})-[f:FAV]->(:AnthrovePost)<-[:REFERENCE{source_post_id: $source_post_id}]-(:Source{domain: $source_domain})
|
|
RETURN COUNT(f) > 0 AS hasRelationship
|
|
`
|
|
|
|
params := map[string]any{
|
|
"anthrove_user_id": anthroveUserID,
|
|
"source_post_id": sourcePostID,
|
|
"source_domain": sourceUrl,
|
|
}
|
|
|
|
result, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if len(result.Records) == 0 {
|
|
return false, fmt.Errorf("no records found")
|
|
}
|
|
|
|
exists, _, err := neo4j.GetRecordValue[bool](result.Records[0], "hasRelationship")
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"relationship_exists": exists,
|
|
"relationship_anthrove_user_id": anthroveUserID,
|
|
"relationship_e621_post_id": "",
|
|
}).Trace("graph: checked user post relationship")
|
|
|
|
return exists, nil
|
|
}
|