Merge pull request #1 from Gururaj26/update-config

Now ssh generator also updates ssh config

constants.js

@@ -1,7 +1,9 @@
 const seperator = '\n' + '---------------------------------------------------' + '\n'
 const urlRegex = /[-a-zA-Z0-9@:%_\+.~#?&//=]{2,256}\.[a-z]{2,4}\b(\/[-a-zA-Z0-9@:%_\+.~#?&//=]*)?/gi;
+const thanksDecorator = '///////////////// Thanks for using ssh-generator ! /////////////////';
 
 module.exports = {
   seperator: seperator,
-  urlRegex: urlRegex
+  urlRegex: urlRegex,
+  thanksDecorator: thanksDecorator
 }

helpers.js

@@ -2,7 +2,11 @@ const chalkPipe = require('chalk-pipe')
 const constants = require('./constants.js')
 const seperator = constants.seperator
 const urlRegex = constants.urlRegex
-var exec = require('child_process').exec
+const thanksDecorator = constants.thanksDecorator
+const exec = require('child_process').exec
+const fs = require('fs');
+var os = require('os');
+const path = require('path')
 
 module.exports = {
   validateUrl: function(name) {
@@ -15,6 +19,25 @@ module.exports = {
     const isValid = (name !== '');
     return isValid || '~ Username is required';
   },
+  updateConfig: function(username, domain){
+    const currentUser = os.userInfo().username;
+    const configPath = `${ path.parse(process.cwd()).root + 'Users/' + currentUser + '/.ssh/config'}`;
+    const host = `${'Host '+ username + '-' + domain}`;
+    const hostName = `${'HostName '+ domain}`
+    const identityFile = `${'IdentityFile ~/.ssh/' + username + '-' + domain}`;
+    const config = `${'\r\n' + host + '\r\n' + '  ' + hostName + '\r\n' + '  ' + identityFile}`;
+    fs.open(configPath, "a+", function(error, fd) {
+      if (error) {
+        console.error("open error:  " + error.message);
+      } else {
+        fs.appendFile(fd, config, function (err) {
+          if (err) throw err;
+          console.log(seperator + 'Saved SSH to config file !' + seperator + thanksDecorator);
+          process.exit(1);
+        });
+      }
+    });
+  },
   generateKeys: function(username, domain, type) {
     let exp = 'ssh-keygen -t rsa -b 4096 -f ~/.ssh/' + username + '-' + domain + ' ' + '-C' + ' ' + '"' + username + '-' + domain + '"';
     let printExp = `${'cat ~/.ssh/' + username + '-' + domain + '.pub'}`;
@@ -26,7 +49,8 @@ module.exports = {
         console.log('stdout: ' + data);
         let subchild = exec(printExp);
         subchild.stdout.on('data', function(data) {
-          console.log(seperator + 'Freshly baked key: Ready to be served !' + seperator, chalkPipe('orange')(data));
+          console.log(seperator + 'Freshly baked key: Ready to be served !' + seperator + chalkPipe('orange')(data));
+          module.exports.updateConfig(username, domain);
         });
         subchild.stderr.on('data', function(data) {
           console.log('stdout: ' + data);