From 6f09343e85ab23c4d81f827bbd47a720f3e2cf7b Mon Sep 17 00:00:00 2001
From: ankonym <ankonym@gmail.com>
Date: Fri, 2 Oct 2015 20:58:29 +0200
Subject: Update rake task to allow generation of multi-use invites

The rake task now takes a second (optional) argument that sets the number of uses per invite code.
If this is omitted, the default number of uses is 1.

(This commit also contains some minor code cleanup that removes some stuff that I'd commented out but not removed.)
---
 lib/tasks/invite_code.rake | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

(limited to 'lib/tasks')

diff --git a/lib/tasks/invite_code.rake b/lib/tasks/invite_code.rake
index f3bafac..6623640 100644
--- a/lib/tasks/invite_code.rake
+++ b/lib/tasks/invite_code.rake
@@ -1,16 +1,25 @@
 
 
 desc "Generate a batch of invite codes"
-task :generate_invites, [:n] => :environment do |task, args|
+task :generate_invites, [:n, :u] => :environment do |task, args|
 
-    codes = args.n
-    codes = codes.to_i
+  codes = args.n
+  codes = codes.to_i
 
-    codes.times do |x|
+  if args.u == nil
+    max_uses = 1
+
+  elsif
+    max_uses = args.u
+    max_uses = max_uses.to_i
+  end
+
+  codes.times do |x|
     x = InviteCode.new
+    x.invite_max_uses = max_uses
     x.save
-    puts "#{x.invite_code} Code generated."
-
+    puts "#{x.invite_code} Code generated with #{x.invite_max_uses} uses."
   end
+
 end
 
-- 
cgit v1.2.3