client-api-builder 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6e9f87a8414f328789843aa8a4cc91736c2453908cae60041e4bc780e43fda62
-  data.tar.gz: ccb71fac620b9beae7a308ae1f575da6ba7850578480b358c26e6191cafe5dfe
+  metadata.gz: '01795148989ed0a14c42cb886fd3ce76e3692d82c388f49adf09b7b7229b928f'
+  data.tar.gz: 4ce828ecc5ee2017e6318ab83e8c1c02e3a557f746a7f3f6fc8d7471cdac2750
 SHA512:
-  metadata.gz: 1139fd2bcf58929790aa3923b4f3fae69f63e4bc23c9e3e59bd785e56997434c287942fb563b7c6f2c7f6689076e00ac07a550a1b198868e2137d46ae33c24de
-  data.tar.gz: c77f20fc74bada0c03764d945506d8ebcf51fb4a0db96ebf2221d0b4d14191a762f08a76ac046f1450f80b5b6f51f19757d074e733e737294870e779f4ba479e
+  metadata.gz: 57b713cda946c263a13f6642d8548442f574eaea1f0374b1e02c7d2f071a3826aeb3e49eaa3a785784eeb7caa671584192f0391e6a508b5dbbd4fff32868063e
+  data.tar.gz: d41bfea574c3399d534455ab7165a9fc6f91c347fd5c6d6634fa67731bb5198ccc843532e06fea853097316f766522daebf875cce855694b87e67bb0ea6771fb

data/client-api-builder.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name        = 'client-api-builder'
-  s.version     = '0.2.2'
+  s.version     = '0.2.3'
   s.licenses    = ['MIT']
   s.summary     = 'Develop Client API libraries faster'
   s.description = 'Utility for constructing API clients'

data/lib/client_api_builder/net_http_request.rb

@@ -28,7 +28,30 @@ module ClientApiBuilder
         request.body = body if body
 
         Net::HTTP.start(uri.hostname, uri.port, connection_options.merge(use_ssl: uri.scheme == 'https')) do |http|
-          http.request(request)
+          http.request(request) do |response|
+            yield response if block_given?
+          end
+        end
+      end
+
+      def stream(method:, uri:, body:, headers:, connection_options:)
+        request(method: method, uri: uri, body: body, headers: headers, connection_options: connection_options) do |response|
+          response.read_body do |chunk|
+            yield chunk
+          end
+        end
+      end
+
+      def stream_to_io(method:, uri:, body:, headers:, connection_options:, io:)
+        stream(method: method, uri: uri, body: body, headers: headers, connection_options: connection_options) do |chunk|
+          io.write chunk
+        end
+      end
+
+      def stream_to_file(method:, uri:, body:, headers:, connection_options:, file:)
+        mode = connection_options.delete(:file_mode) || 'wb'
+        File.open(file, mode) do |io|
+          stream_to_io(method: method, uri: uri, body: body, headers: headers, connection_options: connection_options, io: io)
         end
       end
     end

data/lib/client_api_builder/router.rb

@@ -7,7 +7,7 @@ module ClientApiBuilder
       base.extend InheritanceHelper::Methods
       base.extend ClassMethods
       base.include ::ClientApiBuilder::NetHTTP::Request
-      base.attr_reader :response
+      base.attr_reader :response, :request_options
     end
 
     module ClassMethods
@@ -192,8 +192,18 @@ module ClientApiBuilder
           end
         expected_response_codes.map!(&:to_s)
 
+        stream_param =
+          case options[:stream]
+          when true,
+               :file
+            :file
+          when :io
+            :io
+          end
+
         method_args = named_arguments.map { |arg_name| "#{arg_name}:" }
         method_args += ['body:'] if has_body_param
+        method_args += ["#{stream_param}:"] if stream_param
         method_args += ['**__options__', '&block']
 
         code = "def #{method_name}(" + method_args.join(', ') + ")\n"
@@ -205,9 +215,30 @@ module ClientApiBuilder
         code += "  __body__ = build_body(__body__, __options__)\n"
         code += "  __headers__ = build_headers(__options__)\n"
         code += "  __connection_options__ = build_connection_options(__options__)\n"
-        code += "  @response = request(method: #{http_method.inspect}, uri: __uri__, body: __body__, headers: __headers__, connection_options: __connection_options__)\n"
+        code += "  @request_options = {method: #{http_method.inspect}, uri: __uri__, body: __body__, headers: __headers__, connection_options: __connection_options__}\n"
+        code += "  @request_options[:#{stream_param}] = #{stream_param}\n" if stream_param
+
+        case options[:stream]
+        when true,
+             :file,
+             :io
+          code += "  @response = stream_to_file(**@request_options)\n"
+        when :block
+          code += "  @response = stream(**@request_options, &block)\n"
+        else
+          code += "  @response = request(**@request_options)\n"
+        end
+
         code += "  expected_response_code!(@response, __expected_response_codes__, __options__)\n"
-        code += "  handle_response(@response, __options__, &block)\n"
+
+        if options[:stream] || options[:return] == :response
+          code += "  @response\n"
+        elsif options[:return] == :body
+          code += "  @response.body\n"
+        else
+          code += "  handle_response(@response, __options__, &block)\n"
+        end
+
         code += "end\n"
         code
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: client-api-builder
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Doug Youch
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-07-06 00:00:00.000000000 Z
+date: 2021-07-17 00:00:00.000000000 Z
 dependencies: []
 description: Utility for constructing API clients
 email: dougyouch@gmail.com