@gibme/mikrotik 2.0.0 → 2.0.1

package/dist/index.js

@@ -375,14 +375,27 @@ class Mikrotik extends ssh_1.default {
                     });
                 }
                 this.on('stream', handleStream);
+                const cleanup = () => __awaiter(this, void 0, void 0, function* () {
+                    this.off('stream', handleStream);
+                    yield Mikrotik.cache.del(target);
+                });
                 yield Mikrotik.cache.set(target, target, options.duration); // set our mutex
-                yield this.stream(`/tool bandwidth-test protocol=${options.protocol} ` +
+                const cancel = yield this.stream(`/tool bandwidth-test protocol=${options.protocol} ` +
                     `user=${username} password=${password} ` +
                     `duration=${options.duration}s direction=${options.direction} ` +
                     `address=${target} random-data=${options.random_data ? 'yes' : 'no'} interval=1s`, {
                     separator: '\r\n\r\n'
                 });
-                yield Mikrotik.cache.del(target); // release our mutex
+                this.once('stream_complete', () => __awaiter(this, void 0, void 0, function* () {
+                    yield cleanup();
+                }));
+                this.once('stream_cancelled', () => __awaiter(this, void 0, void 0, function* () {
+                    yield cleanup();
+                    return reject(new Error('Bandwidth Test Cancelled'));
+                }));
+                if (options.timeout) {
+                    setTimeout(cancel, options.timeout);
+                }
             }));
         });
     }

package/dist/types.d.ts

@@ -35,6 +35,7 @@ export declare namespace BandwidthTest {
         protocol: Protocol;
         random_data: boolean;
         callback: (frame: Update) => void;
+        timeout: number;
     }
     export {};
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gibme/mikrotik",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "A simple mikrotik helper/wrapper",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -56,7 +56,7 @@
   },
   "dependencies": {
     "@gibme/cache": "^1.1.5",
-    "@gibme/ssh": "^1.0.2",
+    "@gibme/ssh": "^1.0.3",
     "@types/jsbn": "^1.2.33",
     "ip-address": "^9.0.5",
     "semver": "^7.6.2"