139 lines
3.9 KiB
Go
139 lines
3.9 KiB
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
|
|
}
|