dis 2.0.0 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f7c60e34e3af85dabdb54a45edefc614356194ece8f98fefcc85f9245efbba4d
-  data.tar.gz: e9f450a4147cb7af0812f1755a0ecab8c0aea07c4c6cd425e1bb8fc0a9eb2476
+  metadata.gz: b78f0ff03e09c43d839c8ba8e119c6548ae518469d17d6646767c987f36f6a15
+  data.tar.gz: 3c9d88b627d1eadb871f926a2552e9761eb007442543db0aa65f88623e00f168
 SHA512:
-  metadata.gz: ddc5e8de5bb8d432d0c16a96c5d3fdae656dc2c5051dc9f728b2b6e7021679a9c9ec16b6be36d63bdc838a38e6fd965a649d56267fba4ec644b9f57aae1a41e3
-  data.tar.gz: c82397e472d58508bb953afb5480721ec6f23aa18e70da28b67ba2edcb92fba55315eeeb4879509079ae6883cbc2aab1caa13832f30a4408c76828b13a5d9e98
+  metadata.gz: 7987a1138a986daa040447ff607a07b6bf077a0ea0450274dafbff3b977cca1fee9bdf2121847c9492ffe75fe12b7bb11fc8a056240d473a69129e205ca821af
+  data.tar.gz: 888c70668ce77b47451359d5cfe953793488eb1238749eaa0f7151e6779ed105ff6a1941cd9b8e3d69419c24a349c022fc857cea443bb48b48f102f4711507bf

data/README.md

@@ -125,8 +125,7 @@ document.with_data_file { |path| Vips::Image.new_from_file(path.to_s).avg }
 
 `open_data` returns an open file, valid until you close it — even if
 the content is deleted or evicted from a cache layer meanwhile. Use it
-when the reader outlives the current call stack, such as a response
-body handed to the web server.
+when the reader outlives the current call stack.
 
 ```ruby
 document.open_data { |file| file.read }
@@ -134,6 +133,27 @@ document.open_data { |file| file.read }
 file = document.open_data # caller closes it
 ```
 
+### Sending data from a controller
+
+Include `Dis::Controller` and use `send_dis_data` to stream a record's
+data to the client. It works like `send_file`, but reads through an
+open descriptor rather than a path, so the response is unaffected if
+the content is evicted or deleted while it is being written.
+
+```ruby
+class DocumentsController < ApplicationController
+  include Dis::Controller
+
+  def show
+    send_dis_data(Document.find(params[:id]), disposition: "inline")
+  end
+end
+```
+
+`filename` and `content_type` default to the record's own metadata.
+The full set of options is `filename`, `content_type`, `disposition`
+and `status`.
+
 ## Layers
 
 The underlying storage consists of one or more layers. Each layer

data/lib/dis/controller.rb

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+module Dis
+  # = Dis Controller
+  #
+  # Adds {#send_dis_data} for serving stored data from a controller.
+  #
+  #   class DocumentsController < ApplicationController
+  #     include Dis::Controller
+  #
+  #     def show
+  #       send_dis_data(Document.find(params[:id]))
+  #     end
+  #   end
+  module Controller
+    extend ActiveSupport::Concern
+
+    included do
+      include ActionController::DataStreaming
+    end
+
+    private
+
+    # Sends the record's data to the client. Works like +send_file+,
+    # but reads through an open descriptor rather than a path, so the
+    # response is unaffected if the content is evicted or deleted
+    # while it is being written.
+    #
+    # +filename+ and +content_type+ default to the record's own
+    # metadata.
+    #
+    # @param record [Dis::Model] the record to send data from
+    # @param filename [String, nil] suggested filename
+    # @param content_type [String, nil] the content type
+    # @param disposition [String] +"attachment"+ or +"inline"+
+    # @param status [Integer] the HTTP status code
+    # @return [void]
+    # @raise [Dis::Errors::NotFoundError] if the data is not found
+    #
+    # @example
+    #   send_dis_data(document, disposition: "inline")
+    def send_dis_data(record, filename: nil, content_type: nil,
+                      disposition: "attachment", status: 200)
+      file = record.open_data
+      send_file_headers!(
+        { filename: filename || dis_metadata(record, :filename),
+          type: content_type || dis_metadata(record, :content_type),
+          disposition: }.compact
+      )
+      self.status = status
+      response.headers["Content-Length"] = file.size.to_s
+      self.response_body = Dis::ResponseBody.new(file)
+    end
+
+    def dis_metadata(record, name)
+      attribute = record.class.dis_attributes[name]
+      record[attribute] if attribute
+    end
+  end
+end

data/lib/dis/response_body.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Dis
+  # = Dis Response Body
+  #
+  # Rack body that streams an open file and closes it once the
+  # response has been sent. The file may already be unlinked, so it is
+  # read through the open descriptor and never by path.
+  class ResponseBody
+    CHUNK_SIZE = 16_384
+
+    delegate :closed?, to: :@file
+
+    # @param file [File] an open, readable file
+    def initialize(file)
+      @file = file
+    end
+
+    # Returns the entire contents as a binary string.
+    #
+    # @return [String]
+    def body
+      @file.rewind
+      @file.read
+    end
+
+    # Yields the contents in chunks.
+    #
+    # @yieldparam chunk [String] a chunk of the contents
+    # @return [void]
+    def each
+      @file.rewind
+      while (chunk = @file.read(CHUNK_SIZE))
+        yield chunk
+      end
+    end
+
+    # Closes the underlying file.
+    #
+    # @return [void]
+    def close
+      @file.close unless @file.closed?
+    end
+  end
+end

data/lib/dis/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Dis
-  VERSION = "2.0.0"
+  VERSION = "2.1.0"
 end

data/lib/dis.rb

@@ -6,6 +6,7 @@ require "fog/core"
 require "fog/local"
 require "active_job"
 require "concurrent"
+require "dis/controller"
 require "dis/engine"
 require "dis/errors"
 require "dis/jobs"
@@ -13,6 +14,7 @@ require "dis/logging"
 require "dis/layer"
 require "dis/layers"
 require "dis/model"
+require "dis/response_body"
 require "dis/storage"
 require "dis/validations"
 
@@ -27,6 +29,7 @@ require "dis/validations"
 # configure layers via {Dis::Storage.layers}.
 #
 # @see Dis::Model
+# @see Dis::Controller
 # @see Dis::Storage
 # @see Dis::Layer
 module Dis

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dis
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.1.0
 platform: ruby
 authors:
 - Inge Jørgensen
@@ -110,6 +110,7 @@ files:
 - LICENSE
 - README.md
 - lib/dis.rb
+- lib/dis/controller.rb
 - lib/dis/engine.rb
 - lib/dis/errors.rb
 - lib/dis/jobs.rb
@@ -123,6 +124,7 @@ files:
 - lib/dis/model.rb
 - lib/dis/model/class_methods.rb
 - lib/dis/model/data.rb
+- lib/dis/response_body.rb
 - lib/dis/storage.rb
 - lib/dis/validations.rb
 - lib/dis/validations/data_presence.rb