wetransfer 0.4.4 → 0.9.0.beta

data/lib/we_transfer_client/board_builder.rb

@@ -0,0 +1,33 @@
+class BoardBuilder
+  attr_reader :items
+  class TransferIOError < StandardError; end
+
+  def initialize
+    @items = []
+  end
+
+  def add_file(name:, io:)
+    ensure_io_compliant!(io)
+    @items << FutureFile.new(name: name, io: io)
+  end
+
+  def add_file_at(path:)
+    add_file(name: File.basename(path), io: File.open(path, 'rb'))
+  end
+
+  def add_web_url(url:, title: url)
+    @items << FutureLink.new(url: url, title: title)
+  end
+
+  private
+
+  def ensure_io_compliant!(io)
+    io.seek(0)
+    io.read(1) # Will cause things like Errno::EACCESS to happen early, before the upload begins
+    io.seek(0) # Also rewinds the IO for later uploading action
+    size = io.size # Will cause a NoMethodError
+    raise TransferIOError, "#{File.basename(io)}, given to add_file has a size of 0" if size <= 0
+  rescue NoMethodError
+    raise TransferIOError, "#{File.basename(io)}, given to add_file must respond to seek(), read() and size(), but #{io.inspect} did not"
+  end
+end

data/lib/we_transfer_client/boards.rb

@@ -0,0 +1,74 @@
+module WeTransfer
+  class Client
+    module Boards
+      def create_board_and_upload_items(name:, description:, &block)
+        future_board = create_feature_board(name: name, description: description, &block)
+        remote_board = create_remote_board(board: future_board)
+        remote_board.files.each do |file|
+          check_for_file_duplicates(future_board.files, file)
+          local_file = future_board.files.select { |x| x.name == file.name }.first
+          upload_file(object: remote_board, file: file, io: local_file.io)
+          complete_file!(object: remote_board, file: file)
+        end
+        remote_board
+      end
+
+      def get_board(board:)
+        request_board(board: board)
+      end
+
+      private
+
+      def create_board(name:, description:, &block)
+        future_board = create_feature_board(name: name, description: description, &block)
+        create_remote_board(board: future_board)
+      end
+
+      def add_items(board:, board_builder_class: BoardBuilder)
+        builder = board_builder_class.new
+        yield(builder)
+        add_items_to_remote_board(items: builder.items, remote_board: board)
+      rescue LocalJumpError
+        raise ArgumentError, 'No items where added to the board'
+      end
+
+      def create_feature_board(name:, description:, future_board_class: FutureBoard, board_builder_class: BoardBuilder)
+        builder = board_builder_class.new
+        yield(builder) if block_given?
+        future_board_class.new(name: name, description: description, items: builder.items)
+      end
+
+      def create_remote_board(board:, remote_board_class: RemoteBoard)
+        authorize_if_no_bearer_token!
+        response = faraday.post(
+          '/v2/boards',
+          JSON.pretty_generate(board.to_initial_request_params),
+          auth_headers.merge('Content-Type' => 'application/json')
+        )
+        ensure_ok_status!(response)
+        remote_board = remote_board_class.new(JSON.parse(response.body, symbolize_names: true))
+        board.items.any? ? add_items_to_remote_board(items: board.items, remote_board: remote_board) : remote_board
+      end
+
+      def add_items_to_remote_board(items:, remote_board:)
+        items.group_by(&:class).each do |_, grouped_items|
+          grouped_items.each do |item|
+            item.add_to_board(client: self, remote_board: remote_board)
+          end
+        end
+        remote_board
+      end
+
+      def request_board(board:, remote_board_class: RemoteBoard)
+        authorize_if_no_bearer_token!
+        response = faraday.get(
+          "/v2/boards/#{board.id}",
+          {},
+          auth_headers.merge('Content-Type' => 'application/json')
+        )
+        ensure_ok_status!(response)
+        remote_board_class.new(JSON.parse(response.body, symbolize_names: true))
+      end
+    end
+  end
+end

data/lib/we_transfer_client/future_board.rb

@@ -0,0 +1,32 @@
+class FutureBoard
+  attr_reader :name, :description, :items
+
+  def initialize(name:, description: nil, items: [])
+    @name = name
+    @description = description
+    @items = items
+  end
+
+  def files
+    @items.select { |item| item.class == FutureFile }
+  end
+
+  def links
+    @items.select { |item| item.class == FutureLink }
+  end
+
+  def to_initial_request_params
+    {
+      name: name,
+      description: description,
+    }
+  end
+
+  def to_request_params
+    {
+      name: name,
+      description: description,
+      items: items.map(&:to_request_params),
+    }
+  end
+end

data/lib/we_transfer_client/future_file.rb

@@ -0,0 +1,28 @@
+class FutureFile
+  attr_reader :name, :io
+
+  def initialize(name:, io:)
+    @name = name
+    @io = io
+  end
+
+  def to_request_params
+    {
+      name: @name,
+      size: @io.size,
+    }
+  end
+
+  def add_to_board(client:, remote_board:)
+    client.authorize_if_no_bearer_token!
+    response = client.faraday.post(
+      "/v2/boards/#{remote_board.id}/files",
+      # this needs to be a array with hashes => [{name, filesize}]
+      JSON.pretty_generate([to_request_params]),
+      client.auth_headers.merge('Content-Type' => 'application/json')
+    )
+    client.ensure_ok_status!(response)
+    file_item = JSON.parse(response.body, symbolize_names: true).first
+    remote_board.items << RemoteFile.new(file_item)
+  end
+end

data/lib/we_transfer_client/future_link.rb

@@ -0,0 +1,28 @@
+class FutureLink
+  attr_reader :url, :title
+
+  def initialize(url:, title: url)
+    @url = url
+    @title = title
+  end
+
+  def to_request_params
+    {
+      url: url,
+      title: title,
+    }
+  end
+
+  def add_to_board(client:, remote_board:)
+    client.authorize_if_no_bearer_token!
+    response = client.faraday.post(
+      "/v2/boards/#{remote_board.id}/links",
+      # this needs to be a array with hashes => [{name, filesize}]
+      JSON.pretty_generate([to_request_params]),
+      client.auth_headers.merge('Content-Type' => 'application/json')
+    )
+    client.ensure_ok_status!(response)
+    file_item = JSON.parse(response.body, symbolize_names: true).first
+    remote_board.items << RemoteLink.new(file_item)
+  end
+end

data/lib/we_transfer_client/future_transfer.rb

@@ -1,9 +1,15 @@
-class FutureTransfer < Ks.strict(:name, :description, :items)
-  def to_create_transfer_params
+class FutureTransfer
+  attr_accessor :message, :files
+
+  def initialize(message:, files: [])
+    @message = message
+    @files = files
+  end
+
+  def to_request_params
     {
-      name: name,
-      description: description,
-      items: items.map(&:to_item_request_params),
+      message: message,
+      files: files.map(&:to_request_params),
     }
   end
 end

data/lib/we_transfer_client/remote_board.rb

@@ -0,0 +1,47 @@
+class RemoteBoard
+  ItemTypeError = Class.new(NameError)
+
+  attr_reader :id, :items, :url, :state
+
+  CHUNK_SIZE = 6 * 1024 * 1024
+
+  def initialize(id:, state:, url:, name:, description: '', items: [])
+    @id = id
+    @state = state
+    @url = url
+    @name = name
+    @description = description
+    @items = to_instances(items: items)
+  end
+
+  def prepare_file_upload(client:, file:, part_number:)
+    url = file.request_board_upload_url(client: client, board_id: @id, part_number: part_number)
+    [url, CHUNK_SIZE]
+  end
+
+  def prepare_file_completion(client:, file:)
+    file.complete_board_file(client: client, board_id: @id)
+  end
+
+  def files
+    @items.select { |item| item.class == RemoteFile }
+  end
+
+  def links
+    @items.select { |item| item.class == RemoteLink }
+  end
+
+  private
+
+  def to_instances(items:)
+    items.map do |item|
+      begin
+        remote_class = "Remote#{item[:type].capitalize}"
+        Module.const_get(remote_class)
+          .new(item)
+      rescue NameError
+        raise ItemTypeError, "Cannot instantiate item with type '#{item[:type]}' and id '#{item[:id]}'"
+      end
+    end
+  end
+end

data/lib/we_transfer_client/remote_file.rb

@@ -0,0 +1,55 @@
+class RemoteFile
+  attr_reader :multipart, :name, :id, :url, :type
+
+  def initialize(id:, name:, size:, url: nil, type: 'file', multipart:)
+    @id = id
+    @name = name
+    @size = size
+    @url = url
+    @type = type
+    @size = size
+    multi = Struct.new(*multipart.keys)
+    @multipart = multi.new(*multipart.values)
+  end
+
+  def request_transfer_upload_url(client:, transfer_id:, part_number:)
+    response = client.faraday.get(
+      "/v2/transfers/#{transfer_id}/files/#{@id}/upload-url/#{part_number}",
+      {},
+      client.auth_headers.merge('Content-Type' => 'application/json')
+    )
+    client.ensure_ok_status!(response)
+    JSON.parse(response.body, symbolize_names: true).fetch(:url)
+  end
+
+  def request_board_upload_url(client:, board_id:, part_number:)
+    response = client.faraday.get(
+      "/v2/boards/#{board_id}/files/#{@id}/upload-url/#{part_number}/#{@multipart.id}",
+      {},
+      client.auth_headers.merge('Content-Type' => 'application/json')
+    )
+    client.ensure_ok_status!(response)
+    JSON.parse(response.body, symbolize_names: true).fetch(:url)
+  end
+
+  def complete_transfer_file(client:, transfer_id:)
+    body = {part_numbers: @multipart.part_numbers}
+    response = client.faraday.put(
+      "/v2/transfers/#{transfer_id}/files/#{@id}/upload-complete",
+      JSON.pretty_generate(body),
+      client.auth_headers.merge('Content-Type' => 'application/json')
+    )
+    client.ensure_ok_status!(response)
+    JSON.parse(response.body, symbolize_names: true)
+  end
+
+  def complete_board_file(client:, board_id:)
+    response = client.faraday.put(
+      "/v2/boards/#{board_id}/files/#{@id}/upload-complete",
+      '{}',
+      client.auth_headers.merge('Content-Type' => 'application/json')
+    )
+    client.ensure_ok_status!(response)
+    JSON.parse(response.body, symbolize_names: true)
+  end
+end

data/lib/we_transfer_client/remote_link.rb

@@ -0,0 +1,9 @@
+class RemoteLink
+  attr_reader :type
+  def initialize(id:, url:, type:, meta:)
+    @id = id
+    @url = url
+    @title = meta.fetch(:title)
+    @type = type
+  end
+end

data/lib/we_transfer_client/remote_transfer.rb

@@ -1,2 +1,25 @@
-class RemoteTransfer < Ks.strict(:id, :version_identifier, :state, :shortened_url, :name, :description, :size, :items)
+class RemoteTransfer
+  attr_reader :files, :url, :state, :id
+
+  def initialize(id:, state:, url:, message:, files: [])
+    @id = id
+    @state = state
+    @message = message
+    @url = url
+    @files = files_to_class(files)
+  end
+
+  def prepare_file_upload(client:, file:, part_number:)
+    url = file.request_transfer_upload_url(client: client, transfer_id: @id, part_number: part_number)
+    chunk_size = file.multipart.chunk_size
+    [url, chunk_size]
+  end
+
+  def prepare_file_completion(client:, file:)
+    file.complete_transfer_file(client: client, transfer_id: @id)
+  end
+
+  def files_to_class(files)
+    files.map { |x| RemoteFile.new(x) }
+  end
 end

data/lib/we_transfer_client/transfer_builder.rb

@@ -1,27 +1,20 @@
 class TransferBuilder
-  attr_reader :items
+  attr_reader :files
   class TransferIOError < StandardError; end
 
   def initialize
-    @items = []
+    @files = []
   end
 
   def add_file(name:, io:)
     ensure_io_compliant!(io)
-    @items << FutureFileItem.new(name: name, io: io)
-    true
+    @files << FutureFile.new(name: name, io: io)
   end
 
   def add_file_at(path:)
     add_file(name: File.basename(path), io: File.open(path, 'rb'))
   end
 
-  def add_web_url(url:, title: nil)
-    title ||= url
-    @items << FutureWebItem.new(url: url, title: title)
-    true
-  end
-
   def ensure_io_compliant!(io)
     io.seek(0)
     io.read(1) # Will cause things like Errno::EACCESS to happen early, before the upload begins

data/lib/we_transfer_client/transfers.rb

@@ -0,0 +1,73 @@
+module WeTransfer
+  class Client
+    module Transfers
+      def create_transfer_and_upload_files(message:, &block)
+        future_transfer = create_future_transfer(message: message, &block)
+        remote_transfer = create_remote_transfer(future_transfer)
+        remote_transfer.files.each do |file|
+          check_for_file_duplicates(future_transfer.files, file)
+          local_file = future_transfer.files.select { |x| x.name == file.name }.first
+          upload_file(object: remote_transfer, file: file, io: local_file.io)
+          complete_file!(object: remote_transfer, file: file)
+        end
+        complete_transfer(transfer: remote_transfer)
+      end
+
+      def get_transfer(transfer_id:)
+        request_transfer(transfer_id)
+      end
+
+      private
+
+      def create_transfer(message:, &block)
+        transfer = create_future_transfer(message: message, &block)
+        create_remote_transfer(transfer)
+      end
+
+      def complete_transfer(transfer:)
+        complete_transfer_call(transfer)
+      end
+
+      def create_future_transfer(message:, future_transfer_class: FutureTransfer, transfer_builder_class: TransferBuilder)
+        builder = transfer_builder_class.new
+        yield(builder)
+        future_transfer_class.new(message: message, files: builder.files)
+      rescue LocalJumpError
+        raise ArgumentError, 'No files were added to transfer'
+      end
+
+      def create_remote_transfer(xfer)
+        authorize_if_no_bearer_token!
+        response = faraday.post(
+          '/v2/transfers',
+          JSON.pretty_generate(xfer.to_request_params),
+          auth_headers.merge('Content-Type' => 'application/json')
+        )
+        ensure_ok_status!(response)
+        RemoteTransfer.new(JSON.parse(response.body, symbolize_names: true))
+      end
+
+      def complete_transfer_call(object)
+        authorize_if_no_bearer_token!
+        response = faraday.put(
+          "/v2/transfers/#{object.id}/finalize",
+          '',
+          auth_headers.merge('Content-Type' => 'application/json')
+        )
+        ensure_ok_status!(response)
+        RemoteTransfer.new(JSON.parse(response.body, symbolize_names: true))
+      end
+
+      def request_transfer(transfer_id)
+        authorize_if_no_bearer_token!
+        response = faraday.get(
+          "/v2/transfers/#{transfer_id}",
+          {},
+          auth_headers.merge('Content-Type' => 'application/json')
+        )
+        ensure_ok_status!(response)
+        RemoteTransfer.new(JSON.parse(response.body, symbolize_names: true))
+      end
+    end
+  end
+end