dragonfly 1.3.0 → 1.4.1

data/lib/dragonfly/image_magick/processors/thumb.rb

@@ -1,32 +1,40 @@
+require "dragonfly/image_magick/commands"
+
 module Dragonfly
   module ImageMagick
     module Processors
       class Thumb
+        include ParamValidators
 
         GRAVITIES = {
-          'nw' => 'NorthWest',
-          'n'  => 'North',
-          'ne' => 'NorthEast',
-          'w'  => 'West',
-          'c'  => 'Center',
-          'e'  => 'East',
-          'sw' => 'SouthWest',
-          's'  => 'South',
-          'se' => 'SouthEast'
+          "nw" => "NorthWest",
+          "n" => "North",
+          "ne" => "NorthEast",
+          "w" => "West",
+          "c" => "Center",
+          "e" => "East",
+          "sw" => "SouthWest",
+          "s" => "South",
+          "se" => "SouthEast",
         }
 
         # Geometry string patterns
-        RESIZE_GEOMETRY         = /\A\d*x\d*[><%^!]?\z|\A\d+@\z/ # e.g. '300x200!'
+        RESIZE_GEOMETRY = /\A\d*x\d*[><%^!]?\z|\A\d+@\z/ # e.g. '300x200!'
         CROPPED_RESIZE_GEOMETRY = /\A(\d+)x(\d+)#(\w{1,2})?\z/ # e.g. '20x50#ne'
-        CROP_GEOMETRY           = /\A(\d+)x(\d+)([+-]\d+)?([+-]\d+)?(\w{1,2})?\z/ # e.g. '30x30+10+10'
+        CROP_GEOMETRY = /\A(\d+)x(\d+)([+-]\d+)?([+-]\d+)?(\w{1,2})?\z/ # e.g. '30x30+10+10'
 
-        def update_url(url_attributes, geometry, opts={})
-          format = opts['format']
+        def update_url(url_attributes, geometry, opts = {})
+          format = opts["format"]
           url_attributes.ext = format if format
         end
 
-        def call(content, geometry, opts={})
-          content.process!(:convert, args_for_geometry(geometry), opts)
+        def call(content, geometry, opts = {})
+          validate!(opts["format"], &is_word)
+          validate!(opts["frame"], &is_number)
+          Commands.convert(content, args_for_geometry(geometry), {
+            "format" => opts["format"],
+            "frame" => opts["frame"],
+          })
         end
 
         def args_for_geometry(geometry)
@@ -37,11 +45,11 @@ module Dragonfly
             resize_and_crop_args($1, $2, $3)
           when CROP_GEOMETRY
             crop_args(
-              'width' => $1,
-              'height' => $2,
-              'x' => $3,
-              'y' => $4,
-              'gravity' => $5
+              "width" => $1,
+              "height" => $2,
+              "x" => $3,
+              "y" => $4,
+              "gravity" => $5,
             )
           else raise ArgumentError, "Didn't recognise the geometry string #{geometry}"
           end
@@ -54,26 +62,24 @@ module Dragonfly
         end
 
         def crop_args(opts)
-          raise ArgumentError, "you can't give a crop offset and gravity at the same time" if opts['x'] && opts['gravity']
+          raise ArgumentError, "you can't give a crop offset and gravity at the same time" if opts["x"] && opts["gravity"]
 
-          width   = opts['width']
-          height  = opts['height']
-          gravity = GRAVITIES[opts['gravity']]
-          x       = "#{opts['x'] || 0}"
-          x = '+' + x unless x[/\A[+-]/]
-          y       = "#{opts['y'] || 0}"
-          y = '+' + y unless y[/\A[+-]/]
+          width = opts["width"]
+          height = opts["height"]
+          gravity = GRAVITIES[opts["gravity"]]
+          x = "#{opts["x"] || 0}"
+          x = "+" + x unless x[/\A[+-]/]
+          y = "#{opts["y"] || 0}"
+          y = "+" + y unless y[/\A[+-]/]
 
           "#{"-gravity #{gravity} " if gravity}-crop #{width}x#{height}#{x}#{y} +repage"
         end
 
         def resize_and_crop_args(width, height, gravity)
-          gravity = GRAVITIES[gravity || 'c']
+          gravity = GRAVITIES[gravity || "c"]
           "-resize #{width}x#{height}^^ -gravity #{gravity} -crop #{width}x#{height}+0+0 +repage"
         end
-
       end
     end
   end
 end
-

data/lib/dragonfly/middleware.rb

@@ -10,7 +10,8 @@ module Dragonfly
 
     def call(env)
       response = Dragonfly.app(@dragonfly_app_name).call(env)
-      if response[1]['X-Cascade'] == 'pass'
+      headers = response[1].transform_keys(&:downcase)
+      if headers['x-cascade'] == 'pass'
         @app.call(env)
       else
         response

data/lib/dragonfly/param_validators.rb

@@ -0,0 +1,37 @@
+module Dragonfly
+  module ParamValidators
+    class InvalidParameter < RuntimeError; end
+
+    module_function
+
+    IS_NUMBER = ->(param) {
+      param.is_a?(Numeric) || /\A[\d\.]+\z/ === param
+    }
+
+    IS_WORD = ->(param) {
+      /\A\w+\z/ === param
+    }
+
+    IS_WORDS = ->(param) {
+      /\A[\w ]+\z/ === param
+    }
+
+    def is_number; IS_NUMBER; end
+    def is_word; IS_WORD; end
+    def is_words; IS_WORDS; end
+
+    def validate!(parameter, &validator)
+      return if parameter.nil?
+      raise InvalidParameter unless validator.(parameter)
+    end
+
+    def validate_all!(parameters, &validator)
+      parameters.each { |p| validate!(p, &validator) }
+    end
+
+    def validate_all_keys!(obj, keys, &validator)
+      parameters = keys.map { |key| obj[key] }
+      validate_all!(parameters, &validator)
+    end
+  end
+end

data/lib/dragonfly/response.rb

@@ -1,4 +1,4 @@
-require 'uri'
+require 'cgi'
 require 'rack'
 
 module Dragonfly
@@ -26,10 +26,10 @@ module Dragonfly
         end
       rescue Job::Fetch::NotFound => e
         Dragonfly.warn(e.message)
-        [404, {"Content-Type" => "text/plain"}, ["Not found"]]
+        [404, {"content-type" => "text/plain"}, ["Not found"]]
       rescue RuntimeError => e
         Dragonfly.warn("caught error - #{e.message}")
-        [500, {"Content-Type" => "text/plain"}, ["Internal Server Error"]]
+        [500, {"content-type" => "text/plain"}, ["Internal Server Error"]]
       end
       log_response(response)
       response
@@ -65,8 +65,8 @@ module Dragonfly
 
     def method_not_allowed_headers
       {
-        'Content-Type' => 'text/plain',
-        'Allow' => 'GET, HEAD'
+        'content-type' => 'text/plain',
+        'allow' => 'GET, HEAD'
       }
     end
 
@@ -78,16 +78,16 @@ module Dragonfly
 
     def standard_headers
       {
-        "Content-Type" => job.mime_type,
-        "Content-Length" => job.size.to_s,
-        "Content-Disposition" => filename_string
+        "content-type" => job.mime_type,
+        "content-length" => job.size.to_s,
+        "content-disposition" => filename_string
       }
     end
 
     def cache_headers
       {
-        "Cache-Control" => "public, max-age=31536000", # (1 year)
-        "ETag" => %("#{job.signature}")
+        "cache-control" => "public, max-age=31536000", # (1 year)
+        "etag" => %("#{job.signature}")
       }
     end
 
@@ -99,7 +99,7 @@ module Dragonfly
 
     def filename_string
       return unless job.name
-      filename = request_from_msie? ? URI.encode(job.name) : job.name
+      filename = request_from_msie? ? CGI.escape(job.name) : job.name
       %(filename="#{filename}")
     end
 

data/lib/dragonfly/routed_endpoint.rb

@@ -45,7 +45,7 @@ module Dragonfly
     end
 
     def plain_response(status, message)
-      [status, {"Content-Type" => "text/plain"}, [message]]
+      [status, {"content-type" => "text/plain"}, [message]]
     end
   end
 end

data/lib/dragonfly/server.rb

@@ -59,18 +59,18 @@ module Dragonfly
           response.to_response
         end
       else
-        [404, {'Content-Type' => 'text/plain', 'X-Cascade' => 'pass'}, ['Not found']]
+        [404, {'content-type' => 'text/plain', 'x-cascade' => 'pass'}, ['Not found']]
       end
     rescue Job::NoSHAGiven => e
-      [400, {"Content-Type" => 'text/plain'}, ["You need to give a SHA parameter"]]
+      [400, {"content-type" => 'text/plain'}, ["You need to give a SHA parameter"]]
     rescue Job::IncorrectSHA => e
-      [400, {"Content-Type" => 'text/plain'}, ["The SHA parameter you gave is incorrect"]]
+      [400, {"content-type" => 'text/plain'}, ["The SHA parameter you gave is incorrect"]]
     rescue JobNotAllowed => e
       Dragonfly.warn(e.message)
-      [403, {"Content-Type" => 'text/plain'}, ["Forbidden"]]
+      [403, {"content-type" => 'text/plain'}, ["Forbidden"]]
     rescue Serializer::BadString, Serializer::MaliciousString, Job::InvalidArray => e
       Dragonfly.warn(e.message)
-      [404, {'Content-Type' => 'text/plain'}, ['Not found']]
+      [404, {'content-type' => 'text/plain'}, ['Not found']]
     end
 
     def url_for(job, opts={})
@@ -111,8 +111,8 @@ module Dragonfly
             V
       DRAGONFLY
       [200, {
-        'Content-Type' => 'text/plain',
-        'Content-Size' => body.bytesize.to_s
+        'content-type' => 'text/plain',
+        'content-size' => body.bytesize.to_s
         },
         [body]
       ]

data/lib/dragonfly/version.rb

@@ -1,3 +1,3 @@
 module Dragonfly
-  VERSION = "1.3.0"
+  VERSION = "1.4.1"
 end

data/spec/dragonfly/app_spec.rb

@@ -138,7 +138,7 @@ describe Dragonfly::App do
       }.should raise_error(NotImplementedError)
     end
     it "should correctly call it if the datastore provides it" do
-      @app.datastore.should_receive(:url_for).with('some_uid', :some => :opts).and_return 'http://egg.head'
+      @app.datastore.should_receive(:url_for).with('some_uid', {:some => :opts}).and_return 'http://egg.head'
       @app.remote_url_for('some_uid', :some => :opts).should == 'http://egg.head'
     end
   end

data/spec/dragonfly/configurable_spec.rb

@@ -65,7 +65,7 @@ describe Dragonfly::Configurable do
       configurer = Dragonfly::Configurable::Configurer.new do
         meth :jobby, :nobby
       end
-      obj.should_receive(:jobby).with('beans', :make => 5)
+      obj.should_receive(:jobby).with('beans', {:make => 5})
       obj.should_receive(:nobby).with(['nuts'])
       configurer.configure(obj) do
         jobby 'beans', :make => 5
@@ -79,7 +79,7 @@ describe Dragonfly::Configurable do
       end
       egg = double('egg')
       obj.should_receive(:egg).and_return(egg)
-      egg.should_receive(:jobby).with('beans', :make => 5)
+      egg.should_receive(:jobby).with('beans', {:make => 5})
       configurer.configure(obj) do
         jobby 'beans', :make => 5
       end
@@ -93,7 +93,7 @@ describe Dragonfly::Configurable do
 
     it "provides 'plugin' for using plugins" do
       pluggy = double('plugin')
-      pluggy.should_receive(:call).with(obj, :a, 'few' => ['args'])
+      pluggy.should_receive(:call).with(obj, :a, {'few' => ['args']})
       configurer.configure(obj) do
         plugin pluggy, :a, 'few' => ['args']
       end
@@ -101,7 +101,7 @@ describe Dragonfly::Configurable do
 
     it "allows using 'plugin' with symbols" do
       pluggy = double('plugin')
-      pluggy.should_receive(:call).with(obj, :a, 'few' => ['args'])
+      pluggy.should_receive(:call).with(obj, :a, {'few' => ['args']})
       configurer.register_plugin(:pluggy){ pluggy }
       configurer.configure(obj) do
         plugin :pluggy, :a, 'few' => ['args']

data/spec/dragonfly/content_spec.rb

@@ -320,7 +320,7 @@ describe Dragonfly::Content do
     end
 
     it "allows passing options" do
-      app.datastore.should_receive(:write).with(content, :hello => 'there')
+      app.datastore.should_receive(:write).with(content, {:hello => 'there'})
       content.store(:hello => 'there')
     end
   end

data/spec/dragonfly/file_data_store_spec.rb

@@ -9,11 +9,11 @@ describe Dragonfly::FileDataStore do
   end
 
   def assert_exists(path)
-    File.exists?(path).should be_truthy
+    File.exist?(path).should be_truthy
   end
 
   def assert_does_not_exist(path)
-    File.exists?(path).should be_falsey
+    File.exist?(path).should be_falsey
   end
 
   def assert_contains(dir, filepattern)

data/spec/dragonfly/image_magick/commands_spec.rb

@@ -0,0 +1,98 @@
+require "spec_helper"
+require "dragonfly/image_magick/commands"
+
+describe Dragonfly::ImageMagick::Commands do
+  include Dragonfly::ImageMagick::Commands
+
+  let(:app) { test_app }
+
+  def sample_content(name)
+    Dragonfly::Content.new(app, SAMPLES_DIR.join(name))
+  end
+
+  describe "convert" do
+    let(:image) { sample_content("beach.png") } # 280x355
+
+    it "should allow for general convert commands" do
+      convert(image, "-scale 56x71")
+      image.should have_width(56)
+      image.should have_height(71)
+    end
+
+    it "should allow for general convert commands with added format" do
+      convert(image, "-scale 56x71", "format" => "gif")
+      image.should have_width(56)
+      image.should have_height(71)
+      image.should have_format("gif")
+      image.meta["format"].should == "gif"
+    end
+
+    it "should work for commands with parenthesis" do
+      convert(image, "\\( +clone -sparse-color Barycentric '0,0 black 0,%[fx:h-1] white' -function polynomial 2,-2,0.5 \\) -compose Blur -set option:compose:args 15 -composite")
+      image.should have_width(280)
+    end
+
+    it "should work for files with spaces/apostrophes in the name" do
+      image = Dragonfly::Content.new(app, SAMPLES_DIR.join("mevs' white pixel.png"))
+      convert(image, "-resize 2x2!")
+      image.should have_width(2)
+    end
+
+    it "allows converting specific frames" do
+      gif = sample_content("gif.gif")
+      convert(gif, "-resize 50x50")
+      all_frames_size = gif.size
+
+      gif = sample_content("gif.gif")
+      convert(gif, "-resize 50x50", "frame" => 0)
+      one_frame_size = gif.size
+
+      one_frame_size.should < all_frames_size
+    end
+
+    it "accepts input arguments for convert commands" do
+      image2 = image.clone
+      convert(image, "")
+      convert(image2, "", "input_args" => "-extract 50x50+10+10")
+
+      image.should_not equal_image(image2)
+      image2.should have_width(50)
+    end
+
+    it "allows converting using specific delegates" do
+      expect {
+        convert(image, "", "format" => "jpg", "delegate" => "png")
+      }.to call_command(app.shell, %r{convert png:/[^']+?/beach\.png /[^']+?\.jpg})
+    end
+
+    it "maintains the mime_type meta if it exists already" do
+      convert(image, "-resize 10x")
+      image.meta["mime_type"].should be_nil
+
+      image.add_meta("mime_type" => "image/png")
+      convert(image, "-resize 5x")
+      image.meta["mime_type"].should == "image/png"
+      image.mime_type.should == "image/png" # sanity check
+    end
+
+    it "doesn't maintain the mime_type meta on format change" do
+      image.add_meta("mime_type" => "image/png")
+      convert(image, "", "format" => "gif")
+      image.meta["mime_type"].should be_nil
+      image.mime_type.should == "image/gif" # sanity check
+    end
+  end
+
+  describe "generate" do
+    let (:image) { Dragonfly::Content.new(app) }
+
+    before(:each) do
+      generate(image, "-size 1x1 xc:white", "png")
+    end
+
+    it { image.should have_width(1) }
+    it { image.should have_height(1) }
+    it { image.should have_format("png") }
+    it { image.meta.should == { "format" => "png" } }
+  end
+end

data/spec/dragonfly/image_magick/generators/plain_spec.rb

@@ -1,4 +1,5 @@
-require 'spec_helper'
+require "spec_helper"
+require "dragonfly/param_validators"
 
 describe Dragonfly::ImageMagick::Generators::Plain do
   let (:generator) { Dragonfly::ImageMagick::Generators::Plain.new }
@@ -9,32 +10,32 @@ describe Dragonfly::ImageMagick::Generators::Plain do
     before(:each) do
       generator.call(image, 3, 2)
     end
-    it {image.should have_width(3)}
-    it {image.should have_height(2)}
-    it {image.should have_format('png')}
-    it {image.meta.should == {'format' => 'png', 'name' => 'plain.png'}}
+    it { image.should have_width(3) }
+    it { image.should have_height(2) }
+    it { image.should have_format("png") }
+    it { image.meta.should == { "format" => "png", "name" => "plain.png" } }
   end
 
   describe "specifying the format" do
     before(:each) do
-      generator.call(image, 1, 1, 'format'=> 'gif')
+      generator.call(image, 1, 1, "format" => "gif")
     end
-    it {image.should have_format('gif')}
-    it {image.meta.should == {'format' => 'gif', 'name' => 'plain.gif'}}
+    it { image.should have_format("gif") }
+    it { image.meta.should == { "format" => "gif", "name" => "plain.gif" } }
   end
 
   describe "specifying the colour" do
     it "works with English spelling" do
-      generator.call(image, 1, 1, 'colour' => 'red')
+      generator.call(image, 1, 1, "colour" => "red")
     end
 
     it "works with American spelling" do
-      generator.call(image, 1, 1, 'color' => 'red')
+      generator.call(image, 1, 1, "color" => "red")
     end
 
     it "blows up with a bad colour" do
       expect {
-        generator.call(image, 1, 1, 'colour' => 'lardoin')
+        generator.call(image, 1, 1, "colour" => "lardoin")
       }.to raise_error(Dragonfly::Shell::CommandFailed)
     end
   end
@@ -42,9 +43,34 @@ describe Dragonfly::ImageMagick::Generators::Plain do
   describe "urls" do
     it "updates the url" do
       url_attributes = Dragonfly::UrlAttributes.new
-      generator.update_url(url_attributes, 1, 1, 'format' => 'gif')
-      url_attributes.name.should == 'plain.gif'
+      generator.update_url(url_attributes, 1, 1, "format" => "gif")
+      url_attributes.name.should == "plain.gif"
     end
   end
 
+  describe "param validations" do
+    {
+      "color" => "white -write bad.png",
+      "colour" => "white -write bad.png",
+      "format" => "png -write bad.png",
+    }.each do |opt, value|
+      it "validates bad opts like #{opt} = '#{value}'" do
+        expect {
+          generator.call(image, 1, 1, opt => value)
+        }.to raise_error(Dragonfly::ParamValidators::InvalidParameter)
+      end
+    end
+
+    it "validates width" do
+      expect {
+        generator.call(image, "1 -write bad.png", 1)
+      }.to raise_error(Dragonfly::ParamValidators::InvalidParameter)
+    end
+
+    it "validates height" do
+      expect {
+        generator.call(image, 1, "1 -write bad.png")
+      }.to raise_error(Dragonfly::ParamValidators::InvalidParameter)
+    end
+  end
 end

data/spec/dragonfly/image_magick/generators/plasma_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe Dragonfly::ImageMagick::Generators::Plasma do
   let (:generator) { Dragonfly::ImageMagick::Generators::Plasma.new }
@@ -10,23 +10,42 @@ describe Dragonfly::ImageMagick::Generators::Plasma do
       generator.call(image, 5, 3)
       image.should have_width(5)
       image.should have_height(3)
-      image.should have_format('png')
-      image.meta.should == {'format' => 'png', 'name' => 'plasma.png'}
+      image.should have_format("png")
+      image.meta.should == { "format" => "png", "name" => "plasma.png" }
     end
 
     it "allows changing the format" do
-      generator.call(image, 1, 1, 'format' => 'jpg')
-      image.should have_format('jpeg')
-      image.meta.should == {'format' => 'jpg', 'name' => 'plasma.jpg'}
+      generator.call(image, 1, 1, "format" => "jpg")
+      image.should have_format("jpeg")
+      image.meta.should == { "format" => "jpg", "name" => "plasma.jpg" }
     end
   end
 
   describe "urls" do
     it "updates the url" do
       url_attributes = Dragonfly::UrlAttributes.new
-      generator.update_url(url_attributes, 1, 1, 'format' => 'jpg')
-      url_attributes.name.should == 'plasma.jpg'
+      generator.update_url(url_attributes, 1, 1, "format" => "jpg")
+      url_attributes.name.should == "plasma.jpg"
     end
   end
-end
 
+  describe "param validations" do
+    it "validates format" do
+      expect {
+        generator.call(image, 1, 1, "format" => "png -write bad.png")
+      }.to raise_error(Dragonfly::ParamValidators::InvalidParameter)
+    end
+
+    it "validates width" do
+      expect {
+        generator.call(image, "1 -write bad.png", 1)
+      }.to raise_error(Dragonfly::ParamValidators::InvalidParameter)
+    end
+
+    it "validates height" do
+      expect {
+        generator.call(image, 1, "1 -write bad.png")
+      }.to raise_error(Dragonfly::ParamValidators::InvalidParameter)
+    end
+  end
+end