pdf_watermark 0.1.6 → 0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 523bc19fdfbb123b16f40b89259eae28c0857920
-  data.tar.gz: 24971ecaa3890c432f2adc4555078117c12d95e7
+  metadata.gz: a4b337c7d1c0f23cf1345a9feb4081b8e80c3ff3
+  data.tar.gz: 09f3589967afc9d1e17b593b59769d882dfbe5c1
 SHA512:
-  metadata.gz: 751aeef5401b8ea5275a9fd5e689e603fcb0dcf5e507ff46a6b0abc5b67760b5744b1678eefd979a072ea6df58cdb6aea559d6f578c0913b7212ceee3782a909
-  data.tar.gz: 7460fae2f2873174361d7dae45e82e47114477249cb78adaf7135bf03b4594459af05fe1cc051da96f79f6f286e56ad5d8007649effed7dc83a292ed5af28f60
+  metadata.gz: b0a10564a127ece015b0b14f26c66d4fba19478b9f0e453b57ad2eaef52b4c277a271f7defff1983fc527f3872912c264e9a36d88c926fc76a3c25cec48a1a9e
+  data.tar.gz: 43dfbdb95c8711ed2c0a058185878f060e560fdc42e5115a2357a2b9adfc4d67b6cacf4182cc70b99fcab9084113f7a28fa8b7730221acecac50c2487c038bd2

data/.gitignore

@@ -8,5 +8,6 @@
 /spec/reports/
 /tmp/
 /.idea
-/spec/data/result.pdf
+/spec/data/*
+!/spec/data/test.pdf
 *.gem

data/Gemfile

@@ -1,3 +1,3 @@
-source 'https://ruby.taobao.org/'
+source 'https://gems.ruby-china.org/'
 # Specify your gem's dependencies in pdf_watermark.gemspec
 gemspec

data/lib/pdf_watermark/font.rb

@@ -0,0 +1,47 @@
+require 'hexapdf/font/true_type/font'
+require 'hexapdf/content/canvas'
+module PdfWatermark
+  module Font
+    def width_of(string, font_size)
+      scale = font_size / 1000.0
+      string.codepoints.inject(9) do |memo, code|
+        memo + character_width(code)
+      end * scale
+    end
+
+    def character_width(code)
+      return 0 unless glyph_index[code]
+      return 0.0 if code == 10
+      self[:hmtx][glyph_index[code]].advance_width * scale_factor
+    end
+
+    def glyph_index
+      @glyph_index ||= self[:cmap].preferred_table
+    end
+
+    def scale_factor
+      @scale_factor ||= 1000.0 / self[:head].units_per_em
+    end
+  end
+
+  module Canvas
+    def width_of(string, font_size = nil)
+      self.font.wrapped_font.width_of(string, font_size || self.graphics_state.font_size) +
+        (string.length * self.graphics_state.character_spacing)
+    end
+  end
+
+  module FontDescriptor
+    def perform_validation
+      descent = self[:Descent]
+      if descent && descent > 0
+        self[:Descent] = - descent
+      end
+      super
+    end
+  end
+end
+
+HexaPDF::Font::TrueType::Font.include(PdfWatermark::Font)
+HexaPDF::Content::Canvas.include(PdfWatermark::Canvas)
+HexaPDF::Type::FontDescriptor.prepend(PdfWatermark::FontDescriptor)

data/lib/pdf_watermark/version.rb

@@ -1,3 +1,3 @@
 module PdfWatermark
-  VERSION = "0.1.6"
+  VERSION = "0.2"
 end

data/lib/pdf_watermark/water_mark/base.rb

@@ -0,0 +1,57 @@
+require 'hexapdf'
+require 'hexapdf/utils/math_helpers'
+require 'pdf_watermark/font'
+module PdfWatermark
+  module WaterMark
+    class Base
+      include HexaPDF::Utils::MathHelpers
+
+      def initialize(mark_string, source_path, options)
+        @source_path = source_path
+        @options = options
+        source_size = page_size
+        @content_width = source_size[0] - (options[:margin][1] + options[:margin][3])
+        @content_height = source_size[1] - (options[:margin][0] + options[:margin][2])
+        @angle = options[:angle] == :diagonal ? rad_to_deg(Math.atan(@content_height.to_f/@content_width.to_f)) : options[:angle]
+        @mark_string = mark_string
+
+        @font_size = @options[:font_size]
+        if @font_size.is_a?(String)
+          if @font_size =~ /(\d+[.]?\d*)%/
+            @font_size = ($1.to_f / 100) * @content_width
+            @font_size = [@font_size, @options[:max_font_size]].min
+            @font_size = [@font_size, @options[:min_font_size]].max
+          else
+            @font_size = @font_size.to_i
+          end
+        end
+        @options[:font]
+        @x = @options[:x] || 0
+        @y = @options[:y] || @content_height
+
+        @font = load_font(@options[:font], :watermark_font)
+      end
+
+
+      protected
+
+      def load_font(font_path, name = nil)
+        font_name = name.nil? ? File.basename(font_path, '.ttf') : name
+        map = {}
+        map[font_name] = { none: font_path }
+        document.config['font.map'].merge!(map)
+
+        document.fonts.load(font_name).wrapped_font
+      end
+
+      def page_size
+        box = document.pages[0].box.value
+        [box[2] - box[0], box[3] - box[1]]
+      end
+
+      def document
+        @document ||= HexaPDF::Document.open(@source_path)
+      end
+    end
+  end
+end

data/lib/pdf_watermark/water_mark/repeated.rb

@@ -0,0 +1,42 @@
+require 'pdf_watermark/water_mark/base'
+module PdfWatermark
+  module WaterMark
+    class Repeated < PdfWatermark::WaterMark::Base
+
+      def initialize(*args)
+        super(*args)
+      end
+
+      def render
+        document.pages.each do |page|
+          canvas = page.canvas(type: :overlay)
+          canvas.font(:watermark_font, size: @font_size)
+          canvas.fill_color(@options[:font_color])
+          box_height = @font_size
+          box_width = canvas.width_of(@mark_string)
+          temp_y = @y
+          indent = false
+          offset_x = box_width + @options[:repeat_offset] * @font_size
+          offset_y = box_height + @options[:repeat_offset] * @font_size
+
+
+          canvas.opacity(fill_alpha: @options[:transparent]) do
+            canvas.rotate(@angle, origin: [@content_width/2, @content_height/2]) do
+              while temp_y > 0 do
+                temp_x = indent ? (offset_x / 2.0) : 0
+                while temp_x <= @content_width
+                  canvas.text(@mark_string, at: [temp_x, temp_y])
+                  temp_x += offset_x
+                end
+                temp_y -= offset_y
+                indent = !indent
+              end
+            end
+          end
+        end
+        document
+      end
+    end
+  end
+end
+

data/lib/pdf_watermark/water_mark/single.rb

@@ -0,0 +1,50 @@
+require 'pdf_watermark/water_mark/base'
+module PdfWatermark
+  module WaterMark
+    class Single < PdfWatermark::WaterMark::Base
+
+      def initialize(*args)
+        super(*args)
+      end
+
+      def render
+        document.pages.each do |page|
+          canvas = page.canvas(type: :overlay)
+          canvas.fill_color(@options[:font_color])
+          canvas.font(:watermark_font, size: 18)
+          rad = deg_to_rad(@angle)
+
+          if Math.tan(rad) < @content_height.to_f/@content_width.to_f
+            max_text_width = (@content_width/Math.cos(rad)).abs
+          else
+            max_text_width = (@content_height/Math.sin(rad)).abs
+          end
+          # if user does not provide the font size, scale to fill the whole page
+          if @font_size
+            canvas.font_size(@font_size)
+          else
+            @font_size = calculated_font_size(canvas, @mark_string, @options[:max_font_size], max_text_width)
+            canvas.font_size(@font_size)
+            @x = (@content_width - canvas.width_of(@mark_string))/2 + @options[:margin][3]
+            @y = @content_height / 2 + @options[:margin][2]
+          end
+          canvas.opacity(fill_alpha: @options[:transparent]) do
+            canvas.rotate(@angle, origin: [@content_width/2, @content_height/2]) do
+              canvas.text(@mark_string, at: [@x, @y])
+            end
+          end
+        end
+        document
+      end
+
+      def calculated_font_size(canvas, string, max_font, max_width)
+        text_width = canvas.width_of(string, max_font)
+        if text_width > max_width
+          (max_width - canvas.graphics_state.character_spacing * string.size) / string.size
+        else
+          max_font
+        end
+      end
+    end
+  end
+end

data/lib/pdf_watermark.rb

@@ -1,11 +1,13 @@
 require "pdf_watermark/version"
-require 'pdf_watermark/water_mark'
-require 'combine_pdf'
+require 'pdf_watermark/water_mark/repeated'
+require 'pdf_watermark/water_mark/single'
 module PdfWatermark
 
+  include HexaPDF::Encryption::StandardSecurityHandler::Permissions
+
   LIB_DIR = File.dirname(File.realpath(__FILE__))
-  BASEDIR = File.join(LIB_DIR, '..')
-  FONT_DIR = File.join(LIB_DIR, '..', 'fonts')
+  BASEDIR = File.realpath File.join(LIB_DIR, '..')
+  FONT_DIR = File.realpath File.join(LIB_DIR, '..', 'fonts')
   REPEAT_X_OFFSET = 80
   REPEAT_Y_OFFSET = 80
 
@@ -26,23 +28,25 @@ module PdfWatermark
     }
     options = default.merge(options)
 
-    source_pdf = CombinePDF.load(source, allow_optional_content: true)
-    source_size = page_size(source_pdf.pages[0])
+    if options[:mode].to_sym == :repeat
+      document = PdfWatermark::WaterMark::Repeated.new(mark_string, source, options).render
+    else
+      document = PdfWatermark::WaterMark::Single.new(mark_string, source, options).render
+    end
 
-    options[:source_size] = source_size
-    options[:content_width] ||= source_size[0] - (options[:margin][1] + options[:margin][3])
-    options[:content_height] ||= source_size[1] - (options[:margin][0] + options[:margin][2])
+    if options[:read_only]
+      permissions = 1
+      document.encrypt(name: :Standard, owner_password: options[:password], permissions: permissions)
+    end
 
-    wm = WaterMark.new(mark_string, options).render
-    water_mark_pdf = CombinePDF.parse(wm, allow_optional_content: true).pages[0]
 
-    source_pdf.pages.each do |page|
-      page << water_mark_pdf
-    end
-    if destination.nil?
-      source_pdf.to_pdf
+    if destination
+      document.write(destination)
     else
-      source_pdf.save destination
+      StringIO.open('', 'wb') do |io|
+        document.write(io)
+        io.string
+      end
     end
   end
 

data/pdf_watermark.gemspec

@@ -4,27 +4,26 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'pdf_watermark/version'
 
 Gem::Specification.new do |spec|
-  spec.name          = "pdf_watermark"
-  spec.version       = PdfWatermark::VERSION
-  spec.authors       = ["bruce"]
-  spec.email         = ["shibocuhk@gmail.com"]
+  spec.name = "pdf_watermark"
+  spec.version = PdfWatermark::VERSION
+  spec.authors = ["bruce"]
+  spec.email = ["shibocuhk@gmail.com"]
 
-  spec.summary       = %q{pdf watermark}
-  spec.description   = %q{add watermark to pdf files}
-  spec.homepage      = "http://bruceshi.me"
-  spec.license       = "MIT"
+  spec.summary = %q{pdf watermark}
+  spec.description = %q{add watermark to pdf files}
+  spec.homepage = "http://bruceshi.me"
+  spec.license = "MIT"
 
   # Prevent pushing this gem to RubyGems.org by setting 'allowed_push_host', or
   # delete this section to allow pushing this gem to any host.
 
-  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
-  spec.bindir        = "exe"
-  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.files = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
   spec.test_files = Dir['spec/*_spec.rb']
 
-  spec.add_runtime_dependency('prawn', '~> 2.1', '>= 2.1.0')
-  spec.add_runtime_dependency('combine_pdf', '~> 0.2.34')
+  spec.add_runtime_dependency('hexapdf', '~> 0.3')
 
   spec.add_development_dependency('bundler', '~> 1.10')
   spec.add_development_dependency('rake', '~> 10.0')

metadata

@@ -1,49 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: pdf_watermark
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: '0.2'
 platform: ruby
 authors:
 - bruce
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-02-27 00:00:00.000000000 Z
+date: 2017-05-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: prawn
+  name: hexapdf
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.1'
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 2.1.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '2.1'
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 2.1.0
-- !ruby/object:Gem::Dependency
-  name: combine_pdf
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.2.34
+        version: '0.3'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.2.34
+        version: '0.3'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -147,8 +127,11 @@ files:
 - bin/setup
 - fonts/SourceHanSansSC-Regular.ttf
 - lib/pdf_watermark.rb
+- lib/pdf_watermark/font.rb
 - lib/pdf_watermark/version.rb
-- lib/pdf_watermark/water_mark.rb
+- lib/pdf_watermark/water_mark/base.rb
+- lib/pdf_watermark/water_mark/repeated.rb
+- lib/pdf_watermark/water_mark/single.rb
 - pdf_watermark.gemspec
 homepage: http://bruceshi.me
 licenses:
@@ -170,7 +153,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.5
+rubygems_version: 2.6.8
 signing_key: 
 specification_version: 4
 summary: pdf watermark

data/lib/pdf_watermark/water_mark.rb

@@ -1,129 +0,0 @@
-require 'prawn'
-module PdfWatermark
-  class WaterMark
-    include Prawn::View
-
-    def initialize(mark_string, options)
-      @options = options
-      @content_width = options[:content_width]
-      @content_height = options[:content_height]
-      @angle = options[:angle] == :diagonal ? rad_to_degree(Math.atan(@content_height.to_f/@content_width.to_f)) : options[:angle]
-      @mark_string = mark_string
-      @font_size = @options[:font_size]
-      if @font_size.is_a?(String)
-        if @font_size =~ /(\d+[.]?\d*)%/
-          @font_size = ($1.to_f / 100) * @content_width
-          @font_size = [@font_size, @options[:max_font_size]].min
-          @font_size = [@font_size, @options[:min_font_size]].max
-        else
-          @font_size = @font_size.to_i
-        end
-      end
-
-      @font = @options[:font]
-      @x = @options[:x] || 0
-      @y = @options[:y] || @content_height
-
-    end
-
-    def render
-      case @options[:mode]
-      when :fill
-        fill_content
-      when :repeat
-        repeat_content
-      else
-        repeat_content
-      end
-      document.render
-    end
-
-    def fill_content
-      @max_text_width = @content_width
-
-      rad = degree_to_rad(@angle)
-      if Math.tan(rad) < @content_height.to_f/@content_width.to_f
-        @max_text_width = (@content_width/Math.cos(rad)).abs
-      else
-        @max_text_width = (@content_height/Math.sin(rad)).abs
-      end
-      @font_size ||= calculated_font_size(@mark_string, @options[:max_font_size], @max_text_width)
-
-      font(@options[:font]) do
-        bounding_box([@x, @y], width: @content_width, height: @content_height) do
-          rotate @angle, origin: [@content_width/2, @content_height/2] do
-            transparent(@options[:transparent]) do
-              formatted_text_box(
-                [{ text: @mark_string, color: @options[:font_color] }],
-                at: [@x - (@max_text_width - @content_width) / 2, @y],
-                width: @max_text_width, height: @content_height,
-                align: @options[:align], valign: @options[:valign],
-                size: @font_size,
-              )
-            end
-          end
-        end
-      end
-    end
-
-    def repeat_content
-      font(@font) do
-        transparent(@options[:transparent]) do
-          box_height = @font_size
-          box_width = text_width(@mark_string, @font_size)
-          temp_y = @y
-          indent = false
-          offset_x = box_width + @options[:repeat_offset] * @font_size
-          offset_y = box_height + @options[:repeat_offset] * @font_size
-          bounding_box([@x, @y], width: @content_width, height: @content_height) do
-            while temp_y > 0 do
-              rotate @angle, origin: [@content_width/2, @content_height/2] do
-                temp_x = indent ? (offset_x / 2.0) : 0
-                while temp_x <= @content_width
-                  formatted_text_box(
-                    [{ text: @mark_string, color: @options[:font_color] }],
-                    at: [temp_x, temp_y],
-                    width: box_width, height: box_height,
-                    align: @options[:align], valign: @options[:valign],
-                    size: @font_size,
-                  )
-                  temp_x += offset_x
-                end
-              end
-              temp_y -= offset_y
-              indent = !indent
-            end
-          end
-        end
-      end
-    end
-
-
-    protected
-    def text_width(text, size)
-      document.width_of(text, size: size, margin: 0, left_margin: 0, right_margin: 0).ceil
-
-    end
-
-    def calculated_font_size(text, size, max)
-      text_width = self.text_width(text, size)
-      if text_width > max
-        (max / text.size).floor
-      else
-        size
-      end
-    end
-
-    def document
-      @document ||= Prawn::Document.new(page_size: @options[:source_size], margin: @options[:margin], font: @font)
-    end
-
-    def degree_to_rad(angle)
-      angle * Math::PI / 180
-    end
-
-    def rad_to_degree(rad)
-      rad * 180 / Math::PI
-    end
-  end
-end