UNC-Libraries/hy-c

# frozen_string_literal: true
# [hyc-override] https://github.com/samvera/hydra-head/blob/v12.1.0/hydra-core/app/controllers/concerns/hydra/controller/download_behavior.rb
Hydra::Controller::DownloadBehavior.class_eval do
  protected

  # [hyc-override] Remove from upstream behavior in order to use the default 404 from ApplicationController
  remove_method :render_404

  # [hyc-override] Add extension to files on download
  # Override this if you'd like a different filename
  # @return [String] the filename
  def file_name
    filename = params[:filename] || file.original_name || (asset.respond_to?(:label) && asset.label) || file.id
    filename = CGI.unescape(filename) if Rails.version >= '6.0'
    file_parts = filename.split('.')
    existing_extension = file_parts.length > 1 ? file_parts.last : nil
    vocab_extension = MimeTypeService.label(file.mime_type)
    if vocab_extension.nil? || existing_extension == vocab_extension
      filename
    else
      "#{filename}.#{vocab_extension}"
    end
  end

  def send_range
    _, range = request.headers['HTTP_RANGE'].split('bytes=')
    # [hyc-override] assume client is requesting whole file if no range specified
    range = '0-' if range.nil?
    from, to = range.split('-').map(&:to_i)
    to = file.size - 1 unless to
    length = to - from + 1
    response.headers['Content-Range'] = "bytes #{from}-#{to}/#{file.size}"
    response.headers['Content-Length'] = "#{length}"
    self.status = 206
    prepare_file_headers
    stream_body file.stream(request.headers['HTTP_RANGE'])
  end
end