From cb0b012fac426c58024ded3c27fa915a77c451b4 Mon Sep 17 00:00:00 2001
From: Herrington Darkholme
 <2883231+HerringtonDarkholme@users.noreply.github.com>
Date: Sat, 25 Jul 2020 17:26:21 +0800
Subject: [PATCH] feat: support re-export story from vue

---
 .storybook/custom-presets.js        | 14 ++++++++++++++
 .storybook/export-loader.js         |  7 +++++++
 packages/badge/doc/index.stories.ts |  4 +---
 3 files changed, 22 insertions(+), 3 deletions(-)
 create mode 100644 .storybook/export-loader.js

diff --git a/.storybook/custom-presets.js b/.storybook/custom-presets.js
index 6627cc8630..493c312faa 100644
--- a/.storybook/custom-presets.js
+++ b/.storybook/custom-presets.js
@@ -13,6 +13,20 @@ function webpack(config) {
           loader: require.resolve('vue-loader'),
           options: {},
         },
+        {
+          test: /\.stories\.ts$/,
+          use: [
+              {
+                loader: require.resolve('./export-loader')
+              },
+              {
+                loader: require.resolve('ts-loader'),
+                options: {
+                  transpileOnly: true,
+                },
+              },
+          ]
+        },
         {
           test: /\.(ts|tsx)$/,
           use: [
diff --git a/.storybook/export-loader.js b/.storybook/export-loader.js
new file mode 100644
index 0000000000..381c8e1f16
--- /dev/null
+++ b/.storybook/export-loader.js
@@ -0,0 +1,7 @@
+module.exports = function (content) {
+  return content.replace(
+  /export ?{\s* default as (.+?)}(.+)/g,`
+import _$1 $2
+export const $1 = () => _$1
+`)
+}
diff --git a/packages/badge/doc/index.stories.ts b/packages/badge/doc/index.stories.ts
index a2cdcf529b..3e258d91ff 100644
--- a/packages/badge/doc/index.stories.ts
+++ b/packages/badge/doc/index.stories.ts
@@ -1,7 +1,5 @@
-import Basic from './basic.vue'
+export { default as BasicUsage} from './basic.vue'
 
 export default {
   title: "Badge",
 }
-
-export const BasicUsage = () => Basic