From 33bef7bd93b8a120f05aad9cbbae494bb4cd9ee3 Mon Sep 17 00:00:00 2001
From: Hooray Hu <304327508@qq.com>
Date: Thu, 1 Jun 2023 23:46:17 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=20plop=20=E5=90=AF=E5=8A=A8?=
 =?UTF-8?q?=E8=84=9A=E6=9C=AC=EF=BC=8C=E8=87=AA=E5=8A=A8=E8=AF=BB=E5=8F=96?=
 =?UTF-8?q?=E6=A8=A1=E7=89=88=E7=9B=AE=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plopfile.js | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/plopfile.js b/plopfile.js
index 2103b176..4431083f 100644
--- a/plopfile.js
+++ b/plopfile.js
@@ -1,6 +1,8 @@
+const fs = require('node:fs')
+
 module.exports = function (plop) {
   plop.setWelcomeMessage('请选择需要创建的模式：')
-  plop.setGenerator('page', require('./plop-templates/page/prompt'))
-  plop.setGenerator('component', require('./plop-templates/component/prompt'))
-  plop.setGenerator('store', require('./plop-templates/store/prompt'))
+  fs.readdirSync('./plop-templates').forEach((item) => {
+    plop.setGenerator(item, require(`./plop-templates/${item}/prompt`))
+  })
 }