Expose #to_json on a transfer

Author: arnoFleming

lib/we_transfer/remote_file.rb

@@ -3,7 +3,13 @@ module RemoteFile
     class FileMismatchError < StandardError; end
     class NoIoError < StandardError; end
 
-    class Multipart < ::Ks.strict(:chunks, :chunk_size); end
+    class Multipart < ::Ks.strict(:chunks, :chunk_size)
+      def to_h
+        %i[chunks chunk_size].each_with_object({}) do |prop, memo|
+          memo[prop] = send(prop)
+        end
+      end
+    end
 
     def self.upgrade(files:, files_response:)
       files_response.each do |file_response|

lib/we_transfer/transfer.rb

@@ -5,7 +5,7 @@ class NoFilesAddedError < StandardError; end
 
     extend Forwardable
 
-    attr_reader :files, :id, :state, :url
+    attr_reader :files, :id, :state, :url, :message
 
     class << self
       extend Forwardable
@@ -84,6 +84,19 @@ def as_request_params
       }
     end
 
+    def to_json
+      to_h.to_json
+    end
+
+    def to_h
+      prepared = %i[id state url message].each_with_object({}) do |prop, memo|
+        memo[prop] = send(prop)
+      end
+
+      prepared[:files] = files.map(&:to_h)
+      prepared
+    end
+
     private
 
     def_delegator self, :remote_transfer_params

lib/we_transfer/we_transfer_file.rb

@@ -1,6 +1,6 @@
 module WeTransfer
   class WeTransferFile
-    attr_reader :name, :id, :io, :multipart
+    attr_reader :name, :id, :io, :multipart, :size
 
     def initialize(name: nil, size: nil, io: nil)
       @io = MiniIO.new(io)
@@ -16,5 +16,14 @@ def as_request_params
         size: @size,
       }
     end
+
+    def to_h
+      prepared = %i[name size id].each_with_object({}) do |prop, memo|
+        memo[prop] = send(prop)
+      end
+      prepared[:multipart] = multipart.to_h
+
+      prepared
+    end
   end
 end

spec/features/client_creates_a_transfer_spec.rb

@@ -18,6 +18,8 @@
 
     transfer = client.transfer
 
+    transfer.to_json
+
     expect(transfer.url)
       .to be_nil