From 3138a2964fd4209ee29cc2eb270bd60eded872b6 Mon Sep 17 00:00:00 2001
From: cellarspoon <cellarspoon@riseup.net>
Date: Mon, 10 Jan 2022 09:40:16 +0100
Subject: [PATCH] feat: add invited_by attribute

---
 keycloak_collective_portal/routes/register.py      | 3 +++
 keycloak_collective_portal/templates/register.html | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/keycloak_collective_portal/routes/register.py b/keycloak_collective_portal/routes/register.py
index 4f98323..e2b38c4 100644
--- a/keycloak_collective_portal/routes/register.py
+++ b/keycloak_collective_portal/routes/register.py
@@ -58,7 +58,9 @@ def form_keycloak_register(
     username: str = Form(...),
     email: str = Form(...),
     password: str = Form(...),
+    invited_by: str = Form(...),
 ):
+
     payload = {
         "email": email,
         "username": username,
@@ -74,6 +76,7 @@ def form_keycloak_register(
         "realmRoles": [
             "user_default",
         ],
+        "attributes": {"invited_by": username},
     }
 
     try:
diff --git a/keycloak_collective_portal/templates/register.html b/keycloak_collective_portal/templates/register.html
index ee5268b..db5476c 100644
--- a/keycloak_collective_portal/templates/register.html
+++ b/keycloak_collective_portal/templates/register.html
@@ -20,6 +20,8 @@
     <label for="password">Password:</label>
     <input type="password" name="password" />
 
+    <input type="hidden" name="invited_by" value="{{ username }}"/>
+
     <input type="submit" />
   </form>