p61-project/domain/entities/HabitsTracker.ts

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