safe add handler

controllers/controlSafes.go

@@ -2,13 +2,20 @@ package controllers
 
 import (
 	"fmt"
+	"html"
 	"log"
 	"net/http"
 	"smt/models"
+	"strings"
 
 	"github.com/gin-gonic/gin"
 )
 
+type SafeInput struct {
+	SafeId   int    `db:"SafeId" json:"safeId"`
+	SafeName string `db:"SafeName" json:"safeName"`
+}
+
 // GetSafesHandler provides a list of all safes that a user has access to
 func GetSafesHandler(c *gin.Context) {
 	var UserId int
@@ -44,3 +51,46 @@ func GetAllSafesHandler(c *gin.Context) {
 
 	c.JSON(http.StatusOK, gin.H{"message": "success", "data": safes})
 }
+
+func AddSafeHandler(c *gin.Context) {
+	var input SafeInput
+
+	if err := c.ShouldBindJSON(&input); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	if len(input.SafeName) == 0 {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "no safe name specified"})
+		return
+	}
+
+	s := models.Safe{SafeId: input.SafeId, SafeName: input.SafeName}
+
+	//remove leading/trailing spaces in groupname
+	s.SafeName = html.EscapeString(strings.TrimSpace(s.SafeName))
+
+	// Check if safe already exists
+	testSafe, _ := models.SafeGetByName(s.SafeName)
+	log.Printf("AddSafeHandler checking if safe '%s' already exists\n", s.SafeName)
+
+	if (models.Safe{} == testSafe) {
+		log.Printf("AddSafeHandler confirmed no existing safe name\n")
+	} else {
+		errorString := fmt.Sprintf("attempt to register conflicting safe '%s'", s.SafeName)
+		log.Printf("Register error : '%s'\n", errorString)
+		c.JSON(http.StatusBadRequest, gin.H{"error": errorString})
+		return
+	}
+
+	_, err := s.SafeAdd()
+
+	if err != nil {
+		errString := fmt.Sprintf("error creating safe : '%s'", err)
+		log.Printf("AddSafeHandler %s\n", errString)
+		c.JSON(http.StatusBadRequest, gin.H{"error": errString})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "safe creation success"})
+}