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`))
+  })
 }