From f5e95095c7b346cc2f12ddb20129d785be81755f Mon Sep 17 00:00:00 2001
From: Vargha Csongor <Culisa1023@gmail.com>
Date: Mon, 15 Apr 2024 23:46:11 +0200
Subject: [PATCH] create relationship

---
 backend/handlers/createRelationship.go | 32 +++++++++++++++++++++
 backend/memgraph/createRelationship.go | 39 ++++++++++++++++++++++++++
 2 files changed, 71 insertions(+)
 create mode 100644 backend/handlers/createRelationship.go
 create mode 100644 backend/memgraph/createRelationship.go

diff --git a/backend/handlers/createRelationship.go b/backend/handlers/createRelationship.go
new file mode 100644
index 0000000..71b28eb
--- /dev/null
+++ b/backend/handlers/createRelationship.go
@@ -0,0 +1,32 @@
+package handlers
+
+import (
+	"log"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+	"github.com/vcscsvcscs/GenerationsHeritage/backend/memgraph"
+)
+
+func CreateRelationship(driver neo4j.DriverWithContext) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		var relationship memgraph.Relationship
+		if err := c.ShouldBindJSON(&relationship); err != nil {
+			log.Printf("ip: %s error: %s", c.ClientIP(), err)
+			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+
+			return
+		}
+
+		rec, err := relationship.CreateRelationship(driver)
+		if err != nil {
+			log.Printf("ip: %s error: %s", c.ClientIP(), err)
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "internal server error"})
+
+			return
+		}
+
+		c.JSON(http.StatusCreated, gin.H{"relationship": rec.AsMap()})
+	}
+}
diff --git a/backend/memgraph/createRelationship.go b/backend/memgraph/createRelationship.go
new file mode 100644
index 0000000..8b7de6f
--- /dev/null
+++ b/backend/memgraph/createRelationship.go
@@ -0,0 +1,39 @@
+package memgraph
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+	"golang.org/x/net/context"
+)
+
+func (r *Relationship) CreateRelationship(driver neo4j.DriverWithContext) (*neo4j.Record, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	session := driver.NewSession(ctx, neo4j.SessionConfig{AccessMode: neo4j.AccessModeRead})
+	defer session.Close(ctx)
+
+	if err := r.Verify(); err != nil {
+		return nil, err
+	}
+
+	query := fmt.Sprintf(`MATCH (a:Person), (b:Person)
+	WHERE a.ID = %s AND b.ID = %s`, r.FirstPersonID, r.SecondPersonID)
+
+	if r.Direction == "->" {
+		query = fmt.Sprintf(`%s CREATE (a)-[r:%s {verified: false}]->(b) RETURN r;`, query, r.Relationship)
+	} else if r.Direction == "<-" {
+		query = fmt.Sprintf(`%s CREATE (a)<-[r:%s {verified: false}]-(b) RETURN r;`, query, r.Relationship)
+	} else {
+		query = fmt.Sprintf(`%s CREATE (a)-[r:%s {verified: false}]->(b) RETURN r;`, query, r.Relationship)
+	}
+
+	result, err := session.Run(ctx, query, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return result.Single(ctx)
+}