From ab1d156e043bddcc2571d740eb38af5e9e461c41 Mon Sep 17 00:00:00 2001
From: Vargha Csongor <Culisa1023@gmail.com>
Date: Thu, 10 Apr 2025 21:11:11 +0200
Subject: [PATCH] add relationship api

---
 apps/db-adapter/internal/api/relationship.go | 108 ++++++++++++++++++-
 1 file changed, 107 insertions(+), 1 deletion(-)

diff --git a/apps/db-adapter/internal/api/relationship.go b/apps/db-adapter/internal/api/relationship.go
index 45082f2..b743b5e 100644
--- a/apps/db-adapter/internal/api/relationship.go
+++ b/apps/db-adapter/internal/api/relationship.go
@@ -1,11 +1,117 @@
 package api
 
 import (
+	"context"
+	"fmt"
+	"net/http"
+
 	"github.com/gin-gonic/gin"
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+	"github.com/vcscsvcscs/GenerationsHeritage/apps/db-adapter/internal/api/auth"
+	"github.com/vcscsvcscs/GenerationsHeritage/apps/db-adapter/internal/memgraph"
 	"github.com/vcscsvcscs/GenerationsHeritage/apps/db-adapter/pkg/api"
 )
 
-func (srv *server) CreateRelationship(c *gin.Context, params api.CreateRelationshipParams) {}
+func (srv *server) CreateRelationship(c *gin.Context, params api.CreateRelationshipParams) {
+	var relationship *api.CreateRelationshipJSONRequestBody
+	if err := c.ShouldBindJSON(&relationship); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"msg": err.Error()})
+
+		return
+	}
+
+	session := srv.db.NewSession(c.Request.Context(), neo4j.SessionConfig{})
+	defer closeSession(c.Request.Context(), session, srv.dbOpTimeout)
+
+	actx, aCancel := context.WithTimeout(c.Request.Context(), srv.dbOpTimeout)
+	defer aCancel()
+
+	if err := auth.CouldManagePersonUnknownAdmin(actx, session, *relationship.Id1, params.XUserID); err != nil {
+		if err := auth.CouldManagePersonUnknownAdmin(actx, session, *relationship.Id1, params.XUserID); err != nil {
+			c.JSON(http.StatusUnauthorized, gin.H{"msg": fmt.Sprint("User does not have access to this person", err.Error())})
+
+			return
+		}
+	}
+
+	qctx, qCancel := context.WithTimeout(c.Request.Context(), srv.dbOpTimeout)
+	defer qCancel()
+
+	var relationShipResultRaw any
+	var relationshipError error
+	switch *relationship.Type {
+	case api.CreateRelationshipJSONBodyTypeChild:
+		relationShipResultRaw, relationshipError = session.ExecuteWrite(qctx, memgraph.CreateChildParentRelationship(qctx, *relationship.Id1, *relationship.Id2))
+	case api.CreateRelationshipJSONBodyTypeParent:
+		relationShipResultRaw, relationshipError = session.ExecuteWrite(qctx, memgraph.CreateChildParentRelationship(qctx, *relationship.Id1, *relationship.Id2))
+	case api.CreateRelationshipJSONBodyTypeSibling:
+		relationShipResultRaw, relationshipError = session.ExecuteWrite(qctx, memgraph.CreateSiblingRelationship(qctx, *relationship.Id1, *relationship.Id2))
+	case api.CreateRelationshipJSONBodyTypeSpouse:
+		relationShipResultRaw, relationshipError = session.ExecuteWrite(qctx, memgraph.CreateSpouseRelationship(qctx, *relationship.Id1, *relationship.Id2))
+	default:
+		c.JSON(http.StatusBadRequest, gin.H{"msg": "invalid relationship type"})
+	}
+	if relationshipError != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"msg": relationshipError.Error()})
+
+		return
+	}
+
+	c.JSON(http.StatusOK, relationShipResultRaw)
+}
 
 func (srv *server) GetRelationship(c *gin.Context, id1 int, id2 int, params api.GetRelationshipParams) {
+	session := srv.db.NewSession(c.Request.Context(), neo4j.SessionConfig{})
+	defer closeSession(c.Request.Context(), session, srv.dbOpTimeout)
+
+	actx, aCancel := context.WithTimeout(c.Request.Context(), srv.dbOpTimeout)
+	defer aCancel()
+
+	if err := auth.CouldManagePersonUnknownAdmin(actx, session, id1, params.XUserID); err != nil {
+		if err := auth.CouldManagePersonUnknownAdmin(actx, session, id2, params.XUserID); err != nil {
+			c.JSON(http.StatusUnauthorized, gin.H{"msg": fmt.Sprint("User does not have access to this person", err.Error())})
+
+			return
+		}
+	}
+
+	qctx, qCancel := context.WithTimeout(c.Request.Context(), srv.dbOpTimeout)
+	defer qCancel()
+
+	res, err := session.ExecuteRead(qctx, memgraph.GetRelationship(qctx, id1, id2))
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
+
+		return
+	}
+
+	c.JSON(http.StatusOK, res)
+}
+
+func (srv *server) DeleteRelationship(c *gin.Context, id1 int, id2 int, params api.DeleteRelationshipParams) {
+	session := srv.db.NewSession(c.Request.Context(), neo4j.SessionConfig{})
+	defer closeSession(c.Request.Context(), session, srv.dbOpTimeout)
+
+	actx, aCancel := context.WithTimeout(c.Request.Context(), srv.dbOpTimeout)
+	defer aCancel()
+
+	if err := auth.CouldManagePersonUnknownAdmin(actx, session, id1, params.XUserID); err != nil {
+		if err := auth.CouldManagePersonUnknownAdmin(actx, session, id2, params.XUserID); err != nil {
+			c.JSON(http.StatusUnauthorized, gin.H{"msg": fmt.Sprint("User does not have access to this person", err.Error())})
+
+			return
+		}
+	}
+
+	qctx, qCancel := context.WithTimeout(c.Request.Context(), srv.dbOpTimeout)
+	defer qCancel()
+
+	_, err := session.ExecuteWrite(qctx, memgraph.DeleteRelationship(qctx, id1, id2))
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"msg": err.Error()})
+
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"msg": "relationship deleted"})
 }