feat: allow context prefix to be a boolean

rafamel · Apr 3, 2021 · 405b0ce · 405b0ce
1 parent 7c1310b
commit 405b0ce
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/src/bin/main.ts b/src/bin/main.ts
@@ -97,7 +97,7 @@ export default async function main(
     file: cmd['--file'],
     dir: cmd['--dir'],
     level: cmd['--level'] as LogLevel,
-    prefix: (cmd['--prefix'] ? 'all' : 'none') as PrefixPolicy,
+    prefix: cmd['--prefix'],
     env: (cmd['--env'] || []).reduce((acc, str) => {
       const arr = str.split('=');
       if (arr.length !== 2) {

diff --git a/src/definitions/index.ts b/src/definitions/index.ts
@@ -52,7 +52,7 @@ export interface Context {
    * lines of its *stdout* and *stderr* writes
    * with the stringification of its route.
    */
-  readonly prefix: PrefixPolicy;
+  readonly prefix: PrefixPolicy | boolean;
   /**
    * A *Promise* representing a task's
    * cancellation token.

diff --git a/src/helpers/prefix.ts b/src/helpers/prefix.ts
@@ -2,16 +2,24 @@ import { Context } from '../definitions';
 import { styleString } from './style-string';
 import { stringifyRoute } from './stringify-route';
 import { into } from 'pipettes';
+import { TypeGuard } from 'type-core';
 
 export function getPrefix(
   extra: null | string,
   target: 'print' | 'exec',
   context: Context
 ): string {
   return into(
-    context.route.length ? stringifyRoute(context.route) : '',
-    (prefix) => {
-      if (!prefix || (context.prefix !== target && context.prefix !== 'all')) {
+    {
+      prefix: context.route.length ? stringifyRoute(context.route) : '',
+      policy: TypeGuard.isString(context.prefix)
+        ? context.prefix
+        : context.prefix
+        ? 'all'
+        : 'none'
+    },
+    ({ prefix, policy }) => {
+      if (!prefix || (policy !== target && policy !== 'all')) {
         return extra ? extra + ' ' : null;
       }
       return (