@vbarbarosh/node-helpers 3.16.0 → 3.17.1

package/package.json

@@ -2,7 +2,7 @@
     "author": "Vladimir Barbarosh <vladimir.barbarosh@gmail.com>",
     "license": "MIT",
     "name": "@vbarbarosh/node-helpers",
-    "version": "3.16.0",
+    "version": "3.17.1",
     "description": "A set of helpers for JavaScript/Node.js",
     "files": [
         "src"

package/src/stream_strpbrk.js

@@ -0,0 +1,70 @@
+const stream = require('stream');
+
+/**
+ * Split input stream by consecutive delimiters in `chars`.
+ */
+function stream_strpbrk(chars = '\r\n')
+{
+    const pending = [];
+    return new stream.Transform({
+        objectMode: true,
+        transform: function (str, encoding, next) {
+            str = Buffer.isBuffer(str) ? str.toString() : str;
+
+            for (let off = 0; off < str.length; ) {
+                // skip consecutive delimiters
+                while (off < str.length) {
+                    let stop = true;
+                    for (let i = 0, end = chars.length; i < end; ++i) {
+                        if (str[off] === chars[i]) {
+                            off++;
+                            stop = false;
+                            break;
+                        }
+                    }
+                    if (stop) {
+                        break;
+                    }
+                }
+
+                // find next delimiter
+                const delims = [];
+                for (let i = 0, end = chars.length; i < end; ++i) {
+                    const j = str.indexOf(chars[i], off);
+                    if (j !== -1) {
+                        delims.push(j);
+                    }
+                }
+
+                // no delimiters were found
+                if (!delims.length) {
+                    const tmp = str.slice(off);
+                    if (tmp.length) {
+                        pending.push(tmp);
+                    }
+                    next();
+                    return;
+                }
+
+                delims.sort(fcmp);
+                pending.push(str.slice(off, delims[0]));
+                this.push(''.concat(...pending.splice(0)));
+                off = delims[0];
+            }
+            next();
+        },
+        flush: function (next) {
+            if (pending.length) {
+                this.push(''.concat(...pending.splice(0)));
+            }
+            next();
+        },
+    });
+}
+
+function fcmp(a, b)
+{
+    return a - b;
+}
+
+module.exports = stream_strpbrk;

package/src/stream_strpbrk.test.js

@@ -0,0 +1,28 @@
+const assert = require('assert');
+const stream = require('stream');
+const stream_strpbrk = require('./stream_strpbrk');
+
+const items = [
+    ['empty', '|', [], []],
+    ['| -> empty', '|', ['|'], []],
+    ['|| -> empty', '|', ['||'], []],
+    ['||| -> empty', '|', ['|', '|', '|'], []],
+    ['a → a', '|', ['a'], ['a']],
+    ['|a → a', '|', ['|a'], ['a']],
+    ['a| → a', '|', ['a|'], ['a']],
+    ['a|b → a,b', '|', ['a|b'], ['a', 'b']],
+    ['a|||b → a,b', '|', ['a|||b'], ['a', 'b']],
+    ['abc → abc', '|', ['a', 'b', 'c'], ['abc']],
+    ['a|b;c', '|;', ['a|b;c'], ['a', 'b', 'c']],
+    ['Buffer(a) → a', '|', ['a'].map(Buffer.from), ['a']],
+    ['Buffer(a|b) → a,b', '|', ['a|b'].map(Buffer.from), ['a', 'b']],
+];
+
+describe('stream_strpbrk', function () {
+    items.forEach(function ([title, delimiters, input, expected]) {
+        it(title, async function () {
+            const actual = await stream.Readable.from(input).pipe(stream_strpbrk(delimiters)).toArray();
+            assert.deepStrictEqual(actual, expected);
+        });
+    });
+});