12 files changed, 561 insertions, 0 deletions

diff --git a/api/internal/config/config.go b/api/internal/config/config.go
new file mode 100644
index 0000000..d54e40e
--- /dev/null
+++ b/api/internal/config/config.go
@@ -0,0 +1,17 @@
+package config
+
+import (
+        "fmt"
+        "github.com/spf13/viper"
+)
+
+func Load() (*viper.Viper, error) {
+        v := viper.New()
+        v.SetConfigFile(".env")
+        v.AddConfigPath(".")
+        err := v.ReadInConfig()
+        if err != nil {
+                return nil, fmt.Errorf("error reading .env file: %s", err)
+        }
+        return v, nil
+}
diff --git a/api/internal/controllers/auth.go b/api/internal/controllers/auth.go
new file mode 100644
index 0000000..ab2fbbb
--- /dev/null
+++ b/api/internal/controllers/auth.go
@@ -0,0 +1,79 @@
+package controllers
+
+import (
+        "crypto/rand"
+        "database/sql"
+        "encoding/base64"
+        "errors"
+        "github.com/gin-gonic/gin"
+        "net/http"
+        "water/api/internal/models"
+
+        _ "github.com/mattn/go-sqlite3"
+        "golang.org/x/crypto/bcrypt"
+        "water/api/internal/database"
+)
+
+
+
+// AuthHandler is a function that handles users' authentication. It checks if the request
+// has valid credentials, authenticates the user and sets the user's session.
+// If the authentication is successful, it will allow the user to access protected routes.
+func AuthHandler (c *gin.Context) {
+        username, password, ok := c.Request.BasicAuth()
+        if !ok {
+                c.Header("WWW-Authenticate", `Basic realm="Please enter your username and password."`)
+                c.AbortWithStatus(http.StatusUnauthorized)
+                return
+        }
+
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        var user models.User
+        var preference models.Preference
+
+        row := db.QueryRow("SELECT id as 'id', name, uuid, password FROM Users WHERE name = ?", username)
+        if err := row.Scan(&user.ID, &user.Name, &user.UUID, &user.Password); err != nil {
+                if errors.Is(err, sql.ErrNoRows) {
+                        c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
+                        return
+                }
+        }
+
+        row = db.QueryRow("SELECT id, color, size_id, user_id FROM Preferences where user_id = ?", user.ID)
+        if err := row.Scan(&preference.ID, &preference.Color, &preference.SizeID, &preference.UserID); err != nil {
+                if errors.Is(err, sql.ErrNoRows) {
+                        c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
+                }
+        }
+
+        if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password)); err != nil {
+                c.JSON(http.StatusUnauthorized, gin.H{"error": err.Error()})
+                return
+        }
+
+        // Generate a simple API token
+        apiToken := generateToken()
+        c.JSON(http.StatusOK, gin.H{"token": apiToken, "user": user, "preferences": preference})
+}
+
+
+// generateToken is a helper function used in the AuthHandler. It generates a random token for API authentication.
+// This function creates an empty byte slice of length 32 and fills it with cryptographic random data using the rand.Read function.
+// If an error occurs during the generation, it will return an empty string.
+// The generated cryptographic random data is then encoded into a base64 string and returned.
+func generateToken() string {
+        token := make([]byte, 32)
+        _, err := rand.Read(token)
+        if err != nil {
+                return ""
+        }
+        return base64.StdEncoding.EncodeToString(token)
+}
\ No newline at end of file
diff --git a/api/internal/controllers/preferences.go b/api/internal/controllers/preferences.go
new file mode 100644
index 0000000..a1bcf4f
--- /dev/null
+++ b/api/internal/controllers/preferences.go
@@ -0,0 +1,45 @@
+package controllers
+
+import (
+        "github.com/gin-gonic/gin"
+        "net/http"
+        "database/sql"
+        "water/api/internal/database"
+        "water/api/internal/models"
+)
+
+func GetSizes(c *gin.Context) {
+                db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                }
+        }(db)
+        
+        rows, err := db.Query("SELECT id, size, unit  FROM Sizes")
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
+        defer func(rows *sql.Rows) {
+                err := rows.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(rows)
+
+        var data []models.Size
+
+        for rows.Next() {
+                var size models.Size
+                if err := rows.Scan(&size.ID, &size.Size, &size.Unit); err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+                data = append(data, size)
+        }
+
+        c.JSON(http.StatusOK, data)
+}
diff --git a/api/internal/controllers/stats.go b/api/internal/controllers/stats.go
new file mode 100644
index 0000000..2234787
--- /dev/null
+++ b/api/internal/controllers/stats.go
@@ -0,0 +1,168 @@
+package controllers
+
+import (
+        "database/sql"
+        "github.com/gin-gonic/gin"
+        "net/http"
+        "water/api/internal/database"
+        "water/api/internal/models"
+)
+
+// TODO: add comments to all exported members of package.
+
+// GetAllStatistics connects to the database and queries for all statistics in the database.
+// If none have been found it will return an error, otherwise a 200 code is sent along with the list of statistics.
+func GetAllStatistics(c *gin.Context) {
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        rows, err := db.Query("SELECT s.date, s.quantity, u.uuid, u.name FROM Statistics s INNER JOIN Users u ON u.id = s.user_id")
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
+        defer func(rows *sql.Rows) {
+                err := rows.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(rows)
+
+        var data []models.Statistic
+
+        for rows.Next() {
+                var stat models.Statistic
+                var user models.User
+                if err := rows.Scan(&stat.Date, &stat.Quantity, &user.UUID, &user.Name); err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+                stat.User = user
+                data = append(data, stat)
+        }
+
+        c.JSON(http.StatusOK, data)
+}
+
+func PostNewStatistic(c *gin.Context) {
+        var stat models.StatisticPost
+
+        if err := c.BindJSON(&stat); err != nil {
+                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+                return
+        }
+
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        result, err := db.Exec("INSERT INTO statistics (date, user_id, quantity) values (?, ?, ?)", stat.Date, stat.UserID, stat.Quantity)
+
+        if err != nil {
+                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+        }
+
+        id, err := result.LastInsertId()
+        if err != nil {
+                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+        }
+
+        c.JSON(http.StatusCreated, gin.H{"status": "created", "id": id})
+}
+
+func GetWeeklyStatistics(c *gin.Context) {
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        rows, err := db.Query("SELECT date, total FROM `WeeklyStatisticsView`")
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
+        defer func(rows *sql.Rows) {
+                err := rows.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(rows)
+
+        var data []models.WeeklyStatistic
+        for rows.Next() {
+                var weeklyStat models.WeeklyStatistic
+                if err := rows.Scan(&weeklyStat.Date, &weeklyStat.Total); err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                }
+                data = append(data, weeklyStat)
+        }
+
+        c.JSON(http.StatusOK, data)
+}
+
+func GetDailyUserStatistics(c *gin.Context) {
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        rows, err := db.Query("SELECT name, total FROM DailyUserStatistics")
+
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
+        defer func(rows *sql.Rows) {
+                err := rows.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(rows)
+
+        var data []models.DailyUserTotals
+        for rows.Next() {
+                var stat models.DailyUserTotals
+                if err := rows.Scan(&stat.Name, &stat.Total); err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+                data = append(data, stat)
+        }
+
+        c.JSON(http.StatusOK, data)
+
+}
+
+func GetUserStatistics(c *gin.Context) {
+        c.JSON(http.StatusOK, gin.H{"status": "ok", "uuid": c.Param("uuid")})
+}
+
+func UpdateUserStatistic(c *gin.Context) {
+        c.JSON(http.StatusNoContent, gin.H{"status": "No Content"})
+}
+
+func DeleteUserStatistic(c *gin.Context) {
+        c.JSON(http.StatusNoContent, gin.H{"status": "No Content"})
+}
diff --git a/api/internal/controllers/user.go b/api/internal/controllers/user.go
new file mode 100644
index 0000000..dbb09cf
--- /dev/null
+++ b/api/internal/controllers/user.go
@@ -0,0 +1,68 @@
+package controllers
+
+import (
+        "database/sql"
+        "errors"
+        "log"
+        "net/http"
+        "water/api/internal/database"
+        "water/api/internal/models"
+
+        "github.com/gin-gonic/gin"
+)
+
+func GetUser(c *gin.Context) {
+        c.JSON(http.StatusOK, gin.H{"message": "User found"})
+}
+func GetUserPreferences(c *gin.Context) {
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        var preference models.Preference
+
+        row := db.QueryRow("SELECT id, user_id, color, size_id FROM Preferences WHERE user_id = ?", c.Param("id"))
+        if err := row.Scan(&preference.ID, &preference.UserID, &preference.Color, &preference.SizeID); err != nil {
+                if errors.Is(err, sql.ErrNoRows) {
+                        c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
+                        return
+                } else {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }
+
+        c.JSON(http.StatusOK, preference)
+}
+
+func UpdateUserPreferences(c *gin.Context) {
+        db := database.EstablishDBConnection()
+        defer func(db *sql.DB) {
+                err := db.Close()
+                if err != nil {
+                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                        return
+                }
+        }(db)
+
+        var newPreferences models.Preference
+        if err := c.BindJSON(&newPreferences); err != nil {
+                c.JSON(http.StatusUnprocessableEntity, gin.H{"error": err.Error()})
+                return
+        }
+
+        log.Printf("newPreferences: %v", newPreferences)
+
+        _, err := db.Exec("UPDATE Preferences SET color = ?, size_id = ? WHERE id = ?", newPreferences.Color, newPreferences.SizeID, newPreferences.ID)
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
+
+        c.Status(http.StatusNoContent)
+}
diff --git a/api/internal/database/database.go b/api/internal/database/database.go
new file mode 100644
index 0000000..1866655
--- /dev/null
+++ b/api/internal/database/database.go
@@ -0,0 +1,24 @@
+package database
+
+import (
+        "database/sql"
+        _ "github.com/mattn/go-sqlite3"
+        "log"
+        "path/filepath"
+        "water/api/internal/config"
+)
+
+func EstablishDBConnection() *sql.DB {
+        c, err := config.Load()
+
+        driver := c.GetString("DB_DRIVER")
+        path, err := filepath.Abs(c.GetString("DB_PATH"))
+        if err != nil {
+                log.Fatal("There was and error getting the absolute path of the database.")
+        }
+        db, err := sql.Open(driver, path)
+        if err != nil {
+                panic(err)
+        }
+        return db
+}
\ No newline at end of file
diff --git a/api/internal/middleware/middleware.go b/api/internal/middleware/middleware.go
new file mode 100644
index 0000000..aa27fb8
--- /dev/null
+++ b/api/internal/middleware/middleware.go
@@ -0,0 +1,56 @@
+package middleware
+
+import (
+        "errors"
+        "log"
+        "net/http"
+        "strings"
+
+        "github.com/gin-gonic/gin"
+)
+
+func TokenRequired() gin.HandlerFunc {
+        return func(c *gin.Context) {
+                _, err := checkForTokenInContext(c)
+
+                if err != nil {
+                        c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
+                        c.Abort()
+                        return
+                }
+
+                c.Next()
+        }
+}
+
+func CORSMiddleware() gin.HandlerFunc {
+        return func(c *gin.Context) {
+                c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
+                c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
+                c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
+                c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS, GET, PUT, PATCH")
+
+                if c.Request.Method == "OPTIONS" {
+                        log.Println(c.Request.Header)
+                        c.AbortWithStatus(http.StatusNoContent)
+                        return
+                }
+
+                c.Next()
+        }
+}
+
+func checkForTokenInContext(c *gin.Context) (string, error) {
+        authorizationHeader := c.GetHeader("Authorization")
+        if authorizationHeader == "" {
+                return "", errors.New("authorization header is missing")
+        }
+
+        parts := strings.Split(authorizationHeader, " ")
+
+        if len(parts) != 2 || parts[0] != "Bearer" {
+                return "", errors.New("invalid Authorization header format")
+        }
+
+        return parts[1], nil
+}
diff --git a/api/internal/models/auth.go b/api/internal/models/auth.go
new file mode 100644
index 0000000..fa7dbe4
--- /dev/null
+++ b/api/internal/models/auth.go
@@ -0,0 +1,11 @@
+package models
+
+import "time"
+
+type Token struct {
+        ID        int64     `json:"id"`
+        UserID    int64     `json:"user_id"`
+        Token     string    `json:"token"`
+        CreatedAt time.Time `json:"created_at"`
+        ExpiredAt time.Time `json:"expired_at"`
+}
diff --git a/api/internal/models/preferences.go b/api/internal/models/preferences.go
new file mode 100644
index 0000000..8022099
--- /dev/null
+++ b/api/internal/models/preferences.go
@@ -0,0 +1,14 @@
+package models
+
+type Preference struct {
+        ID     int64  `json:"id"`
+        Color  string `json:"color"`
+        UserID int64  `json:"user_id"`
+        SizeID   int64   `json:"size_id"`
+}
+
+type Size struct {
+        ID   int64  `json:"id"`
+        Size int64  `json:"size"`
+        Unit string `json:"unit"`
+}
diff --git a/api/internal/models/statistics.go b/api/internal/models/statistics.go
new file mode 100644
index 0000000..7dceb3a
--- /dev/null
+++ b/api/internal/models/statistics.go
@@ -0,0 +1,26 @@
+package models
+
+import "time"
+
+type Statistic struct {
+        ID       int64     `json:"id"`
+        Date     time.Time `json:"date"`
+        User     User      `json:"user"`
+        Quantity int       `json:"quantity"`
+}
+
+type StatisticPost struct {
+        Date     time.Time `json:"date"`
+        Quantity int64     `json:"quantity"`
+        UserID   int64     `json:"user_id"`
+}
+
+type WeeklyStatistic struct {
+        Date  string `json:"date"`
+        Total int64  `json:"total"`
+}
+
+type DailyUserTotals struct {
+        Name  string `json:"name"`
+        Total int64  `json:"total"`
+}
diff --git a/api/internal/models/user.go b/api/internal/models/user.go
new file mode 100644
index 0000000..ca5daa4
--- /dev/null
+++ b/api/internal/models/user.go
@@ -0,0 +1,10 @@
+package models
+
+import "github.com/google/uuid"
+
+type User struct {
+        ID       int64     `json:"id"`
+        Name     string    `json:"name"`
+        UUID     uuid.UUID `json:"uuid"`
+        Password string    `json:"-"`
+}
diff --git a/api/internal/router/router.go b/api/internal/router/router.go
new file mode 100644
index 0000000..a71c3e6
--- /dev/null
+++ b/api/internal/router/router.go
@@ -0,0 +1,43 @@
+package router
+
+import (
+        "github.com/gin-gonic/gin"
+        "water/api/internal/controllers"
+        "water/api/internal/middleware"
+)
+
+func SetupRouter() *gin.Engine {
+        // Disable Console Color
+        //      gin.DisableConsoleColor()
+        r := gin.Default()
+        r.Use(middleware.CORSMiddleware())
+        r.Use(gin.Logger())
+        r.Use(gin.Recovery())
+
+        api := r.Group("api/v1")
+
+        api.POST("/auth", controllers.AuthHandler)
+        api.GET("/sizes", middleware.TokenRequired(), controllers.GetSizes)
+        api.PATCH("/user/preferences", controllers.UpdateUserPreferences)
+
+        user := api.Group("/user/:id")
+        user.Use(middleware.TokenRequired())
+        {
+                user.GET("", controllers.GetUser)
+                user.GET("preferences", controllers.GetUserPreferences)
+        }
+
+        stats := api.Group("/stats")
+        stats.Use(middleware.TokenRequired())
+        {
+                stats.GET("", controllers.GetAllStatistics)
+                stats.POST("", controllers.PostNewStatistic)
+                stats.GET("weekly", controllers.GetWeeklyStatistics)
+                stats.GET("daily", controllers.GetDailyUserStatistics)
+                stats.GET("user/:uuid", controllers.GetUserStatistics)
+                stats.PATCH("user/:uuid", controllers.UpdateUserStatistic)
+                stats.DELETE("user/:uuid", controllers.DeleteUserStatistic)
+        }
+
+        return r
+}
\ No newline at end of file