1#!/usr/bin/env ruby
2#
3# Copyright (C) 2017 Denver Gingerich <denver@ossguy.com>
4#
5# This file is part of sgx-catapult.
6#
7# sgx-catapult 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-catapult 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-catapult. If not, see <http://www.gnu.org/licenses/>.
19
20puts "Soprani.ca/MMS Proxy for XMPP - Catapult v0.004\n\n"
21
22require 'goliath'
23require 'net/http'
24require 'redis/connection/hiredis'
25require 'uri'
26
27if ARGV.size != 3 then
28 puts "Usage: mpx-catapult.rb <http_listen_port> " +
29 "<redis_hostname> <redis_port>"
30 exit 0
31end
32
33t = Time.now
34puts "LOG %d.%09d: starting...\n\n" % [t.to_i, t.nsec]
35
36class WebhookHandler < Goliath::API
37 def response(env)
38 puts 'ENV: ' + env.to_s
39 puts 'path: ' + env['REQUEST_PATH']
40 puts 'method: ' + env['REQUEST_METHOD']
41 puts 'BODY: ' + Rack::Request.new(env).body.read
42
43 cred_key = "catapult_cred-"+env['REQUEST_PATH'].split('/', 3)[1]
44
45 # TODO: connect at start of program instead
46 conn = Hiredis::Connection.new
47 begin
48 conn.connect(ARGV[1], ARGV[2].to_i)
49 rescue => e
50 puts 'ERROR: Redis connection failed: ' + e.inspect
51 return [500, {'Content-Type' => 'text/plain'},
52 e.inspect]
53 end
54
55 conn.write ["EXISTS", cred_key]
56 if conn.read == 0
57 conn.disconnect
58
59 puts 'ERROR: invalid path rqst: ' + env['REQUEST_PATH']
60 return [404, {'Content-Type' => 'text/plain'},
61 'not found']
62 end
63
64 conn.write ["LRANGE", cred_key, 0, 3]
65 # we don't actually use users_num, but easier to read so left in
66 user_id, api_token, api_secret, users_num = conn.read
67 conn.disconnect
68
69 uri = URI.parse('https://api.catapult.inetwork.com')
70 http = Net::HTTP.new(uri.host, uri.port)
71 http.use_ssl = true
72 request = ''
73 if env['REQUEST_METHOD'] == 'GET'
74 request = Net::HTTP::Get.new('/v1/users/' + user_id +
75 '/media/' +env['REQUEST_PATH'].split('/', 3)[2])
76 elsif env['REQUEST_METHOD'] == 'HEAD'
77 request = Net::HTTP::Head.new('/v1/users/' + user_id +
78 '/media/' +env['REQUEST_PATH'].split('/', 3)[2])
79 else
80 puts 'ERROR: received non-HEAD/-GET request'
81 return [500, {'Content-Type' => 'text/plain'},
82 e.inspect]
83 end
84 request.basic_auth api_token, api_secret
85 response = http.request(request)
86
87 puts 'API response to send: ' + response.to_s + ' with code ' +
88 response.code + ', body <omitted_due_to_length>'
89
90 if response.code != '200'
91 puts 'ERROR: unexpected return code ' + response.code
92
93 if response.code == '404'
94 return [404, {'Content-Type' => 'text/plain'},
95 'not found']
96 end
97
98 return [response.code, {'Content-Type' => 'text/plain'},
99 'unexpected error']
100 end
101
102 # TODO: maybe need to reflect more headers (multi-part?)
103 [200, {'Content-Length' => response['content-length']},
104 response.body]
105 end
106end
107
108EM.run do
109 server = Goliath::Server.new('0.0.0.0', ARGV[0].to_i)
110 server.api = WebhookHandler.new
111 server.app = Goliath::Rack::Builder.build(server.api.class, server.api)
112 server.logger = Log4r::Logger.new('goliath')
113 server.logger.add(Log4r::StdoutOutputter.new('console'))
114 server.logger.level = Log4r::INFO
115 server.start
116end