1 files changed, 168 insertions, 0 deletions

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"})
+}