chore: expose file name in cli

davidenke · Nov 9, 2024 · aae9796 · aae9796
1 parent 83ed784
commit aae9796
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 6 deletions.
diff --git a/src/cli.spec.ts b/src/cli.spec.ts
@@ -1,9 +1,12 @@
 import { exec as _exec } from 'node:child_process';
+import { rm } from 'node:fs/promises';
 import { promisify } from 'node:util';
 
 const exec = promisify(_exec);
 const cmd = 'node --loader ts-node/esm src/cli.ts';
 
+afterEach(async () => await rm('tmp', { recursive: true, force: true }));
+
 describe('cli', () => {
   it('throws an error if no config path is provided', async () => {
     const runsOrFails = () => exec(`${cmd} -t path/to/target`);
@@ -29,4 +32,14 @@ describe('cli', () => {
     const { stdout } = await exec(`${cmd} -c public/examples/blog.yml -t tmp`);
     expect(stdout).toEqual(expect.stringContaining('> blog schema written to'));
   });
+
+  it('stores with a custom name', async () => {
+    const { stdout } = await exec(`${cmd} -c public/examples/blog.yml -t tmp -n custom.ts`);
+    expect(stdout).toEqual(expect.stringContaining('tmp/custom.ts'));
+  });
+
+  it('replaces placeholders in name', async () => {
+    const { stdout } = await exec(`${cmd} -c public/examples/blog.yml -t tmp -n cfg.%%name%%.ts`);
+    expect(stdout).toEqual(expect.stringContaining('tmp/cfg.blog.ts'));
+  });
 });
diff --git a/src/cli.ts b/src/cli.ts
@@ -22,10 +22,11 @@ const { values } = parseArgs({
   options: {
     config: { type: 'string', short: 'c' },
     target: { type: 'string', short: 't' },
+    naming: { type: 'string', short: 'n' },
     watch: { type: 'boolean', short: 'w' },
   },
 });
-const { config, target, watch: useWatch } = values;
+const { config, target, naming, watch: useWatch } = values;
 
 // check for required arguments
 function fail(message: string, exitCode = 1) {
@@ -45,7 +46,12 @@ function tryOrFail<T>(fn: () => T, error: ERROR, exitCode: false | number = 1):
 }
 
 // read config and transform collections
-export async function loadAndTransformCollections(from?: string, to?: string, isUpdate = false) {
+export async function loadAndTransformCollections(
+  from?: string,
+  to?: string,
+  naming = 'config.%%name%%.ts',
+  isUpdate = false,
+) {
   if (!from) return fail(ERROR.MISSING_CONFIG);
   if (!to) return fail(ERROR.MISSING_TARGET);
 
@@ -57,7 +63,9 @@ export async function loadAndTransformCollections(from?: string, to?: string, is
     collections.map(async collection => {
       // transform collection
       const { cptime } = transformCollection(collection, { zod });
-      const path = resolve(to, `config.${collection.name}.ts`);
+      const keys = { name: collection.name };
+      const name = naming.replace(/%%(\w+)%%/, (_, k: keyof typeof keys) => keys[k] ?? _);
+      const path = resolve(to, name);
 
       // build content and prettify if possible
       const raw = `import { z } from 'astro:content';\n\nexport const schema = ${cptime};\n`;
@@ -89,15 +97,15 @@ if (useWatch) {
   process.on('SIGINT', () => abort.abort());
 
   // run once initially
-  await loadAndTransformCollections(config, target, false);
+  await loadAndTransformCollections(config, target, naming, false);
   console.info('> Watching for changes ...');
 
   // watch for changes
   try {
     const watcher = watch(config!, { encoding: 'utf-8', signal });
     for await (const { eventType } of watcher) {
       if (eventType === 'change') {
-        await loadAndTransformCollections(config, target, true);
+        await loadAndTransformCollections(config, target, naming, true);
       }
     }
   } catch (error: any) {
@@ -106,5 +114,5 @@ if (useWatch) {
     throw error;
   }
 } else {
-  await loadAndTransformCollections(config, target, false);
+  await loadAndTransformCollections(config, target, naming, false);
 }