poppler_pdf_previewer.rb 1004 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# frozen_string_literal: true

module ActiveStorage
  class Previewer::PopplerPDFPreviewer < Previewer
    class << self
      def accept?(blob)
        blob.content_type == "application/pdf" && pdftoppm_exists?
      end

      def pdftoppm_path
        ActiveStorage.paths[:pdftoppm] || "pdftoppm"
      end

      def pdftoppm_exists?
15
        return @pdftoppm_exists if defined?(@pdftoppm_exists)
16 17 18 19 20

        @pdftoppm_exists = system(pdftoppm_path, "-v", out: File::NULL, err: File::NULL)
      end
    end

21
    def preview(**options)
22 23
      download_blob_to_tempfile do |input|
        draw_first_page_from input do |output|
24
          yield io: output, filename: "#{blob.filename.base}.png", content_type: "image/png", **options
25 26 27 28 29 30
        end
      end
    end

    private
      def draw_first_page_from(file, &block)
A
alkesh26 已提交
31
        # use 72 dpi to match thumbnail dimensions of the PDF
32
        draw self.class.pdftoppm_path, "-singlefile", "-cropbox", "-r", "72", "-png", file.path, &block
33 34 35
      end
  end
end