Redirect to the pad when logging in or signing up from a pad

www/common/toolbar.js

@@ -465,9 +465,11 @@ define([
                 window.location.href = '/';
             });
             $userAdmin.find('a.login').click(function (e) {
+                sessionStorage.redirectTo = window.location.href;
                 window.location.href = '/login/';
             });
             $userAdmin.find('a.register').click(function (e) {
+                sessionStorage.redirectTo = window.location.href;
                 window.location.href = '/register/';
             });
 

www/login/main.js

@@ -78,7 +78,17 @@ define([
                                 result.proxy.login_name = result.userName;
                             }
                             Cryptpad.login(result.userHash, result.userName, function () {
-                                document.location.href = '/drive/';
+                                if (sessionStorage.redirectTo) {
+                                    var h = sessionStorage.redirectTo;
+                                    var parser = document.createElement('a');
+                                    parser.href = h;
+                                    if (parser.origin === window.location.origin) {
+                                        delete sessionStorage.redirectTo;
+                                        window.location.href = h;
+                                        return;
+                                    }
+                                }
+                                window.location.href = '/drive/';
                             });
                             return;
                         }

www/register/main.js

@@ -120,7 +120,17 @@ define([
 
             Cryptpad.whenRealtimeSyncs(result.realtime, function () {
                 Cryptpad.login(result.userHash, result.userName, function () {
-                    document.location.href = '/drive/';
+                    if (sessionStorage.redirectTo) {
+                        var h = sessionStorage.redirectTo;
+                        var parser = document.createElement('a');
+                        parser.href = h;
+                        if (parser.origin === window.location.origin) {
+                            delete sessionStorage.redirectTo;
+                            window.location.href = h;
+                            return;
+                        }
+                    }
+                    window.location.href = '/drive/';
                 });
             });
         });