prawn-attachment 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5cedf55c7c97ed821fc7d9a61cd6f88538eb959d17cd2fe11f8a5aaae2d734c0
-  data.tar.gz: 5ad7f5ce585f97f508fda3d6a257f4f7bc47197887914c1424aaa506f4747bad
+  metadata.gz: c82ce1fc502c03d7348f8c72873ae174a82387f42b6f1619ee0405eccd75165b
+  data.tar.gz: 61646a398075b6757fd9682f47754041e12e4bb444956489fbb60bb2757ec059
 SHA512:
-  metadata.gz: c4c2452fa385e45480211f2744e6cfaa7d96131acff1472e8652ef7802500b5c895eeb16358a134520f77492a06db8533abba8d4b3707d9a899ecf8e5384ab36
-  data.tar.gz: '09fdd61fd399f58a78898b15faaea1b4738f1fa7617a0aa8dd1edc9e9bdbbdf02d4fc15bd1208a9d936beefdecbf10f5d604eb09ce079361458a395742357e65'
+  metadata.gz: 76721c29d6d7b918b4ef580b7465b651fc97c9d5579107a842084f8528994854c68d4b6e2a47d8b52325ce4ed74e7934e93ae3680c7ae9ceab22471e8cf99656
+  data.tar.gz: 5e349c0d2f29986ec24c74efbd7b2155b4a7e8623a069c4791da1d0162b361d7db37fb01bf0a86784f6d58390d3aa2d6bef790a3b964ff2c03e2aa83fa98a2e5

data/.rubocop.yml

@@ -1,5 +1,7 @@
 AllCops:
   TargetRubyVersion: 2.5
+  Exclude:
+    - "spec/**/*"
 
 Style/StringLiterals:
   Enabled: true

data/README.md

@@ -28,7 +28,7 @@ require "prawn/attachment"
 
 Prawn::Document.generate("hello.pdf") do
   text "Hello World!"
-  attach "./data.json"
+  attach "data.json", File.open("./data.json")
 end
 ```
 

data/lib/prawn/attachment/version.rb

@@ -2,6 +2,6 @@
 
 module Prawn
   module Attachment
-    VERSION = "0.2.0"
+    VERSION = "0.3.0"
   end
 end

data/lib/prawn/attachment.rb

@@ -19,6 +19,13 @@ module Prawn
       end
     end
 
+    # Invalid data provided
+    class InvalidDataError < StandardError
+      def message
+        "Source data must be an IO object or string"
+      end
+    end
+
     # Attach a file's data to the document. File IO Objects are expected.
     #
     # Arguments:
@@ -42,9 +49,7 @@ module Prawn
     # hidden, then nil is returned.
     #
     def attach(name, src, opts = {})
-      data = src.is_a?(IO) ? src.read : src.b
-      raise NoDataError if data.length.zero?
-
+      data = extract_data_from_source(src)
       opts = prepare_options(name, opts)
 
       # Prepare embeddable representation of the source data
@@ -58,6 +63,19 @@ module Prawn
 
     private
 
+    def extract_data_from_source(src)
+      if src.respond_to?(:read)
+        data = src.read
+      elsif src.respond_to?(:b)
+        data = src.b
+      else
+        raise InvalidDataError
+      end
+      raise NoDataError if data.length.zero?
+
+      data
+    end
+
     def prepare_options(name, opts)
       {
         name: name,

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: prawn-attachment
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - David Lilue
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-03-07 00:00:00.000000000 Z
+date: 2022-03-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: prawn