feat: generator modify scripts

web-infra-dev · Sep 20, 2023 · 610a747 · 610a747
1 parent 31bb519
commit 610a747
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/packages/generator/generators/storybook-generator-next/src/index.ts b/packages/generator/generators/storybook-generator-next/src/index.ts
@@ -1,7 +1,7 @@
 import path from 'path';
 import { GeneratorContext, GeneratorCore } from '@modern-js/codesmith';
 import { AppAPI } from '@modern-js/codesmith-api-app';
-// import { JsonAPI } from '@modern-js/codesmith-api-json';
+import { JsonAPI } from '@modern-js/codesmith-api-json';
 import {
   isTsProject,
   getPackageManager,
@@ -132,6 +132,22 @@ const handleTemplateFile = async (
       : {}),
   };
 
+  const jsonAPI = new JsonAPI(generator);
+  await jsonAPI.update(
+    context.materials.default.get(path.join(appDir, './package.json')),
+    {
+      query: {},
+      update: {
+        $set: {
+          scripts: {
+            'build-storybook': 'storybook build',
+            storybook: 'storybook dev -p 6006',
+          },
+        },
+      },
+    },
+  );
+
   await appApi.runSubGenerator(
     getGeneratorPath(DependenceGenerator, context.config.distTag),
     undefined,