image.rb in glebtv-simple_captcha-0.3.1

- old
+ new

@@ -1,14 +1,40 @@
 require 'tempfile'
 module SimpleCaptcha #:nodoc
   module ImageHelpers #:nodoc
 
+    mattr_accessor :image_styles
+    @@image_styles = {
+      'embosed_silver'  => ['-fill darkblue', '-shade 20x60', '-background white'],
+      'simply_red'      => ['-fill darkred', '-background white'],
+      'simply_green'    => ['-fill darkgreen', '-background white'],
+      'simply_blue'     => ['-fill darkblue', '-background white'],
+      'distorted_black' => ['-fill darkblue', '-edge 10', '-background white'],
+      'all_black'       => ['-fill darkblue', '-edge 2', '-background white'],
+      'charcoal_grey'   => ['-fill darkblue', '-charcoal 5', '-background white'],
+      'almost_invisible' => ['-fill red', '-solarize 50', '-background white']
+    }
+
     DISTORTIONS = ['low', 'medium', 'high']
 
     class << self
 
-      def image_params(color = '#0089d1')
+      def image_params(key = 'simply_blue')
+        image_keys = @@image_styles.keys
+
+        style = begin
+          if key == 'random'
+            image_keys[rand(image_keys.length)]
+          else
+            image_keys.include?(key) ? key : 'simply_blue'
+          end
+        end
+
+        @@image_styles[style]
+      end
+
+      def image_params_from_color(color)
         ["-alpha set -background none -fill \"#{color}\""]
       end
 
       def distortion(key='low')
         key =
@@ -36,15 +62,20 @@
     private
 
       def generate_simple_captcha_image(simple_captcha_key) #:nodoc
         amplitude, frequency = ImageHelpers.distortion(SimpleCaptcha.distortion)
         text = Utils::simple_captcha_new_value(simple_captcha_key)
-        params = ImageHelpers.image_params(SimpleCaptcha.image_color).dup
+        if SimpleCaptcha.image_color.nil?
+          params = ImageHelpers.image_params(SimpleCaptcha.image_style).dup
+        else
+          params = ImageHelpers.image_params_from_color(SimpleCaptcha.image_color).dup
+        end
+
         params << "-size #{SimpleCaptcha.image_size} xc:transparent"
         params << "-gravity \"Center\""
         if params.join(' ').index('-pointsize').nil?
-          params << "-pointsize 35"
+          params << "-pointsize #{SimpleCaptcha.pointsize}"
         end
         dst = Tempfile.new(RUBY_VERSION < '1.9' ? 'simple_captcha.png' : ['simple_captcha', '.png'], SimpleCaptcha.tmp_path)
         dst.binmode
         text.split(//).each_with_index do |letter, index|
           i = -60 + (index*25) + rand(-6..6)
@@ -56,15 +87,15 @@
         (1..10).each do |i|
           params << "-draw \"polyline #{rand(160)},#{rand(61)} #{rand(160)},#{rand(62)}\""
         end
 
         params << "\"#{File.expand_path(dst.path)}\""
-        # puts "convert " +  params.join(' ')
+        # puts "convert " + params.join(' ')
         SimpleCaptcha::Utils::run("convert", params.join(' '))
 
         dst.close
 
         File.expand_path(dst.path)
-        #dst
+                                                            #dst
       end
   end
 end