homebrew-core/Formula/inlets.rb
2019-11-28 12:45:08 -05:00

130 lines
4.1 KiB
Ruby

class Inlets < Formula
desc "Expose your local endpoints to the Internet"
homepage "https://github.com/inlets/inlets"
url "https://github.com/inlets/inlets.git",
:tag => "2.6.3",
:revision => "c2033ecaaef9381d975a3fcadb86011056865fb9"
bottle do
cellar :any_skip_relocation
sha256 "962ea637b1ca6caa9fdd2177ee4ef25eea02a0a1e083e75fc27d6fc03dc27832" => :catalina
sha256 "f3a3903adce74d467c1751cd9b8cb67bd2af3f005202c410dbe4305edbcf2140" => :mojave
sha256 "6cfb002a09e2ad05989fa0c3f95ce7ed2471468ff7f82c4c71f66b8654f0868f" => :high_sierra
end
depends_on "go" => :build
def install
ENV["GOPATH"] = buildpath
(buildpath/"src/github.com/inlets/inlets").install buildpath.children
cd "src/github.com/inlets/inlets" do
commit = Utils.popen_read("git", "rev-parse", "HEAD").chomp
system "go", "build", "-ldflags",
"-s -w -X main.GitCommit=#{commit} -X main.Version=#{version}",
"-a",
"-installsuffix", "cgo", "-o", bin/"inlets"
prefix.install_metafiles
end
end
def cleanup(name, pid)
puts "Tearing down #{name} on PID #{pid}"
Process.kill("TERM", pid)
Process.wait(pid)
end
MOCK_RESPONSE = "INLETS OK".freeze
SECRET_TOKEN = "itsasecret-sssshhhhh".freeze
test do
upstream_server = TCPServer.new(0)
upstream_port = upstream_server.addr[1]
remote_server = TCPServer.new(0)
remote_port = remote_server.addr[1]
upstream_server.close
remote_server.close
puts "Starting mock server on: localhost:#{upstream_port}"
(testpath/"mock_upstream_server.rb").write <<~EOS
require 'socket'
server = TCPServer.new('localhost', #{upstream_port})
loop do
socket = server.accept
request = socket.gets
STDERR.puts request
response = "OK\\n"
shutdown = false
if request.include? "inlets-test"
response = "#{MOCK_RESPONSE}\\n"
shutdown = true
end
socket.print "HTTP/1.1 200 OK\\r\\n" +
"Host: localhost:#{upstream_port}\\r\\n" +
"Content-Type: text/plain\\r\\n" +
"Content-Length: \#\{response.bytesize\}\\r\\n" +
"Connection: close\\r\\n"
socket.print "\\r\\n"
socket.print response
socket.close
if shutdown
puts "Exiting test server"
exit 0
end
end
EOS
mock_upstream_server_pid = fork do
exec "ruby mock_upstream_server.rb"
end
begin
require "uri"
require "net/http"
stable_resource = stable.instance_variable_get(:@resource)
commit = stable_resource.instance_variable_get(:@specs)[:revision]
# Basic --version test
inlets_version = shell_output("#{bin}/inlets version")
assert_match /\s#{commit}$/, inlets_version
assert_match /\s#{version}$/, inlets_version
# Client/Server e2e test
# This test involves establishing a client-server inlets tunnel on the
# remote_port, running a mock server on the upstream_port and then
# testing that we can hit the mock server upstream_port via the tunnel remote_port
puts "Waiting for mock server"
sleep 3
server_pid = fork do
puts "Starting inlets server with port #{remote_port}"
exec "#{bin}/inlets server --port #{remote_port} --token #{SECRET_TOKEN}"
end
client_pid = fork do
puts "Starting inlets client with remote localhost:#{remote_port}, upstream localhost:#{upstream_port}, token: #{SECRET_TOKEN}"
exec "#{bin}/inlets client --remote localhost:#{remote_port} --upstream localhost:#{upstream_port} --token #{SECRET_TOKEN}"
end
puts "Waiting for inlets websocket tunnel"
sleep 3
uri = URI("http://localhost:#{remote_port}/inlets-test")
puts "Querying upstream endpoint via inlets remote: #{uri}"
response = Net::HTTP.get_response(uri)
assert_match MOCK_RESPONSE, response.body
assert_equal response.code, "200"
ensure
cleanup("Mock Server", mock_upstream_server_pid)
cleanup("Inlets Server", server_pid)
cleanup("Inlets Client", client_pid)
end
end
end