mirror of
https://github.com/Freika/dawarich.git
synced 2026-01-10 17:21:38 -05:00
87 lines
2.3 KiB
Ruby
87 lines
2.3 KiB
Ruby
# 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 }
|
|
|
|
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
|
|
}, 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
|