easy_captcha_solver 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b1234c024b893c57f76bff082f670d4b9e602b7b
-  data.tar.gz: fa8c19e2088725784fbbbef9b2dbbaf479f6d636
+  metadata.gz: 2c0efaec755c121dae345e90395b94ad7502f222
+  data.tar.gz: 4b60d856f1c1db88b72e8a6cd2784806478ad18e
 SHA512:
-  metadata.gz: 756f4241d26c040e380c9b05f7a6c94d5e2c5777ca72275356f247eec2eb0d373a254a17ace817987aa0ed39259ab50beadfd09989b5f758f9c0d57ec3962892
-  data.tar.gz: 594942dc6956ae1736b69937b9399aec3ee4b57d1354798737ae821d9be7fd4ae211222d79786dc69793c2dc4d36072a926fce7a936ac8823cb4dec3b83d058f
+  metadata.gz: 0710d76863f1370abdd452a91717e783aa9b09a98a5f2733a1dd28126bffe4a354f638951fbd55b433577f3c282c452beba978eb6d1fdd48f0b4fc4c3ee7fc3e
+  data.tar.gz: bcceed3bc41f98006ef498bfb9687255985a6e6417005bf35fb7c32ab70c4ffc2167f4e0f0d35f4a094188d9f9f3b109811c897d5b8843efdd76513db57c2473

data/lib/easy_captcha_solver.rb

@@ -11,12 +11,22 @@ class EasyCaptchaSolver
     }
 
     @image = options[:image_path] if options[:image_path]
-    @image = agent.get(options[:image_url]).body if options[:image_url]
+
+    # If URL, save a file instead of trying to solve the captcha from memory because of tesseract limitations with .png images
+    if options[:image_url]
+      image = agent.get(options[:image_url]).save!  "./tmp_image"
+
+      # Guess image extension and rename tmp file
+      @image  = "./tmp_image.#{get_image_extension(image)}"
+      File.rename( image,  @image)
+    end
+
     unless @image
       throw Exception.new "A local image path or a image URL must be provided. Example:  easy_c = EasyCaptcha.new( image_url: 'http://www.example.com/captcha_img.jpg')"
     end
 
-    solve_captcha
+    solve_captcha ensure File.delete(@image) if options[:image_url]
+
   end
 
   private
@@ -27,8 +37,27 @@ class EasyCaptchaSolver
       e.language  = :eng
       e.blacklist = '|'
     }
+
     @captcha = e.text_for(@image).strip # => 'ABC'
   end
 end
 
-
+def get_image_extension(local_file_path)
+  png = Regexp.new("\x89PNG".force_encoding("binary"))
+  jpg = Regexp.new("\xff\xd8\xff\xe0\x00\x10JFIF".force_encoding("binary"))
+  jpg2 = Regexp.new("\xff\xd8\xff\xe1(.*){2}Exif".force_encoding("binary"))
+  case IO.read(local_file_path, 10)
+  when /^GIF8/
+    'gif'
+  when /^#{png}/
+    'png'
+  when /^#{jpg}/
+    'jpg'
+  when /^#{jpg2}/
+    'jpg'
+  else
+    mime_type = `file #{local_file_path} --mime-type`.gsub("\n", '') # Works on linux and mac
+    raise UnprocessableEntity, "unknown file type" if !mime_type
+    mime_type.split(':')[1].split('/')[1].gsub('x-', '').gsub(/jpeg/, 'jpg').gsub(/text/, 'txt').gsub(/x-/, '')
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: easy_captcha_solver
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Alex Rodriguez