From 8894c38289dba991a23872a82556a153d043041c Mon Sep 17 00:00:00 2001
From: HF <pixelplanetdev@gmail.com>
Date: Wed, 8 Jan 2020 02:53:10 +0100
Subject: [PATCH] change backup to use commandline args instead of environment

---
 src/backup.js                      | 36 +++++++++++++++++-------------
 src/core/tilesBackup.js            |  6 ++++-
 tools/example-ecosystem-backup.yml |  5 +----
 3 files changed, 26 insertions(+), 21 deletions(-)

diff --git a/src/backup.js b/src/backup.js
index e1ce0d9..26f9808 100644
--- a/src/backup.js
+++ b/src/backup.js
@@ -18,7 +18,6 @@ import bluebird from 'bluebird';
 import process from 'process';
 import { spawn } from 'child_process';
 
-import { DailyCron } from './utils/cron';
 import {
   updateBackupRedis,
   createPngBackup,
@@ -43,15 +42,18 @@ proc.on('exit', (code) => {
 bluebird.promisifyAll(redis.RedisClient.prototype);
 bluebird.promisifyAll(redis.Multi.prototype);
 
-const {
+const [
   CANVAS_REDIS_URL,
   BACKUP_REDIS_URL,
   BACKUP_DIR,
-} = process.env;
+  INTERVAL,
+] = process.argv.slice(2);
+
 if (!CANVAS_REDIS_URL || !BACKUP_REDIS_URL || !BACKUP_DIR) {
-  throw new Error(
-    'You did not set CANVAS_REDIS_URL, BACKUP_REDIS_URL or BACKUP_DIR',
+  console.error(
+    'Usage: node backup.js original_canvas backup_canvas backup_directory',
   );
+  process.exit(1);
 }
 
 const canvasRedis = redis
@@ -77,7 +79,7 @@ function getDateFolder() {
   return backupDir;
 }
 
-function dailyBackup() {
+async function dailyBackup() {
   const backupDir = getDateFolder();
   if (!fs.existsSync(backupDir)) {
     fs.mkdirSync(backupDir);
@@ -89,7 +91,7 @@ function dailyBackup() {
     }
     await updateBackupRedis(canvasRedis, backupRedis, canvases);
     await createPngBackup(backupRedis, canvases, backupDir);
-    console.log('Daily backup done');
+    console.log('Daily full backup done');
   });
 }
 
@@ -106,20 +108,22 @@ function incrementialBackup() {
   );
 }
 
-async function startCronJobs() {
+async function trigger() {
   if (!fs.existsSync(BACKUP_DIR)) {
-    throw new Error(`Backup directory ${BACKUP_DIR} does not exist!`);
+    console.error(`Backup directory ${BACKUP_DIR} does not exist!`);
+    process.exit(1);
   }
   const backupDir = getDateFolder();
   if (!fs.existsSync(backupDir)) {
     await dailyBackup();
+  } else {
+    await incrementialBackup();
   }
-
-  console.log(
-    'Creating one full PNG backup per day and incremential backups every hour.',
-  );
-  DailyCron.hook(dailyBackup);
-  setInterval(incrementialBackup, 15 * 60 * 1000);
+  if (!INTERVAL) {
+    process.exit(0);
+  }
+  setTimeout(trigger, INTERVAL * 60 * 1000);
 }
 
-startCronJobs();
+console.log('Starting backup...');
+trigger();
diff --git a/src/core/tilesBackup.js b/src/core/tilesBackup.js
index e53bc6c..0fb92c9 100644
--- a/src/core/tilesBackup.js
+++ b/src/core/tilesBackup.js
@@ -73,7 +73,11 @@ export async function incrementialBackupRedis(
       fs.mkdirSync(canvasBackupDir);
     }
     const date = new Date();
-    const canvasTileBackupDir = `${canvasBackupDir}/${date.getHours()}${date.getMinutes()}`;
+    let hours = date.getHours();
+    let minutes = date.getMinutes();
+    if (hours < 10) hours = `0${hours}`;
+    if (minutes < 10) minutes = `0${minutes}`;
+    const canvasTileBackupDir = `${canvasBackupDir}/${hours}${minutes}`;
     if (!fs.existsSync(canvasTileBackupDir)) {
       fs.mkdirSync(canvasTileBackupDir);
     }
diff --git a/tools/example-ecosystem-backup.yml b/tools/example-ecosystem-backup.yml
index 3451f27..19941f5 100644
--- a/tools/example-ecosystem-backup.yml
+++ b/tools/example-ecosystem-backup.yml
@@ -1,8 +1,5 @@
 apps:
   - script   : ./backup.js
     name     : 'backup'
+    args     : 'redis://localhost:6379' 'redis://localhost:6380' '/backup/canvas' 15
     node_args: --nouse-idle-notification --expose-gc
-    env:
-      CANVAS_REDIS_URL: 'redis://localhost:6379'
-      BACKUP_REDIS_URL: 'redis://localhost:6380'
-      BACKUP_DIR: '/backups'