From 98350860eb7a166adc5d5c124f3476d473300eb7 Mon Sep 17 00:00:00 2001
From: jendib <benjamin.gam@gmail.com>
Date: Thu, 24 Mar 2016 00:03:29 +0100
Subject: [PATCH] #84: Ask for a TOTP validation code (web UI)

---
 .../webapp/src/app/docs/controller/Login.js   | 22 ++++++++++++++-----
 .../main/webapp/src/partial/docs/login.html   |  9 ++++++++
 docs-web/src/main/webapp/src/style/main.less  |  6 ++++-
 .../src/main/webapp/style/theme/default.less  |  0
 4 files changed, 30 insertions(+), 7 deletions(-)
 delete mode 100644 docs-web/src/main/webapp/style/theme/default.less

diff --git a/docs-web/src/main/webapp/src/app/docs/controller/Login.js b/docs-web/src/main/webapp/src/app/docs/controller/Login.js
index 9c695aa3..9a5fb7ce 100644
--- a/docs-web/src/main/webapp/src/app/docs/controller/Login.js
+++ b/docs-web/src/main/webapp/src/app/docs/controller/Login.js
@@ -4,18 +4,28 @@
  * Login controller.
  */
 angular.module('docs').controller('Login', function($scope, $rootScope, $state, $dialog, User) {
+  $scope.codeRequired = false;
+
+  /**
+   * Login.
+   */
   $scope.login = function() {
     User.login($scope.user).then(function() {
       User.userInfo(true).then(function(data) {
         $rootScope.userInfo = data;
       });
       $state.go('document.default');
-    }, function() {
-      var title = 'Login failed';
-      var msg = 'Username or password invalid';
-      var btns = [{result:'ok', label: 'OK', cssClass: 'btn-primary'}];
-
-      $dialog.messageBox(title, msg, btns);
+    }, function(data) {
+      if (data.data.type == 'ValidationCodeRequired') {
+        // A TOTP validation code is required to login
+        $scope.codeRequired = true;
+      } else {
+        // Login truly failed
+        var title = 'Login failed';
+        var msg = 'Username or password invalid';
+        var btns = [{result: 'ok', label: 'OK', cssClass: 'btn-primary'}];
+        $dialog.messageBox(title, msg, btns);
+      }
     });
   };
 });
\ No newline at end of file
diff --git a/docs-web/src/main/webapp/src/partial/docs/login.html b/docs-web/src/main/webapp/src/partial/docs/login.html
index 4aabda53..d03c33ed 100644
--- a/docs-web/src/main/webapp/src/partial/docs/login.html
+++ b/docs-web/src/main/webapp/src/partial/docs/login.html
@@ -17,6 +17,15 @@
         <input class="form-control" type="password" id="inputPassword" placeholder="Password" ng-model="user.password" />
       </div>
 
+      <span class="help-block" ng-if="codeRequired">
+        A validation code is required
+        <span class="glyphicon glyphicon-question-sign" title="You have activated the two-factor authentication on your account. Please enter a validation code generated by the phone app your configured."></span>
+      </span>
+      <div class="form-group" ng-if="codeRequired">
+        <label class="sr-only" for="inputCode">Validation code</label>
+        <input class="form-control" type="text" id="inputCode" placeholder="Validation code" ng-model="user.code" />
+      </div>
+
       <div class="checkbox">
         <label>
           <input type="checkbox" ng-model="user.remember" /> Remember me
diff --git a/docs-web/src/main/webapp/src/style/main.less b/docs-web/src/main/webapp/src/style/main.less
index 34f76a9f..c9428c93 100644
--- a/docs-web/src/main/webapp/src/style/main.less
+++ b/docs-web/src/main/webapp/src/style/main.less
@@ -240,7 +240,11 @@ input[readonly].share-link {
 .login-box {
   background: rgba(255, 255, 255, 0.5);
   padding: 20px;
-  border-radius: 4px
+  border-radius: 4px;
+
+  .help-block, .checkbox {
+    color: white;
+  }
 }
 
 /* Styling for the ngProgress itself */
diff --git a/docs-web/src/main/webapp/style/theme/default.less b/docs-web/src/main/webapp/style/theme/default.less
deleted file mode 100644
index e69de29b..00000000