flexirest 1.8.1 → 1.8.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c49db36f1d2d0003480f590b89099c921838b25acdca881383654ca88e1fe961
-  data.tar.gz: 05ddbaf9a0ad3819f876585dfec4eeab53e7d746b9ac55ea7b8810a7fea5c3b9
+  metadata.gz: e5882e42d04af9b4712dd46b441168bb9977e8dd14e7e5a9bd870af567395b27
+  data.tar.gz: da50ac5d8b4e980767ac10d4e7b81e47cfaf6814646689d4cf221d08066c3718
 SHA512:
-  metadata.gz: 7c51ca15f2502dfe13a5035fb6ab1566fde2b1bb32fdc5b946e7cae60fcbd778da085dbdc404e73a245312c4ad54e7f5f48161f625ffd130981f4a55a232f219
-  data.tar.gz: b4216957e4ceea4255ca35375986c9fe98c8e865b79a2426b81d7268d1150a67c704615270dc429ffa07e60d6b114d9d5d93a33b75e9c45f9c4e8d9686a28d9d
+  metadata.gz: af1c263325803f45a3c6c79422eb97caee973b18d2ee492f5fdc6183d315817379815686a35caa71acd54f0fda0060b469bac9871c2a3258640eea4c19752e28
+  data.tar.gz: 5f06638ddf0978c0f908e5c79b2e34d0c24d0d362de0e9df2a972561f73dcede626a9947ac901caa612e175f05458fe40d279d881e19a6b22c0f69bac1c371fa

data/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 1.8.2
+
+Features:
+
+- Allowed use of `:form_multipart` as a `request_body_type` to allow uploading files to an API.
+
 ## 1.8.1
 
 Features:

data/docs/body-types.md

@@ -17,6 +17,8 @@ Flexirest::Base.request_body_type = :json
 
 This will also set the header `Content-Type` to `application/x-www-form-urlencoded` by default or `application/json; charset=utf-8` when `:json`. You can override this using the callback `before_request`.
 
+You can also use `:form_multipart` if your API requires file uploads. Any parameters set to `File` like object (supporting `#path` and `#read`) will be automatically uploaded with the parameters, in a normal form submission style.
+
 If you have an API that is inconsistent in its body type requirements, you can also specify it on the individual method mapping:
 
 ```ruby

data/lib/flexirest.rb

@@ -13,6 +13,7 @@ require "flexirest/result_iterator"
 require "flexirest/headers_list"
 require "flexirest/lazy_loader"
 require "flexirest/lazy_association_loader"
+require "flexirest/multipart"
 require "flexirest/json_api_proxy"
 require "flexirest/request"
 require "flexirest/request_delegator"

data/lib/flexirest/multipart.rb

@@ -0,0 +1,81 @@
+# Takes a hash of string and file parameters and returns a string of text
+# formatted to be sent as a multipart form post.
+#
+# Author:: Cody Brimhall <mailto:brimhall@somuchwit.com>
+# Created:: 22 Feb 2008
+# License:: Distributed under the terms of the WTFPL (http://www.wtfpl.net/txt/copying/)
+
+require 'rubygems'
+require 'mime/types'
+require 'cgi'
+
+
+module Flexirest
+  module Multipart
+    VERSION = "1.0.0"
+
+    # Formats a given hash as a multipart form post
+    # If a hash value responds to :string or :read messages, then it is
+    # interpreted as a file and processed accordingly; otherwise, it is assumed
+    # to be a string
+    class Post
+      BOUNDARY = "FLEXIRESTBOUNDARY-20190918-FLEXIRESTBOUNDARY"
+      CONTENT_TYPE = "multipart/form-data; boundary=#{BOUNDARY}"
+      HEADER = {"Content-Type" => CONTENT_TYPE}
+
+      def self.prepare_query(params)
+        fp = []
+
+        params.stringify_keys.each do |k, v|
+          # Are we trying to make a file parameter?
+          if v.respond_to?(:path) and v.respond_to?(:read) then
+            fp.push(FileParam.new(k, v.path, v.read))
+          # We must be trying to make a regular parameter
+          else
+            fp.push(StringParam.new(k, v))
+          end
+        end
+
+        # Assemble the request body using the special multipart format
+        query = fp.collect {|p| "--" + BOUNDARY + "\r\n" + p.to_multipart }.join("") + "--" + BOUNDARY + "--"
+        return query, HEADER
+      end
+    end
+
+    private
+
+    # Formats a basic string key/value pair for inclusion with a multipart post
+    class StringParam
+      attr_accessor :k, :v
+
+      def initialize(k, v)
+        @k = k
+        @v = v
+      end
+
+      def to_multipart
+        return "Content-Disposition: form-data; name=\"#{CGI::escape(k)}\"\r\n\r\n#{v}\r\n"
+      end
+    end
+
+    # Formats the contents of a file or string for inclusion with a multipart
+    # form post
+    class FileParam
+      attr_accessor :k, :filename, :content
+
+      def initialize(k, filename, content)
+        @k = k
+        @filename = filename
+        @content = content
+      end
+
+      def to_multipart
+        # If we can tell the possible mime-type from the filename, use the
+        # first in the list; otherwise, use "application/octet-stream"
+        mime_type = MIME::Types.type_for(filename)[0] || MIME::Types["application/octet-stream"][0]
+        return "Content-Disposition: form-data; name=\"#{CGI::escape(k)}\"; filename=\"#{ filename }\"\r\n" +
+               "Content-Type: #{ mime_type.simplified }\r\n\r\n#{ content }\r\n"
+      end
+    end
+  end
+end

data/lib/flexirest/request.rb

@@ -400,6 +400,8 @@ module Flexirest
       elsif http_method == :get || (http_method == :delete && !@method[:options][:send_delete_body])
         if request_body_type == :form_encoded
           headers["Content-Type"] ||= "application/x-www-form-urlencoded; charset=utf-8"
+        elsif request_body_type == :form_multipart
+          headers["Content-Type"] ||= "multipart/form-data; charset=utf-8"
         elsif request_body_type == :json
           headers["Content-Type"] ||= "application/json; charset=utf-8"
         end
@@ -417,6 +419,23 @@ module Flexirest
           p.to_query
         end
         headers["Content-Type"] ||= "application/x-www-form-urlencoded"
+      elsif request_body_type == :form_multipart
+        headers["Content-Type"] ||= "multipart/form-data; charset=utf-8"
+        @body ||= if params.is_a?(String)
+          params
+        elsif @post_params.is_a?(String)
+          @post_params
+        else
+          p = (params || @post_params || {})
+          if @method[:options][:wrap_root].present?
+            p = {@method[:options][:wrap_root] => p}
+          end
+          data, mp_headers = Flexirest::Multipart::Post.prepare_query(p)
+          mp_headers.each do |k,v|
+            headers[k] = v
+          end
+          data
+        end
       elsif request_body_type == :json
         @body ||= if params.is_a?(String)
           params

data/lib/flexirest/version.rb

@@ -1,3 +1,3 @@
 module Flexirest
-  VERSION = "1.8.1"
+  VERSION = "1.8.2"
 end

data/spec/lib/request_spec.rb

@@ -353,6 +353,28 @@ describe Flexirest::Request do
     ExampleClient.update id:1234, debug:true, test:'foo'
   end
 
+  it "should encode the body in a form-encoded format by default" do
+    body = "--FLEXIRESTBOUNDARY-20190918-FLEXIRESTBOUNDARY\r\n" +
+      "Content-Disposition: form-data; name=\"debug\"\r\n" +
+      "\r\n" +
+      "true\r\n" +
+      "--FLEXIRESTBOUNDARY-20190918-FLEXIRESTBOUNDARY\r\n" +
+      "Content-Disposition: form-data; name=\"test\"\r\n" +
+      "\r\n" +
+      "foo\r\n" +
+      "--FLEXIRESTBOUNDARY-20190918-FLEXIRESTBOUNDARY\r\n" +
+      "Content-Disposition: form-data; name=\"file\"; filename=\"#{File.dirname(__FILE__)}/../../spec/samples/file.txt\"\r\n" +
+      "Content-Type: text/plain\r\n" +
+      "\r\n" +
+      "The quick brown fox jumps over the lazy dog\n\r\n" +
+      "--FLEXIRESTBOUNDARY-20190918-FLEXIRESTBOUNDARY--"
+    expect_any_instance_of(Flexirest::Connection).to receive(:put).with(
+      "/put/1234", body, hash_including(headers: hash_including("Content-Type"=>"multipart/form-data; boundary=FLEXIRESTBOUNDARY-20190918-FLEXIRESTBOUNDARY"))
+    ).and_return(::FaradayResponseMock.new(OpenStruct.new(body:"{\"result\":true}", response_headers:{})))
+    ExampleClient.request_body_type :form_multipart
+    ExampleClient.update id:1234, debug:true, test:'foo', file: File.open("#{File.dirname(__FILE__)}/../../spec/samples/file.txt")
+  end
+
   it "should encode the body in a JSON format if specified" do
     expect_any_instance_of(Flexirest::Connection).to receive(:put).with("/put/1234", %q({"debug":true,"test":"foo"}), an_instance_of(Hash)).and_return(::FaradayResponseMock.new(OpenStruct.new(body:"{\"result\":true}", response_headers:{})))
     ExampleClient.request_body_type :json

data/spec/samples/file.txt

@@ -0,0 +1 @@
+The quick brown fox jumps over the lazy dog

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: flexirest
 version: !ruby/object:Gem::Version
-  version: 1.8.1
+  version: 1.8.2
 platform: ruby
 authors:
 - Andy Jeffries
@@ -293,6 +293,7 @@ files:
 - lib/flexirest/logger.rb
 - lib/flexirest/mapping.rb
 - lib/flexirest/monkey_patching.rb
+- lib/flexirest/multipart.rb
 - lib/flexirest/plain_response.rb
 - lib/flexirest/proxy_base.rb
 - lib/flexirest/recording.rb
@@ -327,6 +328,7 @@ files:
 - spec/lib/ruby_core_spec.rb
 - spec/lib/validation_spec.rb
 - spec/lib/xml_spec.rb
+- spec/samples/file.txt
 - spec/spec_helper.rb
 homepage: https://andyjeffries.co.uk/
 licenses:
@@ -382,4 +384,5 @@ test_files:
 - spec/lib/ruby_core_spec.rb
 - spec/lib/validation_spec.rb
 - spec/lib/xml_spec.rb
+- spec/samples/file.txt
 - spec/spec_helper.rb