feat: add complete workout management system with file parsing
This commit is contained in:
391
internal/workout/handler.go
Normal file
391
internal/workout/handler.go
Normal file
@@ -0,0 +1,391 @@
|
||||
package workout
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"rideaware/internal/config"
|
||||
"rideaware/internal/middleware"
|
||||
)
|
||||
|
||||
type Handler struct {
|
||||
service *Service
|
||||
}
|
||||
|
||||
func NewHandler() *Handler {
|
||||
return &Handler{
|
||||
service: NewService(),
|
||||
}
|
||||
}
|
||||
|
||||
// CreateWorkout POST /api/protected/workouts
|
||||
func (h *Handler) CreateWorkout(w http.ResponseWriter, r *http.Request) {
|
||||
log.Printf("CreateWorkout handler called")
|
||||
claims := r.Context().Value(middleware.UserContextKey).(*config.CustomClaims)
|
||||
|
||||
if claims == nil {
|
||||
log.Printf("Claims is nil")
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusUnauthorized)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "unauthorized"})
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("UserID: %d", claims.UserID)
|
||||
|
||||
var req struct {
|
||||
Title string `json:"title"`
|
||||
Description string `json:"description"`
|
||||
Type string `json:"type"`
|
||||
ScheduledDate string `json:"scheduled_date"`
|
||||
Duration int `json:"duration"`
|
||||
Notes string `json:"notes"`
|
||||
WorkoutData *WorkoutDataJSON `json:"workout_data"`
|
||||
FileType string `json:"file_type"`
|
||||
}
|
||||
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
log.Printf("Decode error: %v", err)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid request body"})
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("CreateWorkout request: %+v", req)
|
||||
|
||||
if req.Title == "" {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "title is required"})
|
||||
return
|
||||
}
|
||||
|
||||
if req.ScheduledDate == "" {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "scheduled_date is required"})
|
||||
return
|
||||
}
|
||||
|
||||
// Parse scheduled date
|
||||
scheduledDate, err := time.Parse("2006-01-02", req.ScheduledDate)
|
||||
if err != nil {
|
||||
log.Printf("Date parse error: %v", err)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid date format"})
|
||||
return
|
||||
}
|
||||
|
||||
// Set default duration if not provided
|
||||
if req.Duration <= 0 {
|
||||
req.Duration = 60
|
||||
}
|
||||
|
||||
// Initialize WorkoutData if nil
|
||||
workoutData := req.WorkoutData
|
||||
if workoutData == nil {
|
||||
workoutData = &WorkoutDataJSON{
|
||||
Segments: []WorkoutSegment{},
|
||||
}
|
||||
}
|
||||
|
||||
// Create workout
|
||||
workout := &Workout{
|
||||
UserID: claims.UserID,
|
||||
Title: req.Title,
|
||||
Description: req.Description,
|
||||
Type: req.Type,
|
||||
Status: "planned",
|
||||
ScheduledDate: scheduledDate,
|
||||
Duration: req.Duration,
|
||||
Notes: req.Notes,
|
||||
FileType: req.FileType,
|
||||
WorkoutData: *workoutData,
|
||||
}
|
||||
|
||||
log.Printf("Creating workout: %+v", workout)
|
||||
|
||||
if err := h.service.repo.CreateWorkout(workout); err != nil {
|
||||
log.Printf("CreateWorkout error: %v", err)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "failed to create workout"})
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("Workout created successfully with ID: %d", workout.ID)
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
json.NewEncoder(w).Encode(workout)
|
||||
}
|
||||
|
||||
// GetWorkouts GET /api/protected/workouts
|
||||
func (h *Handler) GetWorkouts(w http.ResponseWriter, r *http.Request) {
|
||||
claims := r.Context().Value(middleware.UserContextKey).(*config.CustomClaims)
|
||||
|
||||
workouts, err := h.service.GetUserWorkouts(claims.UserID)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "failed to fetch workouts"})
|
||||
return
|
||||
}
|
||||
|
||||
if workouts == nil {
|
||||
workouts = []Workout{}
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(workouts)
|
||||
}
|
||||
|
||||
// GetWorkoutsByMonth GET /api/protected/workouts/month?year=2025&month=11
|
||||
func (h *Handler) GetWorkoutsByMonth(w http.ResponseWriter, r *http.Request) {
|
||||
claims := r.Context().Value(middleware.UserContextKey).(*config.CustomClaims)
|
||||
|
||||
yearStr := r.URL.Query().Get("year")
|
||||
monthStr := r.URL.Query().Get("month")
|
||||
|
||||
year, err := strconv.Atoi(yearStr)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid year"})
|
||||
return
|
||||
}
|
||||
|
||||
month, err := strconv.Atoi(monthStr)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid month"})
|
||||
return
|
||||
}
|
||||
|
||||
workouts, err := h.service.GetWorkoutsByMonth(claims.UserID, year, month)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "failed to fetch workouts"})
|
||||
return
|
||||
}
|
||||
|
||||
if workouts == nil {
|
||||
workouts = []Workout{}
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(workouts)
|
||||
}
|
||||
|
||||
// UpdateWorkout PUT /api/protected/workouts
|
||||
func (h *Handler) UpdateWorkout(w http.ResponseWriter, r *http.Request) {
|
||||
claims := r.Context().Value(middleware.UserContextKey).(*config.CustomClaims)
|
||||
|
||||
idStr := r.URL.Query().Get("id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid workout id"})
|
||||
return
|
||||
}
|
||||
|
||||
var req struct {
|
||||
Title string `json:"title"`
|
||||
Description string `json:"description"`
|
||||
Type string `json:"type"`
|
||||
Status string `json:"status"`
|
||||
Duration int `json:"duration"`
|
||||
Distance float64 `json:"distance"`
|
||||
ElevGain int `json:"elev_gain"`
|
||||
AvgPower int `json:"avg_power"`
|
||||
AvgHR int `json:"avg_hr"`
|
||||
MaxPower int `json:"max_power"`
|
||||
MaxHR int `json:"max_hr"`
|
||||
CaloriesBurned int `json:"calories_burned"`
|
||||
Notes string `json:"notes"`
|
||||
}
|
||||
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid request"})
|
||||
return
|
||||
}
|
||||
|
||||
workout, err := h.service.repo.GetWorkoutByID(uint(id), claims.UserID)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "workout not found"})
|
||||
return
|
||||
}
|
||||
|
||||
if req.Title != "" {
|
||||
workout.Title = req.Title
|
||||
}
|
||||
if req.Description != "" {
|
||||
workout.Description = req.Description
|
||||
}
|
||||
if req.Type != "" {
|
||||
workout.Type = req.Type
|
||||
}
|
||||
if req.Status != "" {
|
||||
workout.Status = req.Status
|
||||
}
|
||||
if req.Duration > 0 {
|
||||
workout.Duration = req.Duration
|
||||
}
|
||||
if req.Distance > 0 {
|
||||
workout.Distance = req.Distance
|
||||
}
|
||||
if req.ElevGain > 0 {
|
||||
workout.ElevGain = req.ElevGain
|
||||
}
|
||||
if req.AvgPower > 0 {
|
||||
workout.AvgPower = req.AvgPower
|
||||
}
|
||||
if req.AvgHR > 0 {
|
||||
workout.AvgHR = req.AvgHR
|
||||
}
|
||||
if req.MaxPower > 0 {
|
||||
workout.MaxPower = req.MaxPower
|
||||
}
|
||||
if req.MaxHR > 0 {
|
||||
workout.MaxHR = req.MaxHR
|
||||
}
|
||||
if req.CaloriesBurned > 0 {
|
||||
workout.CaloriesBurned = req.CaloriesBurned
|
||||
}
|
||||
if req.Notes != "" {
|
||||
workout.Notes = req.Notes
|
||||
}
|
||||
|
||||
if err := h.service.repo.UpdateWorkout(workout); err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "failed to update workout"})
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(workout)
|
||||
}
|
||||
|
||||
// DeleteWorkout DELETE /api/protected/workouts
|
||||
func (h *Handler) DeleteWorkout(w http.ResponseWriter, r *http.Request) {
|
||||
claims := r.Context().Value(middleware.UserContextKey).(*config.CustomClaims)
|
||||
|
||||
idStr := r.URL.Query().Get("id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 32)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "invalid workout id"})
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.service.DeleteWorkout(uint(id), claims.UserID); err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}
|
||||
|
||||
// GetWorkoutTypes GET /api/protected/workout-types
|
||||
func (h *Handler) GetWorkoutTypes(w http.ResponseWriter, r *http.Request) {
|
||||
types := []map[string]interface{}{
|
||||
{"id": 1, "name": "Recovery", "color": "#4285F4", "icon": "🔵"},
|
||||
{"id": 2, "name": "Endurance", "color": "#34A853", "icon": "🟢"},
|
||||
{"id": 3, "name": "Tempo", "color": "#FBBC04", "icon": "🟡"},
|
||||
{"id": 4, "name": "Threshold", "color": "#EA4335", "icon": "🔴"},
|
||||
{"id": 5, "name": "VO2 Max", "color": "#A61C00", "icon": "⭐"},
|
||||
{"id": 6, "name": "Strength", "color": "#800080", "icon": "💪"},
|
||||
{"id": 7, "name": "Race", "color": "#FF1744", "icon": "🏁"},
|
||||
{"id": 8, "name": "Rest", "color": "#CCCCCC", "icon": "😴"},
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(types)
|
||||
}
|
||||
|
||||
// UploadWorkoutFile POST /api/protected/workouts/upload
|
||||
func (h *Handler) UploadWorkoutFile(w http.ResponseWriter, r *http.Request) {
|
||||
claims := r.Context().Value(middleware.UserContextKey).(*config.CustomClaims)
|
||||
|
||||
err := r.ParseMultipartForm(10 << 20) // 10MB max
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "file too large"})
|
||||
return
|
||||
}
|
||||
|
||||
file, handler, err := r.FormFile("file")
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "no file provided"})
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
fileContent := make([]byte, handler.Size)
|
||||
file.Read(fileContent)
|
||||
|
||||
// Parse ZWO file
|
||||
parsedData, err := ParseZWO(fileContent)
|
||||
if err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
// Get scheduled date from form
|
||||
scheduledDateStr := r.FormValue("scheduled_date")
|
||||
scheduledDate, err := time.Parse("2006-01-02", scheduledDateStr)
|
||||
if err != nil {
|
||||
scheduledDate = time.Now()
|
||||
}
|
||||
|
||||
// Create workout with parsed data
|
||||
workout := &Workout{
|
||||
UserID: claims.UserID,
|
||||
Title: parsedData.Name,
|
||||
Description: parsedData.Description,
|
||||
Type: "imported",
|
||||
Status: "planned",
|
||||
ScheduledDate: scheduledDate,
|
||||
Duration: parsedData.TotalDuration,
|
||||
FileType: "zwo",
|
||||
WorkoutData: WorkoutDataJSON{
|
||||
Name: parsedData.Name,
|
||||
Author: parsedData.Author,
|
||||
TotalDuration: parsedData.TotalDuration,
|
||||
Segments: parsedData.Segments,
|
||||
},
|
||||
}
|
||||
|
||||
if err := h.service.repo.CreateWorkout(workout); err != nil {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(map[string]string{"error": "failed to create workout"})
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
json.NewEncoder(w).Encode(workout)
|
||||
}
|
||||
66
internal/workout/model.go
Normal file
66
internal/workout/model.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package workout
|
||||
|
||||
import (
|
||||
"database/sql/driver"
|
||||
"encoding/json"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Workout struct {
|
||||
ID uint `gorm:"primaryKey" json:"id"`
|
||||
UserID uint `gorm:"not null;index" json:"user_id"`
|
||||
Title string `gorm:"not null" json:"title"`
|
||||
Description string `gorm:"default:''" json:"description"`
|
||||
Type string `gorm:"default:''" json:"type"`
|
||||
Status string `gorm:"default:'planned'" json:"status"`
|
||||
ScheduledDate time.Time `gorm:"index" json:"scheduled_date"`
|
||||
Duration int `gorm:"default:0" json:"duration"`
|
||||
Distance float64 `gorm:"default:0" json:"distance"`
|
||||
ElevGain int `gorm:"default:0" json:"elev_gain"`
|
||||
AvgPower int `gorm:"default:0" json:"avg_power"`
|
||||
AvgHR int `gorm:"default:0" json:"avg_hr"`
|
||||
MaxPower int `gorm:"default:0" json:"max_power"`
|
||||
MaxHR int `gorm:"default:0" json:"max_hr"`
|
||||
CaloriesBurned int `gorm:"default:0" json:"calories_burned"`
|
||||
FileType string `gorm:"default:''" json:"file_type"`
|
||||
FileURL string `gorm:"default:''" json:"file_url"`
|
||||
WorkoutData WorkoutDataJSON `gorm:"type:jsonb" json:"workout_data,omitempty"`
|
||||
Notes string `json:"notes"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
type WorkoutDataJSON struct {
|
||||
Name string `json:"name"`
|
||||
Author string `json:"author"`
|
||||
TotalDuration int `json:"total_duration"`
|
||||
Segments []WorkoutSegment `json:"segments"`
|
||||
}
|
||||
|
||||
type WorkoutSegment struct {
|
||||
Type string `json:"type"`
|
||||
Duration int `json:"duration"`
|
||||
PowerLow float64 `json:"power_low"`
|
||||
PowerHigh float64 `json:"power_high"`
|
||||
Power float64 `json:"power"`
|
||||
Cadence int `json:"cadence"`
|
||||
}
|
||||
|
||||
// Scan implements sql.Scanner interface
|
||||
func (w *WorkoutDataJSON) Scan(value interface{}) error {
|
||||
if value == nil {
|
||||
*w = WorkoutDataJSON{}
|
||||
return nil
|
||||
}
|
||||
bytes := value.([]byte)
|
||||
return json.Unmarshal(bytes, &w)
|
||||
}
|
||||
|
||||
// Value implements driver.Valuer interface
|
||||
func (w WorkoutDataJSON) Value() (driver.Value, error) {
|
||||
return json.Marshal(w)
|
||||
}
|
||||
|
||||
func (Workout) TableName() string {
|
||||
return "workouts"
|
||||
}
|
||||
65
internal/workout/repository.go
Normal file
65
internal/workout/repository.go
Normal file
@@ -0,0 +1,65 @@
|
||||
package workout
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"rideaware/pkg/database"
|
||||
"time"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type Repository struct{}
|
||||
|
||||
func NewRepository() *Repository {
|
||||
return &Repository{}
|
||||
}
|
||||
|
||||
func (r *Repository) CreateWorkout(workout *Workout) error {
|
||||
return database.DB.Create(workout).Error
|
||||
}
|
||||
|
||||
func (r *Repository) GetWorkoutByID(id, userID uint) (*Workout, error) {
|
||||
var workout Workout
|
||||
if err := database.DB.Where("id = ? AND user_id = ?", id, userID).
|
||||
First(&workout).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, errors.New("workout not found")
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &workout, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetUserWorkouts(userID uint) ([]Workout, error) {
|
||||
var workouts []Workout
|
||||
if err := database.DB.Where("user_id = ?", userID).
|
||||
Order("scheduled_date DESC").
|
||||
Find(&workouts).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return workouts, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetWorkoutsByDateRange(userID uint, start, end time.Time) ([]Workout, error) {
|
||||
var workouts []Workout
|
||||
if err := database.DB.Where("user_id = ? AND scheduled_date BETWEEN ? AND ?", userID, start, end).
|
||||
Order("scheduled_date ASC").
|
||||
Find(&workouts).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return workouts, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetWorkoutsByMonth(userID uint, year, month int) ([]Workout, error) {
|
||||
start := time.Date(year, time.Month(month), 1, 0, 0, 0, 0, time.UTC)
|
||||
end := start.AddDate(0, 1, 0).Add(-time.Second)
|
||||
return r.GetWorkoutsByDateRange(userID, start, end)
|
||||
}
|
||||
|
||||
func (r *Repository) UpdateWorkout(workout *Workout) error {
|
||||
return database.DB.Model(workout).Updates(workout).Error
|
||||
}
|
||||
|
||||
func (r *Repository) DeleteWorkout(id, userID uint) error {
|
||||
return database.DB.Where("id = ? AND user_id = ?", id, userID).
|
||||
Delete(&Workout{}).Error
|
||||
}
|
||||
88
internal/workout/service.go
Normal file
88
internal/workout/service.go
Normal file
@@ -0,0 +1,88 @@
|
||||
package workout
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Service struct {
|
||||
repo *Repository
|
||||
}
|
||||
|
||||
func NewService() *Service {
|
||||
return &Service{
|
||||
repo: NewRepository(),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Service) CreateWorkout(userID uint, title string, scheduledDate time.Time, duration int) (*Workout, error) {
|
||||
if title == "" {
|
||||
return nil, errors.New("title is required")
|
||||
}
|
||||
|
||||
workout := &Workout{
|
||||
UserID: userID,
|
||||
Title: title,
|
||||
ScheduledDate: scheduledDate,
|
||||
Duration: duration,
|
||||
Status: "planned",
|
||||
}
|
||||
|
||||
if err := s.repo.CreateWorkout(workout); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return workout, nil
|
||||
}
|
||||
|
||||
func (s *Service) GetUserWorkouts(userID uint) ([]Workout, error) {
|
||||
return s.repo.GetUserWorkouts(userID)
|
||||
}
|
||||
|
||||
func (s *Service) GetWorkoutsByMonth(userID uint, year, month int) ([]Workout, error) {
|
||||
return s.repo.GetWorkoutsByMonth(userID, year, month)
|
||||
}
|
||||
|
||||
func (s *Service) UpdateWorkoutStatus(id, userID uint, status string) (*Workout, error) {
|
||||
if status != "planned" && status != "completed" && status != "skipped" {
|
||||
return nil, errors.New("invalid status")
|
||||
}
|
||||
|
||||
workout, err := s.repo.GetWorkoutByID(id, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
workout.Status = status
|
||||
if status == "completed" {
|
||||
workout.UpdatedAt = time.Now()
|
||||
}
|
||||
|
||||
if err := s.repo.UpdateWorkout(workout); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return workout, nil
|
||||
}
|
||||
|
||||
func (s *Service) UpdateWorkoutWithMetrics(id, userID uint, distance float64, avgPower, avgHR int) (*Workout, error) {
|
||||
workout, err := s.repo.GetWorkoutByID(id, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
workout.Distance = distance
|
||||
workout.AvgPower = avgPower
|
||||
workout.AvgHR = avgHR
|
||||
workout.Status = "completed"
|
||||
|
||||
if err := s.repo.UpdateWorkout(workout); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return workout, nil
|
||||
}
|
||||
|
||||
func (s *Service) DeleteWorkout(id, userID uint) error {
|
||||
return s.repo.DeleteWorkout(id, userID)
|
||||
}
|
||||
169
internal/workout/zwo_parser.go
Normal file
169
internal/workout/zwo_parser.go
Normal file
@@ -0,0 +1,169 @@
|
||||
package workout
|
||||
|
||||
import (
|
||||
"encoding/xml"
|
||||
"errors"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type ParsedWorkoutData struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
Author string `json:"author"`
|
||||
SportType string `json:"sport_type"`
|
||||
TotalDuration int `json:"total_duration"`
|
||||
Segments []WorkoutSegment `json:"segments"`
|
||||
}
|
||||
|
||||
type zwoFile struct {
|
||||
Author string `xml:"author,attr"`
|
||||
Name string `xml:"name,attr"`
|
||||
Description string `xml:"description,attr"`
|
||||
SportType string `xml:"sportType,attr"`
|
||||
Workout zwoWorkout `xml:"workout"`
|
||||
}
|
||||
|
||||
type zwoWorkout struct {
|
||||
Warmups []zwoWarmup `xml:"Warmup"`
|
||||
SteadyStates []zwoSteadyState `xml:"SteadyState"`
|
||||
Cooldowns []zwoCooldown `xml:"Cooldown"`
|
||||
Intervals []zwoInterval `xml:"Interval"`
|
||||
Ramps []zwoRamp `xml:"Ramp"`
|
||||
FreeRides []zwoFreeRide `xml:"FreeRide"`
|
||||
}
|
||||
|
||||
type zwoWarmup struct {
|
||||
Duration int `xml:"Duration,attr"`
|
||||
PowerLow float64 `xml:"PowerLow,attr"`
|
||||
PowerHigh float64 `xml:"PowerHigh,attr"`
|
||||
Cadence int `xml:"Cadence,attr"`
|
||||
}
|
||||
|
||||
type zwoSteadyState struct {
|
||||
Duration int `xml:"Duration,attr"`
|
||||
Power float64 `xml:"Power,attr"`
|
||||
Cadence int `xml:"Cadence,attr"`
|
||||
}
|
||||
|
||||
type zwoCooldown struct {
|
||||
Duration int `xml:"Duration,attr"`
|
||||
PowerLow float64 `xml:"PowerLow,attr"`
|
||||
PowerHigh float64 `xml:"PowerHigh,attr"`
|
||||
Cadence int `xml:"Cadence,attr"`
|
||||
}
|
||||
|
||||
type zwoInterval struct {
|
||||
Duration int `xml:"Duration,attr"`
|
||||
PowerLow float64 `xml:"PowerLow,attr"`
|
||||
PowerHigh float64 `xml:"PowerHigh,attr"`
|
||||
Cadence int `xml:"Cadence,attr"`
|
||||
}
|
||||
|
||||
type zwoRamp struct {
|
||||
Duration int `xml:"Duration,attr"`
|
||||
PowerLow float64 `xml:"PowerLow,attr"`
|
||||
PowerHigh float64 `xml:"PowerHigh,attr"`
|
||||
Cadence int `xml:"Cadence,attr"`
|
||||
}
|
||||
|
||||
type zwoFreeRide struct {
|
||||
Duration int `xml:"Duration,attr"`
|
||||
Cadence int `xml:"Cadence,attr"`
|
||||
}
|
||||
|
||||
func ParseZWO(content []byte) (*ParsedWorkoutData, error) {
|
||||
var zwo zwoFile
|
||||
err := xml.Unmarshal(content, &zwo)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to parse ZWO file: %w", err)
|
||||
}
|
||||
|
||||
if zwo.Name == "" {
|
||||
return nil, errors.New("workout name is required")
|
||||
}
|
||||
|
||||
parsed := &ParsedWorkoutData{
|
||||
Name: zwo.Name,
|
||||
Description: zwo.Description,
|
||||
Author: zwo.Author,
|
||||
SportType: zwo.SportType,
|
||||
Segments: []WorkoutSegment{},
|
||||
}
|
||||
|
||||
// Parse warmups
|
||||
for _, w := range zwo.Workout.Warmups {
|
||||
seg := WorkoutSegment{
|
||||
Type: "warmup",
|
||||
Duration: w.Duration,
|
||||
PowerLow: w.PowerLow,
|
||||
PowerHigh: w.PowerHigh,
|
||||
Cadence: w.Cadence,
|
||||
}
|
||||
parsed.Segments = append(parsed.Segments, seg)
|
||||
parsed.TotalDuration += w.Duration
|
||||
}
|
||||
|
||||
// Parse steady states
|
||||
for _, s := range zwo.Workout.SteadyStates {
|
||||
seg := WorkoutSegment{
|
||||
Type: "steadystate",
|
||||
Duration: s.Duration,
|
||||
Power: s.Power,
|
||||
Cadence: s.Cadence,
|
||||
}
|
||||
parsed.Segments = append(parsed.Segments, seg)
|
||||
parsed.TotalDuration += s.Duration
|
||||
}
|
||||
|
||||
// Parse cooldowns
|
||||
for _, c := range zwo.Workout.Cooldowns {
|
||||
seg := WorkoutSegment{
|
||||
Type: "cooldown",
|
||||
Duration: c.Duration,
|
||||
PowerLow: c.PowerLow,
|
||||
PowerHigh: c.PowerHigh,
|
||||
Cadence: c.Cadence,
|
||||
}
|
||||
parsed.Segments = append(parsed.Segments, seg)
|
||||
parsed.TotalDuration += c.Duration
|
||||
}
|
||||
|
||||
// Parse intervals
|
||||
for _, i := range zwo.Workout.Intervals {
|
||||
seg := WorkoutSegment{
|
||||
Type: "interval",
|
||||
Duration: i.Duration,
|
||||
PowerLow: i.PowerLow,
|
||||
PowerHigh: i.PowerHigh,
|
||||
Cadence: i.Cadence,
|
||||
}
|
||||
parsed.Segments = append(parsed.Segments, seg)
|
||||
parsed.TotalDuration += i.Duration
|
||||
}
|
||||
|
||||
// Parse ramps
|
||||
for _, r := range zwo.Workout.Ramps {
|
||||
seg := WorkoutSegment{
|
||||
Type: "ramp",
|
||||
Duration: r.Duration,
|
||||
PowerLow: r.PowerLow,
|
||||
PowerHigh: r.PowerHigh,
|
||||
Cadence: r.Cadence,
|
||||
}
|
||||
parsed.Segments = append(parsed.Segments, seg)
|
||||
parsed.TotalDuration += r.Duration
|
||||
}
|
||||
|
||||
// Parse free rides
|
||||
for _, f := range zwo.Workout.FreeRides {
|
||||
seg := WorkoutSegment{
|
||||
Type: "freeride",
|
||||
Duration: f.Duration,
|
||||
Cadence: f.Cadence,
|
||||
}
|
||||
parsed.Segments = append(parsed.Segments, seg)
|
||||
parsed.TotalDuration += f.Duration
|
||||
}
|
||||
|
||||
return parsed, nil
|
||||
}
|
||||
Reference in New Issue
Block a user