1#!/usr/bin/env ruby
 2#
 3# Copyright (C) 2020  Denver Gingerich <denver@ossguy.com>
 4#
 5# This file is part of sgx-bwmsgsv2.
 6#
 7# sgx-bwmsgsv2 is free software: you can redistribute it and/or modify it under
 8# the terms of the GNU Affero General Public License as published by the Free
 9# Software Foundation, either version 3 of the License, or (at your option) any
10# later version.
11#
12# sgx-bwmsgsv2 is distributed in the hope that it will be useful, but WITHOUT
13# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
14# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
15# details.
16#
17# You should have received a copy of the GNU Affero General Public License along
18# with sgx-bwmsgsv2.  If not, see <http://www.gnu.org/licenses/>.
19
20require 'redis'
21require 'hiredis'
22require 'net/http'
23require 'json'
24require 'time'
25
26$stdout.sync = true
27
28puts "Redis queue to SGX HTTP request translator - for Bandwidth API V2 SGX\n"\
29	"==>> last commit of this version is " + `git rev-parse HEAD` + "\n"
30
31if ARGV.size != 3
32	# note that <redis_queue_suffix> should match h2r-bwmsgsv2's $queue_name
33	puts "Usage: r2s-bwmsgsv2.rb <redis_queue_suffix> <sgx_hostname> "\
34		"<sgx_https_port>"
35	exit 0
36end
37
38["INT", "TERM"].each do |sig|
39	trap(sig) do
40		puts "Translator has terminated due to SIG#{sig}."
41		exit 0
42	end
43end
44
45t = Time.now
46puts "LOG %d.%09d: starting...\n\n" % [t.to_i, t.nsec]
47
48redis = Redis.new(:driver => :hiredis)
49
50while true
51	timestamps_plus_json_blob = redis.brpoplpush('incoming_messages-' +
52		ARGV[0], 'pending_messages-' + ARGV[0])
53	# TODO: add timestamping here, and MUST include a ./tai call with it
54
55	# TODO: print less stuff in here
56	puts "TODO - got some stuff: " + timestamps_plus_json_blob
57
58	# add some timestamps to timestamps_plus_json_blob (our own)
59
60	# TODO: fix so sending timestamps_plus_json_blob to SGX, and via POST
61	response = Net::HTTP.get_response(
62		URI("https://#{ARGV[1]}:#{ARGV[2]}/r2tst.php")
63	)
64
65	puts "TODO - response.code = #{response.code} and body #{response.body}"
66
67	# if SGX gives back anything but response.code 200, retry 5 times-ish
68	# if SGX failed to respond after retries, email or other notify
69	#  and back off with even longer retries (can't do anything else really)
70
71	# if got HTTP 200, then:
72
73	# redis.set(...) with expiration of 3 days, including timestamps
74
75	# confirm ARGV[1] key has one element and e == timestamps_plus_json_blob
76	#  if so, then rpop ARGV[1] and discard
77	#  if not, then email or other notify, and probably exit with failure
78end