Print out count of images in /aiinput on startup (#331)

* Print out count of images in /aiinput on startup
Fixes #330

* Print out count of images in /aiinput on startup
Fixes #330

Author: danecreekphotography

CHANGELOG.md

@@ -2,6 +2,8 @@
 
 ## Unreleased
 
+- Log messages are printed on startup to help confirm the image folder was mounted correctly with Docker.
+  Resolves [issue 330](https://github.com/danecreekphotography/node-deepstackai-trigger/issues/330).
 - Web server now shuts down properly when reloading settings. Resolves [issue 323](https://github.com/danecreekphotography/node-deepstackai-trigger/issues/323).
 - Startup is now re-attempted if there are any failures during launch. Each re-attempt is 30 seconds
   apart and five re-attempts will happen before things are assumed to just be completely broken.

src/TriggerManager.ts

@@ -7,7 +7,9 @@ import * as log from "./Log";
 import * as MQTTManager from "./handlers/mqttManager/MqttManager";
 
 import ITriggerConfigJson from "./types/ITriggerConfigJson";
+import * as fs from "fs";
 import MqttHandlerConfig from "./handlers/mqttManager/MqttHandlerConfig";
+import path from "path";
 import PushbulletConfig from "./handlers/pushbulletManager/PushbulletConfig";
 import PushoverConfig from "./handlers/pushoverManager/PushoverConfig";
 import Rect from "./Rect";
@@ -252,3 +254,32 @@ export function resetOverallStatistics(): ITriggerStatistics {
   MQTTManager.publishStatisticsMessage(triggeredCount, analyzedFilesCount);
   return getOverallStatistics();
 }
+
+/**
+ * Checks the watch folder on each trigger to see if there are images in it. If
+ * not throws a warning.
+ * @returns True if all the watch locations are valid, false otherwise.
+ */
+export function verifyTriggerWatchLocations(): boolean {
+  const invalidWatchLocations = triggers?.filter(trigger => {
+    const watchFolder = path.dirname(trigger.watchPattern);
+
+    let files: string[];
+
+    try {
+      files = fs.readdirSync(watchFolder);
+    } catch (e) {
+      log.warn(
+        "Trigger manager",
+        `Unable to read contents of watch folder ${watchFolder} for trigger ${trigger.name}. Check and make sure the image folder is mounted properly. ${e}`,
+      );
+      return true;
+    }
+
+    log.verbose("Trigger manager", `There are ${files.length} images waiting in ${watchFolder} for ${trigger.name}.`);
+    return false;
+  });
+
+  // If no invalid watch locations were found then we're good to go and return true
+  return invalidWatchLocations.length == 0;
+}

src/main.ts

@@ -22,7 +22,7 @@ import * as WebServer from "./WebServer";
 import npmPackageInfo from "../package.json";
 
 // If startup fails restart is reattempted 5 times every 30 seconds.
-const restartAttemptWaitTime = 5 * 1000;
+const restartAttemptWaitTime = 30 * 1000;
 const maxRestartAttempts = 5;
 
 // The list of settings file watchers, used to stop them on hot reloading of settings.
@@ -83,6 +83,9 @@ async function startup(): Promise<void> {
 
     // Load the trigger configuration.
     triggersFilePath = TriggerManager.loadConfiguration(["/run/secrets/triggers", "/config/triggers.json"]);
+    if (!TriggerManager.verifyTriggerWatchLocations()) {
+      throw Error(`Unable to access one or more watch locations.`);
+    }
 
     // Initialize the other handler managers. MQTT got done earlier
     // since it does double-duty and sends overall status messages for the system.
@@ -120,6 +123,9 @@ async function startup(): Promise<void> {
     // Notify it's not up and running
     await MqttManager.publishServerState("offline", e.message);
 
+    // Shutdown the web server plus other things that may have spun up successfully.
+    await shutdown();
+
     restartAttemptCount++;
 
     // Try starting again in a little bit.
@@ -174,6 +180,11 @@ async function hotLoadTriggers(path: string) {
   await TriggerManager.stopWatching();
 
   TriggerManager.loadConfiguration([path]);
+  if (!TriggerManager.verifyTriggerWatchLocations()) {
+    log.warn("Main", `Unable to access one or more watch locations.`);
+    return;
+  }
+
   TriggerManager.startWatching();
 }