create pssg download job

Gemfile

@@ -81,6 +81,7 @@ gem 'rails_semantic_logger', '~> 4.4'
 gem 'redis'
 gem 'redis-namespace'
 gem 'restforce'
+gem 'rgeo-geojson'
 gem 'ruby-saml'
 gem 'rubyzip', '>= 1.3.0'
 gem 'savon'

Gemfile.lock

@@ -523,6 +523,8 @@ GEM
     rgeo-activerecord (6.1.0)
       activerecord (~> 5.0)
       rgeo (>= 1.0.0)
+    rgeo-geojson (2.1.1)
+      rgeo (>= 1.0.0)
     rspec (3.9.0)
       rspec-core (~> 3.9.0)
       rspec-expectations (~> 3.9.0)
@@ -776,6 +778,7 @@ DEPENDENCIES
   redis
   redis-namespace
   restforce
+  rgeo-geojson
   rspec-rails (~> 3.5)
   rspec_junit_formatter
   rubocop

app/workers/facilities/pssg_download.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require 'rgeo/geo_json'
+
+module Facilities
+  class PSSGDownload
+    include Sidekiq::Worker
+    include SentryLogging
+
+    def perform
+      @drivetime_band_client = Facilities::DrivetimeBandClient.new
+      download_data
+    end
+
+    private
+
+    def create_and_save_drive_time_data(drive_time_data)
+      attributes = drive_time_data.dig('attributes')
+
+      vha_id = extract_vha_id(attributes)
+      facility = BaseFacility.find_facility_by_id(vha_id)
+      return if facility.nil?
+
+      drive_time_band = facility.drivetime_bands.find_or_initialize_by(vha_facility_id: extract_name(attributes))
+      drive_time_band.min = attributes.dig('FromBreak')
+      drive_time_band.max = attributes.dig('ToBreak')
+      drive_time_band.name = attributes.dig('Name')
+      drive_time_band.polygon = extract_polygon(drive_time_data)
+      facility.save
+    end
+
+    def extract_name(attributes)
+      name = attributes.dig('Name')
+      name.partition(':').first.strip!
+    end
+
+    def extract_vha_id(attributes)
+      'vha_' + extract_name(attributes)
+    end
+
+    def extract_polygon(drive_time_data)
+      rings = drive_time_data.dig('geometry', 'rings')
+      geojson = "{\"type\":\"Polygon\",\"coordinates\":#{rings}}"
+      RGeo::GeoJSON.decode(geojson)
+    end
+
+    def download_data
+      offset = 0
+      loop do
+        response = @drivetime_band_client.get_drivetime_bands(offset, 1)
+        break if response.nil?
+
+        response.each(&method(:create_and_save_drive_time_data))
+        offset += 1
+      end
+    end
+  end
+end

config/sidekiq_scheduler.yml

@@ -104,6 +104,12 @@ FacilityLocationBulkUpdateVHA:
   args: ['vha']
   description: "Download and store facility location data"
 
+FacilityLocationVHADriveTime:
+  cron: "55 4 * * * America/New_York"
+  class: Facilities::PSSGDownload
+  args: ['vha']
+  description: "Download and store facility location data"
+
 MaintenanceWindowRefresh:
   cron: "*/3 * * * * America/New_York"
   class: PagerDuty::PollMaintenanceWindows

rakelib/jobs.rake

@@ -52,7 +52,13 @@ namespace :jobs do
   end
 
   desc 'Populate/refresh All facility location types to db cache'
-  task pull_all_facility_location_data: %i[pull_nca_data pull_vba_data pull_vc_data pull_vha_data] do
+  task pull_all_facility_location_data:
+           %i[pull_nca_data pull_vba_data pull_vc_data pull_vha_data pull_drive_time_bands] do
     # run all dependencies
   end
+
+  desc 'Populate/refresh Drive time bands'
+  task pull_drive_time_bands: :environment do
+    Facilities::PSSGDownload.perform_async
+  end
 end