Sha256: 09df2599466ec5bb17d66386d85243043c8bf8f535cc2b10f4e3f4eb35aa9340

Contents?: true

Size: 1.25 KB

Versions: 8

Compression:

Stored size: 1.25 KB

Contents

# frozen_string_literal: true
require "csv"

module ActiveAdmin
  class ResourceController < BaseController

    # This module overrides CSV responses to allow large data downloads.
    # Could be expanded to JSON and XML in the future.
    #
    module Streaming

      def index
        super do |format|
          format.csv { stream_csv }
          yield(format) if block_given?
        end
      end

      protected

      def stream_resource(&block)
        headers["X-Accel-Buffering"] = "no"
        headers["Cache-Control"] = "no-cache"
        headers["Last-Modified"] = Time.current.httpdate

        if ActiveAdmin.application.disable_streaming_in.include? Rails.env
          self.response_body = block[String.new]
        else
          self.response_body = Enumerator.new &block
        end
      end

      def csv_filename
        "#{resource_collection_name.to_s.gsub('_', '-')}-#{Time.zone.now.to_date.to_formatted_s(:default)}.csv"
      end

      def stream_csv
        headers["Content-Type"] = "text/csv; charset=utf-8" # In Rails 5 it's set to HTML??
        headers["Content-Disposition"] = %{attachment; filename="#{csv_filename}"}
        stream_resource &active_admin_config.csv_builder.method(:build).to_proc.curry[self]
      end

    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
activeadmin-3.0.0 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.14.0 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.13.1 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.13.0 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.12.0 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.11.2 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.11.1 lib/active_admin/resource_controller/streaming.rb
activeadmin-2.11.0 lib/active_admin/resource_controller/streaming.rb