feat: save reports to user directories

viasite · Dec 24, 2020 · 783214e · 783214e
1 parent e0e4739
commit 783214e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/scrap-site.js b/src/scrap-site.js
@@ -565,8 +565,15 @@ module.exports = async (baseUrl, options = {}) => {
       }
 
       // copy to local reports
-      const localPath = 'data/reports/' + jsonName;
+      let localDir = 'data/reports/';
+      if (options.socket.uid) {
+        const userDir = options.socket.uid.slice(0, 5);
+        localDir += userDir + '/';
+        if (!fs.existsSync(localDir)) fs.mkdirSync(localDir);
+      }
+      const localPath = localDir + jsonName;
       fs.copyFileSync(jsonPath, localPath);
+
       socketSend(options.socket, 'result', {name: jsonName});
 
       // TODO: error upload 8MB+

diff --git a/src/server.js b/src/server.js
@@ -92,6 +92,8 @@ io.on("connection", (socket) => {
         ? "anonymous user"
         : "user authenticated";
     log(msg, socket);
+
+    sendStats(socket);
   });
 
   socket.on("scan", async ({ url, args }) => {
@@ -115,9 +117,11 @@ io.on("connection", (socket) => {
     opts.webService = true;
     opts.socket = socket;
 
+    // console.log('opts: ', opts);
     program.outBrief(opts);
 
     // try {
+    log('Pending...', socket)
     q.push(async function () {
       return await scan(url, opts);
     });