@alwaysai/device-agent 1.3.0-1 → 1.3.1-1

package/lib/cloud-connection/transaction-manager.d.ts

@@ -1,14 +1,30 @@
+import { LiveUpdatesHandler } from './live-updates-handler';
+import { Publisher } from './publisher';
 export declare class TransactionManager {
-    private txToProject;
-    private projectToTx;
-    constructor();
-    AddTransactionErrors: {
-        TRANSACTION_ID_ONGOING: "TRANSACTION_ID_ONGOING";
-        PROJECT_ID_ONGOING: "PROJECT_ID_ONGOING";
+    private detailsByTx;
+    private detailsByProject;
+    private liveUpdatesHandler;
+    private publisher;
+    private startTransaction;
+    private updateTransaction;
+    constructor(publisher: Publisher, liveUpdatesHandler: LiveUpdatesHandler);
+    Errors: {
+        TRANSACTION_ONGOING: "TRANSACTION_ONGOING";
+        PROJECT_ONGOING: "PROJECT_ONGOING";
+        TRANSACTION_NOT_ONGOING: "TRANSACTION_NOT_ONGOING";
     };
-    addTransaction(txId: string, projectId: string, step?: string): void;
-    getTransactionFromProject(projectId: string): string;
-    getProjectFromTransaction(txId: string): string;
+    runTransactionStep(props: {
+        func: () => Promise<boolean>;
+        projectId: string;
+        txId: string;
+        start: boolean;
+        stepName?: string;
+    }): Promise<void>;
+    getTransactionFromProject(projectId: string): string | undefined;
+    isOngoingTransactionForProjectID(projectId: string): boolean;
+    isOngoingTransaction(txId: string): boolean;
+    isAnyOngoingTransaction(): boolean;
+    getProjectFromTransaction(txId: string): string | undefined;
     completeTransaction(txId: string): void;
 }
 //# sourceMappingURL=transaction-manager.d.ts.map

package/lib/cloud-connection/transaction-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"transaction-manager.d.ts","sourceRoot":"","sources":["../../src/cloud-connection/transaction-manager.ts"],"names":[],"mappings":"AAGA,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,WAAW,CACd;IACL,OAAO,CAAC,WAAW,CAAuD;;IAOnE,oBAAoB;;;MAGxB;IAEI,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM;IAwB7D,yBAAyB,CAAC,SAAS,EAAE,MAAM;IAK3C,yBAAyB,CAAC,IAAI,EAAE,MAAM;IAKtC,mBAAmB,CAAC,IAAI,EAAE,MAAM;CAOxC"}
+{"version":3,"file":"transaction-manager.d.ts","sourceRoot":"","sources":["../../src/cloud-connection/transaction-manager.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAcxC,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,WAAW,CAA0C;IAC7D,OAAO,CAAC,gBAAgB,CAA0C;IAClE,OAAO,CAAC,kBAAkB,CAAqB;IAC/C,OAAO,CAAC,SAAS,CAAY;IAE7B,OAAO,CAAC,gBAAgB;IA+CxB,OAAO,CAAC,iBAAiB;gBAab,SAAS,EAAE,SAAS,EAAE,kBAAkB,EAAE,kBAAkB;IAOjE,MAAM;;;;MAIV;IAEU,kBAAkB,CAAC,KAAK,EAAE;QACrC,IAAI,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7B,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,OAAO,CAAC;QACf,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;IA4CM,yBAAyB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAKhE,gCAAgC,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAI5D,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI3C,uBAAuB,IAAI,OAAO;IAIlC,yBAAyB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAK3D,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;CAmB/C"}

package/lib/cloud-connection/transaction-manager.js

@@ -1,52 +1,121 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TransactionManager = void 0;
+const device_agent_schemas_1 = require("@alwaysai/device-agent-schemas");
+const logger_1 = require("../util/logger");
 const util_1 = require("alwaysai/lib/util");
 const coded_error_1 = require("@carnesen/coded-error");
 class TransactionManager {
-    constructor() {
-        this.txToProject = {};
-        this.projectToTx = {};
-        this.AddTransactionErrors = (0, util_1.keyMirror)({
-            TRANSACTION_ID_ONGOING: null,
-            PROJECT_ID_ONGOING: null
+    constructor(publisher, liveUpdatesHandler) {
+        this.detailsByTx = {};
+        this.detailsByProject = {};
+        this.Errors = (0, util_1.keyMirror)({
+            TRANSACTION_ONGOING: null,
+            PROJECT_ONGOING: null,
+            TRANSACTION_NOT_ONGOING: null
         });
-        this.txToProject = {};
-        this.projectToTx = {};
+        this.detailsByTx = {};
+        this.detailsByProject = {};
+        this.publisher = publisher;
+        this.liveUpdatesHandler = liveUpdatesHandler;
     }
-    addTransaction(txId, projectId, step) {
-        // Check if the Transaction already exists
-        if (this.txToProject[txId]) {
-            const { projectId, step } = this.txToProject[txId];
-            throw new coded_error_1.CodedError(`Transaction ID ${txId} already ongoing! project ID: ${projectId} step: ${step}`, this.AddTransactionErrors.TRANSACTION_ID_ONGOING);
+    startTransaction(txId, projectId, stepName) {
+        // Check if the transaction already exists
+        if (this.detailsByTx[txId]) {
+            const txnDetails = this.detailsByTx[txId];
+            throw new coded_error_1.CodedError(`Transaction ${txId} already ongoing!\n${JSON.stringify(txnDetails, null, 2)}`, this.Errors.TRANSACTION_ONGOING);
+        }
+        // Check if there is any ongoing transactions for project
+        if (this.detailsByProject[projectId]) {
+            const txnDetails = this.detailsByProject[projectId];
+            throw new coded_error_1.CodedError(`Project ${projectId} already has an ongoing transaction!\n${JSON.stringify(txnDetails, null, 2)}`, this.Errors.PROJECT_ONGOING);
+        }
+        // Map the Transaction ID with Project ID
+        const txDetails = {
+            txId,
+            projectId,
+            stepName,
+            start: new Date().toLocaleString()
+        };
+        this.detailsByTx[txId] = txDetails;
+        this.detailsByProject[projectId] = txDetails;
+        logger_1.logger.info(`Started transaction:\n${JSON.stringify(txDetails, null, 2)}`);
+        // send live updates
+        void this.liveUpdatesHandler.enableTransactionStatus({
+            txId
+        });
+    }
+    updateTransaction(txId, stepName) {
+        const txDetails = this.detailsByTx[txId];
+        if (!txDetails) {
+            throw new coded_error_1.CodedError(`Cannot update transaction ${txId} since it doesn't exist!`, this.Errors.TRANSACTION_NOT_ONGOING);
+        }
+        txDetails.stepName = stepName;
+        txDetails.update = new Date().toLocaleString();
+        logger_1.logger.info(`Updated transaction:\n${JSON.stringify(txDetails, null, 2)}`);
+    }
+    async runTransactionStep(props) {
+        const { func, projectId, txId, start, stepName } = props;
+        if (start) {
+            this.startTransaction(txId, projectId, stepName);
         }
         else {
-            // Check if there is any ongoing Transactions for Project ID
-            if (this.projectToTx[projectId]) {
-                const { txId, step } = this.projectToTx[projectId];
-                throw new coded_error_1.CodedError(`This Project ID ${projectId} already has an ongoing transaction! transaction ID: ${txId} step: ${step}`, this.AddTransactionErrors.PROJECT_ID_ONGOING);
-            }
-            else {
-                // Map the Transaction ID with Project ID
-                this.txToProject[txId] = { projectId, step };
-                this.projectToTx[projectId] = { txId, step };
+            this.updateTransaction(txId, stepName);
+        }
+        try {
+            const completed = await func();
+            if (completed) {
+                this.completeTransaction(txId);
+                const successStatusResponsePayload = {
+                    status: device_agent_schemas_1.keyMirrors.statusResponse.success
+                };
+                // Send final status message
+                const message = (0, device_agent_schemas_1.buildToClientStatusResponseMessage)(this.publisher.getClientId(), successStatusResponsePayload, txId);
+                this.publisher.publishToClient(message);
             }
         }
+        catch (e) {
+            const message = e.message;
+            logger_1.logger.error(`Failed to execute cmd for ${projectId}:\n${message}\n${e.stack}`);
+            this.completeTransaction(txId);
+            const failureStatusResponsePayload = {
+                status: device_agent_schemas_1.keyMirrors.statusResponse.failure,
+                message
+            };
+            // Send final status message
+            const failureStatusResponseMessage = (0, device_agent_schemas_1.buildToClientStatusResponseMessage)(this.publisher.getClientId(), failureStatusResponsePayload, txId);
+            this.publisher.publishToClient(failureStatusResponseMessage);
+        }
     }
     getTransactionFromProject(projectId) {
-        const txnDetails = this.projectToTx[projectId];
+        const txnDetails = this.detailsByProject[projectId];
         return txnDetails === null || txnDetails === void 0 ? void 0 : txnDetails.txId;
     }
+    isOngoingTransactionForProjectID(projectId) {
+        return projectId in this.detailsByProject;
+    }
+    isOngoingTransaction(txId) {
+        return txId in this.detailsByTx;
+    }
+    isAnyOngoingTransaction() {
+        return Object.keys(this.detailsByTx).length > 0;
+    }
     getProjectFromTransaction(txId) {
-        const txnDetails = this.txToProject[txId];
+        const txnDetails = this.detailsByTx[txId];
         return txnDetails === null || txnDetails === void 0 ? void 0 : txnDetails.projectId;
     }
     completeTransaction(txId) {
-        const txnDetails = this.txToProject[txId];
-        delete this.txToProject[txId];
-        if (txnDetails !== undefined) {
-            delete this.projectToTx[txnDetails.projectId];
+        const txDetails = this.detailsByTx[txId];
+        if (txDetails === undefined) {
+            throw new coded_error_1.CodedError(`Cannot complete transaction ${txId} since it doesn't exist!`, this.Errors.TRANSACTION_NOT_ONGOING);
         }
+        txDetails.stop = new Date().toLocaleString();
+        logger_1.logger.info(`Completed transaction:\n${JSON.stringify(txDetails, null, 2)}`);
+        delete this.detailsByTx[txId];
+        delete this.detailsByProject[txDetails.projectId];
+        void this.liveUpdatesHandler.disableTransactionStatus({
+            txId
+        });
     }
 }
 exports.TransactionManager = TransactionManager;

package/lib/cloud-connection/transaction-manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"transaction-manager.js","sourceRoot":"","sources":["../../src/cloud-connection/transaction-manager.ts"],"names":[],"mappings":";;;AAAA,4CAA8C;AAC9C,uDAAmD;AAEnD,MAAa,kBAAkB;IAK7B;QAJQ,gBAAW,GACjB,EAAE,CAAC;QACG,gBAAW,GAAoD,EAAE,CAAC;QAOnE,yBAAoB,GAAG,IAAA,gBAAS,EAAC;YACtC,sBAAsB,EAAE,IAAI;YAC5B,kBAAkB,EAAE,IAAI;SACzB,CAAC,CAAC;QAPD,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;IACxB,CAAC;IAOM,cAAc,CAAC,IAAY,EAAE,SAAiB,EAAE,IAAa;QAClE,0CAA0C;QAC1C,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;YAC1B,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,IAAI,wBAAU,CAClB,kBAAkB,IAAI,iCAAiC,SAAS,UAAU,IAAI,EAAE,EAChF,IAAI,CAAC,oBAAoB,CAAC,sBAAsB,CACjD,CAAC;SACH;aAAM;YACL,4DAA4D;YAC5D,IAAI,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE;gBAC/B,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;gBACnD,MAAM,IAAI,wBAAU,CAClB,mBAAmB,SAAS,wDAAwD,IAAI,UAAU,IAAI,EAAE,EACxG,IAAI,CAAC,oBAAoB,CAAC,kBAAkB,CAC7C,CAAC;aACH;iBAAM;gBACL,yCAAyC;gBACzC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;gBAC7C,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aAC9C;SACF;IACH,CAAC;IAEM,yBAAyB,CAAC,SAAiB;QAChD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC/C,OAAO,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,IAAI,CAAC;IAC1B,CAAC;IAEM,yBAAyB,CAAC,IAAY;QAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC1C,OAAO,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,SAAS,CAAC;IAC/B,CAAC;IAEM,mBAAmB,CAAC,IAAY;QACrC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,UAAU,KAAK,SAAS,EAAE;YAC5B,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;SAC/C;IACH,CAAC;CACF;AAxDD,gDAwDC"}
+{"version":3,"file":"transaction-manager.js","sourceRoot":"","sources":["../../src/cloud-connection/transaction-manager.ts"],"names":[],"mappings":";;;AAAA,yEAIwC;AAGxC,2CAAwC;AACxC,4CAA8C;AAC9C,uDAAmD;AAWnD,MAAa,kBAAkB;IAkE7B,YAAY,SAAoB,EAAE,kBAAsC;QAjEhE,gBAAW,GAAuC,EAAE,CAAC;QACrD,qBAAgB,GAAuC,EAAE,CAAC;QAuE3D,WAAM,GAAG,IAAA,gBAAS,EAAC;YACxB,mBAAmB,EAAE,IAAI;YACzB,eAAe,EAAE,IAAI;YACrB,uBAAuB,EAAE,IAAI;SAC9B,CAAC,CAAC;QAVD,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IAC/C,CAAC;IAjEO,gBAAgB,CACtB,IAAY,EACZ,SAAiB,EACjB,QAAiB;QAEjB,0CAA0C;QAC1C,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;YAC1B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC1C,MAAM,IAAI,wBAAU,CAClB,eAAe,IAAI,sBAAsB,IAAI,CAAC,SAAS,CACrD,UAAU,EACV,IAAI,EACJ,CAAC,CACF,EAAE,EACH,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAChC,CAAC;SACH;QAED,yDAAyD;QACzD,IAAI,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE;YACpC,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,IAAI,wBAAU,CAClB,WAAW,SAAS,yCAAyC,IAAI,CAAC,SAAS,CACzE,UAAU,EACV,IAAI,EACJ,CAAC,CACF,EAAE,EACH,IAAI,CAAC,MAAM,CAAC,eAAe,CAC5B,CAAC;SACH;QAED,yCAAyC;QACzC,MAAM,SAAS,GAAG;YAChB,IAAI;YACJ,SAAS;YACT,QAAQ;YACR,KAAK,EAAE,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE;SACnC,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;QACnC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,GAAG,SAAS,CAAC;QAC7C,eAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3E,oBAAoB;QACpB,KAAK,IAAI,CAAC,kBAAkB,CAAC,uBAAuB,CAAC;YACnD,IAAI;SACL,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,IAAY,EAAE,QAAiB;QACvD,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,SAAS,EAAE;YACd,MAAM,IAAI,wBAAU,CAClB,6BAA6B,IAAI,0BAA0B,EAC3D,IAAI,CAAC,MAAM,CAAC,uBAAuB,CACpC,CAAC;SACH;QACD,SAAS,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC9B,SAAS,CAAC,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;QAC/C,eAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IAC7E,CAAC;IAeM,KAAK,CAAC,kBAAkB,CAAC,KAM/B;QACC,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;QACzD,IAAI,KAAK,EAAE;YACT,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;SAClD;aAAM;YACL,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;SACxC;QACD,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,EAAE,CAAC;YAC/B,IAAI,SAAS,EAAE;gBACb,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAC/B,MAAM,4BAA4B,GAA0B;oBAC1D,MAAM,EAAE,iCAAU,CAAC,cAAc,CAAC,OAAO;iBAC1C,CAAC;gBACF,4BAA4B;gBAC5B,MAAM,OAAO,GAAG,IAAA,yDAAkC,EAChD,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAC5B,4BAA4B,EAC5B,IAAI,CACL,CAAC;gBACF,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;aACzC;SACF;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,OAAO,GAAW,CAAC,CAAC,OAAO,CAAC;YAClC,eAAM,CAAC,KAAK,CACV,6BAA6B,SAAS,MAAM,OAAO,KAAK,CAAC,CAAC,KAAK,EAAE,CAClE,CAAC;YAEF,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE/B,MAAM,4BAA4B,GAA0B;gBAC1D,MAAM,EAAE,iCAAU,CAAC,cAAc,CAAC,OAAO;gBACzC,OAAO;aACR,CAAC;YACF,4BAA4B;YAC5B,MAAM,4BAA4B,GAAG,IAAA,yDAAkC,EACrE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAC5B,4BAA4B,EAC5B,IAAI,CACL,CAAC;YACF,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,4BAA4B,CAAC,CAAC;SAC9D;IACH,CAAC;IAEM,yBAAyB,CAAC,SAAiB;QAChD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;QACpD,OAAO,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,IAAI,CAAC;IAC1B,CAAC;IAEM,gCAAgC,CAAC,SAAiB;QACvD,OAAO,SAAS,IAAI,IAAI,CAAC,gBAAgB,CAAC;IAC5C,CAAC;IAEM,oBAAoB,CAAC,IAAY;QACtC,OAAO,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC;IAClC,CAAC;IAEM,uBAAuB;QAC5B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;IAClD,CAAC;IAEM,yBAAyB,CAAC,IAAY;QAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC1C,OAAO,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,SAAS,CAAC;IAC/B,CAAC;IAEM,mBAAmB,CAAC,IAAY;QACrC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,SAAS,KAAK,SAAS,EAAE;YAC3B,MAAM,IAAI,wBAAU,CAClB,+BAA+B,IAAI,0BAA0B,EAC7D,IAAI,CAAC,MAAM,CAAC,uBAAuB,CACpC,CAAC;SACH;QACD,SAAS,CAAC,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;QAC7C,eAAM,CAAC,IAAI,CACT,2BAA2B,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAChE,CAAC;QACF,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAElD,KAAK,IAAI,CAAC,kBAAkB,CAAC,wBAAwB,CAAC;YACpD,IAAI;SACL,CAAC,CAAC;IACL,CAAC;CACF;AA1KD,gDA0KC"}

package/lib/cloud-connection/transaction-manager.test.js

@@ -3,79 +3,226 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const device_agent_schemas_1 = require("@alwaysai/device-agent-schemas");
 const transaction_manager_1 = require("./transaction-manager");
 const uuid_1 = require("uuid");
+const publisher_1 = require("./publisher");
+const mockClient = {
+    publish: jest.fn()
+};
+const clientId = 'test-client';
+const mockLiveUpdatesHandler = {
+    enableTransactionStatus: jest.fn(),
+    disableTransactionStatus: jest.fn()
+};
 describe('Test Transaction Manager', () => {
     let txnMgr;
+    let publisher;
+    const func_complete = jest
+        .fn()
+        .mockResolvedValue(true);
+    const func_incomplete = jest
+        .fn()
+        .mockResolvedValue(false);
     beforeEach(() => {
-        txnMgr = new transaction_manager_1.TransactionManager();
+        publisher = new publisher_1.Publisher(mockClient, clientId);
+        txnMgr = new transaction_manager_1.TransactionManager(publisher, mockLiveUpdatesHandler);
         jest.clearAllMocks();
     });
     function generateRandomProjectId() {
         return (0, uuid_1.v4)();
     }
-    test('Add a new transaction', async () => {
+    test('Start a new transaction', async () => {
         const txId = (0, device_agent_schemas_1.generateTxId)();
         const projectId = generateRandomProjectId();
-        txnMgr.addTransaction(txId, projectId);
+        await txnMgr.runTransactionStep({
+            func: func_incomplete,
+            projectId,
+            txId,
+            start: true
+        });
         expect(txnMgr.getProjectFromTransaction(txId)).toEqual(projectId);
         expect(txnMgr.getTransactionFromProject(projectId)).toEqual(txId);
+        expect(txnMgr.isOngoingTransaction(txId)).toBe(true);
+        expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(true);
+        expect(txnMgr.isAnyOngoingTransaction()).toBe(true);
+    });
+    test('Start a new transaction which completes in one step', async () => {
+        const txId = (0, device_agent_schemas_1.generateTxId)();
+        const projectId = generateRandomProjectId();
+        await txnMgr.runTransactionStep({
+            func: func_complete,
+            projectId,
+            txId,
+            start: true
+        });
+        expect(txnMgr.isOngoingTransaction(txId)).toBe(false);
+        expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(false);
+        expect(txnMgr.isAnyOngoingTransaction()).toBe(false);
+        const msg = JSON.parse(mockClient.publish.mock.calls[0][1]);
+        expect(msg.payload.status).toBe(device_agent_schemas_1.keyMirrors.statusResponse.success);
+    });
+    test('Start a new transaction and complete in second step', async () => {
+        const txId = (0, device_agent_schemas_1.generateTxId)();
+        const projectId = generateRandomProjectId();
+        await txnMgr.runTransactionStep({
+            func: func_incomplete,
+            projectId,
+            txId,
+            start: true,
+            stepName: 'step1'
+        });
+        await txnMgr.runTransactionStep({
+            func: func_complete,
+            projectId,
+            txId,
+            start: false,
+            stepName: 'step2'
+        });
+        expect(txnMgr.isOngoingTransaction(txId)).toBe(false);
+        expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(false);
+        expect(txnMgr.isAnyOngoingTransaction()).toBe(false);
     });
-    test('Add multiple transactions to different projects', async () => {
+    test('Start multiple transactions for different projects', async () => {
         const numTransactions = 3;
         const projectTxIdList = [];
         for (let i = 0; i < numTransactions; i++) {
             const txId = (0, device_agent_schemas_1.generateTxId)();
             const projectId = generateRandomProjectId();
-            txnMgr.addTransaction(txId, projectId);
+            await txnMgr.runTransactionStep({
+                func: func_incomplete,
+                projectId,
+                txId,
+                start: true
+            });
             projectTxIdList.push({ projectId, txId });
         }
         projectTxIdList.forEach(({ txId, projectId }) => {
             expect(txnMgr.getProjectFromTransaction(txId)).toEqual(projectId);
             expect(txnMgr.getTransactionFromProject(projectId)).toEqual(txId);
+            expect(txnMgr.isOngoingTransaction(txId)).toBe(true);
+            expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(true);
+            expect(txnMgr.isAnyOngoingTransaction()).toBe(true);
         });
     });
-    test('Attempt to add existing transaction to a project, results in failure', async () => {
+    test('Attempt to start an ongoing transaction, results in failure', async () => {
         const txId = (0, device_agent_schemas_1.generateTxId)();
         const projectId = generateRandomProjectId();
-        txnMgr.addTransaction(txId, projectId);
+        await txnMgr.runTransactionStep({
+            func: func_incomplete,
+            projectId,
+            txId,
+            start: true,
+            stepName: 'step1'
+        });
         try {
-            txnMgr.addTransaction(txId, projectId);
-            throw new Error('Expected adding transaction to fail!');
+            await txnMgr.runTransactionStep({
+                func: func_incomplete,
+                projectId,
+                txId,
+                start: true,
+                stepName: 'step2'
+            });
+            throw new Error('Expected starting transaction to fail!');
         }
-        catch (exception) {
-            console.log(exception);
-            expect(exception.code).toBe(txnMgr.AddTransactionErrors.TRANSACTION_ID_ONGOING);
+        catch (e) {
+            console.log(e);
+            expect(e.code).toBe(txnMgr.Errors.TRANSACTION_ONGOING);
         }
         expect(txnMgr.getTransactionFromProject(projectId)).toEqual(txId);
         expect(txnMgr.getProjectFromTransaction(txId)).toEqual(projectId);
     });
-    test('Attempt to add a transaction to a project with an ongoing transaction, results in failure', async () => {
+    test('Attempt to continue a transaction that is not ongoing', async () => {
+        const txId = (0, device_agent_schemas_1.generateTxId)();
+        const projectId = generateRandomProjectId();
+        try {
+            await txnMgr.runTransactionStep({
+                func: func_incomplete,
+                projectId,
+                txId,
+                start: false,
+                stepName: 'step1'
+            });
+            throw new Error('Expected continue transaction to fail!');
+        }
+        catch (e) {
+            console.log(e);
+            expect(e.code).toBe(txnMgr.Errors.TRANSACTION_NOT_ONGOING);
+        }
+        expect(txnMgr.isOngoingTransaction(txId)).toBe(false);
+        expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(false);
+        expect(txnMgr.isAnyOngoingTransaction()).toBe(false);
+    });
+    test('Attempt to start a transaction for a project with an ongoing transaction, results in failure', async () => {
         const txId = (0, device_agent_schemas_1.generateTxId)();
         const txId2 = (0, device_agent_schemas_1.generateTxId)();
         const projectId = generateRandomProjectId();
-        txnMgr.addTransaction(txId, projectId);
+        await txnMgr.runTransactionStep({
+            func: func_incomplete,
+            projectId,
+            txId,
+            start: true,
+            stepName: 'step1'
+        });
         try {
-            txnMgr.addTransaction(txId2, projectId);
-            throw new Error('Expected adding transaction to fail!');
+            await txnMgr.runTransactionStep({
+                func: func_incomplete,
+                projectId,
+                txId: txId2,
+                start: true,
+                stepName: 'step2'
+            });
+            throw new Error('Expected start transaction to fail!');
         }
-        catch (exception) {
-            console.log(exception);
-            expect(exception.code).toBe(txnMgr.AddTransactionErrors.PROJECT_ID_ONGOING);
+        catch (e) {
+            console.log(e);
+            expect(e.code).toBe(txnMgr.Errors.PROJECT_ONGOING);
         }
         expect(txnMgr.getTransactionFromProject(projectId)).toEqual(txId);
         expect(txnMgr.getProjectFromTransaction(txId)).toEqual(projectId);
         expect(txnMgr.getProjectFromTransaction(txId2)).toBeUndefined();
     });
-    test('Test remove transaction from queue', async () => {
+    test('Handle error in step function', async () => {
         const txId = (0, device_agent_schemas_1.generateTxId)();
         const projectId = generateRandomProjectId();
-        txnMgr.addTransaction(txId, projectId);
+        await txnMgr.runTransactionStep({
+            func: jest.fn().mockImplementation(() => {
+                throw new Error('Test error!');
+            }),
+            projectId,
+            txId,
+            start: true,
+            stepName: 'step1'
+        });
+        expect(txnMgr.isOngoingTransaction(txId)).toBe(false);
+        expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(false);
+        expect(txnMgr.isAnyOngoingTransaction()).toBe(false);
+        const msg = JSON.parse(mockClient.publish.mock.calls[0][1]);
+        expect(msg.payload.status).toBe(device_agent_schemas_1.keyMirrors.statusResponse.failure);
+    });
+    test('Complete ongoing transaction', async () => {
+        const txId = (0, device_agent_schemas_1.generateTxId)();
+        const projectId = generateRandomProjectId();
+        await txnMgr.runTransactionStep({
+            func: func_incomplete,
+            projectId,
+            txId,
+            start: true,
+            stepName: 'step1'
+        });
         txnMgr.completeTransaction(txId);
         expect(txnMgr.getTransactionFromProject(projectId)).toBeUndefined();
         expect(txnMgr.getProjectFromTransaction(txId)).toBeUndefined();
+        expect(txnMgr.isOngoingTransactionForProjectID(projectId)).toBe(false);
+        expect(txnMgr.isAnyOngoingTransaction()).toBe(false);
     });
     test('Test remove non-existing transaction from queue', async () => {
         const txId = (0, device_agent_schemas_1.generateTxId)();
-        txnMgr.completeTransaction(txId);
+        try {
+            txnMgr.completeTransaction(txId);
+            throw new Error('Expected completTransaction to fail!');
+        }
+        catch (e) {
+            console.log(e);
+            expect(e.code).toBe(txnMgr.Errors.TRANSACTION_NOT_ONGOING);
+        }
         expect(txnMgr.getProjectFromTransaction(txId)).toBeUndefined();
     });
 });

package/lib/cloud-connection/transaction-manager.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"transaction-manager.test.js","sourceRoot":"","sources":["../../src/cloud-connection/transaction-manager.test.ts"],"names":[],"mappings":";;AAAA,yEAA8D;AAC9D,+DAA2D;AAC3D,+BAAoC;AAEpC,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,MAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,IAAI,wCAAkB,EAAE,CAAC;QAClC,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,uBAAuB;QAC9B,OAAO,IAAA,SAAM,GAAE,CAAC;IAClB,CAAC;IAED,IAAI,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,eAAe,GAAG,CAAC,CAAC;QAC1B,MAAM,eAAe,GAAQ,EAAE,CAAC;QAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,EAAE,CAAC,EAAE,EAAE;YACxC,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;YAC5C,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACvC,eAAe,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;SAC3C;QACD,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE;YAC9C,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACvC,IAAI;YACF,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;SACzD;QAAC,OAAO,SAAS,EAAE;YAClB,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvB,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CACzB,MAAM,CAAC,oBAAoB,CAAC,sBAAsB,CACnD,CAAC;SACH;QACD,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QAC3G,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACvC,IAAI;YACF,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;SACzD;QAAC,OAAO,SAAS,EAAE;YAClB,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvB,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CACzB,MAAM,CAAC,oBAAoB,CAAC,kBAAkB,CAC/C,CAAC;SACH;QACD,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACvC,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAEjC,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAEjC,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACjE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"transaction-manager.test.js","sourceRoot":"","sources":["../../src/cloud-connection/transaction-manager.test.ts"],"names":[],"mappings":";;AAAA,yEAIwC;AACxC,+DAA2D;AAC3D,+BAAoC;AACpC,2CAAwC;AAGxC,MAAM,UAAU,GAAG;IACjB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;CACnB,CAAC;AACF,MAAM,QAAQ,GAAG,aAAa,CAAC;AAE/B,MAAM,sBAAsB,GAAG;IAC7B,uBAAuB,EAAE,IAAI,CAAC,EAAE,EAAE;IAClC,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;CACP,CAAC;AAE/B,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,MAA0B,CAAC;IAC/B,IAAI,SAAoB,CAAC;IACzB,MAAM,aAAa,GAA2B,IAAI;SAC/C,EAAE,EAAE;SACJ,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAC3B,MAAM,eAAe,GAA2B,IAAI;SACjD,EAAE,EAAE;SACJ,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAE5B,UAAU,CAAC,GAAG,EAAE;QACd,SAAS,GAAG,IAAI,qBAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,GAAG,IAAI,wCAAkB,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;QACnE,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,uBAAuB;QAC9B,OAAO,IAAA,SAAM,GAAE,CAAC;IAClB,CAAC;IAED,IAAI,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,eAAe;YACrB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,aAAa;YACnB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,MAAM,GAAG,GAAkC,IAAI,CAAC,KAAK,CACnD,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CACpC,CAAC;QACF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iCAAU,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,eAAe;YACrB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,aAAa;YACnB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,KAAK;YACZ,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,eAAe,GAAG,CAAC,CAAC;QAC1B,MAAM,eAAe,GAAQ,EAAE,CAAC;QAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,EAAE,CAAC,EAAE,EAAE;YACxC,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;YAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,SAAS;gBACT,IAAI;gBACJ,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;YACH,eAAe,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;SAC3C;QACD,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE;YAC9C,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,eAAe;YACrB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,IAAI;YACF,MAAM,MAAM,CAAC,kBAAkB,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,SAAS;gBACT,IAAI;gBACJ,KAAK,EAAE,IAAI;gBACX,QAAQ,EAAE,OAAO;aAClB,CAAC,CAAC;YACH,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACf,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;SACxD;QACD,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,IAAI;YACF,MAAM,MAAM,CAAC,kBAAkB,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,SAAS;gBACT,IAAI;gBACJ,KAAK,EAAE,KAAK;gBACZ,QAAQ,EAAE,OAAO;aAClB,CAAC,CAAC;YACH,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACf,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;SAC5D;QACD,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,8FAA8F,EAAE,KAAK,IAAI,EAAE;QAC9G,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,eAAe;YACrB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,IAAI;YACF,MAAM,MAAM,CAAC,kBAAkB,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,SAAS;gBACT,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,IAAI;gBACX,QAAQ,EAAE,OAAO;aAClB,CAAC,CAAC;YACH,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;SACxD;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACf,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;SACpD;QACD,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACtC,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;YACjC,CAAC,CAAC;YACF,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,MAAM,GAAG,GAAkC,IAAI,CAAC,KAAK,CACnD,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CACpC,CAAC;QACF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iCAAU,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,kBAAkB,CAAC;YAC9B,IAAI,EAAE,eAAe;YACrB,SAAS;YACT,IAAI;YACJ,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAEjC,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,gCAAgC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,IAAI,GAAG,IAAA,mCAAY,GAAE,CAAC;QAC5B,IAAI;YACF,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;SACzD;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACf,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;SAC5D;QAED,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACjE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/lib/secure-tunneling/secure-tunneling.d.ts

@@ -0,0 +1,105 @@
+export declare type SecureTunnelPortInfo = {
+    enabled: boolean;
+    type: string;
+    ip: string;
+    port: number;
+};
+export declare type SecureTunnelShadowDesRep = {
+    st_ports: SecureTunnelPortInfo[];
+};
+export declare type SecureTunnelShadowState = {
+    reported?: SecureTunnelShadowDesRep;
+    desired?: SecureTunnelShadowDesRep;
+};
+export declare type SecureTunnelShadowUpdateDelta = {
+    version: number;
+    timestamp: number;
+    state: SecureTunnelShadowDesRep;
+    metadata?: any;
+};
+export declare type SecureTunnelShadowUpdate = {
+    version: number;
+    state: SecureTunnelShadowState;
+};
+declare type SecureTunnelNotificationType = {
+    clientAccessToken: string;
+    region: string;
+    services: string[];
+};
+/**
+ * Handles Secure Tunnel Shadow State
+ */
+export declare class SecureTunnelHandlerSingleton {
+    private static instance;
+    private reported;
+    private httpProxyMap;
+    private localproxyInfo;
+    /**
+     * Initializes private variables of SecureTunnel handler.
+     * private constructor to prevent instantiation from outside
+     */
+    private constructor();
+    /**
+     * Implements the Singleton of the SecureTunnel
+     */
+    static getInstance(): SecureTunnelHandlerSingleton;
+    /**
+     * Stops all proxies started before
+     */
+    destroy(): Promise<void>;
+    /**
+     * Returns current state of SecureTunnel shadow
+     * @returns {SecureTunnelShadowDesRep} - reported state of the SecureTunnel shadow
+     */
+    getSecureTunnelShadow(): SecureTunnelShadowDesRep;
+    /**
+     * Updates current state of SecureTunnel shadow
+     * @param {SecureTunnelShadowUpdateDelta} deltaMsg - delta message, which includes desired state of the SecureTunnel shadow
+     * @return {SecureTunnelShadowDesRep} update reported message to send back to AWS IoT device shadow
+     */
+    syncShadowToDeviceState(deltaMsg: SecureTunnelShadowUpdateDelta): Promise<SecureTunnelShadowDesRep>;
+    /**
+     * Starts SecureTunnel
+     * @param {SecureTunnelNotificationType} message - AWS notification received
+     */
+    secureTunnelNotifyHandler(message: SecureTunnelNotificationType): Promise<void>;
+    /**
+     * Removes reported ports which are do not exist in desired
+     * @param {SecureTunnelPortInfo[]} desiredPorts - desired port config
+     */
+    private CleanupReportedPorts;
+    /**
+     * Starts port proxy mapping process
+     * @param {SecureTunnelPortInfo} portInfo - port info to start the port mapping process for
+     */
+    private startProxyMapping;
+    /**
+     * Stops port proxy mapping process
+     * @param {SecureTunnelPortInfo} portInfo - port info to stop the port mapping process for
+     */
+    private stopProxyMapping;
+    /**
+     * Starts SecureTunnel localproxy process
+     */
+    private startLocalproxy;
+    /**
+     * Stops SecureTunnel localproxy process
+     */
+    private stopLocalproxy;
+    /**
+     * processes and validate notify message
+     * @param {string[]} message - message, which contains: clientAccessToken, region, services
+     */
+    private processNotifyMessage;
+    /**
+     * Downloads SecureTunnel localproxy, if it was not downloaded before
+     */
+    private downloadSecureTunnel;
+    /**
+     * Gets next available localhost port
+     */
+    private getNextAvailablePort;
+    private sortPorts;
+}
+export {};
+//# sourceMappingURL=secure-tunneling.d.ts.map

package/lib/secure-tunneling/secure-tunneling.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"secure-tunneling.d.ts","sourceRoot":"","sources":["../../src/secure-tunneling/secure-tunneling.ts"],"names":[],"mappings":"AAsBA,oBAAY,oBAAoB,GAAG;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,oBAAY,wBAAwB,GAAG;IACrC,QAAQ,EAAE,oBAAoB,EAAE,CAAC;CAClC,CAAC;AAEF,oBAAY,uBAAuB,GAAG;IACpC,QAAQ,CAAC,EAAE,wBAAwB,CAAC;IACpC,OAAO,CAAC,EAAE,wBAAwB,CAAC;CACpC,CAAC;AAEF,oBAAY,6BAA6B,GAAG;IAC1C,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,wBAAwB,CAAC;IAChC,QAAQ,CAAC,EAAE,GAAG,CAAC;CAChB,CAAC;AAEF,oBAAY,wBAAwB,GAAG;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,uBAAuB,CAAC;CAChC,CAAC;AAsBF,aAAK,4BAA4B,GAAG;IAClC,iBAAiB,EAAE,MAAM,CAAC;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAMF;;GAEG;AACH,qBAAa,4BAA4B;IACvC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAA+B;IACtD,OAAO,CAAC,QAAQ,CAA2B;IAC3C,OAAO,CAAC,YAAY,CAAyB;IAC7C,OAAO,CAAC,cAAc,CAA6B;IAEnD;;;OAGG;IACH,OAAO;IAqBP;;OAEG;WACW,WAAW,IAAI,4BAA4B;IAUzD;;OAEG;IACU,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAuBrC;;;OAGG;IACI,qBAAqB,IAAI,wBAAwB;IAOxD;;;;OAIG;IACU,uBAAuB,CAClC,QAAQ,EAAE,6BAA6B,GACtC,OAAO,CAAC,wBAAwB,CAAC;IA0DpC;;;OAGG;IACU,yBAAyB,CACpC,OAAO,EAAE,4BAA4B,GACpC,OAAO,CAAC,IAAI,CAAC;IAmBhB;;;OAGG;YACW,oBAAoB;IAkClC;;;OAGG;YACW,iBAAiB;IA+C/B;;;OAGG;YACW,gBAAgB;IA2C9B;;OAEG;YACW,eAAe;IAuB7B;;OAEG;YACW,cAAc;IA4B5B;;;OAGG;IACH,OAAO,CAAC,oBAAoB;IAmF5B;;OAEG;YACW,oBAAoB;IA8BlC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAgB5B,OAAO,CAAC,SAAS;CAmBlB"}