mm_reader 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 5e249a24e890b035b344001022ff512a55dd9ea6
+  data.tar.gz: b655f367614aa89e029eb84223bfbaf660fdead3
+SHA512:
+  metadata.gz: 2d68c85d654f6a49c0eb19239fa7d00e372954ce7af24e332bdd292c3815e5245ea9b3ebb806bc1c8836af733d49412c4ac46b10bf2aed6b59aa86293113f706
+  data.tar.gz: d120c047d841952894560e0244101ed93ae7cae130b263c4d62897f2b10ea1227d2cea9849bca54031ff1e87780c3dada1077c3ac0e77156180b446861d61b34

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in mm_reader.gemspec
+gemspec

data/README.md

@@ -0,0 +1,55 @@
+# MmReader
+
+Адаптер для работы с базой MaxMind
+
+## Установка
+
+Добавить в Gemfile:
+
+```ruby
+gem 'mm_reader'
+```
+
+Запустить:
+
+    $ bundle
+
+Установать путь к будующему файлу базы sql.
+
+```
+  # config/initializer/mm_reader.db
+
+  require 'mm_reader'
+  MmReader::Connection.current_db = 'path to sqlite.db'
+```
+
+Выполнить задачу на генерацию базы:
+
+    $ rake mm:create MM_PATH='path/to/csv/dir'
+
+## Использование
+
+Запросы к базе.
+
+```
+  MmReader.find(123000, 'Москва', 'Московская область') # => ['10.0.0.0', '10.0.0.1', ...]
+```
+
+
+## Удаление базы
+
+Выполнить задачу на удаление базы:
+
+    $ rake mm:clear
+
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/mm_reader.
+

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "mm_reader"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/mm_reader.rb

@@ -0,0 +1,17 @@
+require "mm_reader/version"
+require 'sqlite3'
+require 'csv'
+require 'ipaddr'
+require 'pry'
+require 'fiber'
+
+module MmReader
+  # Your code goes here...
+  def self.find(postal_code = nil, city = nil, state = nil)
+    MmReader::Query.new(postal_code, city, state).result
+  end
+end
+
+require "mm_reader/connection"
+require "mm_reader/converter"
+require "mm_reader/query"

data/lib/mm_reader/connection.rb

@@ -0,0 +1,12 @@
+module MmReader
+  class Connection
+    class << self
+      attr_accessor :current_db
+
+      def get sqlite_file = MmReader::Connection.current_db
+        raise "Необходимо указать путь к базе sqlite" if sqlite_file.nil?
+        SQLite3::Database.new(sqlite_file)
+      end
+    end
+  end
+end

data/lib/mm_reader/converter.rb

@@ -0,0 +1,146 @@
+module MmReader
+  class Converter
+    def self.clear sqlite_file = MmReader::Connection.current_db
+      FileUtils.rm sqlite_file if File.exists?(sqlite_file)
+    end
+
+    def initialize csv_dir
+      @csv_dir = csv_dir
+    end
+
+    def to_sqlite sqlite_file = MmReader::Connection.current_db
+      @db = MmReader::Connection.get sqlite_file
+      @db.execute('PRAGMA journal_mode=OFF; PRAGMA synchronous=OFF; PRAGMA locking_mode = EXCLUSIVE; PRAGMA count_changes = OFF; PRAGMA cache_size=500000; PRAGMA temp_store = MEMORY; PRAGMA auto_vacuum = NONE;')
+      @db.execute('BEGIN;')
+      puts "import_blocks_ip4"
+      import_blocks_ip4
+      puts "import_blocks_ip6"
+      import_blocks_ip6
+      puts "import_locations_ru"
+      import_locations_ru
+      puts "index_data"
+      index_data
+      puts "drop_temp"
+      drop_temp
+      puts "Commit"
+      @db.execute('COMMIT;')
+    end
+
+    def import_blocks_ip4
+      import_csv "#{@csv_dir}/GeoLite2-City-Blocks-IPv4.csv", "tmp_ip4"
+      
+    end
+
+    def import_blocks_ip6
+      import_csv "#{@csv_dir}/GeoLite2-City-Blocks-IPv6.csv", "tmp_ip6"
+    end
+
+    def import_locations_ru
+      import_csv "#{@csv_dir}/GeoLite2-City-Locations-ru.csv", "tmp_locations"
+    end
+
+    def drop_temp
+      @db.execute("DROP TABLE tmp_ip4")
+      @db.execute("DROP TABLE tmp_ip6")
+      @db.execute("DROP TABLE tmp_locations")
+    end
+
+    def index_data
+      @db.execute("
+        CREATE TABLE ip4_index (
+          network varchar(255), 
+          postal_code varchar(10), 
+          city varchar(255), 
+          state varchar(255)
+        );
+      ")
+      @db.execute("
+        CREATE TABLE ip6_index (
+          network varchar(255), 
+          postal_code varchar(10), 
+          city varchar(255), 
+          state varchar(255)
+        );
+      ")
+
+      @db.execute("
+        INSERT INTO ip4_index (network, postal_code, city, state)
+        SELECT tmp_ip4.network, tmp_ip4.postal_code, tmp_locations.city_name, tmp_locations.subdivision_1_name FROM tmp_ip4 LEFT JOIN tmp_locations ON tmp_ip4.geoname_id=tmp_locations.geoname_id;
+      ")
+
+      @db.execute("
+        INSERT INTO ip6_index (network, postal_code, city, state)
+        SELECT tmp_ip6.network, tmp_ip6.postal_code, tmp_locations.city_name, tmp_locations.subdivision_1_name FROM tmp_ip6 LEFT JOIN tmp_locations ON tmp_ip6.geoname_id=tmp_locations.geoname_id;
+      ")
+      @db.execute "CREATE INDEX postal_code4_idx ON ip4_index (postal_code);"
+      @db.execute "CREATE INDEX postal_code6_idx ON ip6_index (postal_code);"
+
+    end
+
+    def import_csv from_file, to_table
+      provider = prepare_data from_file
+      columns = provider.transfer
+      schema = provider.transfer
+      @db.execute "CREATE TABLE #{to_table} (#{schema});"
+
+      loop do
+        stop = insert_bulk to_table, columns, provider
+        break if stop
+      end
+    end
+
+    def insert_bulk to_table, columns, provider, limit  = 10000
+      begin
+        stop = false
+        data = []
+        limit.times{ data << provider.transfer }
+        stop
+      rescue FiberError
+        stop = true
+      ensure
+        @db.execute "INSERT INTO #{to_table} (#{columns}) VALUES #{data_mask_map(data)};", data
+        stop
+      end
+    end
+
+    def prepare_data from_file
+      row_fiber = nil
+      iterator = Fiber.new do
+        header = nil
+        CSV.foreach(from_file, col_sep: ',') do |row|
+          row_fiber.transfer row
+        end
+      end
+
+      row_fiber = Fiber.new do
+        headers = iterator.transfer
+        Fiber.yield headers.join(', ') # columns
+        first_row = iterator.transfer
+        schema = headers.map.with_index{ |col, index| "#{col} #{col_type(first_row[index])}" }.join(', ')
+        Fiber.yield schema # schema
+
+        Fiber.yield first_row # first_row
+
+        loop do
+          Fiber.yield iterator.transfer # other
+        end
+      end
+    end
+
+    TYPES = {
+      String => 'varchar(255)',
+      Fixnum => 'integer'
+    }
+
+    def col_type data
+      TYPES[data.class] || TYPES[String]
+    end
+
+    def data_mask_map data
+      row = Array.new(data.first.size, '?').join(', ')
+      rows = ("(#{row}), " * data.size)
+      rows.chomp!(', ')
+      rows
+    end
+  end
+end

data/lib/mm_reader/query.rb

@@ -0,0 +1,32 @@
+module MmReader
+  class Query
+    attr_accessor :postal, :city, :state
+
+    def initialize postal, city, state
+      @postal = postal
+      @city = city
+      @state = state
+    end
+
+    def query_result
+      db = MmReader::Connection.get
+      result = db.execute('SELECT DISTINCT network FROM ip4_index WHERE postal_code = ?', [@postal])
+      return result if result.size > 0
+
+      result = db.execute('SELECT DISTINCT network FROM ip4_index WHERE city = ? and state = ?', [@city, @state])
+      return result if result.size > 0
+
+      result = db.execute('SELECT DISTINCT network FROM ip4_index WHERE city = ?', [@city])
+      return result if result.size > 0
+
+      result = db.execute('SELECT DISTINCT network FROM ip4_index WHERE state = ?', [@state])
+      return result if result.size > 0
+
+      return []
+    end
+    def result
+      query_result.flatten.map{ |mask| IPAddr.new(mask).to_range.map(&:to_s) }.flatten
+    end
+
+  end
+end

data/lib/mm_reader/version.rb

@@ -0,0 +1,3 @@
+module MmReader
+  VERSION = "0.1.0"
+end

data/lib/tasks/convert.rake

@@ -0,0 +1,12 @@
+namespace :mm do
+  task :create do
+    raise "Вы должны указать путь к CSV базе MaxMind в переменной окружения MM_PATH" unless ENV[:MM_PATH]
+    raise "Не верно указан путь к XML базе ФИАС в переменной окружения MM_PATH" unless File.exist? ENV[:MM_PATH]
+    db = Converter.new(ENV[:MM_PATH])
+    db.to_sqlite
+  end
+
+  task :clear do
+    Converter.clear
+  end
+end

data/mm_reader.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'mm_reader/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "mm_reader"
+  spec.version       = MmReader::VERSION
+  spec.authors       = ["Andrey Bolshov", "uno4ki"]
+  spec.email         = ["asnow.dev@gmail.com", "i.have@no.mail"]
+
+  spec.summary       = %q{Конвертор базы MaxMind.}
+  spec.description   = %q{Конвертор базы MaxMind.}
+  spec.homepage      = "http://github.com/the-furnish/mm_reader"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "sqlite3", "~> 1.3"
+  spec.add_development_dependency "bundler", "~> 1.12"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency 'pry'
+  spec.add_development_dependency 'ruby-prof'
+  spec.add_development_dependency 'pry-byebug'
+  spec.add_development_dependency 'pry-stack_explorer'
+end

metadata

@@ -0,0 +1,171 @@
+--- !ruby/object:Gem::Specification
+name: mm_reader
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Andrey Bolshov
+- uno4ki
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-08-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  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: ruby-prof
+  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: pry-byebug
+  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: pry-stack_explorer
+  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'
+description: Конвертор базы MaxMind.
+email:
+- asnow.dev@gmail.com
+- i.have@no.mail
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/mm_reader.rb
+- lib/mm_reader/connection.rb
+- lib/mm_reader/converter.rb
+- lib/mm_reader/query.rb
+- lib/mm_reader/version.rb
+- lib/tasks/convert.rake
+- mm_reader.gemspec
+homepage: http://github.com/the-furnish/mm_reader
+licenses: []
+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.6.6
+signing_key: 
+specification_version: 4
+summary: Конвертор базы MaxMind.
+test_files: []