dawarich/app/services/imports/create.rb

120 lines
3.4 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
class Imports::Create
2025-07-02 14:22:40 -04:00
include Imports::Broadcaster
attr_reader :user, :import
def initialize(user, import)
@user = user
@import = import
end
def call
2025-06-28 06:22:56 -04:00
import.update!(status: :processing)
2025-07-02 14:22:40 -04:00
broadcast_status_update
2025-06-28 06:22:56 -04:00
2025-08-22 13:10:40 -04:00
temp_file_path = Imports::SecureFileDownloader.new(import.file).download_to_temp_file
2025-08-23 10:07:15 -04:00
source = if import.source.nil? || should_detect_source?
detect_source_from_file(temp_file_path)
else
import.source
end
2025-08-23 10:37:53 -04:00
import.update!(source: source)
2025-08-22 13:10:40 -04:00
importer(source).new(import, user.id, temp_file_path).call
schedule_stats_creating(user.id)
schedule_visit_suggesting(user.id, import)
update_import_points_count(import)
User.reset_counters(user.id, :points)
rescue StandardError => e
2025-06-28 06:22:56 -04:00
import.update!(status: :failed)
2025-07-02 14:22:40 -04:00
broadcast_status_update
2025-06-28 06:22:56 -04:00
2025-08-21 12:42:45 -04:00
ExceptionReporter.call(e, 'Import failed')
create_import_failed_notification(import, user, e)
2025-06-28 06:22:56 -04:00
ensure
2025-08-22 13:10:40 -04:00
if temp_file_path && File.exist?(temp_file_path)
File.unlink(temp_file_path)
end
2025-07-02 14:22:40 -04:00
if import.processing?
import.update!(status: :completed)
broadcast_status_update
end
end
private
def importer(source)
2025-08-23 13:09:18 -04:00
raise ArgumentError, 'Import source cannot be nil' if source.nil?
2025-08-22 16:14:03 -04:00
case source.to_s
2025-04-23 17:36:16 -04:00
when 'google_semantic_history' then GoogleMaps::SemanticHistoryImporter
when 'google_phone_takeout' then GoogleMaps::PhoneTakeoutImporter
when 'google_records' then GoogleMaps::RecordsStorageImporter
2025-02-22 17:14:23 -05:00
when 'owntracks' then OwnTracks::Importer
when 'gpx' then Gpx::TrackImporter
2025-04-23 17:36:16 -04:00
when 'geojson' then Geojson::Importer
when 'immich_api', 'photoprism_api' then Photos::Importer
2025-08-22 16:14:03 -04:00
else
raise ArgumentError, "Unsupported source: #{source}"
end
end
def update_import_points_count(import)
Import::UpdatePointsCountJob.perform_later(import.id)
end
def schedule_stats_creating(user_id)
2024-12-06 10:52:36 -05:00
import.years_and_months_tracked.each do |year, month|
Stats::CalculatingJob.perform_later(user_id, year, month)
end
end
def schedule_visit_suggesting(user_id, import)
return unless user.safe_settings.visits_suggestions_enabled?
min_max = import.points.pluck('MIN(timestamp), MAX(timestamp)').first
return if min_max.compact.empty?
start_at = Time.zone.at(min_max[0])
end_at = Time.zone.at(min_max[1])
VisitSuggestingJob.perform_later(user_id:, start_at:, end_at:)
end
def create_import_failed_notification(import, user, error)
2025-04-19 07:18:39 -04:00
message = import_failed_message(import, error)
Notifications::Create.new(
user:,
kind: :error,
title: 'Import failed',
2025-04-19 07:18:39 -04:00
content: message
).call
end
2025-04-19 07:18:39 -04:00
2025-08-23 10:07:15 -04:00
def should_detect_source?
# Don't override API-based sources that can't be reliably detected
!%w[immich_api photoprism_api].include?(import.source)
end
2025-08-22 13:10:40 -04:00
def detect_source_from_file(temp_file_path)
2025-08-22 14:40:06 -04:00
detector = Imports::SourceDetector.new_from_file_header(temp_file_path)
2025-08-23 10:07:15 -04:00
2025-08-22 13:10:40 -04:00
detector.detect_source!
end
2025-04-19 07:18:39 -04:00
def import_failed_message(import, error)
if DawarichSettings.self_hosted?
"Import \"#{import.name}\" failed: #{error.message}, stacktrace: #{error.backtrace.join("\n")}"
else
"Import \"#{import.name}\" failed, please contact us at hi@dawarich.com"
end
end
end