Implement invite codes for user registration

1 files changed, 10 insertions, 2 deletions

diff --git a/register.php b/register.php
index d7f5fed..44ab879 100755
--- a/register.php
+++ b/register.php
@@ -10,7 +10,9 @@
 	<h2>Register an account</h2>
 	<form action="register.php" method="post">
 		<label for="user_name">Username: </label><br>
-		<input type="text" name="user_name"><br>
+		<input type="text" name="user_name" id="user_name"><br>
+        <label for="invite_code">Invite Code: </label><br>
+        <input type="text" name="invite_code" id="invite_code"><br>
 		<label for="user_pass">Password: </label><br>
 		<input type="password" name="user_pass"><br>
 		<label for="user_pass_check">Re-enter password: </label><br>
@@ -27,9 +29,15 @@ if ($_SERVER['REQUEST_METHOD'] == 'POST') {
 	$form = new RegisterForm();
 	$username = $form->validate_username($_POST['user_pass']);
 	$password = $form->validate_password($_POST['user_pass'], $_POST['user_pass_check']);
+	$invite_code = $form->validate_invite_code($_POST['invite_code']);
 
-    $form->on_success(function () use ($username, $password) {
+    $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();
+
         echo '<p class="success">Account successfully registered! You can now <a href="signin.php">sign in</a></p>';
     });
 }