refactor: export collect as function for programmatic use

push-based · Jun 19, 2023 · 83390d1 · 83390d1
2 parents 938bea8 + af6d04d
commit 83390d1
Show file tree

Hide file tree

Showing 7 changed files with 68 additions and 37 deletions.
diff --git a/packages/cli/src/lib/commands/assert/command-impl.ts b/packages/cli/src/lib/commands/assert/command-impl.ts
@@ -0,0 +1,9 @@
+import {RcJsonAsArgv} from "../../types";
+
+import {logVerbose} from "../../core/loggin";
+import {readBudgets} from "./utils/budgets";
+
+export async function runAssertCommand(argv: RcJsonAsArgv) {
+  logVerbose(readBudgets());
+  return Promise.resolve();
+}
diff --git a/packages/cli/src/lib/commands/assert/index.ts b/packages/cli/src/lib/commands/assert/index.ts
@@ -21,7 +21,3 @@ export const assertCommand: YargsCommandObject = {
   }
 };
 
-export async function run(argv: Partial<RcJson>): Promise<void> {
-  logVerbose(readBudgets());
-  return Promise.resolve();
-}
diff --git a/packages/cli/src/lib/commands/collect/command-impl.ts b/packages/cli/src/lib/commands/collect/command-impl.ts
@@ -0,0 +1,19 @@
+import {RcJson, RcJsonAsArgv} from "../../types";
+import {getCollectCommandOptionsFromArgv} from "./utils/params";
+import {logVerbose} from "../../core/loggin";
+import {run} from "../../core/processing/behaviors";
+import {collectRcJson} from "../init/processes/collect-rc-json";
+import {startServerIfNeededAndExecute} from "./utils/serve-command";
+import {collectReports} from "./processes/collect-reports";
+
+export async function runCollectCommand(argv: RcJsonAsArgv) {
+  const cfg = getCollectCommandOptionsFromArgv(argv);
+  logVerbose('Collect options: ', cfg);
+  await run([
+    collectRcJson,
+    (cfg: RcJson) =>
+      startServerIfNeededAndExecute(() => collectReports(cfg)
+          .then()
+        , cfg.collect)
+  ])(cfg);
+}
diff --git a/packages/cli/src/lib/commands/collect/index.ts b/packages/cli/src/lib/commands/collect/index.ts
@@ -1,12 +1,7 @@
-import { YargsCommandObject } from '../../core/yargs/types';
-import { logVerbose } from '../../core/loggin/index';
-import { COLLECT_OPTIONS } from './options';
-import { startServerIfNeededAndExecute } from './utils/serve-command';
-import { collectRcJson } from '../init/processes/collect-rc-json';
-import { run } from '../../core/processing/behaviors';
-import { collectReports } from './processes/collect-reports';
-import { RcJson } from '../../types';
-import { getCollectCommandOptionsFromArgv } from './utils/params';
+import {YargsCommandObject} from '../../core/yargs/types';
+import {logVerbose} from '../../core/loggin/index';
+import {COLLECT_OPTIONS} from './options';
+import {runCollectCommand} from "./command-impl";
 
 export const collectUserFlowsCommand: YargsCommandObject = {
   command: 'collect',
@@ -15,15 +10,7 @@ export const collectUserFlowsCommand: YargsCommandObject = {
   module: {
     handler: async (argv: any) => {
       logVerbose(`run "collect" as a yargs command with args:`);
-      const cfg = getCollectCommandOptionsFromArgv(argv);
-      logVerbose('Collect options: ', cfg);
-      await run([
-        collectRcJson,
-        (cfg: RcJson) =>
-          startServerIfNeededAndExecute(() => collectReports(cfg)
-              .then()
-            , cfg.collect)
-      ])(cfg);
+      await runCollectCommand(argv);
     }
   }
 };
diff --git a/packages/cli/src/lib/commands/init/command-impl.ts b/packages/cli/src/lib/commands/init/command-impl.ts
@@ -0,0 +1,29 @@
+import {RcJsonAsArgv} from "../../types";
+
+import {log, logVerbose} from "../../core/loggin";
+import {run} from "../../core/processing/behaviors";
+import {collectRcJson} from "../init/processes/collect-rc-json";
+import {getGlobalOptionsFromArgv} from "../../global/utils";
+import {getInitCommandOptionsFromArgv} from "./utils";
+import {updateRcJson} from "./processes/update-rc-json";
+import {handleFlowGeneration} from "./processes/generate-userflow";
+import {handleGhWorkflowGeneration} from "./processes/generate-workflow";
+import {handleBudgetsGeneration} from "./processes/generate-lh-budgets";
+import {SETUP_CONFIRM_MESSAGE} from "./constants";
+
+export async function runInitCommand(argv: RcJsonAsArgv) {
+  const { interactive } = getGlobalOptionsFromArgv(argv);
+  const { generateFlow, generateGhWorkflow, generateBudgets, lhr, ...cfg } = getInitCommandOptionsFromArgv(argv);
+  logVerbose('Init options: ', { interactive, generateFlow, generateGhWorkflow, generateBudgets,lhr, ...cfg });
+
+  await run([
+    collectRcJson,
+    updateRcJson,
+    handleFlowGeneration({ interactive: !!interactive, generateFlow }),
+    handleGhWorkflowGeneration({ generateGhWorkflow }),
+    handleBudgetsGeneration({ generateBudgets, lhr }),
+  ])(cfg );
+  log(SETUP_CONFIRM_MESSAGE);
+  // @TODO move to constants
+  log('To execute a user flow run `npx user-flow` or `npx user-flow collect`');
+}
diff --git a/packages/cli/src/lib/commands/init/index.ts b/packages/cli/src/lib/commands/init/index.ts
@@ -10,6 +10,7 @@ import { handleFlowGeneration } from './processes/generate-userflow';
 import { getGlobalOptionsFromArgv } from '../../global/utils';
 import { handleGhWorkflowGeneration } from './processes/generate-workflow';
 import { handleBudgetsGeneration } from './processes/generate-lh-budgets';
+import {runInitCommand} from "./command-impl";
 
 export const initCommand: YargsCommandObject = {
   command: 'init',
@@ -18,20 +19,7 @@ export const initCommand: YargsCommandObject = {
   module: {
     handler: async (argv: any) => {
       logVerbose(`run "init" as a yargs command`);
-      const { interactive } = getGlobalOptionsFromArgv(argv);
-      const { generateFlow, generateGhWorkflow, generateBudgets, lhr, ...cfg } = getInitCommandOptionsFromArgv(argv);
-      logVerbose('Init options: ', { interactive, generateFlow, generateGhWorkflow, generateBudgets,lhr, ...cfg });
-
-      await run([
-        collectRcJson,
-        updateRcJson,
-        handleFlowGeneration({ interactive: !!interactive, generateFlow }),
-        handleGhWorkflowGeneration({ generateGhWorkflow }),
-        handleBudgetsGeneration({ generateBudgets, lhr }),
-      ])(cfg );
-      log(SETUP_CONFIRM_MESSAGE);
-      // @TODO move to constants
-      log('To execute a user flow run `npx user-flow` or `npx user-flow collect`');
+      await runInitCommand(argv);
     }
   }
 };

diff --git a/packages/cli/src/lib/index.ts b/packages/cli/src/lib/index.ts
@@ -28,4 +28,7 @@ export { DEFAULT_COLLECT_UF_PATH } from './commands/collect/options/ufPath.const
 export { DEFAULT_PERSIST_OUT_PATH } from './commands/collect/options/outPath.constant';
 export { ReportFormat } from './commands/collect/options/types';
 export { createReducedReport } from './commands/collect/utils/report/utils';
-export { enrichReducedReportWithBaseline } from './commands/collect/utils/report/utils';
+export {enrichReducedReportWithBaseline} from './commands/collect/utils/report/utils';
+export {runInitCommand} from './commands/init/command-impl';
+export {runCollectCommand} from './commands/collect/command-impl';
+export {runAssertCommand} from './commands/assert/command-impl';