genny 0.3.0 → 0.4.0

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    ZDUwYjhjNDY5MmZlYmIwYjZmMTYzNTU5MzQyOWQxOWM4YzdhZjA3YQ==
+    NjMzY2FjYWYxN2UzOTE1ZTdmOTVjMmU1M2Q5N2MwMmEzOTNkMjEzNg==
   data.tar.gz: !binary |-
-    ZGE2ZjhmYjhmMGZiMjcyNDFmZTEwNTUwNTRjZGQ4NmEwZWJhNDgxNA==
+    MGZlNjNlYTE3NjkwZGM5ZmExYjk5NmFlZmZhMGU3NTU3YWMxOTlkNQ==
 SHA512:
   metadata.gz: !binary |-
-    ZDI1MjVkNTNjMDNmMzQxMDdmMTZkOTU4NzhlN2UyYjcwZDlkZTBiNmQ0Mjdk
-    MWE5YzE2OGU0NmJjNGM0Yjc4MjU1MmFkMzkwYjAzY2MxNDI4NGNlYTRkMGQ4
-    ZTFjMmJjOWU4NDI2OTAyYTc2Njg2YThkYzYxMzNmZDI2YWY2ZTk=
+    NGYxNmM0N2MyYWJiMGFkOTNlYTU2NzI5Y2QzMzNiNTU3MjIwNzJkODg0ZDc4
+    ZjE0ZTQ4OWUzMDU0NmNiYjI3MGU4NTY3ODg5MGFkNTIzZGVmMzdmN2E5NWE3
+    YTVkOWFiMWRkZGJlNTRlYTI2ZWFlZjYyZWRlM2RmOTZlY2JjMDU=
   data.tar.gz: !binary |-
-    NGFkY2I0ZTVkMWIwYjEyZWYxNGZhNjNjMmRlZDVkNTc0OWY0OWMxZGY2ZjBl
-    ODZmNWRhNjY5MTE5YWM0NWQ1YWRkODU1ODAxNDQxYzE2MjAyMWJhN2YwYWQx
-    MGMzYTI0NDhiODhlYjAyZWU2MmQ2MDllYzIyMWRhMGUwZjQ1ZjY=
+    MmVjNWYwMDk4MGY3Nzg5ZjQ0MWFiYjU4OTg1OTMxYTEyNGQwZTg0ZTY2ZjYx
+    ODdlYjkyMWJhZDkxOGIyMmViMWZjNmU3YjViZDNmYTI3MGE3OTFiNWE2YzQw
+    YWU4Nzc2ZGEyYjAwZDg0ZGI1OGYwMDliNDhlODNmNmVhOTFiYzU=

data/VERSION

@@ -1 +1 @@
-0.3.0
+0.4.0

data/lib/genny/regexp.rb

@@ -2,13 +2,88 @@ require "genny/base"
 
 module Genny
   module Regexp
+    DEFAULT_OPTS = {
+      min: 2,
+      max: 5
+    }
     # A bit unecessary, but who knows, maybe someone will have use for it
     def self.genny(_opts = {})
       %r{.*}
     end
 
     def genny(opts = {})
-      raise NotImplementedError, "I gotta get some sleep. Genny can't make strings which fit Regexps... yet."
+      raise NotImplementedError, "This regular expression doesn't seem to support the `source` method" unless self.respond_to?(:source)
+      begin
+        str = generate_from_regexp_string(self.source, DEFAULT_OPTS.merge(opts))
+        raise unless str.match(self)
+      rescue
+        raise NotImplementedError, "Looks like Genny doesn't support the regular expression #{self.source}#{str.nil? ? '.' : ", it made '#{str}'."} Feel like adding support? https://github.com/blinkboxbooks/genny"
+      end
+      str
+    end
+
+    private
+
+    def generate_from_regexp_string(string, opts = {})
+      string.sub!(/^\^?(.+?)\$?$/, '\1')
+      string.scan(/(?:\[([^\]]+)\]|\((?:\?<[^>]+>|\?:)?(.*)\)|\\([*+?\\\{\}\[\]WwDdSs.^$])|([^*+?\{\}\[\]]))(\*\??|\+\??|\{(\d*),?(\d*)\})?/).map do |group|
+        ranges, subexpr, special_char, char, limit, min, max = group
+        if !ranges.nil?
+          chars = array_from_ranges(ranges)
+          get_char = proc { chars.sample }
+        elsif !subexpr.nil?
+          get_char = proc { generate_from_regexp_string(subexpr, opts) }
+        elsif !special_char.nil?
+          get_char = special_char_parser(special_char)
+        else
+          get_char = proc { char }
+        end
+
+        min = max = "1" if limit.nil?
+        if min.nil? || min.empty?
+          max = (max.nil? || max.empty?) ? opts[:max] : max.to_i
+          min = [opts[:min], max].min
+        elsif max.empty?
+          min = max = min.to_i
+        else
+          min = min.to_i
+          max = max.to_i
+        end
+        num = max == min ? min : Random.rand(max - min) + min
+        
+        num.times.inject("") do |str, _|
+          str << get_char.call
+        end
+      end.join
+    end
+
+    def array_from_ranges(ranges)
+      ranges.scan(/(([^\\])-(.)|.)/).inject([]) do |all, match|
+        if match[2].nil?
+          all.push(match.first)
+        else
+          all.push(*(match[1]..match[2]).to_a)
+        end
+      end
+    end
+
+    def special_char_parser(special_char)
+      case special_char
+      when "w"
+        proc { (('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a + %w{_}).sample }
+      when "W"
+        proc { ((" ".."/").to_a + ("[".."`").to_a + ("{".."~").to_a - %w{_}).sample }
+      when "d"
+        proc { ("0".."9").to_a.sample }
+      when "D"
+        proc { (('a'..'z').to_a + ('A'..'Z').to_a).sample }
+      when "s"
+        proc { " " }
+      when "S"
+        proc { (' '..'~').to_a .sample }
+      else
+        proc { special_char }
+      end
     end
   end
 end

data/lib/genny/string.rb

@@ -17,6 +17,9 @@ module Genny
       opts = Genny.symbolize(opts)
       viable_formats = @@formats.keys & [*opts[:format]]
       return @@formats[viable_formats.sample].call(opts) unless viable_formats.empty?
+      if (re = ::Regexp.new(opts[:format]) rescue false)
+        return re.extend(Genny::Regexp).genny
+      end
       guess = @@hints.inject(nil) { |guess, hint|
         break guess if !(guess = hint.call(opts)).nil?
       } if opts[:hint]

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: genny
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - JP Hastings-Spital
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-03 00:00:00.000000000 Z
+date: 2014-11-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake