scrypted-detection-trainer 0.1.8 → 0.1.9

package/out/main.nodejs.js

@@ -1549,8 +1549,13 @@ var __importStar = (this && this.__importStar) || (function () {
         return result;
     };
 })();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", ({ value: true }));
 const sdk_1 = __importStar(__webpack_require__(/*! @scrypted/sdk */ "./node_modules/@scrypted/sdk/dist/src/index.js"));
+const fs_1 = __importDefault(__webpack_require__(/*! fs */ "fs"));
+const path_1 = __importDefault(__webpack_require__(/*! path */ "path"));
 const { systemManager, deviceManager, mediaManager } = sdk_1.default;
 // ─── Constants ────────────────────────────────────────────────────────────────
 const LABELS = ['person', 'animal', 'face', 'vehicle', 'plate', 'package', 'discard'];
@@ -1571,10 +1576,14 @@ class DetectionTrainer extends sdk_1.ScryptedDeviceBase {
         this.lastCapture = new Map();
         // Map<captureId, CaptureRecord>
         this.captures = new Map();
-        // Map<captureId, jpegBuffer>
-        this.images = new Map();
         // Active event listeners
         this.listeners = [];
+        // Use a stable directory inside the plugin's volume
+        this.imgDir = path_1.default.join(process.env.SCRYPTED_PLUGIN_VOLUME || '/tmp', 'detection-trainer-images');
+        try {
+            fs_1.default.mkdirSync(this.imgDir, { recursive: true });
+        }
+        catch { }
         this.loadState();
         this.registerListeners();
     }
@@ -1592,25 +1601,48 @@ class DetectionTrainer extends sdk_1.ScryptedDeviceBase {
         catch (e) {
             this.console.warn('Could not load captures from storage:', e);
         }
-        // images are stored as individual items
+        // Clean up any old base64 image entries from previous versions
         for (const [id] of this.captures) {
-            const raw = this.storage.getItem(`img:${id}`);
-            if (raw)
-                this.images.set(id, Buffer.from(raw, 'base64'));
+            try {
+                this.storage.removeItem(`img:${id}`);
+            }
+            catch { }
         }
     }
     saveCaptures() {
-        this.storage.setItem('captures', JSON.stringify([...this.captures.values()]));
+        try {
+            this.storage.setItem('captures', JSON.stringify([...this.captures.values()]));
+        }
+        catch (e) {
+            this.console.warn('Could not save captures:', e);
+        }
+    }
+    imgPath(id) {
+        return path_1.default.join(this.imgDir, `${id}.jpg`);
     }
     saveImage(id, buf) {
-        this.storage.setItem(`img:${id}`, buf.toString('base64'));
+        try {
+            fs_1.default.writeFileSync(this.imgPath(id), buf);
+        }
+        catch (e) {
+            this.console.warn(`Could not save image ${id}:`, e);
+        }
+    }
+    loadImage(id) {
+        try {
+            const p = this.imgPath(id);
+            if (fs_1.default.existsSync(p))
+                return fs_1.default.readFileSync(p);
+        }
+        catch { }
+        return undefined;
     }
     deleteCapture(id) {
-        const old = this.storage.getItem(`img:${id}`);
-        if (old)
-            this.storage.removeItem(`img:${id}`);
+        try {
+            fs_1.default.unlinkSync(this.imgPath(id));
+        }
+        catch { }
         this.captures.delete(id);
-        this.images.delete(id);
         this.saveCaptures();
     }
     // ── Settings ─────────────────────────────────────────────────────────────
@@ -1744,7 +1776,6 @@ class DetectionTrainer extends sdk_1.ScryptedDeviceBase {
             reviewed: false,
         };
         this.captures.set(id, record);
-        this.images.set(id, jpeg);
         this.saveImage(id, jpeg);
         this.saveCaptures();
         this.console.log(`Captured ${best.className} (${Math.round((best.score || 0) * 100)}%) from ${cameraName} [${this.captures.size} total]`);
@@ -1755,8 +1786,8 @@ class DetectionTrainer extends sdk_1.ScryptedDeviceBase {
         const path = url.pathname.replace(request.rootPath, '');
         // Serve image
         if (path.startsWith('/img/')) {
-            const id = path.slice(5);
-            const img = this.images.get(id);
+            const id = path.slice(5).replace(/[^a-zA-Z0-9_\-]/g, ''); // sanitize
+            const img = this.loadImage(id);
             if (!img)
                 return response.send('Not found', { code: 404 });
             return response.send(img, { headers: { 'Content-Type': 'image/jpeg', 'Cache-Control': 'max-age=3600' } });
@@ -1827,7 +1858,7 @@ class DetectionTrainer extends sdk_1.ScryptedDeviceBase {
             // Build a simple tarball-like structure as JSON for download
             const dataset = [];
             for (const record of labeled) {
-                const img = this.images.get(record.id);
+                const img = this.loadImage(record.id);
                 if (!img)
                     continue;
                 const fname = `${record.id}`;
@@ -2449,6 +2480,17 @@ setInterval(loadPending, 30_000);
 exports["default"] = DetectionTrainer;
 
 
+/***/ },
+
+/***/ "fs"
+/*!*********************!*\
+  !*** external "fs" ***!
+  \*********************/
+(module) {
+
+"use strict";
+module.exports = require("fs");
+
 /***/ },
 
 /***/ "module"
@@ -2460,6 +2502,17 @@ exports["default"] = DetectionTrainer;
 "use strict";
 module.exports = require("module");
 
+/***/ },
+
+/***/ "path"
+/*!***********************!*\
+  !*** external "path" ***!
+  \***********************/
+(module) {
+
+"use strict";
+module.exports = require("path");
+
 /***/ }
 
 /******/ 	});