package internal import ( "context" "fmt" "git.dragse.it/anthrove/anthrove-graph-sdk.git/pkg/models" "github.com/neo4j/neo4j-go-driver/v5/neo4j" log "github.com/sirupsen/logrus" ) func CreateUserNodeWithSourceRelation(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID, sourceDomain string, userID string, username string) (*models.AnthroveUser, error) { query := ` MATCH (userNode:User {user_id: $anthrove_user_id}) MATCH (sourceNode:Source {domain: $source_domain}) MERGE (userNode)-[r:HAS_ACCOUNT_AT {username: $source_user_name, user_id: $source_user_id}]->(sourceNode) ` params := map[string]any{ "anthrove_user_id": anthroveUserID, "source_user_id": userID, "source_user_name": username, "source_domain": sourceDomain, } _, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer) if err != nil { return nil, err } var anthroveUserRelationship []models.AnthroveUserRelationship anthroveUserRelationship = append(anthroveUserRelationship, models.AnthroveUserRelationship{ UserID: userID, Username: username, ScrapeTimeInterval: "", Source: models.AnthroveSource{ DisplayName: "", Domain: sourceDomain, Icon: "", }, }) anthroveUser := models.AnthroveUser{ UserID: anthroveUserID, Relationship: anthroveUserRelationship, } log.WithFields(log.Fields{ "anthrove_user_id": anthroveUserID, "source_user_id": userID, "source_user_name": username, "source_domain": sourceDomain, }).Trace("graph: crated user with relationship") return &anthroveUser, nil } func GetUserFavoritesCount(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID) (int64, error) { var userFavoriteCount int64 query := ` MATCH (userNode:User {user_id: anthrove_user_id}) MATCH (userNode)-[:FAV]->(favPost:AnthrovePost) MATCH (sourceNode)-[:REFERENCE]->(favPost) RETURN count(favPost) AS FavoritePostsCount ` params := map[string]any{ "anthrove_user_id": anthroveUserID, } result, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer) if err != nil { return 0, err } if len(result.Records) == 0 { // no matches -> user does not exist, return count 0 return userFavoriteCount, err } record := result.Records[0] userFavoriteCount, _, err = neo4j.GetRecordValue[int64](record, "FavoritePostsCount") if err != nil { return userFavoriteCount, err } log.WithFields(log.Fields{ "anthrove_user_id": anthroveUserID, "anthrove_user_fav_count": userFavoriteCount, }).Trace("graph: get user favorite count") return userFavoriteCount, nil } func GetUserSourceLink(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID) (map[string]models.AnthroveUserRelationship, error) { userSource := make(map[string]models.AnthroveUserRelationship) query := ` MATCH (user:User{user_id: anthrove_user_id})-[r:HAS_ACCOUNT_AT]->(s:Source) RETURN toString(r.user_id) AS sourceUserID, toString(r.username) AS sourceUsername, s.display_name as sourceDisplayName; ` params := map[string]any{ "anthrove_user_id": anthroveUserID, } 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("user has no relations") } for i := range result.Records { record := result.Records[i] sourceName, _, err := neo4j.GetRecordValue[string](record, "sourceDisplayName") if err != nil { return nil, err } sourceUserID, _, err := neo4j.GetRecordValue[string](record, "sourceUserID") if err != nil { return nil, err } sourceUsername, _, err := neo4j.GetRecordValue[string](record, "sourceUsername") if err != nil { return nil, err } anthroveSourceUser := models.AnthroveUserRelationship{ UserID: sourceUserID, Username: sourceUsername, } userSource[sourceName] = anthroveSourceUser } log.WithFields(log.Fields{ "anthrove_user_id": anthroveUserID, "anthrove_data": userSource, }).Trace("graph: get user favorite count") return userSource, nil }