102 lines
3.4 KiB
Go
102 lines
3.4 KiB
Go
package model
|
|
|
|
import (
|
|
"database/sql"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Habit struct {
|
|
ID int64 `db:"id" json:"id"`
|
|
UserID int64 `db:"user_id" json:"user_id"`
|
|
Name string `db:"name" json:"name"`
|
|
Description string `db:"description" json:"description"`
|
|
Color string `db:"color" json:"color"`
|
|
Icon string `db:"icon" json:"icon"`
|
|
Frequency string `db:"frequency" json:"frequency"`
|
|
TargetDays []int `db:"-" json:"target_days"`
|
|
TargetCount int `db:"target_count" json:"target_count"`
|
|
ReminderTime sql.NullString `db:"reminder_time" json:"-"`
|
|
ReminderTimeStr *string `db:"-" json:"reminder_time"`
|
|
IsArchived bool `db:"is_archived" json:"is_archived"`
|
|
CreatedAt time.Time `db:"created_at" json:"created_at"`
|
|
UpdatedAt time.Time `db:"updated_at" json:"updated_at"`
|
|
}
|
|
|
|
func (h *Habit) ProcessForJSON() {
|
|
if h.ReminderTime.Valid && h.ReminderTime.String != "" {
|
|
timeStr := h.ReminderTime.String
|
|
// Handle formats like "0000-01-01T19:00:00Z" or "19:00:00"
|
|
if strings.Contains(timeStr, "T") {
|
|
// Parse as timestamp and extract time
|
|
t, err := time.Parse(time.RFC3339, timeStr)
|
|
if err == nil {
|
|
formatted := t.Format("15:04")
|
|
h.ReminderTimeStr = &formatted
|
|
return
|
|
}
|
|
}
|
|
// Handle "19:00:00" format
|
|
if len(timeStr) >= 5 {
|
|
formatted := timeStr[:5]
|
|
h.ReminderTimeStr = &formatted
|
|
}
|
|
}
|
|
}
|
|
|
|
type HabitLog struct {
|
|
ID int64 `db:"id" json:"id"`
|
|
HabitID int64 `db:"habit_id" json:"habit_id"`
|
|
UserID int64 `db:"user_id" json:"user_id"`
|
|
Date time.Time `db:"date" json:"date"`
|
|
Count int `db:"count" json:"count"`
|
|
Note string `db:"note" json:"note"`
|
|
CreatedAt time.Time `db:"created_at" json:"created_at"`
|
|
}
|
|
|
|
type CreateHabitRequest struct {
|
|
Name string `json:"name"`
|
|
Description string `json:"description,omitempty"`
|
|
Color string `json:"color,omitempty"`
|
|
Icon string `json:"icon,omitempty"`
|
|
Frequency string `json:"frequency,omitempty"`
|
|
TargetDays []int `json:"target_days,omitempty"`
|
|
TargetCount int `json:"target_count,omitempty"`
|
|
ReminderTime *string `json:"reminder_time,omitempty"`
|
|
}
|
|
|
|
type UpdateHabitRequest struct {
|
|
Name *string `json:"name,omitempty"`
|
|
Description *string `json:"description,omitempty"`
|
|
Color *string `json:"color,omitempty"`
|
|
Icon *string `json:"icon,omitempty"`
|
|
Frequency *string `json:"frequency,omitempty"`
|
|
TargetDays []int `json:"target_days,omitempty"`
|
|
TargetCount *int `json:"target_count,omitempty"`
|
|
ReminderTime *string `json:"reminder_time,omitempty"`
|
|
IsArchived *bool `json:"is_archived,omitempty"`
|
|
}
|
|
|
|
type LogHabitRequest struct {
|
|
Date string `json:"date,omitempty"`
|
|
Count int `json:"count,omitempty"`
|
|
Note string `json:"note,omitempty"`
|
|
}
|
|
|
|
type HabitStats struct {
|
|
HabitID int64 `json:"habit_id"`
|
|
TotalLogs int `json:"total_logs"`
|
|
CurrentStreak int `json:"current_streak"`
|
|
LongestStreak int `json:"longest_streak"`
|
|
CompletionPct float64 `json:"completion_pct"`
|
|
ThisWeek int `json:"this_week"`
|
|
ThisMonth int `json:"this_month"`
|
|
}
|
|
|
|
type OverallStats struct {
|
|
TotalHabits int `json:"total_habits"`
|
|
ActiveHabits int `json:"active_habits"`
|
|
TodayCompleted int `json:"today_completed"`
|
|
WeeklyAvg float64 `json:"weekly_avg"`
|
|
}
|