rhenium/plum

# frozen-string-literal: true

module Plum
  class Client
    DEFAULT_CONFIG = {
      https: true,
      hostname: nil,
      verify_mode: OpenSSL::SSL::VERIFY_PEER,
      ssl_context: nil,
      http2_settings: {},
      user_agent: "plum/#{Plum::VERSION}",
      auto_decode: true,
    }.freeze

    attr_reader :host, :port, :config
    attr_reader :socket, :session

    # Creates a new HTTP client and starts communication.
    # A shorthand for `Plum::Client.new(args).start(&block)`
    def self.start(host, port = nil, config = {}, &block)
      client = self.new(host, port, config)
      client.start(&block)
    end

    # Creates a new HTTP client.
    # @param host [String | IO] the host to connect, or IO object.
    # @param port [Integer] the port number to connect
    # @param config [Hash<Symbol, Object>] the client configuration
    def initialize(host, port = nil, config = {})
      if host.is_a?(IO)
        @socket = host
      else
        @socket = nil
        @host = host
        @port = port || (config[:https] ? 443 : 80)
      end
      @config = DEFAULT_CONFIG.merge(hostname: host).merge!(config)
      @started = false
    end

    # Starts communication.
    # If block passed, waits for asynchronous requests and closes the connection after calling the block.
    def start(&block)
      raise IOError, "Session already started" if @started
      _start
      if block_given?
        begin
          ret = yield(self)
          resume
          return ret
        ensure
          close
        end
      end
      self
    end

    # Resume communication with the server until all running requests are complete.
    def resume
      @session.succ until @session.empty?
    end

    # Closes the connection immediately.
    def close
      @session.close if @session
    ensure
      @socket.close if @socket
    end

    # Creates a new HTTP request.
    # @param headers [Hash<String, String>] the request headers
    # @param body [String] the request body
    # @param options [Hash<Symbol, Object>] request options
    # @param block [Proc] if passed, it will be called when received response headers.
    def request(headers, body, options = {}, &block)
      raise ArgumentError, ":method and :path headers are required" unless headers[":method"] && headers[":path"]
      @session.request(headers, body, @config.merge(options), &block)
    end

    # @!method get
    # @!method head
    # @!method delete
    # @param path [String] the absolute path to request (translated into :path header)
    # @param options [Hash<Symbol, Object>] the request options
    # @param block [Proc] if specified, calls the block when finished
    # Shorthand method for `#request`
    %w(GET HEAD DELETE).each { |method|
      define_method(:"#{method.downcase}") do |path, options = {}, &block|
        _request_helper(method, path, nil, options, &block)
      end
    }

    # @!method post
    # @!method put
    # @param path [String] the absolute path to request (translated into :path header)
    # @param body [String] the request body
    # @param options [Hash<Symbol, Object>] the request options
    # @param block [Proc] if specified, calls the block when finished
    # Shorthand method for `#request`
    %w(POST PUT).each { |method|
      define_method(:"#{method.downcase}") do |path, body, options = {}, &block|
        _request_helper(method, path, body, options, &block)
      end
    }

    private
    # @return [Boolean] http2 nego?
    def _connect
      @socket = TCPSocket.open(@host, @port)

      if @config[:https]
        ctx = @config[:ssl_context] || new_ssl_ctx
        @socket = OpenSSL::SSL::SSLSocket.new(@socket, ctx)
        @socket.hostname = @config[:hostname]
        @socket.sync_close = true
        @socket.connect
        if ctx.verify_mode != OpenSSL::SSL::VERIFY_NONE
          @socket.post_connection_check(@config[:hostname])
        end

        @socket.alpn_protocol == "h2"
      end
    end

    def _start
      @started = true
      nego = @socket || _connect

      if @config[:https]
        klass = nego ? ClientSession : LegacyClientSession
      else
        klass = UpgradeClientSession
      end

      @session = klass.new(@socket, @config)
    end

    def new_ssl_ctx
      ctx = OpenSSL::SSL::SSLContext.new
      ctx.ssl_version = :TLSv1_2
      ctx.verify_mode = @config[:verify_mode]
      cert_store = OpenSSL::X509::Store.new
      cert_store.set_default_paths
      ctx.cert_store = cert_store
      ctx.ciphers = "ALL:!" + SSLSocketServerConnection::CIPHER_BLACKLIST.join(":!")
      ctx.alpn_protocols = ["h2", "http/1.1"]
      ctx
    end

    def _request_helper(method, path, body, options, &block)
      base = { ":method" => method,
               ":path" => path,
               "user-agent" => @config[:user_agent] }
      base.merge!(options[:headers]) if options[:headers]
      request(base, body, options, &block)
    end
  end
end