[FEAT] Refactor API main file and models

This commit refactors the `main.go` file in the API directory, as well as the related models in the `models.go` file. The changes include:

- Reordering imports and removing unnecessary imports
- Fixing error messages to be more descriptive
- Handling database connections more efficiently with deferred closures
- Handling errors and returning appropriate error responses
- Adding proper JSON bindings for POST requests
- Adding new views in the database scripts for aggregated statistics and daily user statistics

No changes were made to imports and requires.

2 files changed, 138 insertions, 95 deletions

diff --git a/api/main.go b/api/main.go
index 57feb09..17a3c3a 100644
--- a/api/main.go
+++ b/api/main.go
@@ -1,18 +1,17 @@
 package main
 
 import (
-        "net/http"
         "crypto/rand"
-        "encoding/base64"
         "database/sql"
-        "strings"
+        "encoding/base64"
         "errors"
         "log"
+        "net/http"
+        "strings"
 
         "github.com/gin-gonic/gin"
         _ "github.com/mattn/go-sqlite3"
         "golang.org/x/crypto/bcrypt"
-        "water/api/lib"
 )
 
 func CORSMiddleware() gin.HandlerFunc {
@@ -22,15 +21,12 @@ func CORSMiddleware() gin.HandlerFunc {
                 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")
 
-                log.Println("I am here")
-
                 if c.Request.Method == "OPTIONS" {
                         log.Println(c.Request.Header)
-                        c.AbortWithStatus(204)
+                        c.AbortWithStatus(http.StatusNoContent)
                         return
                 }
 
-                log.Println(c.Request.Header)
                 c.Next()
         }
 }
@@ -38,7 +34,10 @@ func CORSMiddleware() gin.HandlerFunc {
 // generatToken will g
 func generateToken() string {
         token := make([]byte, 32)
-        rand.Read(token)
+        _, err := rand.Read(token)
+        if err != nil {
+                return ""
+        }
         return base64.StdEncoding.EncodeToString(token)
 }
 
@@ -53,13 +52,13 @@ func establishDBConnection() *sql.DB {
 func checkForTokenInContext(c *gin.Context) (string, error) {
         authorizationHeader := c.GetHeader("Authorization")
         if authorizationHeader == "" {
-                return "", errors.New("Authorization header is missing")
+                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 "", errors.New("invalid Authorization header format")
         }
 
         return parts[1], nil
@@ -98,15 +97,21 @@ func setupRouter() *gin.Engine {
                 }
 
                 db := establishDBConnection()
-                defer db.Close()
+                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
-                var size models.Size
+                var user User
+                var preference Preference
+                var size Size
 
                 row := db.QueryRow("SELECT name, uuid, password, color, size, unit FROM Users u INNER JOIN Preferences p ON p.user_id = u.id INNER JOIN Sizes s ON p.size_id = s.id WHERE u.name = ?", username)
                 if err := row.Scan(&user.Name, &user.UUID, &user.Password, &preference.Color, &size.Size, &size.Unit); err != nil {
-                        if err == sql.ErrNoRows {
+                        if errors.Is(err, sql.ErrNoRows) {
                                 c.AbortWithStatus(http.StatusUnauthorized)
                                 return
                         }
@@ -129,128 +134,147 @@ func setupRouter() *gin.Engine {
         {
                 stats.GET("/", func(c *gin.Context) {
                         db := establishDBConnection()
-                        defer db.Close()
+                        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(500, gin.H{"error": err.Error()})
+                                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
                                 return
                         }
-                        defer rows.Close()
+                        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
+                        var data []Statistic
 
                         for rows.Next() {
-                                var stat models.Statistic
-                                var user models.User
+                                var stat Statistic
+                                var user User
                                 if err := rows.Scan(&stat.Date, &stat.Quantity, &user.UUID, &user.Name); err != nil {
-                                        c.JSON(500, gin.H{"error": err.Error()})
+                                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
                                         return
                                 }
                                 stat.User = user
                                 data = append(data, stat)
                         }
 
+                        c.JSON(http.StatusOK, data)
+                })
 
-                        // TODO: return to this and figure out how to best collect the data you are looking for for each user (zach and parker)
-                        rows, err = db.Query("SELECT date(s.date), SUM(s.quantity) as total FROM Statistics s WHERE s.date >= date('now', '-7 days') GROUP BY DATE(s.date)")
-                        if err != nil {
-                                c.JSON(500, gin.H{"error": err.Error()})
+                stats.POST("/", func(c *gin.Context) {
+                        var stat StatisticPost
+
+                        if err := c.BindJSON(&stat); err != nil {
+                                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
                                 return
                         }
-                        defer rows.Close()
 
-                        var dailySummaries []models.DailySummary
-                        for rows.Next() {
-                                var summary models.DailySummary
-                                if err := rows.Scan(&summary.Date, &summary.Total); err != nil {
-                                        c.JSON(500, gin.H{"error": err.Error()})
+                        db := establishDBConnection()
+                        defer func(db *sql.DB) {
+                                err := db.Close()
+                                if err != nil {
+                                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
                                         return
                                 }
-                                dailySummaries = append(dailySummaries, summary)
-                        }
+                        }(db)
 
-                        c.JSON(http.StatusOK, gin.H{"stats": data, "totals": dailySummaries})
-                        rows, err = db.Query("SELECT s.date, SUM(s.quantity) as total, u.uuid, u.name FROM Statistics s INNER JOIN Users u ON u.id = s.user_id WHERE s.date >= date('now', '-7 days') GROUP BY s.date, s.user_id")
+                        result, err := db.Exec("INSERT INTO statistics (date, user_id, quantity) values (?, ?, ?)", stat.Date, stat.UserID, stat.Quantity)
 
                         if err != nil {
-                                c.JSON(500, gin.H{"error": err.Error()})
-                                return
+                                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
                         }
-                        defer rows.Close()
 
-                        var totals []interface{}
-                        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(500, gin.H{"error": err.Error()})
-                                        return
-                                }
-                                stat.User = user
-                                totals = append(totals, stat)
+                        id, err := result.LastInsertId()
+                        if err != nil {
+                                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
                         }
 
-                        c.JSON(http.StatusOK, gin.H{"stats": data, "totals": totals})
+                        c.JSON(http.StatusCreated, gin.H{"status": "created", "id": id})
                 })
 
-                stats.POST("/", func(c *gin.Context) {
-                        var stat models.Statistic
+                stats.GET("weekly/", func(c *gin.Context) {
+                        db := establishDBConnection()
+                        defer func(db *sql.DB) {
+                                err := db.Close()
+                                if err != nil {
+                                        c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                                        return
+                                }
+                        }(db)
 
-                        if err := c.BindJSON(&stat); err != nil {
-                                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+                        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 []WeeklyStatistic
+                        for rows.Next() {
+                                var weeklyStat 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)
+                })
 
+                stats.GET("totals/", func(c *gin.Context) {
                         db := establishDBConnection()
-                        defer db.Close()
+                        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, 1, stat.Quantity)
+                        rows, err := db.Query("SELECT name, total FROM DailyUserStatistics")
 
                         if err != nil {
-                                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+                                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)
 
-                        id, err := result.LastInsertId()
-                        if err != nil {
-                                c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+                        var data []DailyUserTotals
+                        for rows.Next() {
+                                var stat 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.StatusCreated, gin.H{"status": "created", "id": id})
-                })
+                        c.JSON(http.StatusOK, data)
 
-                stats.GET("/totals/", func(c *gin.Context) {
-                        c.JSON(http.StatusOK, gin.H{"status": "ok"})
                 })
 
-                //              stats.GET("/totals/", func(c *gin.Context) {
-                //                      db := establishDBConnection()
-                //                      defer db.Close()
-                //
-                //                      rows, err := db.Query("SELECT s.date, SUM(s.quantity) as total, u.uuid, u.name FROM Statistics s INNER JOIN Users u ON u.id = s.user_id WHERE s.date >= date('now', '-7 days') GROUP BY s.date, s.user_id")
-                //
-                //                      if err != nil {
-                //                              c.JSON(500, gin.H{"error": err.Error()})
-                //                              return
-                //                      }
-                //                      defer rows.Close()
-                //
-                //                      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(500, gin.H{"error": err.Error()})
-                //                                      return
-                //                              }
-                //                              stat.User = user
-                //                              data = append(data, stat)
-                //                      }
-                //
-                //                      c.JSON(http.StatusOK, data)
-                //
-                //              })
-
                 stats.GET("user/:uuid", func(c *gin.Context) {
                         c.JSON(http.StatusOK, gin.H{"status": "ok", "uuid": c.Param("uuid")})
                 })
@@ -270,5 +294,8 @@ func setupRouter() *gin.Engine {
 func main() {
         r := setupRouter()
         // Listen and Server in 0.0.0.0:8080
-        r.Run(":8080")
+        err := r.Run(":8080")
+        if err != nil {
+                return
+        }
 }
diff --git a/api/lib/models.go b/api/models.go
index f959519..0845d1d 100644
--- a/api/lib/models.go
+++ b/api/models.go
@@ -1,4 +1,4 @@
-package models
+package main
 
 import (
     "time"
@@ -12,6 +12,12 @@ type Statistic struct {
     Quantity int `json:"quantity"`
 }
 
+type StatisticPost struct {
+    Date time.Time `json:"date"`
+    Quantity int64 `json:"quantity"`
+    UserID int64 `json:"user_id"`
+}
+
 type User struct {
     ID int64 `json:"-"`
     Name string `json:"name"`
@@ -39,3 +45,13 @@ type Size struct {
     Size int64 `json:"size"`
     Unit string `json:"unit"`
 }
+
+type WeeklyStatistic struct {
+    Date string `json:"date"`
+    Total int64 `json:"total"`
+}
+
+type DailyUserTotals struct {
+    Name string `json:"name"`
+    Total int64 `json:"total"`
+}
\ No newline at end of file