💥 feat: support umi@3 app (#106)

* 💥 feat: support umi@3 app * update snapshot
umijs · Apr 20, 2020 · cfb00c7 · cfb00c7
1 parent e4e9a00
commit cfb00c7
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 20 deletions.
diff --git a/cli.js b/cli.js
@@ -26,8 +26,12 @@ if (!semver.satisfies(process.version, '>= 8.0.0')) {
 const name = args._[0] || '';
 const { type } = args;
 delete args.type;
-run({
-  name,
-  type,
-  args,
-});
+
+(async () => {
+  await run({
+    name,
+    type,
+    args,
+  });
+  process.exit(0);
+})();
diff --git a/lib/generators/ant-design-pro/index.js b/lib/generators/ant-design-pro/index.js
@@ -40,17 +40,28 @@ class AntDesignProGenerator extends BasicGenerator {
       };
     } else {
       const prompts = [
+        {
+          name: 'nextPro',
+          type: 'confirm',
+          message: '🧙 Be the first to experience the new umi@3 ?',
+          default: false,
+          when: () => {
+            // 为了测试，没有任何用处
+            process.send && process.send({ type: 'prompt' });
+            process.emit('message', { type: 'prompt' });
+            return true;
+          },
+        },
         {
           name: 'language',
           type: 'list',
           message: '🤓 Which language do you want to use?',
           choices: ['TypeScript', 'JavaScript'],
           default: 'TypeScript',
-          filter: value => {
-            // 为了测试，没有任何用处
+          when: ({ nextPro }) => {
             process.send && process.send({ type: 'prompt' });
             process.emit('message', { type: 'prompt' });
-            return value;
+            return !nextPro;
           },
         },
         {
@@ -59,18 +70,20 @@ class AntDesignProGenerator extends BasicGenerator {
           message: '🚀 Do you need all the blocks or a simple scaffold?',
           choices: ['simple', 'complete'],
           default: 'simple',
-          filter: value => {
-            // 为了测试，没有任何用处
+          when: ({ nextPro }) => {
             process.send && process.send({ type: 'prompt' });
             process.emit('message', { type: 'prompt' });
-            return value;
+            return !nextPro;
           },
         },
         {
           name: 'nextAntd',
           type: 'confirm',
-          when: ({ allBlocks }) => {
-            return allBlocks === 'simple';
+          when: ({ allBlocks, nextPro }) => {
+            process.send && process.send({ type: 'prompt' });
+            process.emit('message', { type: 'prompt' });
+            console.log(nextPro);
+            return allBlocks === 'simple' && !nextPro;
           },
           message: '🦄 Time to use better, faster and latest antd@4!',
           default: true,
@@ -83,7 +96,7 @@ class AntDesignProGenerator extends BasicGenerator {
   }
 
   async writing() {
-    const { language, allBlocks, nextAntd } = this.prompts;
+    const { language = 'TypeScript', allBlocks, nextPro, nextAntd } = this.prompts;
 
     const isTypeScript = language === 'TypeScript';
     const projectName = this.opts.name || this.opts.env.cwd;
@@ -101,14 +114,18 @@ class AntDesignProGenerator extends BasicGenerator {
       gitArgs.push('--branch', 'antd@3');
     }
 
+    if (nextPro) {
+      gitArgs.push('--branch', 'v5');
+    }
+
     // all-blocks 分支上包含了所有的区块
     if (allBlocks === 'complete') {
       log(`🙈 complete mode can only use the version of antd@4`);
       gitArgs.push('--branch', 'all-blocks');
     }
 
     // Set branch if provided
-    if (this.opts.args.branch) {
+    if (this.opts.args.branch && !nextPro) {
       gitArgs.push('--branch', this.opts.args.branch);
     }
 

diff --git a/lib/run.js b/lib/run.js
@@ -67,7 +67,7 @@ const run = async config => {
 
   try {
     return runGenerator(`./generators/${type}`, config);
-  } catch(e) {
+  } catch (e) {
     console.error(chalk.red(`> Generate failed`), e);
     process.exit(1);
   }

diff --git a/test/__snapshots__/index.test.js.snap b/test/__snapshots__/index.test.js.snap
@@ -9,12 +9,15 @@ exports[`test umi-create test generate antd pro project from gitee.org 1`] = `
   block           - Create a umi block. 
   library         - Create a library with umi. 
   plugin          - Create a umi plugin. [41D[41C[2K[1A[2K[1A[2K[1A[2K[1A[2K[1A[2K[1A[2K[1A[2K[G? Select the boilerplate type ant-design-pro[44D[44C
+? 🧙 Be the first to experience the new umi@3 ? (y/N) [54D[54C[2K[G? 🧙 Be the first to experience the new umi@3 ? No[50D[50C
+
 ? 🤓 Which language do you want to use? (Use arrow keys)
 ❯ TypeScript 
   JavaScript [13D[13C[2K[1A[2K[1A[2K[G? 🤓 Which language do you want to use? TypeScript[50D[50C
 ? 🚀 Do you need all the blocks or a simple scaffold? (Use arrow keys)
 ❯ simple 
   complete [11D[11C[2K[1A[2K[1A[2K[G? 🚀 Do you need all the blocks or a simple scaffold? simple[60D[60C
+false
 ? 🦄 Time to use better, faster and latest antd@4! (Y/n) [57D[57C[2K[G? 🦄 Time to use better, faster and latest antd@4! Yes[54D[54C
 ❯ 🚚 clone success
 ❯ Clean up...
@@ -31,6 +34,8 @@ exports[`test umi-create test generate antd pro project from github 1`] = `
   block           - Create a umi block. 
   library         - Create a library with umi. 
   plugin          - Create a umi plugin. [41D[41C[2K[1A[2K[1A[2K[1A[2K[1A[2K[1A[2K[1A[2K[1A[2K[G? Select the boilerplate type ant-design-pro[44D[44C
+? 🧙 Be the first to experience the new umi@3 ? (y/N) [54D[54C[2K[G? 🧙 Be the first to experience the new umi@3 ? No[50D[50C
+
 ? 🤓 Which language do you want to use? (Use arrow keys)
 ❯ TypeScript 
   JavaScript [13D[13C[2K[1A[2K[1A[2K[G? 🤓 Which language do you want to use? 
@@ -39,6 +44,7 @@ exports[`test umi-create test generate antd pro project from github 1`] = `
 ? 🚀 Do you need all the blocks or a simple scaffold? (Use arrow keys)
 ❯ simple 
   complete [11D[11C[2K[1A[2K[1A[2K[G? 🚀 Do you need all the blocks or a simple scaffold? simple[60D[60C
+false
 ? 🦄 Time to use better, faster and latest antd@4! (Y/n) [57D[57C[2K[G? 🦄 Time to use better, faster and latest antd@4! Yes[54D[54C
 ❯ 🚚 clone success
 ❯ [Sylvanas] Prepare js environment...

diff --git a/test/index.test.js b/test/index.test.js
@@ -23,6 +23,8 @@ describe('test umi-create', () => {
       .beforeScript(path.join(fixtures, 'mock_github.js'))
       .waitForPrompt()
       .write('\n')
+      .write('\n')
+      .write('\n')
       .writeKey('DOWN', 'ENTER')
       .write('\n')
       .write('\n')
@@ -51,6 +53,8 @@ describe('test umi-create', () => {
       .write('\n')
       .write('\n')
       .write('\n')
+      .write('\n')
+      .write('\n')
       .end();
 
     expect(winEOL(response.stdout).replace(/>/g, '❯')).toMatchSnapshot();
@@ -78,6 +82,9 @@ describe('typescript', () => {
       .write('\n')
       .write('\n')
       .write('\n')
+      .write('\n')
+      .write('\n')
+      .write('\n')
       .end();
 
     expect(response.code).toBe(0);
@@ -102,13 +109,14 @@ describe('typescript', () => {
         .beforeScript(path.join(fixtures, 'mock_github.js'))
         .waitForPrompt()
         .write('\n')
+        .write('\n')
+        .write('\n')
         // ts
         .write('\n')
         // all-blocks
         .writeKey('DOWN', 'ENTER')
         .write('\n')
         .end();
-
       expect(response.code).toBe(0);
       expect(fs.existsSync(path.join(temp, '/src/pages/exception/403/index.tsx'))).toBeTruthy();
     });
@@ -127,14 +135,18 @@ describe('typescript', () => {
       })
         .beforeScript(path.join(fixtures, 'mock_github.js'))
         .waitForPrompt()
-        .write('\n')
+        .writeKey('ENTER')
+        .writeKey('ENTER')
+        .writeKey('ENTER')
         // js
         .writeKey('DOWN', 'ENTER')
         // all blocks
         .writeKey('DOWN', 'ENTER')
-        .write('\n')
+        .writeKey('ENTER')
+        .writeKey('ENTER')
         .end();
-
+      console.log(response);
+      console.log(temp);
       expect(response.code).toBe(0);
       expect(fs.existsSync(path.join(temp, 'tsconfig.json'))).toBeFalsy();
       expect(fs.existsSync(path.join(temp, '/src/pages/exception/403/index.jsx'))).toBeTruthy();