dawarich/app/models/import.rb

# frozen_string_literal: true

class Import < ApplicationRecord
  belongs_to :user
  has_many :points, dependent: :destroy

  has_one_attached :file

  # Flag to skip background processing during user data import
  attr_accessor :skip_background_processing

  after_commit -> { Import::ProcessJob.perform_later(id) unless skip_background_processing }, on: :create
  after_commit :remove_attached_file, on: :destroy
  before_commit :recalculate_stats, on: :destroy, if: -> { points.exists? }

  validates :name, presence: true, uniqueness: { scope: :user_id }
  validate :file_size_within_limit, if: -> { user.trial? }
  validate :import_count_within_limit, if: -> { user.trial? }

  enum :status, { created: 0, processing: 1, completed: 2, failed: 3, deleting: 4 }

  enum :source, {
    google_semantic_history: 0, owntracks: 1, google_records: 2,
    google_phone_takeout: 3, gpx: 4, immich_api: 5, geojson: 6, photoprism_api: 7,
    user_data_archive: 8, kml: 9
  }, allow_nil: true

  def process!
    if user_data_archive?
      process_user_data_archive!
    else
      Imports::Create.new(user, self).call
    end
  end

  def process_user_data_archive!
    Users::ImportDataJob.perform_later(id)
  end

  def reverse_geocoded_points_count
    points.reverse_geocoded.count
  end

  def years_and_months_tracked
    points.order(:timestamp).pluck(:timestamp).map do |timestamp|
      time = Time.zone.at(timestamp)
      [time.year, time.month]
    end.uniq
  end

  def migrate_to_new_storage
    return if file.attached?

    raw_file = File.new(raw_data)

    file.attach(io: raw_file, filename: name, content_type: 'application/json')
  end

  private

  def remove_attached_file
    file.purge_later
  end

  def file_size_within_limit
    return unless file.attached?

    return unless file.blob.byte_size > 11.megabytes

    errors.add(:file, 'is too large. Trial users can only upload files up to 10MB.')
  end

  def import_count_within_limit
    return unless new_record?

    existing_imports_count = user.imports.count
    return unless existing_imports_count >= 5

    errors.add(:base, 'Trial users can only create up to 5 imports. Please subscribe to import more files.')
  end

  def recalculate_stats
    years_and_months_tracked.each do |year, month|
      Stats::CalculatingJob.perform_later(user.id, year, month)
    end
  end
end