consign 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 8c62636b6885f9f8bff11cae0039659e05ce5808744d52fca7afee29d9616183
+  data.tar.gz: 73a23bf8e80a50b3d0f1e12d6323887a761aa5eb8ddb0d2d5ed73fbe93e9c562
+SHA512:
+  metadata.gz: b690e81fe0be9e0dc41a213db28dc094f78dd5a209301b0aabae1c3ce65b37cc0c4b0c5a82b0a68e269fa7be4ca06a61821003a547e608f0e649ff01ac877392
+  data.tar.gz: 5b9f940ceb11132378cd6ded5840a32ce80f015d74019ebd0c6878117f54e15415c8d867b2197030bbfd37b81692ebee14963bd238dd247c841797025c55ea42

data/lib/consign/railtie.rb

@@ -0,0 +1,27 @@
+module Consign
+  class Railtie < ::Rails::Railtie
+    config.before_configuration do
+      Consign.api_key = ENV["CONSIGN_API_KEY"]
+
+      host = Consign.local_mode ? ".lvh.me" : ".consign.dev"
+      Rails.application.config.hosts << host
+    end
+
+    server do
+      if Rails.env.development? && Consign.api_key.present?
+        Thread.new { Consign::Tunnel.start(port: detect_port) }
+      else
+        puts "[Consign] Tunnel disabled. Set CONSIGN_API_KEY to enable." if Rails.env.development?
+      end
+    end
+
+    private
+      def self.detect_port
+        if (idx = ARGV.index("-p") || ARGV.index("--port"))
+          ARGV[idx + 1].to_i
+        else
+          ENV.fetch("PORT", 3000).to_i
+        end
+      end
+  end
+end

data/lib/consign/tunnel.rb

@@ -0,0 +1,178 @@
+require "socket"
+require "net/http"
+require "uri"
+require "json"
+require "openssl"
+
+module Consign
+  class Tunnel
+    MAX_RETRIES = 10
+    RETRY_DELAY = 1
+
+    def self.start(port: nil)
+      new(port: port).start
+    end
+
+    def initialize(port: nil)
+      @local_port = port || ENV.fetch("PORT", "3000").to_i
+    end
+
+    def start
+      log "🔐 Authenticating with #{control_url}..."
+
+      if config = fetch_configuration
+        log "🔌 Connecting to tunnel server..."
+        run_tunnel_loop(config)
+      else
+        log "❌ Failed to authenticate after multiple attempts.", :error
+      end
+    end
+
+    private
+      def fetch_configuration
+        attempt_with_retries(max: MAX_RETRIES) do
+          response = post_authentication
+
+          if response.is_a?(Net::HTTPSuccess)
+            JSON.parse(response.body)
+          else
+            raise "Authentication failed: #{response.code}"
+          end
+        end
+      end
+
+      def run_tunnel_loop(config)
+        loop do
+          begin
+            connect_and_serve(config)
+          rescue Interrupt
+            break
+          rescue StandardError => e
+            log "⚠️  Tunnel disconnected: #{e.message}. Reconnecting...", :warn
+            sleep RETRY_DELAY
+          end
+        end
+      end
+
+      def connect_and_serve(config)
+        socket = open_socket(config["server_host"], config["server_port"])
+        return unless perform_handshake(socket, config)
+
+        log_connected(config)
+
+        while line = socket.gets
+          handle_request(socket, JSON.parse(line))
+        end
+      ensure
+        socket&.close
+      end
+
+      def open_socket(host, port)
+        socket = TCPSocket.new(host, port)
+        if port == 443
+          ctx = OpenSSL::SSL::SSLContext.new
+          socket = OpenSSL::SSL::SSLSocket.new(socket, ctx)
+          socket.hostname = host
+          socket.connect
+        end
+        socket
+      end
+
+      def perform_handshake(socket, config)
+        req = "GET /_connect HTTP/1.1\r\n" \
+              "Host: #{config['server_host']}\r\n" \
+              "Connection: Upgrade\r\n" \
+              "Upgrade: consign\r\n" \
+              "X-Consign-Subdomain: #{config['subdomain']}\r\n" \
+              "X-Consign-Token: #{config['otp']}\r\n" \
+              "\r\n"
+        socket.write(req)
+
+        read_handshake_response(socket)
+      end
+
+      def read_handshake_response(socket)
+        buffer = ""
+        while (line = socket.gets)
+          buffer += line
+          break if line == "\r\n"
+        end
+        buffer.start_with?("HTTP/1.1 101")
+      end
+
+      def handle_request(socket, req)
+        Thread.new do
+          begin
+            response = forward_to_local(req)
+            socket.puts({ id: req['id'], status: response.code.to_i, headers: response.to_hash, body: response.body }.to_json)
+          rescue StandardError => e
+            socket.puts({ id: req['id'], status: 502, headers: {}, body: "Bad Gateway: #{e.message}" }.to_json)
+          end
+        end
+      end
+
+      def forward_to_local(req)
+        uri = URI("http://127.0.0.1:#{@local_port}#{req['url']}")
+
+        Net::HTTP.start(uri.hostname, uri.port) do |http|
+          local_req = Net::HTTPGenericRequest.new(req['method'], true, true, req['url'])
+          req['headers']&.each { |k, v| local_req[k] = v.first }
+
+          if req['host']
+            local_req['Host'] = req['host']
+            local_req['X-Forwarded-Host'] = req['host']
+            local_req['X-Forwarded-Proto'] = "http" # or https if we were terminating TLS locally
+            local_req['X-Forwarded-Port'] = "80"
+          end
+
+          local_req.body = req['body'] if req['body']
+          http.request(local_req)
+        end
+      end
+
+      def post_authentication
+        uri = URI("#{control_url}/api/v1/tunnels")
+        req = Net::HTTP::Post.new(uri)
+        req["Authorization"] = "Token token=\"#{Consign.api_key}\""
+
+        Net::HTTP.start(uri.hostname, uri.port, use_ssl: uri.scheme == "https") do |http|
+          http.request(req)
+        end
+      end
+
+      def attempt_with_retries(max:)
+        attempts = 0
+        begin
+          yield
+        rescue StandardError => e
+          attempts += 1
+          if attempts < max
+            sleep RETRY_DELAY
+            retry
+          end
+          nil
+        end
+      end
+
+      def control_url
+        ENV.fetch("CONSIGN_URL", default_host)
+      end
+
+      def default_host
+        Consign.local_mode ? "http://localhost:3000" : "https://www.consign.dev"
+      end
+
+      def log_connected(config)
+        display_host = config["server_host"]
+        display_host = "lvh.me" if display_host == "127.0.0.1"
+        display_host = display_host.sub(/^tunnel\./, "") if display_host.start_with?("tunnel.")
+
+        log "✅ Tunnel active: https://#{config['subdomain']}.#{display_host}"
+      end
+
+      def log(msg, level = :info)
+        output = "[Consign] #{msg}"
+        defined?(Rails) ? Rails.logger.send(level, output) : puts(output)
+      end
+  end
+end

data/lib/consign/version.rb

@@ -0,0 +1,3 @@
+module Consign
+  VERSION = "0.1.1"
+end

data/lib/consign.rb

@@ -0,0 +1,15 @@
+require "consign/version"
+require "consign/railtie"
+require "consign/tunnel"
+
+module Consign
+  mattr_accessor :api_key
+  mattr_accessor :local_mode
+
+  @@environment = :production
+  @@local_mode = false
+
+  def self.configure
+    yield self
+  end
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification
+name: consign
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Consign Team
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.0'
+description: Expose your local Rails server to the world with a public subdomain.
+email:
+- hello@consign.dev
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/consign.rb
+- lib/consign/railtie.rb
+- lib/consign/tunnel.rb
+- lib/consign/version.rb
+homepage: https://www.consign.dev
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://www.consign.dev
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 4.0.3
+specification_version: 4
+summary: Zero-config tunneling for Rails
+test_files: []