From 4157ebb38457340825f3de36bbf021b46a616235 Mon Sep 17 00:00:00 2001
From: Piero Toffanin <pt@masseranolabs.com>
Date: Mon, 24 May 2021 14:57:51 -0400
Subject: [PATCH] Winbundle script/command

---
 .dockerignore        |   2 +
 .gitignore           |   4 +-
 SOURCE               |   1 +
 package.json         |   3 +-
 scripts/winbundle.js | 131 +++++++++++++++++++++++++++++++++++++++++++
 5 files changed, 139 insertions(+), 2 deletions(-)
 create mode 100644 SOURCE
 create mode 100644 scripts/winbundle.js

diff --git a/.dockerignore b/.dockerignore
index 0cc72dd..c554bbb 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -1,3 +1,5 @@
 node_modules
 tests
 tmp
+nodeodm.exe
+dist
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
index 11fb8de..ece1052 100644
--- a/.gitignore
+++ b/.gitignore
@@ -43,4 +43,6 @@ jspm_packages
 .vscode
 
 package-lock.json
-apps/
\ No newline at end of file
+apps/
+nodeodm.exe
+dist/
\ No newline at end of file
diff --git a/SOURCE b/SOURCE
new file mode 100644
index 0000000..39388f9
--- /dev/null
+++ b/SOURCE
@@ -0,0 +1 @@
+NodeODM is free software. You can download the source code from https://github.com/OpenDroneMap/NodeODM/
\ No newline at end of file
diff --git a/package.json b/package.json
index 2cd72b7..55980fe 100644
--- a/package.json
+++ b/package.json
@@ -4,7 +4,8 @@
   "description": "REST API to access ODM",
   "main": "index.js",
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "test": "echo \"Error: no test specified\" && exit 1",
+    "winbundle": "node scripts/winbundle.js"
   },
   "repository": {
     "type": "git",
diff --git a/scripts/winbundle.js b/scripts/winbundle.js
new file mode 100644
index 0000000..4d43609
--- /dev/null
+++ b/scripts/winbundle.js
@@ -0,0 +1,131 @@
+const fs = require('fs');
+const spawnSync = require('child_process').spawnSync;
+const path = require('path');
+const request = require('request');
+const async = require('async');
+const nodeUnzip = require('node-unzip-2');
+const archiver = require('archiver');
+
+const download = function(uri, filename, callback) {
+    console.log(`Downloading ${uri}`);
+    request.head(uri, function(err, res, body) {
+        if (err) callback(err);
+        else{
+            request(uri).pipe(fs.createWriteStream(filename)).on('close', callback);
+        }
+    });
+};
+
+function downloadApp(destFolder, appUrl, cb){
+    if (!fs.existsSync(destFolder)) fs.mkdirSync(destFolder, { recursive: true });
+    else {
+        cb();
+        return;
+    }
+
+    let zipPath = path.join(destFolder, "download.zip");
+    let _called = false;
+
+    const done = (err) => {
+        if (!_called){ // Bug in nodeUnzip, makes this get called twice
+            if (fs.existsSync(zipPath)) fs.unlinkSync(zipPath);
+            _called = true; 
+            cb(err);
+        }
+    };
+    download(appUrl, zipPath, err => {
+        if (err) done(err);
+        else{
+            // Unzip
+            console.log(`Extracting ${zipPath}`);
+            fs.createReadStream(zipPath).pipe(nodeUnzip.Extract({ path: destFolder }))
+                .on('close', done)
+                .on('error', done);
+        }
+    });
+}
+
+async.series([
+    cb => {
+        // Cleanup directories
+        console.log("Cleaning up folders");
+        for (let dir of ["data", "tmp"]){
+            for (let entry of fs.readdirSync(dir)){
+                if (entry !== ".gitignore"){
+                    console.log(`Removing ${dir}/${entry}`);
+                    fs.rmdirSync(path.join(dir, entry), { recursive: true });
+                }
+            }
+        }
+        cb();
+    },
+
+    cb => {
+        downloadApp(path.join("apps", "7z"), "https://github.com/OpenDroneMap/NodeODM/releases/download/v2.1.0/7z19.zip", cb);
+    },
+    cb => {
+        downloadApp(path.join("apps", "unzip"), "https://github.com/OpenDroneMap/NodeODM/releases/download/v2.1.0/unzip600.zip", cb);
+    },
+    cb => {
+        console.log("Building executable");
+        const code = spawnSync('nexe.cmd', ['index.js', '-t', 'windows-x64-12.16.3', '-o', 'nodeodm.exe'], { stdio: "pipe"}).status;
+
+        if (code === 0) cb();
+        else cb(new Error(`nexe returned non-zero error code: ${code}`));
+    },
+    cb => {
+        // Zip
+        const outFile = path.join("dist", "nodeodm.zip");
+        if (!fs.existsSync("dist")) fs.mkdirSync("dist");
+        if (fs.existsSync(outFile)) fs.unlinkSync(outFile);
+
+        let output = fs.createWriteStream(outFile);
+        let archive = archiver.create('zip', {
+            zlib: { level: 5 } // Sets the compression level (1 = best speed since most assets are already compressed)
+        });
+
+        archive.on('finish', () => {
+            console.log("Done!");
+            cb();
+        });
+
+        archive.on('error', err => {
+            console.error(`Could not archive .zip file: ${err.message}`);
+            cb(err);
+        });
+
+        const files = [
+            "apps",
+            "data",
+            "helpers",
+            "public",
+            "scripts",
+            "tmp",
+            "config-default.json",
+            "LICENSE",
+            "SOURCE",
+            "package.json",
+            "nodeodm.exe"
+        ];
+
+        archive.pipe(output);
+        files.forEach(file => {
+            console.log(`Adding ${file}`);
+            let stat = fs.lstatSync(file);
+            if (stat.isFile()){
+                archive.file(file, {name: path.basename(file)});
+            }else if (stat.isDirectory()){
+                archive.directory(file, path.basename(file));
+            }else{
+                logger.error(`Could not add ${file}`);
+            }
+        });
+
+        archive.finalize();
+    }
+], (err) => {
+    if (err) console.log(`Bundle failed: ${err}`);
+    else console.log("Bundle ==> dist/nodeodm.zip");
+});
+
+