russian_post 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 2b004d7e8725e3d084956ead13b4e170b135309a
+  data.tar.gz: 1e9a5e5a24281d852341963fc6d7270602cc2822
+SHA512:
+  metadata.gz: cc8680522729c6b95a06754a8a213610af79a77cc2ab12f9d6e4106cde61dd5e4b0e864b150f228ae3ba3198c214602d8d256ac17ddf0a953e9c1bcfaede08c9
+  data.tar.gz: f62d87d7981d586d74b5968e3f3e6a845bfd3503366b38ff5af093dca3057701dcf55dfc9ce7ef72e162b2f4e670389fe77b6443177fd36b7b7464589fd98365

data/.gitignore

@@ -0,0 +1,2 @@
+app.rb
+views

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,29 @@
+PATH
+  remote: .
+  specs:
+    russian_post (0.0.1)
+      chunky_png (~> 1.2)
+      excon (~> 0.20)
+      nokogiri (~> 1.5)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.3.3)
+    chunky_png (1.2.7)
+    crack (0.3.2)
+    excon (0.20.1)
+    nokogiri (1.5.9)
+    rake (10.0.3)
+    webmock (1.11.0)
+      addressable (>= 2.2.7)
+      crack (>= 0.3.2)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.3)
+  rake
+  russian_post!
+  webmock (~> 1.11)

data/README.md

@@ -0,0 +1,29 @@
+# Russian Post Tools
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'russian_post'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install russian_post
+
+## Usage
+
+### Captcha recognizer
+
+```ruby
+require 'russian_post'
+
+RussianPost::Captcha.for_url(captcha_url).text
+# => 96950
+
+RussianPost::Captcha.for_data(png_blob).text
+# => 43455
+```

data/Rakefile

@@ -0,0 +1,10 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |i|
+ i.libs << 'test'
+ i.test_files = FileList['test/**/*_test.rb']
+ i.verbose = true
+end
+
+desc "Run tests"
+task :default => :test

data/lib/russian_post/captcha/pattern.rb

@@ -0,0 +1,26 @@
+
+class RussianPost::Captcha::Pattern
+
+  TARGET_MATCH_RATE = 1.0
+
+  attr_reader :points, :character
+
+  def initialize(points, character)
+    @points, @character = points, character
+  end
+
+  def match?(image, x, y)
+    matching = 0
+
+    points.each do |px, py|
+      cx, cy = x + px, y + py
+
+      break if cx >= image.width || cy >= image.height || image[cx, cy] == 255
+
+      matching += 1
+    end
+
+    matching / points.size >= TARGET_MATCH_RATE
+  end
+
+end

data/lib/russian_post/captcha/patterns.rb

@@ -0,0 +1,67 @@
+require 'russian_post/captcha/pattern'
+
+class RussianPost::Captcha::Patterns
+  attr_reader :patterns
+
+  class << self
+
+    def built_in
+      @builtin_patterns ||= RussianPost::Captcha::Patterns.new(File.expand_path('characters.dat', File.dirname(__FILE__)))
+    end
+
+  end
+
+  def initialize(file)
+    @patterns = load_patterns(file)
+  end
+
+  def each(&block)
+    @patterns.each(&block)
+  end
+
+  def add(pattern)
+    @patterns << pattern
+  end
+
+  def find(image, x, y)
+    each do |pattern|
+      return pattern if pattern.match?(image, x, y)
+    end
+
+    return false
+  end
+
+  private 
+
+  def serialize(patterns)
+    array_hash = Hash.new{ |hash, key| hash[key] = []}
+
+    data = patterns.inject(array_hash) do |result, pattern|
+      result[pattern.character] << pattern.points
+      result
+    end
+
+    Marshal.dump(data)
+  end
+
+  def deserialize(data)
+    raw_data = Marshal.load(data)
+
+    raw_data.inject([]) do |result, data|
+      character, patterns = data
+
+      result += patterns.map { |points| RussianPost::Captcha::Pattern.new(points, character) }
+    end
+  end
+
+  def save_patterns(file, patterns)
+    File.open(file, 'wb') { |file| file.write(serialize(patterns)) }
+  end
+
+  def load_patterns(file)
+    raise "Specified patterns file not exists (#{file})" unless File.exists?(file)
+
+    deserialize(File.open(file, 'rb') { |file| file.read })
+  end
+
+end

data/lib/russian_post/captcha.rb

@@ -0,0 +1,99 @@
+require 'excon'
+require 'chunky_png'
+
+module RussianPost
+
+  class Captcha
+
+    require 'russian_post/captcha/patterns'
+
+    attr_reader :patterns
+
+    class << self
+
+      def for_url(url, patterns = nil)
+        RussianPost::Captcha.new(url, nil, patterns)
+      end
+
+      def for_data(data, patterns = nil)
+        RussianPost::Captcha.new(nil, data, patterns)
+      end
+
+    end
+
+    def image
+      @data ||= fetch_image
+    end
+
+    def text
+      @recognized_text ||= prepare_text(recognize)
+    end
+
+    def valid?
+      text.size == 5
+    end
+
+    def recognize
+      @recognize ||= recognize!
+    end
+
+    def recognize!
+
+      captcha_image = grayscale(image)
+
+      results = {}
+
+      for x in 0..captcha_image.width - 2
+        for y in 0..captcha_image.height - 2
+          result = patterns.find(captcha_image, x, y)
+
+          results[x] = result if result
+        end
+      end
+
+      Hash[results.sort].values
+    end
+
+    private 
+
+    def initialize(url = nil, data = nil, patterns = nil) # private constructor
+      @url = url
+      @data = ChunkyPNG::Image.from_blob(data) if data
+      @patterns ||= RussianPost::Captcha::Patterns.built_in 
+    end
+
+    def prepare_text(results)
+      results.map(&:character).join('')
+    end
+
+    def grayscale(captcha_image)
+      captcha_image.pixels.map! { |color| color = ChunkyPNG::Color.grayscale_teint(color); color = (color >= 255) ? 255 : 0 ; color }
+
+      captcha_image
+    end
+
+    def fetch_image
+      raise "URL not specified" unless @url
+
+      data = Excon.get(@url).body
+
+
+
+      unless data.start_with?("\x89PNG".force_encoding("ASCII-8BIT")) # not png header
+        if data =~ /<input id=\"key\" name=\"key\" value=\"([0-9]+)\"\/>/ # tough security huh
+          data = Excon.post(@url, body: "key=#{$1}").body
+        end
+
+        if data =~ /Object moved/ # faulty captcha
+          raise "Russian Post captcha service error"
+        end
+      end
+
+      File.open('test.png', 'wb') { |f| f.write(data) }
+
+      ChunkyPNG::Image.from_blob(data)
+    end
+
+  end
+
+end

data/lib/russian_post/russian_post.rb

@@ -0,0 +1,11 @@
+require 'chunky_png'
+
+module RussianPost
+
+  require 'russian_post/version'
+
+  require 'russian_post/captcha'
+  
+  require 'russian_post/tracking'
+
+end

data/lib/russian_post/version.rb

@@ -0,0 +1,3 @@
+module RussianPost
+  VERSION = "0.0.1"
+end

data/russian_post.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+require 'russian_post/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "russian_post"
+  spec.version       = RussianPost::VERSION
+  spec.authors       = ["t3hk0d3"]
+  spec.email         = ["clouster@yandex.ru"]
+  spec.description   = %q{RussianPost tools}
+  spec.summary       = %q{RussianPost tools}
+  spec.homepage      = "http://github.com/t3hk0d3/russian_post"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "excon", "~> 0.20"
+  spec.add_dependency "chunky_png", "~> 1.2"
+  spec.add_dependency "nokogiri", "~> 1.5"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "webmock", "~> 1.11"
+end

data/test/russian_post/captcha/pattern_test.rb

@@ -0,0 +1,19 @@
+require 'test_helper'
+
+class RussianPost::Captcha::PatternTest < Test::Unit::TestCase
+
+  def setup
+    @image = load_captcha_image
+
+    @points = [[5, 0], [6, 0], [7, 0], [8, 0], [10, 0], [4, 1], [10, 1], [10, 2], [10, 3], [5, 4], [10, 4], [1, 5], [5, 5], [10, 5],
+    [5, 6], [10, 6], [0, 7], [5, 7], [0, 8], [3, 8], [4, 8], [0, 9], [11, 9], [11, 10], [0, 11], [1, 11], [2, 11], [3, 11], [5, 11],
+    [5, 12], [10, 13], [5, 14], [6, 14], [7, 14], [8, 14], [9, 14]]
+
+    @pattern = RussianPost::Captcha::Pattern.new(@points, '4')
+  end
+
+  def test_find_pattern 
+    assert @pattern.match?(@image, 4, 3)
+  end
+
+end

data/test/russian_post/captcha/patterns_test.rb

@@ -0,0 +1,20 @@
+require 'test_helper'
+
+class RussianPost::Captcha::PatternsTest < Test::Unit::TestCase
+
+  def setup
+    @image = load_captcha_image
+
+    @patterns = RussianPost::Captcha::Patterns.built_in # @FIXME use mock patterns instead
+    
+  end
+
+  def test_builtin_patterns 
+    assert @patterns.patterns.size > 0
+  end
+
+  def test_find
+    assert_equal '4', @patterns.find(@image, 4, 3).character
+  end
+
+end

data/test/russian_post/captcha_test.rb

@@ -0,0 +1,40 @@
+require 'test_helper'
+
+module RussianPost
+
+  class CaptchaTest < Test::Unit::TestCase
+
+    def setup 
+      @url = 'http://www.russianpost.ru/CaptchaService/CaptchaImage.ashx?Id=361256433'
+    end 
+
+    def test_recognize 
+      captcha = RussianPost::Captcha.for_data(load_captcha_file)
+
+      assert_equal '49396', captcha.text
+    end
+
+    def test_fetch
+      stub_get = stub_request(:get, @url).to_return(:body => load_captcha_file, :headers => { 'Content-Type' => 'image/png' })
+
+      RussianPost::Captcha.for_url(@url).recognize!
+
+      assert_requested(stub_get)
+    end
+
+    def test_cunning_fetch
+      @tough_security = "<html><head></head><body onload=\"document.myform.submit();\"><form method=\"post\" name=\"myform\" style=\"visibility:hidden;\"><input id=\"key\" name=\"key\" value=\"117413\"/><input type=\"submit\"/></form></body></html>\r\n\r\n"
+
+      stub_get = stub_request(:get, @url).to_return(:body => @tough_security, :headers => { 'Content-Type' => 'text/html' })
+      stub_post = stub_request(:post, @url).with(:body => 'key=117413').to_return(:body => load_captcha_file, :headers => { 'Content-Type' => 'image/png' })
+
+      RussianPost::Captcha.for_url(@url).recognize!
+
+      assert_requested(stub_get)
+      assert_requested(stub_post)
+
+    end
+
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,23 @@
+$:.unshift File.expand_path("../lib/russian_post/", File.dirname(__FILE__))
+
+require 'test/unit'
+require 'webmock/test_unit'
+
+require 'russian_post'
+
+class Test::Unit::TestCase
+
+  def load_captcha_file
+    File.open('test/fixtures/captcha.png', 'rb') { |f| f.read }
+  end
+
+
+  def load_captcha_image
+      image = ChunkyPNG::Image.from_file('test/fixtures/captcha.png')
+      image.pixels.map! { |color| color = ChunkyPNG::Color.grayscale_teint(color); color = (color >= 255) ? 255 : 0 ; color }
+
+      image
+  end
+
+end
+

metadata

@@ -0,0 +1,151 @@
+--- !ruby/object:Gem::Specification
+name: russian_post
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- t3hk0d3
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: excon
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.20'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.20'
+- !ruby/object:Gem::Dependency
+  name: chunky_png
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.11'
+description: RussianPost tools
+email:
+- clouster@yandex.ru
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- lib/russian_post/captcha.rb
+- lib/russian_post/captcha/characters.dat
+- lib/russian_post/captcha/pattern.rb
+- lib/russian_post/captcha/patterns.rb
+- lib/russian_post/russian_post.rb
+- lib/russian_post/version.rb
+- russian_post.gemspec
+- test/fixtures/captcha.png
+- test/russian_post/captcha/pattern_test.rb
+- test/russian_post/captcha/patterns_test.rb
+- test/russian_post/captcha_test.rb
+- test/test_helper.rb
+homepage: http://github.com/t3hk0d3/russian_post
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: RussianPost tools
+test_files:
+- test/fixtures/captcha.png
+- test/russian_post/captcha/pattern_test.rb
+- test/russian_post/captcha/patterns_test.rb
+- test/russian_post/captcha_test.rb
+- test/test_helper.rb
+has_rdoc: