@miso.ai/server-commons 0.6.0-beta.1 → 0.6.2-beta.0

package/package.json

@@ -20,5 +20,5 @@
     "uuid": "^9.0.0",
     "yargs": "^17.5.1"
   },
-  "version": "0.6.0-beta.1"
+  "version": "0.6.2-beta.0"
 }

package/src/stream/diff.js

@@ -0,0 +1,96 @@
+import { Transform } from 'stream';
+
+const OUTPUT = {
+  BOTH: 0,
+  PLUS: 1,
+  MINUS: 2,
+};
+
+export default class DiffStream extends Transform {
+
+  constructor(data, {
+    objectMode,
+    readableObjectMode,
+    writableObjectMode,
+    ...options
+  } = {}) {
+    super({
+      readableObjectMode: !!writableObjectMode || (readableObjectMode !== undefined ? !!readableObjectMode : objectMode !== undefined ? !!objectMode : false),
+      writableObjectMode: writableObjectMode !== undefined ? !!writableObjectMode : objectMode !== undefined ? !!objectMode : false,
+    });
+    if (writableObjectMode && readableObjectMode !== undefined && !readableObjectMode) {
+      throw new Error(`Cannot have writableObjectMode = true and readableObjectMode = false.`);
+    }
+    this._options = this._normalizeOptions(options);
+    this._baseDataSet = new Set(data);
+    this._inputDataSet = new Set();
+  }
+
+  _normalizeOptions({
+    output,
+    ...options
+  }) {
+    return {
+      output: this._normalizeOutputOptions(output),
+      ...options,
+    };
+  }
+
+  _normalizeOutputOptions(output = 'default') {
+    switch (output) {
+      case 'plus':
+        return OUTPUT.PLUS;
+      case 'minus':
+        return OUTPUT.MINUS;
+      case 'default':
+        return OUTPUT.BOTH;
+      default:
+        throw new Error(`Unrecognized output mode: ${output}`);
+    }
+  }
+
+  async _transform(value, _) {
+    if (value instanceof Buffer) {
+      value = value.toString();
+    }
+    // dedupe
+    const input = this._inputDataSet;
+    if (input.has(value)) {
+      return;
+    }
+    input.add(value);
+
+    const base = this._baseDataSet;
+    if (base.has(value)) {
+      base.delete(value);
+    } else {
+      this._output(true, value);
+    }
+  }
+
+  async _flush(done) {
+    for (const value of this._baseDataSet) {
+      this._output(false, value);
+    }
+    done();
+  }
+
+  _output(plus, value) {
+    switch (this._options.output) {
+      case OUTPUT.BOTH:
+        this.push(this.readableObjectMode ? [plus, value] : `${plus ? '+' : '-'} ${value}`);
+        break;
+      case OUTPUT.PLUS:
+        if (plus) {
+          this.push(value);
+        }
+        break;
+      case OUTPUT.MINUS:
+        if (!plus) {
+          this.push(value);
+        }
+        break;
+    }
+  }
+
+}

package/src/stream/index.js

@@ -3,3 +3,4 @@ export { default as BufferedReadStream } from './buffered-read.js';
 export { default as BufferedWriteStream } from './buffered-write.js';
 export { default as OutputStream } from './output.js';
 export { default as LogUpdateStream } from './log-update.js';
+export { default as DiffStream } from './diff.js';

package/src/stream/misc.js

@@ -60,3 +60,19 @@ export function transform(fn, { transform: _, ...options } = {}) {
     },
   });
 }
+
+export function take(n, { transform: _, ...options } = {}) {
+  let count = 0;
+  return new Transform({
+    ...options,
+    transform (chunk, _, next) {
+      if (count > n) {
+        next(undefined);
+        this.end();
+      } else {
+        next(undefined, chunk);
+      }
+      count++;
+    }
+  });
+}

package/src/string.js

@@ -30,3 +30,11 @@ export function padLeft(val, num, str) {
 export function padRight(val, num, str) {
   return pad(false, val, num, str);
 }
+
+export function encodeURIIfNecessary(str) {
+  return typeof str === 'string' && !str.includes('%') ? encodeURI(str) : str;
+}
+
+export function encodeURIComponentIfNecessary(str) {
+  return typeof str === 'string' && !str.includes('%') ? encodeURIComponent(str) : str;
+}