view family tree

backend/handlers/viewFamilyTree.go

@@ -1 +1,55 @@
 package handlers
+
+import (
+	"context"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+)
+
+func ViewFamiliyTree(driver neo4j.DriverWithContext) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		defer cancel()
+
+		session := driver.NewSession(ctx, neo4j.SessionConfig{AccessMode: neo4j.AccessModeRead})
+		defer session.Close(ctx)
+
+		id := c.Query("id")
+		if id == "" {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "id is required"})
+
+			return
+		}
+
+		query := `
+		MATCH (n:Person {ID: $person_id})-[p:Parent*1..]->(family:Person)
+		OPTIONAL MATCH (family)-[c:Child]->(children:Person)
+		WITH family, p, children, c, n
+		OPTIONAL MATCH (children)<-[p2:Parent]-(OtherParents:Person)
+		WITH family, p, children, c, OtherParents, p2,n
+		OPTIONAL MATCH (family)-[s:Spouse]-(spouse:Person)
+		RETURN family, p, children, c, OtherParents, p2, spouse, s, n`
+
+		result, err := session.Run(ctx, query, map[string]any{"person_id": id})
+		if err != nil {
+			log.Printf("ip: %s error: %s", c.ClientIP(), err)
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "internal server error"})
+
+			return
+		}
+
+		rec, err := result.Single(ctx)
+		if err != nil {
+			log.Printf("ip: %s error: %s", c.ClientIP(), err)
+			c.JSON(http.StatusNotFound, gin.H{"error": "could not find family tree for person with id: " + id})
+
+			return
+		}
+
+		c.JSON(200, rec.AsMap()["n"])
+	}
+}

backend/main.go

@@ -52,6 +52,7 @@ func main() {
 	router.DELETE("/relationship", handlers.DeleteRelationship(memgraphDriver))
 	router.PUT("/relationship", handlers.VerifyRelationship(memgraphDriver))
 	router.POST("/createRelationshipAndPerson", handlers.CreateRelationshipAndPerson(memgraphDriver))
+	router.GET("/familyTree", handlers.ViewFamiliyTree(memgraphDriver))
 
 	server := utilities.SetupHttpsServer(router, *cert, *key, *httpsPort, *httpPort, requestTimeout)