From b41b6952289052736e503696edfe40078245d2f4 Mon Sep 17 00:00:00 2001
From: Son <son@son>
Date: Tue, 26 Oct 2021 12:06:16 +0200
Subject: [PATCH] include the referral code when user signs up via SIWSL

---
 .../templates/oauth/authorize_nonlogin_user.html    | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/app/oauth/templates/oauth/authorize_nonlogin_user.html b/app/oauth/templates/oauth/authorize_nonlogin_user.html
index ab7da8f0..c06caea1 100644
--- a/app/oauth/templates/oauth/authorize_nonlogin_user.html
+++ b/app/oauth/templates/oauth/authorize_nonlogin_user.html
@@ -32,10 +32,15 @@
           <a href="{{ url_for('auth.login', next=next) }}" class="btn btn-success">
             Login
           </a>
-
-          <a href="{{ url_for('auth.register', next=next) }}" class="btn btn-info">
-            Sign Up
-          </a>
+          {% if client.referral %}
+            <a href="{{ url_for('auth.register', slref=client.referral.code, next=next) }}" class="btn btn-info">
+              Sign Up
+            </a>
+          {% else %}
+            <a href="{{ url_for('auth.register', next=next) }}" class="btn btn-info">
+              Sign Up
+            </a>
+          {% endif %}
         </div>
       </div>