Просмотр исходного кода

Merge pull request #181 from developStorm/webauthn-patch-1

Auto activate WebAuthn authentication
Son Nguyen Kim 5 лет назад
Родитель
Сommit
4d87df01a3
2 измененных файлов с 7 добавлено и 0 удалено
  1. 4 0
      app/auth/templates/auth/fido.html
  2. 3 0
      app/auth/views/fido.py

+ 4 - 0
app/auth/templates/auth/fido.html

@@ -62,6 +62,10 @@
         $("#btnVerifyKey").click(verifyKey);
       </script>
 
+      {% if auto_activate %}
+        <script>$('document').ready(verifyKey());</script>
+      {% endif %}
+        
     </div>
   </div>
 

+ 3 - 0
app/auth/views/fido.py

@@ -35,6 +35,7 @@ def fido():
         flash("Only user with security key linked should go to this page", "warning")
         return redirect(url_for("auth.login"))
 
+    auto_activate = True
     fido_token_form = FidoTokenForm()
 
     next_url = request.args.get("next")
@@ -69,6 +70,7 @@ def fido():
         except Exception as e:
             LOG.error(f"An error occurred in WebAuthn verification process: {e}")
             flash("Key verification failed.", "warning")
+            auto_activate = False
         else:
             user.fido_sign_count = new_sign_count
             db.session.commit()
@@ -101,4 +103,5 @@ def fido():
         fido_token_form=fido_token_form,
         webauthn_assertion_options=webauthn_assertion_options,
         enable_otp=user.enable_otp,
+        auto_activate=auto_activate,
     )