short-url-creator 0.1.0 → 0.1.1

data/lib/short-url-creator.rb

@@ -15,6 +15,7 @@ class ShortUrlCreator
     @count = 0
     @a = Array.new(7, -1)
     @k = 0
+    @short_url = []
   end 
 
   def iterate_chars(array_size)
@@ -33,13 +34,17 @@ class ShortUrlCreator
     k = a.length  
     if  (k > 1)  
       if ((@h[a[k-2]] + @h[a[k-1]]) != '10')
-        puts buffer.reverse 
+        @short_url << buffer.reverse 
       end
     else
-      puts buffer.reverse
+      @short_url << buffer.reverse
     end
   end
 
+  def generate(count)
+    get_short_url(count)
+    @short_url.join
+  end
   def get_short_url(count)
     if count >  @array_size
       new_count = count - @array_size

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: short-url-creator
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors: []