From 6516e0dfd24508b838ba184c5015fbb42f6e6012 Mon Sep 17 00:00:00 2001
From: Nick O'Leary <nick.oleary@gmail.com>
Date: Wed, 10 Jan 2018 17:37:41 +0000
Subject: [PATCH] Allow a user to pick existing sshkeys from ~/.ssh

---
 editor/js/ui/projects/projectUserSettings.js  | 258 ++++++++++++------
 editor/js/ui/projects/projects.js             |   8 +-
 editor/sass/projects.scss                     |   9 +
 editor/sass/userSettings.scss                 |   6 +-
 red/api/editor/sshkeys.js                     |   2 +-
 .../localfilesystem/projects/Project.js       |   4 +-
 .../storage/localfilesystem/sshkeys.js        |  62 +++--
 .../storage/localfilesystem/sshkeys_spec.js   |  50 ++--
 8 files changed, 268 insertions(+), 131 deletions(-)

diff --git a/editor/js/ui/projects/projectUserSettings.js b/editor/js/ui/projects/projectUserSettings.js
index 021d47736..a4501bb0f 100644
--- a/editor/js/ui/projects/projectUserSettings.js
+++ b/editor/js/ui/projects/projectUserSettings.js
@@ -44,32 +44,44 @@ RED.projects.userSettings = (function() {
     function createSSHKeySection(pane) {
         var container = $('<div class="user-settings-section"></div>').appendTo(pane);
         var popover;
-        var title = $('<h4></h4>').text("SSH Keys").appendTo(container);
+        var title = $('<h3></h3>').text("SSH Keys").appendTo(container);
+        var subtitle = $('<div style="color:#aaa;"></div>').appendTo(container).text("Allows you to create secure connections to remote git repositories.");
 
-        var addKeyButton = $('<button class="editor-button editor-button-small" style="float: right; margin-right: 10px;">generate key</button>')
-            .appendTo(title)
+        var addKeyButton = $('<button class="editor-button editor-button-small" style="float: right; margin-right: 10px;">add key</button>')
+            .appendTo(subtitle)
             .click(function(evt) {
                 addKeyButton.attr('disabled',true);
+                saveButton.attr('disabled',true);
+                // bg.children().removeClass("selected");
+                // addLocalButton.click();
                 addKeyDialog.slideDown(200);
                 keyNameInput.focus();
-                saveButton.attr('disabled',true);
             });
 
         var validateForm = function() {
-            var validName = /^[a-zA-Z0-9\-_]+$/.test(keyNameInput.val());
-            var passphrase = passphraseInput.val();
-            var validPassphrase = passphrase.length === 0 || passphrase.length >= 8;
+            var valid = /^[a-zA-Z0-9\-_]+$/.test(keyNameInput.val());
+            keyNameInput.toggleClass('input-error',keyNameInputChanged&&!valid);
 
-            saveButton.attr('disabled',!validName || !validPassphrase);
-            keyNameInput.toggleClass('input-error',keyNameInputChanged&&!validName);
-            passphraseInput.toggleClass('input-error',!validPassphrase);
-            if (!validPassphrase) {
-                passphraseInputSubLabel.text("Passphrase too short");
-            } else if (passphrase.length === 0) {
-                passphraseInputSubLabel.text("Optional");
-            } else {
-                passphraseInputSubLabel.text("");
-            }
+            // var selectedButton = bg.find(".selected");
+            // if (selectedButton[0] === addLocalButton[0]) {
+            //     valid = valid && localPublicKeyPathInput.val().length > 0 && localPrivateKeyPathInput.val().length > 0;
+            // } else if (selectedButton[0] === uploadButton[0]) {
+            //     valid = valid && publicKeyInput.val().length > 0 && privateKeyInput.val().length > 0;
+            // } else if (selectedButton[0] === generateButton[0]) {
+                var passphrase = passphraseInput.val();
+                var validPassphrase = passphrase.length === 0 || passphrase.length >= 8;
+                passphraseInput.toggleClass('input-error',!validPassphrase);
+                if (!validPassphrase) {
+                    passphraseInputSubLabel.text("Passphrase too short");
+                } else if (passphrase.length === 0) {
+                    passphraseInputSubLabel.text("Optional");
+                } else {
+                    passphraseInputSubLabel.text("");
+                }
+                valid = valid && validPassphrase;
+            // }
+
+            saveButton.attr('disabled',!valid);
 
             if (popover) {
                 popover.close();
@@ -79,27 +91,88 @@ RED.projects.userSettings = (function() {
 
         var row = $('<div class="user-settings-row"></div>').appendTo(container);
         var addKeyDialog = $('<div class="projects-dialog-list-dialog"></div>').hide().appendTo(row);
-        $('<div class="projects-dialog-list-dialog-header">').text('Generate SSH Key').appendTo(addKeyDialog);
+        $('<div class="projects-dialog-list-dialog-header">').text('Add SSH Key').appendTo(addKeyDialog);
         var addKeyDialogBody = $('<div>').appendTo(addKeyDialog);
+
+        row = $('<div class="user-settings-row"></div>').appendTo(addKeyDialogBody);
+        $('<div style="color:#aaa;"></div>').appendTo(row).text("Generate a new public/private key pair");
+        // var bg = $('<div></div>',{class:"button-group", style:"text-align: center"}).appendTo(row);
+        // var addLocalButton = $('<button class="editor-button toggle selected">use local key</button>').appendTo(bg);
+        // var uploadButton = $('<button class="editor-button toggle">upload key</button>').appendTo(bg);
+        // var generateButton = $('<button class="editor-button toggle">generate key</button>').appendTo(bg);
+        // bg.children().click(function(e) {
+        //     e.preventDefault();
+        //     if ($(this).hasClass("selected")) {
+        //         return;
+        //     }
+        //     bg.children().removeClass("selected");
+        //     $(this).addClass("selected");
+        //     if (this === addLocalButton[0]) {
+        //         addLocalKeyPane.show();
+        //         generateKeyPane.hide();
+        //         uploadKeyPane.hide();
+        //     } else if (this === uploadButton[0]) {
+        //         addLocalKeyPane.hide();
+        //         generateKeyPane.hide();
+        //         uploadKeyPane.show();
+        //     } else if (this === generateButton[0]){
+        //         addLocalKeyPane.hide();
+        //         generateKeyPane.show();
+        //         uploadKeyPane.hide();
+        //     }
+        //     validateForm();
+        // })
+
+
         row = $('<div class="user-settings-row"></div>').appendTo(addKeyDialogBody);
         $('<label for=""></label>').text('Name').appendTo(row);
+        var keyNameInputChanged = false;
         var keyNameInput = $('<input type="text">').appendTo(row).on("change keyup paste",function() {
             keyNameInputChanged = true;
             validateForm();
         });
-        var keyNameInputChanged = false;
         $('<label class="projects-edit-form-sublabel"><small>Must contain only A-Z 0-9 _ -</small></label>').appendTo(row).find("small");
 
-        row = $('<div class="user-settings-row"></div>').appendTo(addKeyDialogBody);
+        var generateKeyPane = $('<div>').appendTo(addKeyDialogBody);
+        row = $('<div class="user-settings-row"></div>').appendTo(generateKeyPane);
         $('<label for=""></label>').text('Passphrase').appendTo(row);
-        passphraseInput = $('<input type="password">').appendTo(row).on("change keyup paste",validateForm);
+        var passphraseInput = $('<input type="password">').appendTo(row).on("change keyup paste",validateForm);
         var passphraseInputSubLabel = $('<label class="projects-edit-form-sublabel"><small>Optional</small></label>').appendTo(row).find("small");
 
+        // var addLocalKeyPane = $('<div>').hide().appendTo(addKeyDialogBody);
+        // row = $('<div class="user-settings-row"></div>').appendTo(addLocalKeyPane);
+        // $('<label for=""></label>').text('Public key').appendTo(row);
+        // var localPublicKeyPathInput = $('<input type="text">').appendTo(row).on("change keyup paste",validateForm);
+        // $('<label class="projects-edit-form-sublabel"><small>Public key file path, for example: ~/.ssh/id_rsa.pub</small></label>').appendTo(row).find("small");
+        // row = $('<div class="user-settings-row"></div>').appendTo(addLocalKeyPane);
+        // $('<label for=""></label>').text('Private key').appendTo(row);
+        // var localPrivateKeyPathInput = $('<input type="text">').appendTo(row).on("change keyup paste",validateForm);
+        // $('<label class="projects-edit-form-sublabel"><small>Private key file path, for example: ~/.ssh/id_rsa</small></label>').appendTo(row).find("small");
+        //
+        // var uploadKeyPane = $('<div>').hide().appendTo(addKeyDialogBody);
+        // row = $('<div class="user-settings-row"></div>').appendTo(uploadKeyPane);
+        // $('<label for=""></label>').text('Public key').appendTo(row);
+        // var publicKeyInput = $('<textarea>').appendTo(row).on("change keyup paste",validateForm);
+        // $('<label class="projects-edit-form-sublabel"><small>Paste in public key contents, for example: ~/.ssh/id_rsa.pub</small></label>').appendTo(row).find("small");
+        // row = $('<div class="user-settings-row"></div>').appendTo(uploadKeyPane);
+        // $('<label for=""></label>').text('Private key').appendTo(row);
+        // var privateKeyInput = $('<textarea>').appendTo(row).on("change keyup paste",validateForm);
+        // $('<label class="projects-edit-form-sublabel"><small>Paste in private key contents, for example: ~/.ssh/id_rsa</small></label>').appendTo(row).find("small");
+
+
+
+
         var hideEditForm = function() {
             addKeyButton.attr('disabled',false);
             addKeyDialog.hide();
+
             keyNameInput.val("");
+            keyNameInputChanged = false;
             passphraseInput.val("");
+            // localPublicKeyPathInput.val("");
+            // localPrivateKeyPathInput.val("");
+            // publicKeyInput.val("");
+            // privateKeyInput.val("");
             if (popover) {
                 popover.close();
                 popover = null;
@@ -118,11 +191,24 @@ RED.projects.userSettings = (function() {
                 evt.preventDefault();
                 var spinner = utils.addSpinnerOverlay(addKeyDialog).addClass('projects-dialog-spinner-contain');
                 var payload = {
-                    name: keyNameInput.val(),
-                    comment: gitEmailInput.val(),
-                    password: passphraseInput.val(),
-                    size: 4096
+                    name: keyNameInput.val()
                 };
+
+                var selectedButton = bg.find(".selected");
+                if (selectedButton[0] === addLocalButton[0]) {
+                    payload.type = "local";
+                    payload.publicKeyPath = localPublicKeyPathInput.val();
+                    payload.privateKeyPath = localPrivateKeyPathInput.val();
+                } else if (selectedButton[0] === uploadButton[0]) {
+                    payload.type = "upload";
+                    payload.publicKey = publicKeyInput.val();
+                    payload.privateKey = privateKeyInput.val();
+                } else if (selectedButton[0] === generateButton[0]) {
+                    payload.type = "generate";
+                    payload.comment = gitEmailInput.val();
+                    payload.password = passphraseInput.val();
+                    payload.size = 4096;
+                }
                 var done = function(err) {
                     spinner.remove();
                     if (err) {
@@ -181,6 +267,7 @@ RED.projects.userSettings = (function() {
             $('<button class="editor-button editor-button-small">Copy to clipboard</button>')
                 .appendTo(formButtons)
                 .click(function(evt) {
+                    evt.stopPropagation();
                     evt.preventDefault();
                     document.getSelection().selectAllChildren(keyBox[0]);
                     var ret = document.execCommand('copy');
@@ -189,7 +276,7 @@ RED.projects.userSettings = (function() {
 
             return row;
         }
-        var keyList = $('<ol>').appendTo(row).editableList({
+        var keyList = $('<ol class="projects-dialog-ssh-key-list">').appendTo(row).editableList({
             height: 'auto',
             addButton: false,
             scrollOnAdd: false,
@@ -210,66 +297,67 @@ RED.projects.userSettings = (function() {
 
                 var tools = $('<span class="button-row entry-tools">').appendTo(container);
                 var expandedRow;
-                $('<button class="editor-button editor-button-small"><i class="fa fa-eye"></i></button>')
-                    .appendTo(tools)
-                    .click(function(e) {
-                        if (expandedRow) {
-                            expandedRow.slideUp(200,function() {
-                                expandedRow.remove();
-                                expandedRow = null;
-                            })
-                        } else {
-                            expandedRow = expandKey(container,entry);
-                        }
+                row.click(function(e) {
+                    if (expandedRow) {
+                        expandedRow.slideUp(200,function() {
+                            expandedRow.remove();
+                            expandedRow = null;
+                        })
+                    } else {
+                        expandedRow = expandKey(container,entry);
+                    }
                     })
-                $('<button class="editor-button editor-button-small"><i class="fa fa-trash"></i></button>')
-                    .appendTo(tools)
-                    .click(function(e) {
-                        var spinner = utils.addSpinnerOverlay(row).addClass('projects-dialog-spinner-contain');
-                        var notification = RED.notify("Are you sure you want to delete the SSH key '"+entry.name+"'? This cannot be undone.", {
-                            type: 'warning',
-                            modal: true,
-                            fixed: true,
-                            buttons: [
-                                {
-                                    text: RED._("common.label.cancel"),
-                                    click: function() {
-                                        spinner.remove();
-                                        notification.close();
-                                    }
-                                },
-                                {
-                                    text: "Delete key",
-                                    click: function() {
-                                        notification.close();
-                                        var url = "settings/user/keys/"+entry.name;
-                                        var options = {
-                                            url: url,
-                                            type: "DELETE",
-                                            responses: {
-                                                200: function(data) {
-                                                    row.fadeOut(200,function() {
-                                                        keyList.editableList('removeItem',entry);
-                                                        setTimeout(spinner.remove, 100);
-                                                        if (keyList.editableList('length') === 0) {
-                                                            keyList.editableList('addItem',emptyItem);
-                                                        }
-                                                    });
-                                                },
-                                                400: {
-                                                    'unexpected_error': function(error) {
-                                                        console.log(error);
-                                                        spinner.remove();
-                                                    }
-                                                },
-                                            }
+                if (!entry.system) {
+                    $('<button class="editor-button editor-button-small"><i class="fa fa-trash"></i></button>')
+                        .appendTo(tools)
+                        .click(function(e) {
+                            e.stopPropagation();
+                            var spinner = utils.addSpinnerOverlay(row).addClass('projects-dialog-spinner-contain');
+                            var notification = RED.notify("Are you sure you want to delete the SSH key '"+entry.name+"'? This cannot be undone.", {
+                                type: 'warning',
+                                modal: true,
+                                fixed: true,
+                                buttons: [
+                                    {
+                                        text: RED._("common.label.cancel"),
+                                        click: function() {
+                                            spinner.remove();
+                                            notification.close();
+                                        }
+                                    },
+                                    {
+                                        text: "Delete key",
+                                        click: function() {
+                                            notification.close();
+                                            var url = "settings/user/keys/"+entry.name;
+                                            var options = {
+                                                url: url,
+                                                type: "DELETE",
+                                                responses: {
+                                                    200: function(data) {
+                                                        row.fadeOut(200,function() {
+                                                            keyList.editableList('removeItem',entry);
+                                                            setTimeout(spinner.remove, 100);
+                                                            if (keyList.editableList('length') === 0) {
+                                                                keyList.editableList('addItem',emptyItem);
+                                                            }
+                                                        });
+                                                    },
+                                                    400: {
+                                                        'unexpected_error': function(error) {
+                                                            console.log(error);
+                                                            spinner.remove();
+                                                        }
+                                                    },
+                                                }
+                                            }
+                                            utils.sendRequest(options);
                                         }
-                                        utils.sendRequest(options);
                                     }
-                                }
-                            ]
+                                ]
+                            });
                         });
-                    });
+                }
                 if (entry.expand) {
                     expandedRow = expandKey(container,entry);
                 }
@@ -288,7 +376,11 @@ RED.projects.userSettings = (function() {
                             key.expand = true;
                         }
                         keyList.editableList('addItem',key);
-                    })
+                    });
+                    if (keyList.editableList('length') === 0) {
+                        keyList.editableList('addItem',emptyItem);
+                    }
+
                 }
             })
         }
diff --git a/editor/js/ui/projects/projects.js b/editor/js/ui/projects/projects.js
index bc7d5c40f..7cf1a35c7 100644
--- a/editor/js/ui/projects/projects.js
+++ b/editor/js/ui/projects/projects.js
@@ -1232,7 +1232,7 @@ RED.projects = (function() {
             //data-i18n="[placeholder]menu.label.searchInput"
             delay: 200,
             change: function() {
-                filterTerm = $(this).val();
+                filterTerm = $(this).val().toLowerCase();
                 list.editableList('filter');
                 if (selectedListItem && !selectedListItem.is(":visible")) {
                     selectedListItem.children().children().removeClass('selected');
@@ -1241,6 +1241,12 @@ RED.projects = (function() {
                     if (options.select) {
                         options.select(selectedListItem.children().data('data'));
                     }
+                } else {
+                    selectedListItem = list.children(":visible").first();
+                    selectedListItem.children().children().addClass('selected');
+                    if (options.select) {
+                        options.select(selectedListItem.children().data('data'));
+                    }
                 }
                 ensureSelectedIsVisible();
             }
diff --git a/editor/sass/projects.scss b/editor/sass/projects.scss
index e9ac97f82..bae399898 100644
--- a/editor/sass/projects.scss
+++ b/editor/sass/projects.scss
@@ -780,6 +780,15 @@ div.projects-dialog-ssh-public-key {
     }
 }
 
+.projects-dialog-ssh-key-list {
+    li {
+        cursor: pointer;
+    }
+    li:hover {
+        background: #f3f3f3;
+    }
+}
+
 .projects-dialog-list {
     position: relative;
     .red-ui-editableList-container {
diff --git a/editor/sass/userSettings.scss b/editor/sass/userSettings.scss
index b51186508..d8f7fc99b 100644
--- a/editor/sass/userSettings.scss
+++ b/editor/sass/userSettings.scss
@@ -73,7 +73,11 @@
         display: table;
         clear: both;
     }
-    .uneditable-input, input {
+    .uneditable-input, input, textarea {
         width: calc(100% - 150px);
     }
+    textarea {
+        resize: none;
+        height: 10em;
+    }
 }
diff --git a/red/api/editor/sshkeys.js b/red/api/editor/sshkeys.js
index f4c8a3982..ee43880c3 100644
--- a/red/api/editor/sshkeys.js
+++ b/red/api/editor/sshkeys.js
@@ -84,7 +84,7 @@ module.exports = {
         app.post("/", needsPermission("settings.write"), function(req,res) {
             var username = getUsername(req.user);
             // console.log('req.body:', req.body);
-            if ( req.body && req.body.name ) {
+            if ( req.body && req.body.name && /^[a-zA-Z0-9\-_]+$/.test(req.body.name)) {
                 runtime.storage.sshkeys.generateSSHKey(username, req.body)
                 .then(function(name) {
                     // console.log('generate key --- success  name:', name);
diff --git a/red/runtime/storage/localfilesystem/projects/Project.js b/red/runtime/storage/localfilesystem/projects/Project.js
index 7bf631492..27b9e419b 100644
--- a/red/runtime/storage/localfilesystem/projects/Project.js
+++ b/red/runtime/storage/localfilesystem/projects/Project.js
@@ -949,7 +949,9 @@ function getProject(name) {
 function listProjects() {
     return fs.readdir(projectsDir).then(function(fns) {
         var dirs = [];
-        fns.sort().filter(function(fn) {
+        fns.sort(function(A,B) {
+            return A.toLowerCase().localeCompare(B.toLowerCase());
+        }).filter(function(fn) {
             var fullPath = fspath.join(projectsDir,fn);
             if (fn[0] != ".") {
                 var stats = fs.lstatSync(fullPath);
diff --git a/red/runtime/storage/localfilesystem/sshkeys.js b/red/runtime/storage/localfilesystem/sshkeys.js
index 4125d2bd6..daaa0b570 100644
--- a/red/runtime/storage/localfilesystem/sshkeys.js
+++ b/red/runtime/storage/localfilesystem/sshkeys.js
@@ -23,12 +23,14 @@ var settings;
 var runtime;
 var log;
 var sshkeyDir;
+var userSSHKeyDir;
 
 function init(_settings, _runtime) {
     settings = _settings;
     runtime = _runtime;
     log = runtime.log;
     sshkeyDir = fspath.join(settings.userDir, "projects", ".sshkeys");
+    userSSHKeyDir = fspath.join(process.env.HOME || process.env.USERPROFILE || process.env.HOMEPATH, ".ssh");
     // console.log('sshkeys.init()');
     return createSSHKeyDirectory();
 }
@@ -38,12 +40,23 @@ function createSSHKeyDirectory() {
 }
 
 function listSSHKeys(username) {
-    var startStr = username + '_';
-    // console.log('sshkeyDir:', sshkeyDir);
-    return fs.readdir(sshkeyDir).then(function(fns) {
+    return listSSHKeysInDir(sshkeyDir,username + '_').then(function(customKeys) {
+        return listSSHKeysInDir(userSSHKeyDir).then(function(existingKeys) {
+            existingKeys.forEach(function(k){
+                k.system = true;
+                customKeys.push(k);
+            })
+            return customKeys;
+        });
+    });
+}
+
+function listSSHKeysInDir(dir,startStr) {
+    startStr = startStr || "";
+    return fs.readdir(dir).then(function(fns) {
         var ret = fns.sort()
             .filter(function(fn) {
-                var fullPath = fspath.join(sshkeyDir,fn);
+                var fullPath = fspath.join(dir,fn);
                 if (fn.length > 2 || fn[0] != ".") {
                     var stats = fs.lstatSync(fullPath);
                     if (stats.isFile()) {
@@ -73,6 +86,10 @@ function listSSHKeys(username) {
                 name: filename
             };
         });
+    }).then(function(result) {
+        return result;
+    }).catch(function() {
+        return []
     });
 }
 
@@ -81,7 +98,13 @@ function getSSHKey(username, name) {
     .then(function(publicSSHKeyPath) {
         return fs.readFile(publicSSHKeyPath, 'utf-8');
     }).catch(function() {
-        return null;
+        var privateKeyPath = fspath.join(userSSHKeyDir,name);
+        var publicKeyPath = privateKeyPath+".pub";
+        return checkFilePairExist(privateKeyPath,publicKeyPath).then(function() {
+            return fs.readFile(publicKeyPath, 'utf-8');
+        }).catch(function() {
+            return null
+        });
     });
 }
 
@@ -116,29 +139,29 @@ function checkExistSSHKeyFiles(username, name) {
     var sshKeyFileBasename = username + '_' + name;
     var privateKeyFilePath = fspath.join(sshkeyDir, sshKeyFileBasename);
     var publicKeyFilePath  = fspath.join(sshkeyDir, sshKeyFileBasename + '.pub');
-    return Promise.all([
-        fs.access(privateKeyFilePath, (fs.constants || fs).R_OK),
-        fs.access(publicKeyFilePath , (fs.constants || fs).R_OK)
-    ])
-    .then(function() {
-        return true;
-    })
-    .catch(function() {
-        return false;
-    });
+    return checkFilePairExist(privateKeyFilePath,publicKeyFilePath)
+        .then(function() {
+            return true;
+        })
+        .catch(function() {
+            return false;
+        });
 }
 
 function checkSSHKeyFileAndGetPublicKeyFileName(username, name) {
     var sshKeyFileBasename = username + '_' + name;
     var privateKeyFilePath = fspath.join(sshkeyDir, sshKeyFileBasename);
     var publicKeyFilePath  = fspath.join(sshkeyDir, sshKeyFileBasename + '.pub');
+    return checkFilePairExist(privateKeyFilePath,publicKeyFilePath).then(function() {
+        return publicKeyFilePath;
+    });
+}
+
+function checkFilePairExist(privateKeyFilePath,publicKeyFilePath) {
     return Promise.all([
         fs.access(privateKeyFilePath, (fs.constants || fs).R_OK),
         fs.access(publicKeyFilePath , (fs.constants || fs).R_OK)
     ])
-    .then(function() {
-        return publicKeyFilePath;
-    });
 }
 
 function deleteSSHKeyFiles(username, name) {
@@ -149,9 +172,6 @@ function deleteSSHKeyFiles(username, name) {
         fs.remove(privateKeyFilePath),
         fs.remove(publicKeyFilePath)
     ])
-    .then(function() {
-        return true;
-    });
 }
 
 function generateSSHKeyPair(name, privateKeyPath, comment, password, size) {
diff --git a/test/red/runtime/storage/localfilesystem/sshkeys_spec.js b/test/red/runtime/storage/localfilesystem/sshkeys_spec.js
index 723ce9e10..094afc39b 100644
--- a/test/red/runtime/storage/localfilesystem/sshkeys_spec.js
+++ b/test/red/runtime/storage/localfilesystem/sshkeys_spec.js
@@ -33,15 +33,20 @@ describe("storage/localfilesystem/sshkeys", function() {
             trace: function() { }
         }
     };
+    var oldHOME;
+
     beforeEach(function(done) {
+        oldHOME = process.env.HOME;
+        process.env.HOME = "/tmp/doesnt/exist";
         fs.remove(userDir,function(err) {
             fs.mkdir(userDir,done);
         });
     });
     afterEach(function(done) {
+        process.env.HOME = oldHOME;
         fs.remove(userDir,done);
     });
-    
+
     it('should create sshkey directory when sshkey initializes', function(done) {
         var sshkeyDirPath = path.join(userDir, 'projects', '.sshkeys');
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
@@ -82,8 +87,8 @@ describe("storage/localfilesystem/sshkeys", function() {
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
                 for(var filename of filenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");
                 }
                 sshkeys.listSSHKeys(username).then(function(retObj) {
                     retObj.should.be.instanceOf(Array).and.have.lengthOf(filenameList.length);
@@ -110,11 +115,11 @@ describe("storage/localfilesystem/sshkeys", function() {
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
                 for(var filename of filenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");
                 }
                 for(var filename of onlyPrivateKeyFilenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
                 }
                 sshkeys.listSSHKeys(username).then(function(retObj) {
                     retObj.should.be.instanceOf(Array).and.have.lengthOf(filenameList.length);
@@ -144,11 +149,11 @@ describe("storage/localfilesystem/sshkeys", function() {
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
                 for(var filename of filenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");
                 }
                 for(var filename of directoryList) {
-                    fs.ensureDirSync(path.join(sshkeyDirPath,filename));        
+                    fs.ensureDirSync(path.join(sshkeyDirPath,filename));
                 }
                 sshkeys.listSSHKeys(username).then(function(retObj) {
                     retObj.should.be.instanceOf(Array).and.have.lengthOf(filenameList.length);
@@ -179,12 +184,12 @@ describe("storage/localfilesystem/sshkeys", function() {
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
                 for(var filename of filenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");
                 }
                 for(var filename of otherUserFilenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename+".pub"),"","utf8");
                 }
                 sshkeys.listSSHKeys(username).then(function(retObj) {
                     retObj.should.be.instanceOf(Array).and.have.lengthOf(filenameList.length);
@@ -215,12 +220,12 @@ describe("storage/localfilesystem/sshkeys", function() {
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
                 for(var filename of filenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),"","utf8");
                 }
                 for(var filename of otherUserFilenameList) {
-                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename),"","utf8");        
-                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename+".pub"),"","utf8");        
+                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename),"","utf8");
+                    fs.writeFileSync(path.join(sshkeyDirPath,otherUsername+"_"+filename+".pub"),"","utf8");
                 }
                 sshkeys.listSSHKeys(username).then(function(retObj) {
                     retObj.should.be.instanceOf(Array).and.have.lengthOf(filenameList.length);
@@ -438,8 +443,8 @@ describe("storage/localfilesystem/sshkeys", function() {
         var fileContent = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAACAQD3a+sgtgzSbbliWxmOq5p6+H/mE+0gjWfLWrkIVmHENd1mifV4uCmIHAR2NfuadUYMQ3+bQ90kpmmEKTMYPsyentsKpHQZxTzG7wOCAIpJnbPTHDMxEJhVTaAwEjbVyMSIzTTPfnhoavWIBu0+uMgKDDlBm+RjlgkFlyhXyCN6UwFrIUUMH6Gw+eQHLiooKIl8ce7uDxIlt+9b7hFCU+sQ3kvuse239DZluu6+8buMWqJvrEHgzS9adRFKku8nSPAEPYn85vDi7OgVAcLQufknNgs47KHBAx9h04LeSrFJ/P5J1b//ItRpMOIme+O9d1BR46puzhvUaCHLdvO9czj+OmW+dIm+QIk6lZIOOMnppG72kZxtLfeKT16ur+2FbwAdL9ItBp4BI/YTlBPoa5mLMxpuWfmX1qHntvtGc9wEwS1P7YFfmF3XiK5apxalzrn0Qlr5UmDNbVIqJb1OlbC0w03Z0oktti1xT+R2DGOLWM4lBbpXDHV1BhQ7oYOvbUD8Cnof55lTP0WHHsOHlQc/BGDti1XA9aBX/OzVyzBUYEf0pkimsD0RYo6aqt7QwehJYdlz9x1NBguBffT0s4NhNb9IWr+ASnFPvNl2sw4XH/8U0J0q8ZkMpKkbLM1Zdp1Fv00GF0f5UNRokai6uM3w/ccantJ3WvZ6GtctqytWrw== \n";
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
-                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),fileContent,"utf8");        
+                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),fileContent,"utf8");
                 sshkeys.getSSHKey(username, filename).then(function(retObj) {
                     retObj.should.be.equal(fileContent);
                     done();
@@ -461,10 +466,9 @@ describe("storage/localfilesystem/sshkeys", function() {
         var fileContent = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAACAQD3a+sgtgzSbbliWxmOq5p6+H/mE+0gjWfLWrkIVmHENd1mifV4uCmIHAR2NfuadUYMQ3+bQ90kpmmEKTMYPsyentsKpHQZxTzG7wOCAIpJnbPTHDMxEJhVTaAwEjbVyMSIzTTPfnhoavWIBu0+uMgKDDlBm+RjlgkFlyhXyCN6UwFrIUUMH6Gw+eQHLiooKIl8ce7uDxIlt+9b7hFCU+sQ3kvuse239DZluu6+8buMWqJvrEHgzS9adRFKku8nSPAEPYn85vDi7OgVAcLQufknNgs47KHBAx9h04LeSrFJ/P5J1b//ItRpMOIme+O9d1BR46puzhvUaCHLdvO9czj+OmW+dIm+QIk6lZIOOMnppG72kZxtLfeKT16ur+2FbwAdL9ItBp4BI/YTlBPoa5mLMxpuWfmX1qHntvtGc9wEwS1P7YFfmF3XiK5apxalzrn0Qlr5UmDNbVIqJb1OlbC0w03Z0oktti1xT+R2DGOLWM4lBbpXDHV1BhQ7oYOvbUD8Cnof55lTP0WHHsOHlQc/BGDti1XA9aBX/OzVyzBUYEf0pkimsD0RYo6aqt7QwehJYdlz9x1NBguBffT0s4NhNb9IWr+ASnFPvNl2sw4XH/8U0J0q8ZkMpKkbLM1Zdp1Fv00GF0f5UNRokai6uM3w/ccantJ3WvZ6GtctqytWrw== \n";
         localfilesystem.init(mockSettings, mockRuntime).then(function() {
             sshkeys.init(mockSettings, mockRuntime).then(function() {
-                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");        
-                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),fileContent,"utf8");        
-                sshkeys.deleteSSHKey(username, filename).then(function(retObj) {
-                    retObj.should.be.true();
+                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename),"","utf8");
+                fs.writeFileSync(path.join(sshkeyDirPath,username+"_"+filename+".pub"),fileContent,"utf8");
+                sshkeys.deleteSSHKey(username, filename).then(function() {
                     fs.existsSync(path.join(sshkeyDirPath,username+'_'+filename)).should.be.false();
                     fs.existsSync(path.join(sshkeyDirPath,username+'_'+filename+'.pub')).should.be.false();
                     done();