const BackupManager = require('influx-backup'); const TEST_DB = "testDB"; const manager = new BackupManager({db: TEST_DB}); // Restore DB api app.post('/restore', (req, res) => { var restore_path = ""; // create a temp directory to use for the backup manager.createDir() .then((dir) => { restore_path = dir; // Use multer to get the file from the req // and store it in the temp dir created var Storage = multer.diskStorage({ destination: restore_path, filename: function(req, file, callback) { callback(null, file.originalname); } }); // "restore" is the name attr of input file in html var upload = multer({storage: Storage}).single("restore"); return multerPromise(upload, req, res) }) .then((file) => { //now the file is stored correctly, I can start the restore command if(file){ return manager.restore(restore_path, file.originalname) }else{ throw Error("No file provided"); } }) //the backup has been restored in a backup database, load the backup in the main database .then(() => manager.loadBackup()) .then(() => res.json({success:true, message: "Backup restored successfully"})) .catch(err => { console.log(err); res.json({success:false, message: err.message}) }); }); // Backup DB api app.get('/backup', (req, res) => { manager.backup() .then((file) => { //zip file is ready, send it to the client var stream = fs.createReadStream(file); res.setHeader('Content-disposition', 'attachment; filename=' + file.split('/').pop()); stream.once("close", function () { stream.destroy(); // makesure stream closed, not close if download aborted. //IMPORTANT: remove backup files and zip manager.deleteDir(path.dirname(file)); }); stream.pipe(res); }) .catch(err => { res.json({success: false, message: err}); }); }); // Check testDB exists, if not create one and start app listening on port 8000 influx.getDatabaseNames() .then(names => { if (!names.includes(TEST_DB)) { return influx.createDatabase(TEST_DB) } }) .then(() => manager.init()) .then(() => { app.listen(8000, function () { console.log('Listening on port 8000') }) }) .catch(err => { console.error(`Error creating Influx database!`) })