From 700e084101fe789433b48f17e555121f39c79bb8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Garc=C3=ADa?=
 <dani-garcia@users.noreply.github.com>
Date: Fri, 25 Jan 2019 18:50:57 +0100
Subject: [PATCH] Add 2FA icon to admin panel

---
 src/api/identity.rs                 | 5 +----
 src/db/models/two_factor.rs         | 1 +
 src/mail.rs                         | 2 +-
 src/static/templates/admin/page.hbs | 3 +++
 4 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/api/identity.rs b/src/api/identity.rs
index b6092a3..8638796 100644
--- a/src/api/identity.rs
+++ b/src/api/identity.rs
@@ -151,10 +151,7 @@ fn twofactor_auth(
     device: &mut Device,
     conn: &DbConn,
 ) -> ApiResult<Option<String>> {
-    let twofactors_raw = TwoFactor::find_by_user(user_uuid, conn);
-    // Remove u2f challenge twofactors (impl detail)
-    let twofactors: Vec<_> = twofactors_raw.iter().filter(|tf| tf.type_ < 1000).collect();
-
+    let twofactors = TwoFactor::find_by_user(user_uuid, conn);
     let providers: Vec<_> = twofactors.iter().map(|tf| tf.type_).collect();
 
     // No twofactor token if twofactor is disabled
diff --git a/src/db/models/two_factor.rs b/src/db/models/two_factor.rs
index fd1d2c7..1086f0e 100644
--- a/src/db/models/two_factor.rs
+++ b/src/db/models/two_factor.rs
@@ -100,6 +100,7 @@ impl TwoFactor {
     pub fn find_by_user(user_uuid: &str, conn: &DbConn) -> Vec<Self> {
         twofactor::table
             .filter(twofactor::user_uuid.eq(user_uuid))
+            .filter(twofactor::type_.lt(1000)) // Filter implementation types
             .load::<Self>(&**conn)
             .expect("Error loading twofactor")
     }
diff --git a/src/mail.rs b/src/mail.rs
index 1d9e45f..0920f9d 100644
--- a/src/mail.rs
+++ b/src/mail.rs
@@ -125,7 +125,7 @@ pub fn send_invite_confirmed(address: &str, org_name: &str, config: &MailConfig)
 fn send_email(address: &str, subject: &str, body: &str, config: &MailConfig) -> EmptyResult {
     let email = EmailBuilder::new()
         .to(address)
-        .from((config.smtp_from.clone(), config.smtp_from_name.clone()))
+        .from((config.smtp_from.as_str(), config.smtp_from_name.as_str()))
         .subject(subject)
         .header(("Content-Type", "text/html"))
         .body(body)
diff --git a/src/static/templates/admin/page.hbs b/src/static/templates/admin/page.hbs
index 538eda8..5d1d9c7 100644
--- a/src/static/templates/admin/page.hbs
+++ b/src/static/templates/admin/page.hbs
@@ -10,6 +10,9 @@
                     <div class="row justify-content-between">
                         <div class="col">
                             <strong>{{Name}}</strong>
+                            {{#if TwoFactorEnabled}}
+                            <span class="badge badge-success ml-2">2FA</span>
+                            {{/if}}
                             <span class="d-block">{{Email}}</span>
                         </div>
                         <div class="col">