MQTT_WeatherGateway/src/server.rb

84 lines
2.2 KiB
Ruby

# frozen_string_literal: true
require 'bundler'
Bundler.require
Dotenv.load
Dotenv.require_keys('MOJI_APPCODE', 'IOT_MQTT_HOST', 'IOT_MQTT_PORT', 'IOT_MQTT_SSL')
wxapi = MojiWeather::Api::RestClient.new(app_code: ENV['MOJI_APPCODE'])
def extract_device_id(topic)
devid = %r{iot/weather/(.*)/request}.match(topic)
if devid.nil?
nil
else
devid[1]
end
end
begin
client = MQTT::Client.new
client.host = ENV['IOT_MQTT_HOST']
client.port = ENV['IOT_MQTT_PORT'].to_i
unless ENV['IOT_MQTT_SSL'].nil? || ENV['IOT_MQTT_SSL'] == 'false' || ENV['IOT_MQTT_SSL'] == 'no'
client.ssl = true
client.cert_file = ENV['IOT_MQTT_SSL_CERT_FILE']
client.key_file = ENV['IOT_MQTT_SSL_KEY_FILE']
client.ca_file = ENV['IOT_MQTT_SSL_CA_FILE']
end
client.connect
client.subscribe('iot/weather/#')
client.get do |topic, payload|
# this method also checks if this is from a request topic.
dev_id = extract_device_id(topic)
unless dev_id.nil?
puts "[#{dev_id}] <- #{payload.length}B"
# decode CBOR object, retrieve request.
dev_req = CBOR.decode(payload)
wx_cond = case dev_req['type']
when 'condition'
MojiWeather::Api::ApiType::CONDITION
when 'aqi'
MojiWeather::Api::ApiType::AQI
when 'forecast24'
MojiWeather::Api::ApiType::FORECAST_24HRS
end
req_params = if !dev_req['city_id'].nil?
{ city_id: dev_req['city_id'] }
elsif !dev_req['location'].nil?
{ location: { lat: dev_req['location']['lat'], lon: dev_req['location']['lon'] } }
else
puts "[#{dev_id}] not a valid request"
next
end
# Request external service for weather information
api_resp = wxapi.query(wx_cond, req_params)
# Encode CBOR object
# Publish to response topic.
resp = api_resp.to_cbor
puts "[#{dev_id}] -> #{resp.length}B"
client.publish("iot/weather/#{dev_id}/response", resp)
end
end
rescue SystemExit, Interrupt
puts "Interrupt caught, client [#{client}] disconnect."
client.disconnect
rescue StandardError => e
p e
end