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

package/package.json

@@ -17,7 +17,8 @@
     "js-yaml": "^4.1.0",
     "log-update": "^5.0.1",
     "toml": "^3.0.0",
-    "uuid": "^9.0.0"
+    "uuid": "^9.0.0",
+    "yargs": "^17.5.1"
   },
-  "version": "0.6.0-beta.0"
+  "version": "0.6.0"
 }

package/src/index.js

@@ -9,6 +9,7 @@ export * as sink from './sink/index.js';
 export * as buffer from './buffer/index.js';
 export * as stream from './stream/index.js';
 export * as log from './log/index.js';
+export * as yargs from './yargs.js';
 
 export { default as Resolution } from './resolution.js';
 export { default as TaskQueue } from './task-queue.js';

package/src/rate-limiting-queue.js

@@ -61,12 +61,8 @@ export default class RateLimitingQueue {
     }, this._interval);
   }
 
-  async _exec(fn, res) {
-    try {
-      res.resolve(await fn());
-    } catch(error) {
-      res.reject(error);
-    }
+  _exec(fn, res) {
+    res.resolveWith(fn);
   }
 
-}
+}

package/src/resolution.js

@@ -9,7 +9,7 @@ export default class Resolution {
     Object.freeze(this);
   }
 
-  async execute(fn) {
+  async resolveWith(fn) {
     try {
       this.resolve(await fn());
     } catch(error) {

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/stream/output.js

@@ -17,6 +17,7 @@ export default class OutputStream extends Writable {
   }
 
   _write(record, _, next) {
+    //console.error(record);
     this._out.write(this._format(record) + '\n');
     next();
   }
@@ -24,7 +25,7 @@ export default class OutputStream extends Writable {
 }
 
 function defaultObjectModeFormat(v) {
-  typeof v === 'object' ? JSON.stringify(v) : `${v}`;
+  return typeof v === 'object' ? JSON.stringify(v) : `${v}`;
 }
 
 function defaultNonObjectModeFormat(v) {

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;
+}

package/src/yargs.js

@@ -0,0 +1,81 @@
+import 'dotenv/config';
+import _yargs from 'yargs/yargs';
+import { hideBin } from 'yargs/helpers';
+
+function hasDefaultCommand(args) {
+  let args0 = args[0];
+  const command = typeof args0 === 'object' ? args0.command : typeof args0 === 'string' ? args0 : undefined;
+  if (!command) {
+    return false; // this should not happen?
+  }
+  for (const c of Array.isArray(command) ? command : [command]) {
+    if (c.startsWith('*') || c.startsWith('$0')) {
+      return true;
+    }
+  }
+  return false;
+}
+
+class Tracker {
+
+  constructor() {}
+
+  get hasDefaultCommand() {
+    return !!this._hasDefaultCommand;
+  }
+
+  shim(yargs) {
+    const _command = yargs.command;
+    yargs.command = (...args) => {
+      if (hasDefaultCommand(args)) {
+        this._hasDefaultCommand = true;
+      }
+      return _command.apply(yargs, args);
+    };
+    return yargs;
+  }
+
+}
+
+export function build(fn) {
+  handleEpipe();
+  const yargs = _yargs(hideBin(process.argv));
+
+  const tracker = new Tracker();
+  tracker.shim(yargs);
+
+  fn(yargs);
+
+  if (!tracker.hasDefaultCommand) {
+    yargs.command('*', '', () => {}, () => yargs.showHelp())
+  }
+
+  return yargs
+    .help()
+    .fail(handleFail)
+    .parse();
+}
+
+let _epipeHandled = false;
+
+export function handleEpipe() {
+  if (_epipeHandled) {
+    return;
+  }
+  _epipeHandled = true;
+  process.stdout.on('error', err => err.code == 'EPIPE' && process.exit(0));
+}
+
+export function handleFail(msg, err) {
+  if (err) {
+    throw err;
+  }
+  console.error(msg);
+  process.exit(1);
+}
+
+export function coerceToArray(arg) {
+  return Array.isArray(arg) ? arg :
+    typeof arg === 'string' ? arg.split(',') :
+    arg === undefined || arg === null ? [] : [arg];
+}