pickaname 0.1.3 → 0.1.8

data/lib/pickaname.rb

@@ -1,46 +1,113 @@
 require "pickaname/version"
+require 'timeout'
 
 module Pickaname
   class Error < StandardError; end
 
   class Robot
-    PREFIX      = File.foreach("lib/data/prefix.txt").map { |line| line.split(' ') }
-    SUFFIX       = File.foreach("lib/data/suffix.txt").map { |line| line.split(' ') }
-    FUNNY       = File.foreach("lib/data/funny.txt").map { |line| line.split(' ') }
+    PREFIX      = File.foreach(File.join(File.dirname(__FILE__), 'data', 'prefix.txt')).map { |line| line.split(' ') }
+    SUFFIX       = File.foreach(File.join(File.dirname(__FILE__), 'data', 'suffix.txt')).map { |line| line.split(' ') }
+    FUNNY       = File.foreach(File.join(File.dirname(__FILE__), 'data', 'funny.txt')).map { |line| line.split(' ') }
+    COMMON       = File.foreach(File.join(File.dirname(__FILE__), 'data', 'common.txt')).map { |line| line.split(' ') }
+    DARK       = File.foreach(File.join(File.dirname(__FILE__), 'data', 'dark.txt')).map { |line| line.split(' ') }
+
+    FIRSTNAME = File.foreach(File.join(File.dirname(__FILE__), 'data', 'firstname.txt')).map { |line| line.split(' ') }
+    LASTNAME = File.foreach(File.join(File.dirname(__FILE__), 'data', 'lastname.txt')).map { |line| line.split(' ') }
+
+
     PREFIX_SIZE = 1
     SUFFIX_SIZE  = 1
 
+    TIMEOUT_IN_SECONDS = 5
+
     attr_reader :name
 
     def initialize()
-      @name = random_string
+      @name = random_str
+    end
+
+    # def self.pseudo
+    #   @name = PREFIX.sample(PREFIX_SIZE).join << SUFFIX.sample(SUFFIX_SIZE).join
+    # end
+
+    def self.common(length: nil)
+      begin
+        Timeout::timeout(TIMEOUT_IN_SECONDS) do
+          @name = COMMON.sample(PREFIX_SIZE).join << COMMON.sample(SUFFIX_SIZE).join
+          while length
+            break if @name.length == length
+            @name = COMMON.sample(PREFIX_SIZE).join << COMMON.sample(SUFFIX_SIZE).join
+          end
+          return @name
+        end
+      rescue Timeout::Error
+        random_letters(length: length)
+      end
     end
 
-    def self.pseudo
-      @name = PREFIX.sample(PREFIX_SIZE).join << SUFFIX.sample(SUFFIX_SIZE).join
+    def self.funny(length: nil)
+      begin
+        Timeout::timeout(TIMEOUT_IN_SECONDS) do
+          @name = COMMON.sample(PREFIX_SIZE).join << FUNNY.sample(SUFFIX_SIZE).join
+          while length
+            break if @name.length == length
+            @name = COMMON.sample(PREFIX_SIZE).join << FUNNY.sample(SUFFIX_SIZE).join
+          end
+          return @name
+        end
+      rescue Timeout::Error
+        random_letters(length: length)
+      end
     end
 
-    def self.funny
-      @name = FUNNY.sample(PREFIX_SIZE).join << FUNNY.sample(SUFFIX_SIZE).join
+    def self.dark(length: nil)
+      begin
+        Timeout::timeout(TIMEOUT_IN_SECONDS) do
+          @name = COMMON.sample(PREFIX_SIZE).join << DARK.sample(SUFFIX_SIZE).join
+          while length
+            break if @name.length == length
+            @name = COMMON.sample(PREFIX_SIZE).join << DARK.sample(SUFFIX_SIZE).join
+          end
+          return @name
+        end
+      rescue Timeout::Error
+        random_letters(length: length)
+      end
     end
 
-    def self.random(record = Robot.new)
-      @name = record.random_string
+    def self.celebrity(length: nil)
+      begin
+        Timeout::timeout(TIMEOUT_IN_SECONDS) do
+          @name = FIRSTNAME.sample(PREFIX_SIZE).join << LASTNAME.sample(SUFFIX_SIZE).join
+          while length
+            break if @name.length == length
+            @name = FIRSTNAME.sample(PREFIX_SIZE).join << LASTNAME.sample(SUFFIX_SIZE).join
+          end
+          return @name
+        end
+      rescue Timeout::Error
+        random_letters(length: length)
+      end
     end
 
-    def random_string
-      down   = ('a'..'z').to_a
-      up     = ('A'..'Z').to_a
-      digits = ('0'..'9').to_a
-      [extract1(down), extract1(up), extract1(digits)].
-        concat(((down+up+digits).sample(8))).shuffle.join
+    def self.random_letters(length: 8)
+      record = Robot.new
+      begin
+        Timeout::timeout(TIMEOUT_IN_SECONDS) do
+          @name = record.random_str(length)
+          while length
+            break if @name.length == length
+            @name = record.random_str(length)
+          end
+          return @name
+        end
+      rescue Timeout::Error
+        puts "Timeout: No results found"
+      end
     end
 
-    def extract1(arr)
-      i = arr.size.times.to_a.sample
-      c = arr[i]
-      arr.delete_at(i)
-      c
+    def random_str(length = 8)
+      return Array.new(length){[*"a".."z", *"0".."9"].sample}.join
     end
   end
 end

data/lib/pickaname/version.rb

@@ -1,3 +1,3 @@
 module Pickaname
-  VERSION = "0.1.3"
+  VERSION = "0.1.8"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pickaname
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.8
 platform: ruby
 authors:
 - Karan
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-08-26 00:00:00.000000000 Z
+date: 2020-09-02 00:00:00.000000000 Z
 dependencies: []
 description: Randomly generates a name
 email:
@@ -24,9 +24,14 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- _config.yml
 - bin/console
 - bin/setup
+- lib/data/common.txt
+- lib/data/dark.txt
+- lib/data/firstname.txt
 - lib/data/funny.txt
+- lib/data/lastname.txt
 - lib/data/prefix.txt
 - lib/data/suffix.txt
 - lib/pickaname.rb