id_shuffler 0.0.0 → 0.0.1

data/ext/id_shuffler/id_shuffler.c

@@ -93,18 +93,18 @@ static VALUE r_base32_unshuffle(VALUE self, VALUE shuffled_str, VALUE key_str) {
     lfsr_num = lfsr_num & 32767;
     l = (r ^ lfsr_num);
     r = l_orig;
-	}
+  }
   // return the resulting original number as a ruby fixnum
   unsigned long original;
-	original = (l << 15) + r;
+  original = (l << 15) + r;
   return ULONG2NUM(original);
 }
 
 void Init_id_shuffler(void) {
   // add two class methods to our IdShuffler class: shuffle and unshuffle, each
   // of which take the object to be [unshuffled] and the key.
-	VALUE klass = rb_define_class("IdShuffler", rb_cObject);
-	rb_define_singleton_method(klass, "shuffle", r_base32_shuffle, 2);
-	rb_define_singleton_method(klass, "unshuffle", r_base32_unshuffle, 2);
+  VALUE klass = rb_define_class("IdShuffler", rb_cObject);
+  rb_define_singleton_method(klass, "shuffle_with_raw_key", r_base32_shuffle, 2);
+  rb_define_singleton_method(klass, "unshuffle_with_raw_key", r_base32_unshuffle, 2);
 }
 

data/lib/id_shuffler.rb

@@ -18,11 +18,23 @@ class IdShuffler
   end
 
   def shuffle(i)
-    self.class.shuffle(i.to_i, @key)
+    self.class.shuffle_with_raw_key(i.to_i, @key)
   end
 
   def unshuffle(s)
-    self.class.unshuffle(s.to_s, @key)
+    self.class.unshuffle_with_raw_key(s.to_s, @key)
+  end
+
+  def self.shuffle(i, key)
+    @keycache ||= {}
+    @keycache[key] ||= Digest::MD5.digest(key)
+    shuffle_with_raw_key(i.to_i, @keycache[key])
+  end
+
+  def self.unshuffle(s, key)
+    @keycache ||= {}
+    @keycache[key] ||= Digest::MD5.digest(key)
+    unshuffle_with_raw_key(s.to_s, @keycache[key])
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: id_shuffler
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.1
   prerelease: 
 platform: ruby
 authors: