100 lines
2.5 KiB
TypeScript
100 lines
2.5 KiB
TypeScript
import type { GoalFrequency } from "./Goal"
|
|
import type { Habit } from "./Habit"
|
|
import { HabitHistory } from "./HabitHistory"
|
|
import type { HabitProgress } from "./HabitProgress"
|
|
|
|
export interface HabitsTrackerData {
|
|
habitsHistory: {
|
|
[key in GoalFrequency]: HabitHistory[]
|
|
}
|
|
}
|
|
|
|
export class HabitsTracker implements HabitsTrackerData {
|
|
public habitsHistory: HabitsTrackerData["habitsHistory"]
|
|
|
|
public constructor(options: HabitsTrackerData) {
|
|
const { habitsHistory } = options
|
|
this.habitsHistory = habitsHistory
|
|
}
|
|
|
|
public static default(): HabitsTracker {
|
|
return new HabitsTracker({
|
|
habitsHistory: {
|
|
daily: [],
|
|
weekly: [],
|
|
monthly: [],
|
|
},
|
|
})
|
|
}
|
|
|
|
public addHabit(habit: Habit): void {
|
|
this.habitsHistory[habit.goal.frequency].push(
|
|
new HabitHistory({
|
|
habit,
|
|
progressHistory: [],
|
|
}),
|
|
)
|
|
}
|
|
|
|
public editHabit(habit: Habit): void {
|
|
const habitHistory = this.getHabitHistoryById(habit.id)
|
|
if (habitHistory == null) {
|
|
return
|
|
}
|
|
habitHistory.habit = habit
|
|
}
|
|
|
|
public updateHabitProgress(habitProgress: HabitProgress): void {
|
|
const habitHistory = this.getHabitHistoryById(habitProgress.habitId)
|
|
if (habitHistory == null) {
|
|
return
|
|
}
|
|
const habitProgressSaved = habitHistory.progressHistory.find((progress) => {
|
|
return progress.id === habitProgress.id
|
|
})
|
|
if (habitProgressSaved == null) {
|
|
habitHistory.progressHistory = [
|
|
...habitHistory.progressHistory,
|
|
habitProgress,
|
|
]
|
|
return
|
|
}
|
|
habitProgressSaved.goalProgress = habitProgress.goalProgress
|
|
habitProgressSaved.date = habitProgress.date
|
|
}
|
|
|
|
public getAllHabitsHistory(): HabitHistory[] {
|
|
return [
|
|
...this.habitsHistory.daily,
|
|
...this.habitsHistory.weekly,
|
|
...this.habitsHistory.monthly,
|
|
]
|
|
}
|
|
|
|
public getHabitHistoryById(id: Habit["id"]): HabitHistory | undefined {
|
|
return this.getAllHabitsHistory().find((habitHistory) => {
|
|
return habitHistory.habit.id === id
|
|
})
|
|
}
|
|
|
|
public getHabitsHistoriesByDate({
|
|
selectedDate,
|
|
frequency,
|
|
}: {
|
|
selectedDate: Date
|
|
frequency: GoalFrequency
|
|
}): HabitHistory[] {
|
|
return this.habitsHistory[frequency].filter((habitItem) => {
|
|
const startDate = new Date(habitItem.habit.startDate)
|
|
startDate.setHours(0, 0, 0, 0)
|
|
|
|
return (
|
|
startDate <= selectedDate &&
|
|
(habitItem.habit.endDate == null ||
|
|
(habitItem.habit.endDate != null &&
|
|
habitItem.habit.endDate >= selectedDate))
|
|
)
|
|
})
|
|
}
|
|
}
|