76 lines
2.9 KiB
Ruby
76 lines
2.9 KiB
Ruby
class FetchIpapiDataJob < ApplicationJob
|
|
queue_as :default
|
|
|
|
# Fetches IPAPI enrichment data for a NetworkRange
|
|
# @param network_range_id [Integer] ID of the tracking NetworkRange (usually /24)
|
|
def perform(network_range_id:)
|
|
tracking_network = NetworkRange.find_by(id: network_range_id)
|
|
return unless tracking_network
|
|
|
|
# Use the network address (first IP in range) as the representative IP
|
|
sample_ip = tracking_network.network_address.split('/').first
|
|
|
|
Rails.logger.info "Fetching IPAPI data for #{tracking_network.cidr} using IP #{sample_ip}"
|
|
|
|
ipapi_data = Ipapi.lookup(sample_ip)
|
|
|
|
if ipapi_data.present? && !ipapi_data.key?('error')
|
|
# Check if IPAPI returned a different route than our tracking network
|
|
ipapi_route = ipapi_data.dig('asn', 'route')
|
|
target_network = tracking_network
|
|
|
|
if ipapi_route.present? && ipapi_route != tracking_network.cidr
|
|
# IPAPI returned a different CIDR - find or create that network range
|
|
Rails.logger.info "IPAPI returned different route: #{ipapi_route} (requested: #{tracking_network.cidr})"
|
|
|
|
target_network = NetworkRange.find_or_create_by(network: ipapi_route) do |nr|
|
|
nr.source = 'api_imported'
|
|
nr.creation_reason = "Created from IPAPI lookup for #{tracking_network.cidr}"
|
|
end
|
|
|
|
Rails.logger.info "Storing IPAPI data on correct network: #{target_network.cidr}"
|
|
end
|
|
|
|
# Store data on the target network (wherever IPAPI said it belongs)
|
|
target_network.set_network_data(:ipapi, ipapi_data)
|
|
target_network.last_api_fetch = Time.current
|
|
target_network.save!
|
|
|
|
# Mark the tracking network as having been queried, with the CIDR that was returned
|
|
tracking_network.mark_ipapi_queried!(target_network.cidr)
|
|
|
|
Rails.logger.info "Successfully fetched IPAPI data for #{tracking_network.cidr} (stored on #{target_network.cidr})"
|
|
|
|
# Broadcast to the tracking network
|
|
broadcast_ipapi_update(tracking_network, ipapi_data)
|
|
else
|
|
Rails.logger.warn "IPAPI returned error for #{tracking_network.cidr}: #{ipapi_data}"
|
|
# Still mark as queried to avoid retrying immediately
|
|
tracking_network.mark_ipapi_queried!(tracking_network.cidr)
|
|
end
|
|
rescue => e
|
|
Rails.logger.error "Failed to fetch IPAPI data for network_range #{network_range_id}: #{e.message}"
|
|
Rails.logger.error e.backtrace.join("\n")
|
|
ensure
|
|
# Always clear the fetching status when done
|
|
tracking_network&.clear_fetching_status!(:ipapi)
|
|
end
|
|
|
|
private
|
|
|
|
def broadcast_ipapi_update(network_range, ipapi_data)
|
|
# Broadcast to a stream specific to this network range
|
|
Turbo::StreamsChannel.broadcast_replace_to(
|
|
"network_range_#{network_range.id}",
|
|
target: "ipapi_data_section",
|
|
partial: "network_ranges/ipapi_data",
|
|
locals: {
|
|
ipapi_data: ipapi_data,
|
|
network_range: network_range,
|
|
parent_with_ipapi: nil,
|
|
ipapi_loading: false
|
|
}
|
|
)
|
|
end
|
|
end
|