diff options
-rw-r--r-- | app/models/invite_code.rb | 8 | ||||
-rw-r--r-- | lib/tasks/invite_code.rake | 1 |
2 files changed, 8 insertions, 1 deletions
diff --git a/app/models/invite_code.rb b/app/models/invite_code.rb index 701b663..30d32d6 100644 --- a/app/models/invite_code.rb +++ b/app/models/invite_code.rb @@ -1,3 +1,5 @@ +require 'base64' +require 'securerandom' class InviteCode < CouchRest::Model::Base use_database 'invite_codes' @@ -13,10 +15,14 @@ class InviteCode < CouchRest::Model::Base def initialize(attributes = {}, options = {}) super(attributes, options) - write_attribute('invite_code', CouponCode.generate) if new? + write_attribute('invite_code', generate_invite) if new? end + def generate_invite + Base64.encode64(SecureRandom.random_bytes).downcase.gsub(/[0oil1+_\/]/,'')[0..7].scan(/..../).join('-') + end + def set_invite_code(code) write_attribute(:invite_code, code) end diff --git a/lib/tasks/invite_code.rake b/lib/tasks/invite_code.rake index b379d3a..fa397bb 100644 --- a/lib/tasks/invite_code.rake +++ b/lib/tasks/invite_code.rake @@ -19,3 +19,4 @@ task :generate_invites, [:n, :u] => :environment do |task, args| end end + |