From 504eb12a1d9241ee704156f15c55d36dfe4c8cd7 Mon Sep 17 00:00:00 2001
From: barbapapazes <e.soubiran25@gmail.com>
Date: Fri, 11 Oct 2024 10:27:02 +0200
Subject: [PATCH] feat(cli): support custom npm scripts prefix

---
 src/node/init/init.ts | 43 ++++++++++++++++++++++++++++++++++---------
 1 file changed, 34 insertions(+), 9 deletions(-)

diff --git a/src/node/init/init.ts b/src/node/init/init.ts
index a1ec8b5ca81c..55b85341bb5f 100644
--- a/src/node/init/init.ts
+++ b/src/node/init/init.ts
@@ -26,6 +26,8 @@ export interface ScaffoldOptions {
   theme: ScaffoldThemeType
   useTs: boolean
   injectNpmScripts: boolean
+  addNpmScriptsPrefix?: boolean
+  npmScriptsPrefix?: string
 }
 
 const getPackageManger = () => {
@@ -36,7 +38,7 @@ const getPackageManger = () => {
 export async function init(root: string | undefined) {
   intro(bold(cyan('Welcome to VitePress!')))
 
-  const options: ScaffoldOptions = await group(
+  const options: ScaffoldOptions = (await group(
     {
       root: async () => {
         if (root) return root
@@ -93,7 +95,25 @@ export async function init(root: string | undefined) {
       injectNpmScripts: () =>
         confirm({
           message: 'Add VitePress npm scripts to package.json?'
+        }),
+
+      addNpmScriptsPrefix: ({ results }) => {
+        if (!results.injectNpmScripts) return Promise.resolve(false)
+
+        return confirm({
+          message: 'Add a prefix for VitePress npm scripts?',
+          initialValue: true
+        })
+      },
+
+      npmScriptsPrefix: ({ results }) => {
+        if (!results.addNpmScriptsPrefix) return Promise.resolve('docs')
+
+        return text({
+          message: 'Prefix for VitePress npm scripts:',
+          placeholder: 'docs'
         })
+      }
     },
     {
       onCancel: () => {
@@ -101,7 +121,7 @@ export async function init(root: string | undefined) {
         process.exit(0)
       }
     }
-  )
+  )) as ScaffoldOptions
 
   outro(scaffold(options))
 }
@@ -112,7 +132,9 @@ export function scaffold({
   description = 'A VitePress Site',
   theme,
   useTs,
-  injectNpmScripts
+  injectNpmScripts,
+  addNpmScriptsPrefix = true,
+  npmScriptsPrefix = 'docs'
 }: ScaffoldOptions): string {
   const resolvedRoot = path.resolve(root)
   const templateDir = path.resolve(
@@ -200,15 +222,18 @@ export function scaffold({
   const tip = tips.length ? yellow([`\n\nTips:`, ...tips].join('\n- ')) : ``
 
   if (injectNpmScripts) {
-    const scripts = {
-      'docs:dev': `vitepress dev${dir}`,
-      'docs:build': `vitepress build${dir}`,
-      'docs:preview': `vitepress preview${dir}`
-    }
+    const scripts: Record<string, string> = {}
+
+    const prefix = addNpmScriptsPrefix ? `${npmScriptsPrefix}:` : ''
+
+    scripts[`${prefix}dev`] = `vitepress dev${dir}`
+    scripts[`${prefix}build`] = `vitepress build${dir}`
+    scripts[`${prefix}preview`] = `vitepress preview${dir}`
+
     Object.assign(userPkg.scripts || (userPkg.scripts = {}), scripts)
     fs.writeFileSync(pkgPath, JSON.stringify(userPkg, null, 2))
     return `Done! Now run ${cyan(
-      `${getPackageManger()} run docs:dev`
+      `${getPackageManger()} run ${prefix}dev`
     )} and start writing.${tip}`
   } else {
     const pm = getPackageManger()