Fetch numbers from local inventory

Amolith created 2 years ago

References: https://soprani.ca/todo/323
Signed-off-by: Amolith <[e-mail]>

Change summary

lib/tel_selections.rb       | 32 ++++++++++++++++++++++++++------
test/test_tel_selections.rb | 29 +++++++++++++++++++++++++++++
2 files changed, 55 insertions(+), 6 deletions(-)

Detailed changes

lib/tel_selections.rb

@@ -87,8 +87,8 @@ class TelSelections
 					q.iris_query
 					.merge(enableTNDetail: true, LCA: false)
 					.merge(Quantity.for(form).iris_query),
-					fallback: q.fallback,
-					memcache: memcache
+					q.sql_query,
+					fallback: q.fallback, memcache: memcache, db: db
 				)
 			end
 
@@ -101,30 +101,50 @@ class TelSelections
 				"810"
 			end
 
-			def initialize(iris_query, fallback: [], memcache: MEMCACHE)
+			def initialize(
+				iris_query, sql_query, fallback: [], memcache: MEMCACHE, db: DB
+			)
 				@iris_query = iris_query
+				@sql_query = sql_query
 				@fallback = fallback
 				@memcache = memcache
+				@db = db
 			end
 
 			def tns
 				Command.log.debug("BandwidthIris::AvailableNumber.list", @iris_query)
 				unless (result = fetch_cache)
-					result = BandwidthIris::AvailableNumber.list(@iris_query)
+					result =
+						BandwidthIris::AvailableNumber.list(@iris_query) +
+						fetch_local_inventory.sync
 				end
 				return next_fallback if result.empty? && !@fallback.empty?
 
 				result.map { |tn| Tn.new(**tn) }
 			end
 
+			def fetch_local_inventory
+				@db.query_defer(@sql_query[0], @sql_query[1..-1]).then { |rows|
+					rows.map { |row|
+						{
+							full_number: row["tel"].sub(/\A\+1/, ""),
+							city: row["locality"],
+							state: row["region"]
+						}
+					}
+				}
+			end
+
 			def next_fallback
 				@memcache.set(cache_key, CBOR.encode([]), 43200)
+				fallback = @fallback.shift
 				self.class.new(
-					@fallback.shift.iris_query.merge(
+					fallback.iris_query.merge(
 						enableTNDetail: true, quantity: @iris_query[:quantity]
 					),
+					fallback.sql_query,
 					fallback: @fallback,
-					memcache: @memcache
+					memcache: @memcache, db: @db
 				).tns
 			end
 

test/test_tel_selections.rb

@@ -122,6 +122,35 @@ class TelSelectionsTest < Minitest::Test
 			)
 		end
 		em :test_fallback
+
+		def test_local_inventory
+			stub_request(
+				:get,
+				"https://dashboard.bandwidth.com/v1.0/accounts//availableNumbers" \
+					"?city=Kitchener-Waterloo&enableTNDetail=true&lCA=false&" \
+					"quantity=10&state=ON"
+			).to_return(status: 200, body: "")
+
+			db = FakeDB.new(
+				["ON", "Kitchener-Waterloo"] => [{
+					"tel" => "[phone]",
+					"region" => "ON",
+					"locality" => "Kitchener-Waterloo"
+				}]
+			)
+			form = Blather::Stanza::X.new
+			form.fields = [{ var: "q", value: "Kitchener, ON" }]
+			tns = execute_command do
+				TelSelections::ChooseTel::AvailableNumber
+					.for(form, db: db, memcache: FakeMemcache.new)
+					.tns
+			end
+			assert_equal(
+				["(226) 555-12345 (Kitchener-Waterloo, ON)"],
+				tns.map(&:to_s)
+			)
+		end
+		em :test_local_inventory
 	end
 
 	class TnTest < Minitest::Test