91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
package workout
|
|
|
|
import (
|
|
"api/middleware"
|
|
"api/utils"
|
|
"database/sql"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"firebase.google.com/go/auth"
|
|
)
|
|
|
|
func NewWorkout(db *sql.DB) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
var dateStr = r.FormValue("date")
|
|
var typeStr = r.FormValue("type")
|
|
var setsStr = r.FormValue("sets")
|
|
var repsStr = r.FormValue("reps")
|
|
|
|
if dateStr == "" || typeStr == "" || setsStr == "" || repsStr == "" {
|
|
http.Error(w, "Missing required fields", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
date, err := time.Parse("2006-01-02", dateStr)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
sets, err := strconv.Atoi(setsStr)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
reps, err := strconv.Atoi(repsStr)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
token := r.Context().Value(middleware.TOKEN_KEY).(*auth.Token)
|
|
|
|
_, err = db.Exec("INSERT INTO workout (user_id, date, type, sets, reps) VALUES (?, ?, ?, ?, ?)", token.UID, date, typeStr, sets, reps)
|
|
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetWorkouts(db *sql.DB) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
token := r.Context().Value(middleware.TOKEN_KEY).(*auth.Token)
|
|
var userId = token.UID
|
|
|
|
rows, err := db.Query("SELECT date, type, sets, reps FROM workout WHERE user_id = ?", userId)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
var workouts = make([]map[string]interface{}, 0)
|
|
for rows.Next() {
|
|
var date string
|
|
var workoutType string
|
|
var sets int
|
|
var reps int
|
|
|
|
err = rows.Scan(&date, &workoutType, &sets, &reps)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
workout := map[string]interface{}{
|
|
"date": date,
|
|
"type": workoutType,
|
|
"sets": sets,
|
|
"reps": reps,
|
|
}
|
|
workouts = append(workouts, workout)
|
|
}
|
|
|
|
utils.WriteJSON(w, workouts)
|
|
}
|
|
}
|