@alwaysai/device-agent 0.0.12 → 0.0.14

package/src/cloud-connection/cmd-status.ts

@@ -0,0 +1,52 @@
+export type CmdStatusType = 'idle' | 'in_progress';
+
+export class CmdStatus {
+  private projectId: string;
+  private status: CmdStatusType;
+
+  constructor(projectId: string, status: CmdStatusType) {
+    this.projectId = projectId;
+    this.status = status;
+  }
+
+  public getProjectId() {
+    return this.projectId;
+  }
+
+  public update(status: CmdStatusType) {
+    this.status = status;
+  }
+
+  public getStatus(): CmdStatusType {
+    return this.status;
+  }
+}
+
+export class CmdStatusManager {
+  private apps: { [projectId: string]: CmdStatus } = {};
+
+  public update(projectId: string, status: CmdStatusType) {
+    if (projectId in this.apps) {
+      this.apps[projectId].update(status);
+    } else {
+      const cmdStatus = new CmdStatus(projectId, status);
+      this.apps[projectId] = cmdStatus;
+    }
+  }
+
+  public getAppCmdStatus(projectId: string) {
+    if (projectId in this.apps) {
+      return this.apps[projectId].getStatus();
+    }
+    throw new Error(`No status for ${projectId}`);
+  }
+
+  public anyCmdInProgress() {
+    for (const projectId in this.apps) {
+      if (this.apps[projectId].getStatus() === 'in_progress') {
+        return true;
+      }
+    }
+    return false;
+  }
+}