dawarich/app/services/gpx/track_importer.rb

96 lines
2.5 KiB
Ruby
Raw Normal View History

2025-02-22 17:14:23 -05:00
# frozen_string_literal: true
2025-04-12 07:06:32 -04:00
require 'rexml/document'
2025-02-22 17:14:23 -05:00
class Gpx::TrackImporter
include Imports::Broadcaster
2025-08-22 13:10:40 -04:00
attr_reader :import, :user_id, :file_path
2025-02-22 17:14:23 -05:00
2025-08-22 13:10:40 -04:00
def initialize(import, user_id, file_path = nil)
2025-02-22 17:14:23 -05:00
@import = import
@user_id = user_id
2025-08-22 13:10:40 -04:00
@file_path = file_path
2025-02-22 17:14:23 -05:00
end
def call
2025-08-22 13:10:40 -04:00
file_content = if file_path && File.exist?(file_path)
File.read(file_path)
else
Imports::SecureFileDownloader.new(import.file).download_with_verification
end
2025-04-23 17:27:55 -04:00
json = Hash.from_xml(file_content)
2025-02-22 17:14:23 -05:00
2025-04-23 17:27:55 -04:00
tracks = json['gpx']['trk']
tracks_arr = tracks.is_a?(Array) ? tracks : [tracks]
2025-02-22 17:14:23 -05:00
2025-04-23 17:27:55 -04:00
points = tracks_arr.map { parse_track(_1) }.flatten.compact
points_data = points.map { prepare_point(_1) }.compact
2025-03-23 13:37:10 -04:00
2025-04-23 17:27:55 -04:00
bulk_insert_points(points_data)
2025-02-22 17:14:23 -05:00
end
private
def parse_track(track)
return if track['trkseg'].blank?
segments = track['trkseg']
segments_array = segments.is_a?(Array) ? segments : [segments]
segments_array.compact.map { |segment| segment['trkpt'] }
end
2025-03-23 13:37:10 -04:00
def prepare_point(point)
2025-02-22 17:14:23 -05:00
return if point['lat'].blank? || point['lon'].blank? || point['time'].blank?
{
lonlat: "POINT(#{point['lon'].to_d} #{point['lat'].to_d})",
altitude: point['ele'].to_i,
timestamp: Time.parse(point['time']).to_i,
import_id: import.id,
velocity: speed(point),
raw_data: point,
user_id: user_id,
created_at: Time.current,
updated_at: Time.current
}
end
def bulk_insert_points(batch)
unique_batch = batch.uniq { |record| [record[:lonlat], record[:timestamp], record[:user_id]] }
# rubocop:disable Rails/SkipsModelValidations
Point.upsert_all(
unique_batch,
unique_by: %i[lonlat timestamp user_id],
returning: false,
on_duplicate: :skip
)
# rubocop:enable Rails/SkipsModelValidations
broadcast_import_progress(import, unique_batch.size)
2025-02-22 17:14:23 -05:00
rescue StandardError => e
create_notification("Failed to process GPX track: #{e.message}")
end
def create_notification(message)
Notification.create!(
user_id: user_id,
title: 'GPX Import Error',
content: message,
kind: :error
)
end
def speed(point)
return if point['extensions'].blank?
2025-07-26 07:32:13 -04:00
value = point.dig('extensions', 'speed')
extensions = point.dig('extensions', 'TrackPointExtension')
value ||= extensions.is_a?(Hash) ? extensions.dig('speed') : nil
2025-07-26 07:37:50 -04:00
value&.to_f&.round(1) || 0.0
2025-02-22 17:14:23 -05:00
end
end