Add POST /user/{userId}/goals (#26)

Closes #4

Reviewed-on: #26

backend/api_test.go

@@ -2,8 +2,9 @@ package main
 
 import (
 	"testing"
+
+	"github.com/gavv/httpexpect/v2"
 )
-import "github.com/gavv/httpexpect/v2"
 
 func startServer(t *testing.T) *httpexpect.Expect {
 	config := ServerConfig{
@@ -56,3 +57,95 @@ func TestGetUserGoalsBadId(t *testing.T) {
 	e := startServer(t)
 	e.GET("/user/bad-id/goals").Expect().Status(400)
 }
+
+func TestCreateUserGoal(t *testing.T) {
+	e := startServer(t)
+
+	// Create a new goal
+	requestBody := map[string]interface{}{
+		"name":   "Test Goal",
+		"target": 5000,
+		"weight": 10,
+	}
+
+	response := e.POST("/user/1/goals").
+		WithJSON(requestBody).
+		Expect().
+		Status(201).
+		JSON().Object()
+
+	// Verify the response has an ID
+	response.ContainsKey("id")
+	goalId := response.Value("id").Number().Raw()
+
+	// Verify the goal exists in the list of goals
+	goals := e.GET("/user/1/goals").
+		Expect().
+		Status(200).
+		JSON().Array()
+
+	goals.Length().IsEqual(1)
+
+	goal := goals.Value(0).Object()
+	goal.Value("id").IsEqual(goalId)
+	goal.Value("name").IsEqual("Test Goal")
+	goal.Value("target").IsEqual(5000)
+	goal.Value("weight").IsEqual(10)
+	goal.Value("progress").IsEqual(0)
+}
+
+func TestCreateUserGoalNoUser(t *testing.T) {
+	e := startServer(t)
+
+	requestBody := map[string]interface{}{
+		"name":   "Test Goal",
+		"target": 5000,
+		"weight": 10,
+	}
+
+	e.POST("/user/999/goals").
+		WithJSON(requestBody).
+		Expect().
+		Status(404)
+}
+
+func TestCreateUserGoalInvalidInput(t *testing.T) {
+	e := startServer(t)
+
+	// Test with empty name
+	requestBody := map[string]interface{}{
+		"name":   "",
+		"target": 5000,
+		"weight": 10,
+	}
+
+	e.POST("/user/1/goals").
+		WithJSON(requestBody).
+		Expect().
+		Status(400)
+
+	// Test with missing fields
+	invalidRequest := map[string]interface{}{
+		"target": 5000,
+	}
+
+	e.POST("/user/1/goals").
+		WithJSON(invalidRequest).
+		Expect().
+		Status(400)
+}
+
+func TestCreateUserGoalBadId(t *testing.T) {
+	e := startServer(t)
+
+	requestBody := map[string]interface{}{
+		"name":   "Test Goal",
+		"target": 5000,
+		"weight": 10,
+	}
+
+	e.POST("/user/bad-id/goals").
+		WithJSON(requestBody).
+		Expect().
+		Status(400)
+}

backend/db.go

@@ -2,8 +2,9 @@ package main
 
 import (
 	"errors"
-	"gitea.seeseepuff.be/seeseemelk/mysqlite"
 	"log"
+
+	"gitea.seeseepuff.be/seeseemelk/mysqlite"
 )
 
 type Db struct {
@@ -86,3 +87,44 @@ func (db *Db) GetUserGoals(userId int) ([]Goal, error) {
 	}
 	return goals, nil
 }
+
+func (db *Db) CreateGoal(userId int, goal *CreateGoalRequest) (int, error) {
+	// Check if user exists before attempting to create a goal
+	exists, err := db.UserExists(userId)
+	if err != nil {
+		return 0, err
+	}
+	if !exists {
+		return 0, errors.New("user does not exist")
+	}
+
+	tx, err := db.db.Begin()
+	if err != nil {
+		return 0, err
+	}
+	defer tx.Rollback()
+
+	// Insert the new goal
+	err = tx.Query("insert into goals (user_id, name, target, progress, weight) values (?, ?, ?, 0, ?)").
+		Bind(userId, goal.Name, goal.Target, goal.Weight).
+		Exec()
+
+	if err != nil {
+		return 0, err
+	}
+
+	// Get the last inserted ID
+	var lastId int
+	err = tx.Query("select last_insert_rowid()").ScanSingle(&lastId)
+	if err != nil {
+		return 0, err
+	}
+
+	// Commit the transaction
+	err = tx.Commit()
+	if err != nil {
+		return 0, err
+	}
+
+	return lastId, nil
+}

backend/dto.go

@@ -24,3 +24,13 @@ type Goal struct {
 	Progress int    `json:"progress"`
 	Weight   int    `json:"weight"`
 }
+
+type CreateGoalRequest struct {
+	Name   string `json:"name"`
+	Target int    `json:"target"`
+	Weight int    `json:"weight"`
+}
+
+type CreateGoalResponse struct {
+	ID int `json:"id"`
+}

backend/main.go

@@ -4,11 +4,12 @@ import (
 	"context"
 	"embed"
 	"errors"
-	"github.com/gin-gonic/gin"
 	"log"
 	"net/http"
 	"os"
 	"strconv"
+
+	"github.com/gin-gonic/gin"
 )
 
 //go:embed migrations/*.sql
@@ -92,6 +93,46 @@ func getUserGoals(c *gin.Context) {
 	c.IndentedJSON(http.StatusOK, goals)
 }
 
+func createUserGoal(c *gin.Context) {
+	userIdStr := c.Param("userId")
+	userId, err := strconv.Atoi(userIdStr)
+	if err != nil {
+		log.Printf("Invalid user ID: %v", err)
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid user ID"})
+		return
+	}
+
+	// Parse request body
+	var goalRequest CreateGoalRequest
+	if err := c.ShouldBindJSON(&goalRequest); err != nil {
+		log.Printf("Error parsing request body: %v", err)
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request body"})
+		return
+	}
+
+	// Validate request
+	if goalRequest.Name == "" {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Goal name cannot be empty"})
+		return
+	}
+
+	// Create goal in database
+	goalId, err := db.CreateGoal(userId, &goalRequest)
+	if err != nil {
+		log.Printf("Error creating goal: %v", err)
+		if err.Error() == "user does not exist" {
+			c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
+		} else {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "Could not create goal"})
+		}
+		return
+	}
+
+	// Return created goal ID
+	response := CreateGoalResponse{ID: goalId}
+	c.IndentedJSON(http.StatusCreated, response)
+}
+
 /*
 *
 Initialises the database, and then starts the server.
@@ -105,6 +146,7 @@ func start(ctx context.Context, config *ServerConfig) {
 	router.GET("/api/users", getUsers)
 	router.GET("/api/user/:userId", getUser)
 	router.GET("/api/user/:userId/goals", getUserGoals)
+	router.POST("/api/user/:userId/goals", createUserGoal)
 
 	srv := &http.Server{
 		Addr:    ":" + config.Port,