User hash upgrade

onewheelskyward · Sep 11, 2020 · 78b25f0 · 78b25f0
1 parent 6249908
commit 78b25f0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 10 deletions.
diff --git a/lib/lita/handlers/irc_handlers.rb b/lib/lita/handlers/irc_handlers.rb
@@ -358,19 +358,24 @@ def handle_ansi_aqi(response)
         show = response.matches[0][0]
       else
         Lita.logger.debug response.user.name
-        if response.user.name == 'lampd1'
-          show = '38447'
-        elsif response.user.name == 'aaronpk' or response.user.name == 'zenlinux'
-          show = '43023'
-        elsif response.user.name == 'djwong'
-          show = '61137'
-        elsif response.user.name == 'philtor'
-          show = '35221'
-        else
+        users = {lampd1: 38447,
+                 aaronpk: 43023,
+                 zenlinux: 43023,
+                 djwong: 61137,
+                 philtor: 35221,
+                 agb: 34409}
+        users.keys.each do |u|
+          if u.to_s == response.user.name
+            show = users[u]
+          end
+        end
+
+        unless show
           Lita.logger.debug "Defaulting to pdx"
           show = "9814"
         end
       end
+
       resp = RestClient.get "https://www.purpleair.com/json?show=#{show}"
       aqi = JSON.parse resp
       label = aqi['results'][0]['Label']

diff --git a/spec/lita/handlers/forecast_io_spec.rb b/spec/lita/handlers/forecast_io_spec.rb
@@ -566,7 +566,7 @@ def mock_up(filename)
   it 'aqis' do
     mock_up 'aqi'
     send_command 'ansiaqi'
-    expect(replies.last).to eq("AQI report for Portland, OR: PM2.5 \u00030875.1\u0003 |\u000308_\u000304▅\u000305████\u0003| \u000305322.93\u0003 max: \u000305322.93\u0003 \u000314(7 day average to 10 min average)\u0003")
+    expect(replies.last).to eq("AQI report for PSU STAR LAB SEL: PM2.5 \u00030875.1\u0003 |\u000308_\u000304▅\u000305████\u0003| \u000305322.93\u0003 max: \u000305322.93\u0003 \u000314(7 day average to 10 min average)\u0003")
   end
 
 end