email-assistant/parsers/package-tracking/main.rb

46 lines
1.4 KiB
Ruby

require 'sinatra'
require 'tracking_number'
set :bind, '0.0.0.0'
set :port, 3000
def get_tracking_url(tracker)
tracking_urls = {
:ups => 'https://wwwapps.ups.com/WebTracking/track?track=yes&trackNums=%s',
:dhl => 'http://www.dhl.com/en/express/tracking.html?brand=DHL&AWB=%s',
:fedex => 'https://www.fedex.com/apps/fedextrack/?tracknumbers=%s',
:ontrac => 'http://www.ontrac.com/trackingres.asp?tracking_number=%s',
:usps => 'https://m.usps.com/m/TrackConfirmAction_detail?tLabels=%s',
}
tracking_url = tracking_urls[tracker.courier_code]
if tracking_url != nil
tracking_url = tracking_url % tracker.tracking_number
end
return tracking_url
end
# Simple status endpoint on root
get '/' do
'OK'
end
# Standard parser api receives PUT {"message": "Email body"} /parse
# Returns [{"token": "extracted token", "type": "token type", "metadata": {}]
post '/parse' do
body = JSON.parse(request.body.read)
trackers = TrackingNumber.search(body['message'])
results = []
for tracker in trackers do
results.push({
:token => tracker.tracking_number,
:type => 'SHIPPING',
:metadata => {
:carrier_name => tracker.courier_name,
:tracking_url => get_tracking_url(tracker),
}
})
end
JSON.dump(results)
end