diff --git a/backend/handlers/updatePerson.go b/backend/handlers/updatePerson.go index 5ac8282..f543df7 100644 --- a/backend/handlers/updatePerson.go +++ b/backend/handlers/updatePerson.go @@ -1 +1,48 @@ package handlers + +import ( + "encoding/json" + "log" + "net/http" + + "github.com/gin-gonic/gin" + "github.com/neo4j/neo4j-go-driver/v5/neo4j" + "github.com/vcscsvcscs/GenerationsHeritage/backend/memgraph" +) + +func UpdatePerson(driver neo4j.DriverWithContext) gin.HandlerFunc { + return func(c *gin.Context) { + if c.Request.Body == nil || c.ContentType() != "application/json" { + log.Printf("ip: %s error: request body is empty or content type is not application/json", c.ClientIP()) + c.JSON(http.StatusBadRequest, gin.H{"error": "content type must be application/json and request body must not be empty"}) + + return + } + + var person memgraph.Person + err := json.NewDecoder(c.Request.Body).Decode(&person) + if err != nil { + log.Printf("ip: %s error: %s", c.ClientIP(), err) + c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request body"}) + + return + } + + if person.ID != "" { + log.Printf("ip: %s error: %s", c.ClientIP(), err) + c.JSON(http.StatusBadRequest, gin.H{"error": "no person ID provided"}) + + return + } + + rec, err := person.UpdatePerson(driver) + if err != nil { + log.Printf("ip: %s error: %s", c.ClientIP(), err) + c.JSON(http.StatusNotFound, gin.H{"error": "could not update person with information provided"}) + + return + } + + c.JSON(http.StatusOK, gin.H{"person": rec.AsMap()}) + } +} diff --git a/backend/memgraph/update_person.go b/backend/memgraph/update_person.go new file mode 100644 index 0000000..27dcba8 --- /dev/null +++ b/backend/memgraph/update_person.go @@ -0,0 +1,30 @@ +package memgraph + +import ( + "fmt" + "time" + + "github.com/neo4j/neo4j-go-driver/v5/neo4j" + "golang.org/x/net/context" +) + +func (p *Person) UpdatePerson(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.AccessModeWrite}) + defer session.Close(ctx) + + if err := p.Verify(); err != nil { + return nil, err + } + + query := fmt.Sprintf("MATCH (n:Person {ID: '%s'}) SET n += {%s} RETURN n;", p.ID, p.ToString()) + + result, err := session.Run(ctx, query, nil) + if err != nil { + return nil, err + } + + return result.Single(ctx) +}