bitter_domain 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5648f4c353c1bd404ffe48dac0d496c908544ec42ad7683385a5c7b36653f65b
-  data.tar.gz: 411b8eae843f42168ce7e7e538e7c32c564a059177dad0532ce155cfc01ab7b1
+  metadata.gz: 50cc7524ec85308d7779171078697a4375c44b4cd064de0bcb854d000e72d225
+  data.tar.gz: 72f92a81a31b21799d83b7dc3c2da4773168feded8a0322ac2d6db7f25cc2c2a
 SHA512:
-  metadata.gz: f34d05508e9019453ad7b09c98fe523be8ee5beb206811caf22ca743b53113025cc33a1b7595a0821c0a059ba4bb715cd843d585113f2d6656b0092a4c0cf9fd
-  data.tar.gz: aa0da73a4ab66b582dd49571b28d365cbe706192668a5680ad381ad60b836b4a7a3b034c759bd2a1b13d3b0bb86cdf859ae1dc9df950b65e64fcd07843029eb7
+  metadata.gz: 599de2c0dd413a33d9e74f7fc6365cfcc4644e8af0c513af7e0c0601d03ac1c8f0e313c7313c930a179d7429a1077c8aa48faae7a53aa8ff38b52e3c327fd840
+  data.tar.gz: 9e37fe354980e43e04c887a3b4393ee0e33b1cdf9adff87c4b017e65cd9e46aa5f846678869022c01af4215929358aca623a8b07adc046fd46b3efb754844290

data/Gemfile.lock

@@ -3,6 +3,7 @@ PATH
   specs:
     bitter_domain (0.1.0)
       colorize
+      public_suffix
       whois (~> 4.0.8)
       whois-parser (~> 1.1.0)
 
@@ -51,6 +52,7 @@ GEM
     pry (0.12.2)
       coderay (~> 1.1.0)
       method_source (~> 0.9.0)
+    public_suffix (3.0.3)
     rake (10.5.0)
     rb-fsevent (0.10.3)
     rb-inotify (0.10.0)

data/bin/bitter_domain

@@ -13,18 +13,16 @@ class CLI < Thor
   method_option :url, aliases: ['-u'], desc: 'url to generate shifts for', required: true
   method_option :verbose, aliases: ['-v'], desc: 'print verbose output'
   def gen_shifts()
-    host, ext = options[:url].split(".")
-    shifter = BitterDomain::BitShifter.new(host)
-    shifted_domains = shifter.get_shifted_domains()
-      .map { |dom| "#{dom}.#{ext}" }
+    mapper = BitterDomain::DomainMapper.new(options[:url])
+    shifted_domains = mapper.gen_shifts
 
     if options["flips-only".to_sym]
-      shifted_domains.each { |shift| $stdout.puts "#{shift}" }
+      mapper.print_shifts
     else
       checker = BitterDomain::DomainChecker.new(shifted_domains)
       checker.test_domains
-      checker.retry if options[:retry]
-      options[:verbose] ? checker.print_verbose : checker.print_available
+      mapper.check_domains(options[:retry])
+      options[:verbose] ? mapper.print_verbose : mapper.print_available
     end
   end
 end

data/bitter_domain.gemspec

@@ -39,6 +39,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "colorize"
   spec.add_dependency "whois", "~> 4.0.8"
   spec.add_dependency "whois-parser", "~> 1.1.0"
+  spec.add_dependency "public_suffix"
 
   spec.add_development_dependency "bundler", "~> 1.17"
   spec.add_development_dependency "rake", "~> 10.0"

data/lib/bitter_domain.rb

@@ -1,5 +1,6 @@
-require "bitter_domain/version"
-require "bitter_domain/bit_shifter"
-require "bitter_domain/domain_checker"
 module BitterDomain
+  require "bitter_domain/version"
+  require "bitter_domain/bit_shifter"
+  require "bitter_domain/domain_checker"
+  require "bitter_domain/domain_mapper"
 end

data/lib/bitter_domain/bit_shifter.rb

@@ -6,6 +6,7 @@ module BitterDomain
 
     def initialize(domain)
       @domain = domain
+      @shifts = []
     end
 
     def valid_domain?(domain)
@@ -15,13 +16,12 @@ module BitterDomain
     def gen_shifts(byte)
       # left-shift 1 through each bit of the byte
       # and XOR to flip a single one at each position
-      shifts = []
       8.times do |i|
         shifted = byte ^ (1 << i)
-        shifts.push(shifted)
+        @shifts.push(shifted)
       end
 
-      shifts
+      @shifts
     end
 
     def get_shifted_domains

data/lib/bitter_domain/domain_mapper.rb

@@ -0,0 +1,84 @@
+require 'uri/http'
+require 'public_suffix'
+
+module BitterDomain
+  class DomainMapper
+    attr_reader :shifted_domains
+    attr_reader :domain
+
+    def initialize(domain)
+      @domain = parse_domain(domain)
+      @checker = nil
+      @shifted_domains = []
+    end
+
+    def gen_shifts
+      host = @domain.sld
+      shifter = BitterDomain::BitShifter.new(host)
+      self.checker = nil
+      @shifted_domains = shifter.get_shifted_domains
+        .map { |dom| "#{dom}.#{@domain.tld}" }
+    end
+
+    def print_shifts
+      self.shifted_domains.each { |shift| $stdout.puts "#{shift}" }
+    end
+
+    def domain=(domain)
+      self.domain = parse_domain(domain)
+      gen_shifts
+    end
+
+    def check_domains(attempt_retry=false)
+      self.checker = BitterDomain::DomainChecker.new(@shifted_domains)
+      self.checker.test_domains
+      
+      self.checker.retry if attempt_retry
+
+      self.checker.available
+    end
+
+    def get_available
+      self.checker.available
+    end
+
+    def get_errors
+      self.checker.errors
+    end
+
+    def get_tested
+      self.checker.tested
+    end
+
+    def retry
+      self.checker.retry
+    end
+
+    def print_verbose
+      self.checker.print_verbose
+    end
+
+    def print_available
+      self.checker.print_available
+    end
+
+    def query_shifts
+      gen_shifts
+      check_domains
+
+      get_available
+    end
+
+
+    protected
+    
+    attr_accessor :checker
+    def parse_domain(url)
+      uri = URI.parse(url)
+      uri = URI.parse("http://#{url}") if uri.scheme.nil?
+      host = uri.host.downcase
+
+      PublicSuffix.parse(host)
+    end
+  end
+end

data/lib/bitter_domain/version.rb

@@ -1,3 +1,3 @@
 module BitterDomain
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bitter_domain
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - axylos
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-04-11 00:00:00.000000000 Z
+date: 2019-05-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: public_suffix
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -160,6 +174,7 @@ files:
 - lib/bitter_domain.rb
 - lib/bitter_domain/bit_shifter.rb
 - lib/bitter_domain/domain_checker.rb
+- lib/bitter_domain/domain_mapper.rb
 - lib/bitter_domain/version.rb
 homepage: https://github.com/Axylos/squat_logs
 licenses: