Day14 - PDF 加浮水印

前言

工作上在处理 API 取得的 PDF 後，要在上面增加浮水印，进而研究与纪录的处理方式

实作

会需要用到 combine_pdf 和 prawn 这两个 Gem 处理

# Gemfile

gem 'combine_pdf', '~> 1.0', '>= 1.0.21'
gem 'prawn', '~> 2.4'

将增加浮水写成一个方法，范例的 PDF ，和实作的 pr

# app/features/pdf/add_watermark_pdf.rb

module Pdf
  class AddWatermarkPdf

    # input_file = Rails.root.join('data/river_demo_watermark.pdf')
    # output_file = input_file.dirname.join('done.pdf')
    # watermark_texts = %w[river_1 river_2 river_3]
    def execute(input_file, output_file, watermark_texts = [])
      tmp_watermark_file(input_file)              # 暂存的浮水印 PDF path
      generate_watermark_content(watermark_texts) # 产浮水印的 PDF
      combine_pdf(input_file, output_file)        # 将 input_file 与浮水印 PDF 合并
      FileUtils.rm_rf(@tmp_watermark_file)        # 移除暂存的浮水印 PDF
    end

    private

    def tmp_watermark_file(input_file)
      @tmp_watermark_file ||= input_file.dirname.join('tmp_watermark.pdf')
    end

    # 可自行设定字型大小、颜色等
    # Prawn Documentation: https://prawnpdf.org/api-docs
    # Prawn example: https://prawnpdf.org/manual.pdf
    def generate_watermark_content(watermark_texts)
      position, amount_per_page = watermark_args
      Prawn::Document.generate(@tmp_watermark_file) do
        watermark_texts.each_with_index do |watermark_text, index|
          page = index / amount_per_page + 1
          coordinate = position[index % amount_per_page]
          start_new_page if page != 1 && (index % amount_per_page).zero?
          go_to_page(page)
          float do
            bounding_box(coordinate, height: 14, width: 48) do
              fill_color '000000'
              fill_rectangle [0, bounds.height], bounds.width, bounds.height
              stroke_bounds
              fill_color 'FFFFFF'
              text watermark_text.to_s, align: :center, valign: :center, size: 12
            end
          end
        end
      end
    end

    # 浮水印在页面上的位置、每页的数量
    def watermark_args
      position = [[0, 769], [0, 392]] # xy 轴座标 (需自行计算适合的位置)
      amount_per_page = 2             # 每页的浮水印数量

      [position, amount_per_page]
    end

    def combine_pdf(input_file, output_file)
      pdf_pages = CombinePDF.load(@tmp_watermark_file).pages
      pdf = CombinePDF.load(input_file)
      pdf.pages.each_with_index { |page, index| page << pdf_pages[index] }
      pdf.save(output_file)
    end
  end
end

完成上述步骤後，接着在 rails console 输入以下，便大功告成

# rails console

input_file = Rails.root.join('data/river_demo_watermark.pdf')
output_file = input_file.dirname.join('done.pdf')
watermark_texts = %w[river_1 river_2 river_3]
Pdf::AddWatermarkPdf.new.execute(input_file, output_file, watermark_texts)