added functionality to restore to main db

badhan.js

@@ -37,6 +37,11 @@ yargs.command({
             describe: 'The backup having the specified timestamp will be restored',
             demandOption: true,
             type: 'number'
+        },
+        production: {
+            describe: 'If true, then the backup will be restored to the production database',
+            type: 'boolean',
+            default: false,
         }
     },
     handler: controllers.restoreController

controllers.js

@@ -115,6 +115,13 @@ const getBackupList = async ()=>{
 const restoreController = async (argv) => {
     console.log("restore command initiated");
     console.log("time: ",argv.time);
+    console.log("restore to production: ",argv.production);
+
+    let mongoURI = process.env.MONGODB_URI_TEST;
+
+    if(argv.production===true){
+        mongoURI = process.env.MONGODB_URI_PROD;
+    }
 
     console.log("fetching backups from cloud...")
     let backupList = await getBackupList();
@@ -140,7 +147,7 @@ const restoreController = async (argv) => {
     let backupProcess = spawn('./mongorestore', [
         '--drop',
         '--dir=backup/'+argv.time+'/Badhan',
-        process.env.MONGODB_URI_TEST
+        mongoURI
     ]);
 
     backupProcess.on('exit', async (code, signal) => {