From 312263219e8c3ae78b241104deccea4a82d767a2 Mon Sep 17 00:00:00 2001
From: cflip <36554078+cflip@users.noreply.github.com>
Date: Sat, 21 Aug 2021 09:54:36 -0600
Subject: Add user generation numbers

---
 register.php | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'register.php')

diff --git a/register.php b/register.php
index 44ab879..e286f01 100755
--- a/register.php
+++ b/register.php
@@ -32,12 +32,13 @@ if ($_SERVER['REQUEST_METHOD'] == 'POST') {
 	$invite_code = $form->validate_invite_code($_POST['invite_code']);
 
     $form->on_success(function () use ($username, $password, $invite_code) {
-        User::register($username, $password);
-
         $inviter = new User();
         $inviter->get_by_invite_code($invite_code);
         $inviter->update_invite_code();
 
+        $generation = $inviter->generation + 1;
+        User::register($username, $password, $generation);
+
         echo '<p class="success">Account successfully registered! You can now <a href="signin.php">sign in</a></p>';
     });
 }
-- 
cgit v1.2.3