genebrand 0.1.0

data/lib/genebrand/command.rb

@@ -0,0 +1,14 @@
+module Genebrand
+  class Command
+    class << self
+      def subclasses
+        @subclasses ||= []
+      end
+
+      def inherited(base)
+        subclasses << base
+        super(base)
+      end
+    end
+  end
+end

data/lib/genebrand/commands/new.rb

@@ -0,0 +1,142 @@
+module Genebrand
+  class New < Command
+    class << self
+      require 'json'
+
+      def load_command(p)
+        p.command(:new) do |c|
+          c.syntax 'new'
+          c.description 'Generates new brand name'
+          c.alias(:brand)
+
+          c.action do |args, options|
+            process(args, options)
+          end
+        end
+      end
+
+      def process(_args, _options)
+        puts "Hello! Let's generate some brands!".cyan
+        puts
+
+        brand = []
+        stopit = false
+        @gen = Genebrand::Generator.new
+
+        catch :parts_done do
+          loop do
+            choose do |menu|
+              menu.prompt = 'What should we add?'.yellow
+
+              menu.choice('String') do
+                brand.push(type: :word,
+                           word: ask('Enter word (English, no spaces and punctiation, 1-10 symbols)')) do |q|
+                  q.validate = /\A[a-zA-Z\d]{1,10}\z/
+                end
+                puts
+              end
+
+              menu.choice('Specific part of speech') { brand.push(pickpart) }
+
+              if brand.length > 1
+                menu.choice('Enough, show me some brands!') { throw :parts_done }
+              end
+            end
+          end
+        end
+
+        @gen.generate(brand)
+      end
+
+      def pickpart
+        puts
+
+        data = { type: :part }
+        part = ''
+
+        choose do |menu|
+          menu.prompt = 'What part of speech should we add?'.yellow
+
+          menu.choice("Noun (#{@gen.words['noun'].count} in db)") { part = 'noun' }
+          menu.choice("Plural (#{@gen.words['plur'].count} in db)") { part = 'plur' }
+          menu.choice("Verb participle (#{@gen.words['verb_part'].count} in db)") { part = 'verb_part' }
+          menu.choice("Verb transitive (#{@gen.words['verb_trans'].count} in db)") { part = 'verb_trans' }
+          menu.choice("Verb intransitive (#{@gen.words['verb_intrans'].count} in db)") { part = 'verb_intrans' }
+          menu.choice("Adjective (#{@gen.words['adj'].count} in db)") { part = 'adj' }
+        end
+
+        data[:part] = part
+
+        addfilters(data)
+      end
+
+      def check_filter_conflict(data)
+        filters = data[:filters]
+        if !filters[:minlen].nil? && !filters[:maxlen].nil?
+          if filters[:minlen] > filters[:maxlen]
+            filters[:minlen] = nil
+            filters[:maxlen] = nil
+            Genebrand::Logger.error 'Minimum length should be lesser than maximum, min and max filters was reset'
+            return false
+          end
+        end
+        true
+      end
+
+      def addfilters(data)
+        puts
+
+        data[:filters] = {} if data[:filters].nil?
+
+        catch :filter_done do
+          loop do
+            puts
+            choose do |menu|
+              menu.prompt = 'Choose filters for this words'.yellow
+
+              check_filter_conflict(data)
+
+              if data[:filters][:minlen].nil?
+                menu.choice('Minimum length') do
+                  data[:filters][:minlen] = ask('Enter minimum word length (between 1 and 10)', Integer) do |q|
+                    q.in = 1..10
+                  end
+                end
+              end
+
+              if data[:filters][:maxlen].nil?
+                menu.choice('Maximum length') do
+                  data[:filters][:maxlen] = ask('Enter maximum word length (between 1 and 10)', Integer) do |q|
+                    q.in = 1..10
+                  end
+                end
+              end
+
+              if data[:filters][:starts].nil?
+                menu.choice('Starts with') do
+                  data[:filters][:starts] = ask('Enter symbols word should start with')
+                end
+              end
+
+              if data[:filters][:ends].nil?
+                menu.choice('Ends with') do
+                  data[:filters][:ends] = ask('Enter symbols word should end with')
+                end
+              end
+
+              if data[:filters][:contains].nil?
+                menu.choice('Contains') do
+                  data[:filters][:contains] = ask('Enter symbols word should contain')
+                end
+              end
+
+              menu.choice('Enough filters, next') { throw :filter_done if check_filter_conflict(data) }
+            end
+          end
+        end
+
+        data
+      end
+    end
+  end
+end

data/lib/genebrand/generator.rb

@@ -0,0 +1,135 @@
+module Genebrand
+  class Generator
+    require 'resolv'
+
+    attr_reader :words
+
+    def initialize
+      @words = JSON.parse(File.read(File.join(Gem::Specification.find_by_name('genebrand').gem_dir, 'lib/data/posinfo.json')))
+    end
+
+    def prettyoutput(domain)
+      data = '['
+
+      # A bit hacky, but pretty fast method to guess domain available or not
+      resolv = Resolv::DNS.open
+      data << (resolv.getresources("#{domain}.com", Resolv::DNS::Resource::IN::NS).count == 0 ? "com".green : "com".red)
+      data << ' '
+      data << (resolv.getresources("#{domain}.new", Resolv::DNS::Resource::IN::NS).count == 0 ? "net".green : "net".red)
+      data << ' '
+      data << (resolv.getresources("#{domain}.org", Resolv::DNS::Resource::IN::NS).count == 0 ? "org".green : "org".red)
+      data << "]\t"
+      data << domain.bold
+
+      return data
+    end
+
+    def generate(info)
+      out = []
+
+      puts
+      puts 'Generating brands with these parameters:'.cyan
+      puts
+      i = 1
+
+      info.each do |item|
+        if item[:type] == :word
+          puts "#{i}. Word: #{item[:word]}".green
+        elsif item[:type] == :part
+          puts "#{i}. Part of speech: #{item[:part]}".green
+          puts 'Filters:'
+          item[:filters].each do |filter, value|
+            if filter == :minlen
+              puts "Minimum length: #{value}"
+            elsif filter == :maxlen
+              puts "Maximum length: #{value}"
+            elsif filter == :starts
+              puts "Starts with: #{value}"
+            elsif filter == :ends
+              puts "Ends with: #{value}"
+            elsif filter == :contains
+              puts "Contains: #{value}"
+            end
+          end
+        end
+
+        i += 1
+      end
+
+      puts
+      puts 'Press any key to continue'.cyan
+      puts
+
+      puts 'Fetching variants...'
+      gener = []
+      wordscount = 0
+      info.each do |item|
+        if item[:type] == :word
+          wordscount += 1
+          gener.push(item[:word])
+        elsif item[:type] == :part
+          parts = @words[item[:part]]
+          item[:filters].each do |filter, value|
+            if filter == :minlen
+              parts = parts.select { |i| i[/^.{#{value},}$/] }
+            elsif filter == :maxlen
+              parts = parts.select { |i| i[/^.{,#{value}}$/] }
+            elsif filter == :starts
+              parts = parts.select { |i| i[/^#{value}.*$/i] }
+            elsif filter == :ends
+              parts = parts.select { |i| i[/^.*#{value}$/i] }
+            elsif filter == :contains
+              parts = parts.select { |i| i[/^.*(#{value}).*$/i] }
+            end
+          end
+
+          if parts.count > 0
+            gener.push(parts)
+          else
+            Genebrand::Logger.warning "0 variants for that part, will be skipped"
+          end
+        end
+      end
+
+      approx = 1
+
+      gener.each do |item|
+        if item.is_a? Array
+          approx *= item.count
+        end
+      end
+
+      #arrdata.reverse!
+
+      puts 'Available variants: '.yellow + approx.to_s.bold
+      puts
+
+      puts "Whois info\tBrand"
+
+      finish = false
+      i = 0
+      while not finish
+        itemd = ''
+
+        gener.each_with_index do |item, index|
+          if item.is_a? Array
+            ind = rand(gener[index].count)
+            itemd << gener[index][ind].capitalize
+          else
+            itemd << gener[index].capitalize
+          end
+        end
+
+        puts prettyoutput(itemd)
+
+        i += 1
+
+        if i%15 == 0
+          puts
+          puts "Press any key to see next variants".cyan
+          gets
+        end
+      end
+    end
+  end
+end

data/lib/genebrand/logger.rb

@@ -0,0 +1,23 @@
+module Genebrand
+  class Logger
+    def self.error(text)
+      puts "ERROR: #{text}".red
+    end
+
+    def self.warning(text)
+      puts "WARNING: #{text}".yellow
+    end
+
+    def self.info(text)
+      puts text.bold.cyan
+    end
+
+    def self.hint(text)
+      puts text.bold
+    end
+
+    def self.log(text)
+      puts text
+    end
+  end
+end

data/lib/genebrand/posparser.rb

@@ -0,0 +1,49 @@
+module Genebrand
+  class PosParser
+    require 'json'
+    require 'fileutils'
+
+    def initialize
+      @parsed = {}
+      @table = {}
+      # Сущ
+      @table['N'] = @parsed['noun'] = []
+      # Мн. число
+      @table['P'] = @parsed['plur'] = []
+      # Глаг. прич, пер, непер
+      @table['V'] = @parsed['verb_part'] = []
+      @table['t'] = @parsed['verb_trans'] = []
+      @table['i'] = @parsed['verb_intrans'] = []
+      # Прилаг
+      @table['A'] = @parsed['adj'] = []
+    end
+
+    def parse(filename)
+      unless File.exist?(filename)
+        fail "File not found: #{filename}"
+        return
+      end
+
+      File.open(filename, 'r').each_line do |line|
+        data = line.split("\t")
+
+        if !is_numeric?(data[0]) && (!/\A[a-zA-Z0-9]{2,10}\z/.match(data[0]).nil?)
+          data[1].split('').each do |partofsp|
+            @table[partofsp].push(data[0]) if @table.key?(partofsp)
+          end
+        end
+      end
+
+      @parsed
+    end
+
+    def parseandsave(filename)
+      FileUtils.mkdir_p 'lib/data'
+      File.open('lib/data/posinfo.json', 'w+') { |f| f.write(parse(filename).to_json) }
+    end
+
+    def is_numeric?(obj)
+      obj.to_s.match(/\A[+-]?\d+?(\.\d+)?\Z/).nil? ? false : true
+    end
+  end
+end

data/lib/genebrand/version.rb

@@ -0,0 +1,3 @@
+module Genebrand
+  VERSION = '0.1.0'
+end

data/lib/genebrand.rb

@@ -0,0 +1,25 @@
+$LOAD_PATH.unshift File.dirname(__FILE__)
+
+def require_all(path)
+  glob = File.join(File.dirname(__FILE__), path, '*.rb')
+  Dir[glob].each do |f|
+    require f
+  end
+end
+
+require 'rubygems'
+require 'colorator'
+require 'highline/import'
+
+module Genebrand
+  require 'genebrand/command'
+  require_all 'genebrand/commands'
+
+  autoload :PosParser, 'genebrand/posparser'
+  autoload :VERSION, 'genebrand/version'
+  autoload :Logger, 'genebrand/logger'
+  autoload :Generator, 'genebrand/generator'
+
+  class << self
+  end
+end

metadata

@@ -0,0 +1,149 @@
+--- !ruby/object:Gem::Specification
+name: genebrand
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Andrey Viktorov
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-06-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorator
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: mercenary
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !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.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+description: CLI brand names generator
+email:
+- andv@outlook.com
+executables:
+- genebrand
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/genebrand
+- genebrand.gemspec
+- lib/data/posinfo.json
+- lib/genebrand.rb
+- lib/genebrand/command.rb
+- lib/genebrand/commands/new.rb
+- lib/genebrand/generator.rb
+- lib/genebrand/logger.rb
+- lib/genebrand/posparser.rb
+- lib/genebrand/version.rb
+homepage: https://github.com/andreyviktorov/genebrand
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+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.4.7
+signing_key: 
+specification_version: 4
+summary: Genebrand is a small brands generator
+test_files: []