From e6c452d000c303374d2c6b36cd8c8e1aa8a6ba2e Mon Sep 17 00:00:00 2001
From: aronwk-aaron <aronwk.aaron@gmail.com>
Date: Fri, 17 Nov 2023 00:55:18 -0600
Subject: [PATCH 1/3] feat: add recaptcha support

---
 app/__init__.py                       |  55 ++++++--
 app/forms.py                          |   5 +
 app/settings_example.py               |  10 ++
 app/templates/admin/dashboard.html.j2 | 188 --------------------------
 app/templates/main/about.html.j2      |   1 -
 5 files changed, 58 insertions(+), 201 deletions(-)
 delete mode 100644 app/templates/admin/dashboard.html.j2

diff --git a/app/__init__.py b/app/__init__.py
index e14672a..5ba9792 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -204,19 +204,23 @@ def register_settings(app):
     # Load environment specific settings
     app.config['TESTING'] = False
     app.config['DEBUG'] = False
+    app.config['SQLALCHEMY_ENGINE_OPTIONS'] = {
+        "pool_pre_ping": True,
+        "pool_size": 10,
+        "max_overflow": 2,
+        "pool_recycle": 300,
+        "pool_pre_ping": True,
+        "pool_use_lifo": True
+    }
 
-    # always pull these two from the env
     app.config['SECRET_KEY'] = os.getenv(
         'APP_SECRET_KEY',
         app.config['APP_SECRET_KEY']
-
     )
     app.config['SQLALCHEMY_DATABASE_URI'] = os.getenv(
         'APP_DATABASE_URI',
         app.config['APP_DATABASE_URI']
     )
-
-    # try to get overides, otherwise just use what we have already
     app.config['USER_ENABLE_REGISTER'] = os.getenv(
         'USER_ENABLE_REGISTER',
         app.config['USER_ENABLE_REGISTER']
@@ -241,14 +245,6 @@ def register_settings(app):
         'USER_REQUIRE_INVITATION',
         app.config['USER_REQUIRE_INVITATION']
     )
-    app.config['SQLALCHEMY_ENGINE_OPTIONS'] = {
-        "pool_pre_ping": True,
-        "pool_size": 10,
-        "max_overflow": 2,
-        "pool_recycle": 300,
-        "pool_pre_ping": True,
-        "pool_use_lifo": True
-    }
     app.config['MAIL_SERVER'] = os.getenv(
         'MAIL_SERVER',
         app.config['MAIL_SERVER']
@@ -312,6 +308,41 @@ def register_settings(app):
         app.config['CACHE_LOCATION']
     )
 
+    # Recaptcha settings
+    if "RECAPTCHA_ENABLE" not in app.config:
+        app.config['RECAPTCHA_ENABLE'] = False
+    app.config['RECAPTCHA_ENABLE'] = os.getenv(
+        'RECAPTCHA_ENABLE',
+        app.config['RECAPTCHA_ENABLE']
+    )
+    if "RECAPTCHA_PUBLIC_KEY" not in app.config:
+        app.config['RECAPTCHA_PUBLIC_KEY'] = ''
+    app.config['RECAPTCHA_PUBLIC_KEY'] = os.getenv(
+        'RECAPTCHA_PUBLIC_KEY',
+        app.config['RECAPTCHA_PUBLIC_KEY']
+    )
+    if "RECAPTCHA_PRIVATE_KEY" not in app.config:
+        app.config['RECAPTCHA_PRIVATE_KEY'] = ''
+    app.config['RECAPTCHA_PRIVATE_KEY'] = os.getenv(
+        'RECAPTCHA_PRIVATE_KEY',
+        app.config['RECAPTCHA_PRIVATE_KEY']
+    )
+    # Optional
+    app.config['RECAPTCHA_API_SERVER'] = os.getenv(
+        'RECAPTCHA_API_SERVER',
+        app.config['RECAPTCHA_API_SERVER']
+    )
+    app.config['RECAPTCHA_PARAMETERS'] = os.getenv(
+        'RECAPTCHA_PARAMETERS',
+        app.config['RECAPTCHA_PARAMETERS']
+    )
+    if "RECAPTCHA_DATA_ATTRS" not in app.config:
+        app.config['RECAPTCHA_DATA_ATTRS'] = {'theme': 'white', 'size': 'invisible'}
+    app.config['RECAPTCHA_DATA_ATTRS'] = os.getenv(
+        'RECAPTCHA_DATA_ATTRS',
+        app.config['RECAPTCHA_DATA_ATTRS']
+    )
+
 
 
 def gm_level(gm_level):
diff --git a/app/forms.py b/app/forms.py
index 51a4a1f..5898611 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -1,4 +1,5 @@
 from flask_wtf import FlaskForm
+from flask_wtf.recaptcha import RecaptchaField
 from flask import current_app
 
 from flask_user.forms import (
@@ -84,6 +85,10 @@ class CustomRegisterForm(FlaskForm):
     ])
 
     invite_token = HiddenField('Token')
+    
+    # Use recaptcha if config enables recaptcha
+    if current_app.config["ENABLE_RECAPTCHA"]:
+        recaptcha = RecaptchaField()
 
     submit = SubmitField('Register')
 
diff --git a/app/settings_example.py b/app/settings_example.py
index df7a992..a449e51 100644
--- a/app/settings_example.py
+++ b/app/settings_example.py
@@ -61,3 +61,13 @@
 
 # Option will be removed once this feature is full implemeted
 ENABLE_CHAR_XML_UPLOAD = False
+
+# Recaptcha settings
+# See: https://flask-wtf.readthedocs.io/en/1.2.x/form/#recaptcha
+RECAPTCHA_ENABLE = False
+RECAPTCHA_PUBLIC_KEY = ''
+RECAPTCHA_PRIVATE_KEY = ''
+# Optional
+# RECAPTCHA_API_SERVER = ''
+# RECAPTCHA_PARAMETERS = ''
+RECAPTCHA_DATA_ATTRS = {'theme': 'white', 'size': 'invisible'}
diff --git a/app/templates/admin/dashboard.html.j2 b/app/templates/admin/dashboard.html.j2
deleted file mode 100644
index 3a90294..0000000
--- a/app/templates/admin/dashboard.html.j2
+++ /dev/null
@@ -1,188 +0,0 @@
-{% extends "bootstrap/base.html" %}
-{% block title %}Key Creation{% endblock %}
-
-{% block navbar %}
-<nav class="navbar navbar-default">
-  <div class="container-fluid">
-    <div class="navbar-header">
-      <a class="navbar-brand" href="{{ url_for('dashboard') }}">Dashboard</a>
-    </div>
-    <ul class="nav navbar-nav">
-    </ul>
-    <ul class="nav navbar-nav" style="float: right">
-      <li class="active"><a href="#">Welcome {{ current_user.username }}!</a></li>
-      <li><a href="{{ url_for('logout') }}">Logout</a></li>
-    </ul>
-  </div>
-</nav>
-{% endblock navbar %}}
-
-{% block content %}
-{# LOGO #}
-<div class="container" style="margin-top: 50px">
-
-    {# Display logo #}
-    <div style="margin-bottom: 50px">
-        <img
-                src="{{ url_for('static', filename=resources.LOGO) }}"
-                class="center-block img-responsive mx-auto d-block"
-                alt="Logo"
-                width="100"
-                height="100"
-        >
-    </div>
-
-</div>
-
-{# Key creation #}
-<div class="container">
-    <div class="text-center">
-        <h3>Key Creation</h3>
-    </div>
-
-    <div class="col-lg-3">
-    </div>
-    <div class="col-lg-6">
-
-    {# If the error value is set, display the error in red text #}
-    {% if error %}
-        <div class="alert alert-danger">
-            {{ error }}
-        </div>
-    {% endif %}
-
-    {# If the message value is set, display the message in green text #}
-    {% if message %}
-        <div class="alert alert-success">
-            {{ message }}
-        </div>
-    {% endif %}
-
-    {# Form which takes in Admin Username, Admin Password, and the amount of keys to create. #}
-    <form action="{{ url_for('dashboard') }}" method="post">
-        {# Key count input #}
-        <div class="form-group">
-            <label for="key_count">Generate keys</label>
-            <input type="number" class="form-control" name="key_count" placeholder="Enter number of keys...">
-            <small class="form-text text-muted">Number of keys to create.</small>
-        </div>
-        
-        {# Submit button #}
-        <div class="form-group">
-            <button type="submit" class="btn btn-primary">Generate Keys</button>
-        </div>
-    </form>
-
-    {# If the keys value is set, create a list for each key in keys #}
-    {% if keys %}
-        <div class="alert alert-success">
-            <ul>
-                {% for key in keys %}
-                    <li>{{ key }}</li>
-                {% endfor %}
-            </ul>
-        </div>
-    {% endif %}
-    </div>
-    <div class="col-lg-3">
-    </div>
-    </div>
-</div>
-
-{# Activity graphs #}
-<div class="container">
-
-    <div class="text-center">
-        <h3>Activity</h3>
-    </div>
-
-    <div class="col-lg-3">
-    </div>
-
-    <div class="col-lg-6">
-
-    <canvas id="sessions_graph" width="400" height="400"></canvas>
-    <canvas id="play_time_graph" width="400" height="400"></canvas>
-    <canvas id="zone_play_time_graph" width="400" height="400"></canvas>
-
-    </div>
-
-    <div class="col-lg-3">
-    </div>
-</div>
-{% endblock %}
-
-{% block scripts %}
-<script src="https://cdn.jsdelivr.net/npm/chart.js@3.5.1/dist/chart.min.js"></script>
-<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.5.1/jquery.min.js"></script>
-
-<script>
-// Make a get request to the server to load the activity data
-$.get("{{ url_for('load_activities') }}", function(data) {
-
-});
-
-// Make a get request to the server to get the activity data for "sessions"
-$.get("{{ url_for('activity_data', name='sessions') }}", function(data) {
-    // Load data as a json object
-    data = JSON.parse(data);
-    var ctx = document.getElementById('sessions_graph').getContext('2d');
-    var myChart = new Chart(ctx, {
-        type: 'line',
-        data: {
-            labels: data.labels,
-            datasets: data.datasets
-        },
-        options: {
-            scales: {
-                y: {
-                    beginAtZero: true
-                }
-            }
-        }
-    });
-});
-
-// Make a get request to the server to get the activity data for "play_time"
-$.get("{{ url_for('activity_data', name='play_time') }}", function(data) {
-    // Load data as a json object
-    data = JSON.parse(data);
-    var ctx = document.getElementById('play_time_graph').getContext('2d');
-    var myChart = new Chart(ctx, {
-        type: 'line',
-        data: {
-            labels: data.labels,
-            datasets: data.datasets
-        },
-        options: {
-            scales: {
-                y: {
-                    beginAtZero: true
-                }
-            }
-        }
-    });
-});
-
-// Make a get request to the server to get the activity data for "zone_play_time"
-$.get("{{ url_for('activity_data', name='zone_play_time') }}", function(data) {
-    // Load data as a json object
-    data = JSON.parse(data);
-    var ctx = document.getElementById('zone_play_time_graph').getContext('2d');
-    var myChart = new Chart(ctx, {
-        type: 'line',
-        data: {
-            labels: data.labels,
-            datasets: data.datasets
-        },
-        options: {
-            scales: {
-                y: {
-                    beginAtZero: true
-                }
-            }
-        }
-    });
-});
-</script>
-{% endblock scripts %}}
\ No newline at end of file
diff --git a/app/templates/main/about.html.j2 b/app/templates/main/about.html.j2
index 0e297f1..3cec998 100644
--- a/app/templates/main/about.html.j2
+++ b/app/templates/main/about.html.j2
@@ -82,7 +82,6 @@
         </div>
       {% endif %}
 
-
       <div class="row">
         <div class="col text-right">
           Source

From 7633053490eff99945bddda556aa5adccbb7a1e3 Mon Sep 17 00:00:00 2001
From: aronwk-aaron <aronwk.aaron@gmail.com>
Date: Fri, 17 Nov 2023 16:38:01 -0600
Subject: [PATCH 2/3] it works

---
 app/__init__.py                               | 29 ++++----
 app/forms.py                                  | 68 ++++++-------------
 app/templates/flask_user/login.html           |  7 +-
 .../flask_user/login_or_register.html         | 10 +++
 app/templates/flask_user/register.html        |  5 ++
 5 files changed, 56 insertions(+), 63 deletions(-)

diff --git a/app/__init__.py b/app/__init__.py
index 5ba9792..756b8ed 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -328,20 +328,21 @@ def register_settings(app):
         app.config['RECAPTCHA_PRIVATE_KEY']
     )
     # Optional
-    app.config['RECAPTCHA_API_SERVER'] = os.getenv(
-        'RECAPTCHA_API_SERVER',
-        app.config['RECAPTCHA_API_SERVER']
-    )
-    app.config['RECAPTCHA_PARAMETERS'] = os.getenv(
-        'RECAPTCHA_PARAMETERS',
-        app.config['RECAPTCHA_PARAMETERS']
-    )
-    if "RECAPTCHA_DATA_ATTRS" not in app.config:
-        app.config['RECAPTCHA_DATA_ATTRS'] = {'theme': 'white', 'size': 'invisible'}
-    app.config['RECAPTCHA_DATA_ATTRS'] = os.getenv(
-        'RECAPTCHA_DATA_ATTRS',
-        app.config['RECAPTCHA_DATA_ATTRS']
-    )
+    if "RECAPTCHA_API_SERVER" in app.config:
+        app.config['RECAPTCHA_API_SERVER'] = os.getenv(
+            'RECAPTCHA_API_SERVER',
+            app.config['RECAPTCHA_API_SERVER']
+        )
+    if "RECAPTCHA_PARAMETERS" in app.config:
+        app.config['RECAPTCHA_PARAMETERS'] = os.getenv(
+            'RECAPTCHA_PARAMETERS',
+            app.config['RECAPTCHA_PARAMETERS']
+        )
+    if "RECAPTCHA_DATA_ATTRS" in app.config:
+        app.config['RECAPTCHA_DATA_ATTRS'] = os.getenv(
+            'RECAPTCHA_DATA_ATTRS',
+            app.config['RECAPTCHA_DATA_ATTRS']
+        )
 
 
 
diff --git a/app/forms.py b/app/forms.py
index 5898611..30d9be3 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -1,18 +1,15 @@
-from flask_wtf import FlaskForm
-from flask_wtf.recaptcha import RecaptchaField
+from flask_wtf import FlaskForm, Recaptcha, RecaptchaField
 from flask import current_app
 
 from flask_user.forms import (
     unique_email_validator,
-    password_validator,
-    unique_username_validator
+    LoginForm,
+    RegisterForm
 )
 from flask_user import UserManager
 from wtforms.widgets import TextArea, NumberInput
 from wtforms import (
     StringField,
-    HiddenField,
-    PasswordField,
     BooleanField,
     SubmitField,
     validators,
@@ -37,35 +34,21 @@ def validate_play_key(form, field):
         field.data = PlayKey.key_is_valid(key_string=field.data)
     return
 
+def validate_recaptcha(form, field):
+    current_app.logger.info("start validating with recaptcha")
+    if current_app.config["RECAPTCHA_ENABLE"]:
+        current_app.logger.info("validating with recaptcha")
+        return Recaptcha()
+    current_app.logger.info("skipping validating with recaptcha")
+    return True
+
 
 class CustomUserManager(UserManager):
     def customize(self, app):
         self.RegisterFormClass = CustomRegisterForm
+        self.LoginFormClass = CustomLoginForm
 
-
-class CustomRegisterForm(FlaskForm):
-    """Registration form"""
-    next = HiddenField()
-    reg_next = HiddenField()
-
-    # Login Info
-    email = StringField(
-        'E-Mail',
-        validators=[
-            Optional(),
-            validators.Email('Invalid email address'),
-            unique_email_validator,
-        ]
-    )
-
-    username = StringField(
-        'Username',
-        validators=[
-            DataRequired(),
-            unique_username_validator,
-        ]
-    )
-
+class CustomRegisterForm(RegisterForm):
     play_key_id = StringField(
         'Play Key',
         validators=[
@@ -73,25 +56,14 @@ class CustomRegisterForm(FlaskForm):
             validate_play_key,
         ]
     )
+    recaptcha = RecaptchaField(
+        validators=[validate_recaptcha]
+    )
 
-    password = PasswordField('Password', validators=[
-        DataRequired(),
-        password_validator,
-        validators.length(max=40, message="The maximum length of the password is 40 characters due to game client limitations")
-    ])
-    retype_password = PasswordField('Retype Password', validators=[
-        validators.EqualTo('password', message='Passwords did not match'),
-        validators.length(max=40, message="The maximum length of the password is 40 characters due to game client limitations")
-    ])
-
-    invite_token = HiddenField('Token')
-    
-    # Use recaptcha if config enables recaptcha
-    if current_app.config["ENABLE_RECAPTCHA"]:
-        recaptcha = RecaptchaField()
-
-    submit = SubmitField('Register')
-
+class CustomLoginForm(LoginForm):
+    recaptcha = RecaptchaField(
+        validators=[validate_recaptcha]
+    )
 
 class CreatePlayKeyForm(FlaskForm):
 
diff --git a/app/templates/flask_user/login.html b/app/templates/flask_user/login.html
index c8d7e3d..2353edd 100644
--- a/app/templates/flask_user/login.html
+++ b/app/templates/flask_user/login.html
@@ -60,7 +60,12 @@ <h1>{%trans%}Sign in{%endtrans%}</h1>
 
     {# Remember me #}
     {% if user_manager.USER_ENABLE_REMEMBER_ME %}
-    {{ render_checkbox_field(login_form.remember_me, tabindex=130) }}
+        {{ render_checkbox_field(login_form.remember_me, tabindex=130) }}
+    {% endif %}
+
+    {# recaptcha #}
+    {% if config.RECAPTCHA_ENABLE %}
+        {{ render_field(form.recaptcha, tabindex=250) }}
     {% endif %}
 
     {# Submit button #}
diff --git a/app/templates/flask_user/login_or_register.html b/app/templates/flask_user/login_or_register.html
index e51a6f1..59fd755 100644
--- a/app/templates/flask_user/login_or_register.html
+++ b/app/templates/flask_user/login_or_register.html
@@ -29,6 +29,11 @@ <h1>{%trans%}Sign in{%endtrans%}</h1>
     {{ render_checkbox_field(login_form.remember_me, tabindex=130) }}
     {% endif %}
 
+    {# recaptcha #}
+    {% if config.RECAPTCHA_ENABLE %}
+        {{ form.recaptcha }}
+    {% endif %}
+
     {# Submit button #}
     {{ render_submit_field(login_form.submit, tabindex=180) }}
 </form>
@@ -63,6 +68,11 @@ <h1>{%trans%}Register{%endtrans%}</h1>
         {{ render_field(register_form.retype_password, tabindex=240) }}
     {% endif %}
 
+    {# recaptcha #}
+    {% if config.RECAPTCHA_ENABLE %}
+        {{ register_form.recaptcha }}
+    {% endif %}
+
     {{ render_submit_field(register_form.submit, tabindex=280) }}
 </form>
 
diff --git a/app/templates/flask_user/register.html b/app/templates/flask_user/register.html
index f6a6e28..99bc9d3 100644
--- a/app/templates/flask_user/register.html
+++ b/app/templates/flask_user/register.html
@@ -47,6 +47,11 @@ <h1>{%trans%}Register{%endtrans%}</h1>
         {{ render_field(form.retype_password, tabindex=240) }}
     {% endif %}
 
+    {# recaptcha #}
+    {% if config.RECAPTCHA_ENABLE %}
+        {{ render_field(form.recaptcha, tabindex=250) }}
+    {% endif %}
+
     {{ render_submit_field(form.submit, tabindex=280) }}
 </form>
 

From b7e48bb656c1ac0c46442f1a5c596d684bc6f4dc Mon Sep 17 00:00:00 2001
From: aronwk-aaron <aronwk.aaron@gmail.com>
Date: Fri, 17 Nov 2023 17:40:01 -0600
Subject: [PATCH 3/3] extend the recaptha validator to make it optional

---
 app/forms.py | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/app/forms.py b/app/forms.py
index 30d9be3..1926724 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -34,13 +34,11 @@ def validate_play_key(form, field):
         field.data = PlayKey.key_is_valid(key_string=field.data)
     return
 
-def validate_recaptcha(form, field):
-    current_app.logger.info("start validating with recaptcha")
-    if current_app.config["RECAPTCHA_ENABLE"]:
-        current_app.logger.info("validating with recaptcha")
-        return Recaptcha()
-    current_app.logger.info("skipping validating with recaptcha")
-    return True
+class CustomRecaptcha(Recaptcha):
+    def __call__(self, form, field):
+        if not current_app.config.get("RECAPTCHA_ENABLE", False):
+            return True
+        return super(CustomRecaptcha, self).__call__(form, field)
 
 
 class CustomUserManager(UserManager):
@@ -57,12 +55,12 @@ class CustomRegisterForm(RegisterForm):
         ]
     )
     recaptcha = RecaptchaField(
-        validators=[validate_recaptcha]
+        validators=[CustomRecaptcha()]
     )
 
 class CustomLoginForm(LoginForm):
     recaptcha = RecaptchaField(
-        validators=[validate_recaptcha]
+        validators=[CustomRecaptcha()]
     )
 
 class CreatePlayKeyForm(FlaskForm):