dawarich/app/models/stat.rb

# frozen_string_literal: true

class Stat < ApplicationRecord
  validates :year, :month, presence: true

  belongs_to :user

  def distance_by_day
    monthly_points = points
    calculate_daily_distances(monthly_points)
  end

  def self.year_distance(year, user)
    stats_by_month = where(year:, user:).order(:month).index_by(&:month)

    (1..12).map do |month|
      month_name = Date::MONTHNAMES[month]
      distance = stats_by_month[month]&.distance || 0

      [month_name, distance]
    end
  end

  def self.year_cities_and_countries(year, user)
    start_at = DateTime.new(year).beginning_of_year
    end_at = DateTime.new(year).end_of_year

    points = user.tracked_points.without_raw_data.where(timestamp: start_at..end_at)

    data = CountriesAndCities.new(points).call

    {
      countries: data.map { _1[:country] }.uniq.count,
      cities: data.sum { _1[:cities].count }
    }
  end

  def points
    user.tracked_points
        .without_raw_data
        .where(timestamp: timespan)
        .order(timestamp: :asc)
  end

  private

  def timespan
    DateTime.new(year, month).beginning_of_month..DateTime.new(year, month).end_of_month
  end

  def calculate_daily_distances(monthly_points)
    timespan.to_a.map.with_index(1) do |day, index|
      daily_points = filter_points_for_day(monthly_points, day)
      distance = calculate_distance(daily_points)
      [index, distance.round(2)]
    end
  end

  def filter_points_for_day(points, day)
    beginning_of_day = day.beginning_of_day.to_i
    end_of_day = day.end_of_day.to_i

    points.select { |p| p.timestamp.between?(beginning_of_day, end_of_day) }
  end

  def calculate_distance(points)
    points.each_cons(2).sum do |point1, point2|
      DistanceCalculator.new(point1, point2).call
    end
  end
end