mm_machine 2.3.4 → 2.3.6

package/drive.js

@@ -1,10 +1,17 @@
-const {
-  Mod
-} = require('./mod');
+const prettier = require('@prettier/sync');
 require('mm_tpl');
 let {
   conf
 } = require('mm_config');
+const {
+  Mod
+} = require('./mod');
+
+if (!$.prettyCode) {
+  $.prettyCode = function (code, options) {
+    return prettier.format(code, options);
+  };
+}
 
 /**
  * 驱动基础类
@@ -95,6 +102,7 @@ Drive.prototype.loadConfig = function (file) {
 /**
  * 设置配置
  * @param {object} config 配置对象
+ * @returns {object} 配置项
  */
 Drive.prototype.setConfig = function (config) {
   if (config) {
@@ -153,27 +161,13 @@ Drive.prototype._getTplDir = function () {
   return this.getParent().getTplDir();
 };
 
-/**
- * 创建脚本
- * @param {string} file 文件路径
- * @param {object} model 模板数据
- * @param {string} tpl_dir 模板文件路径
- * @returns {boolean} 创建是否成功
- */
-Drive.prototype._createScriptFile = function (file, model = {}, tpl_dir = '') {
-  var tpl = this.getScriptTpl(tpl_dir);
-  var content = $.tpl.render(tpl, model);
-  file.saveText(content);
-  return file;
-};
-
 /**
  * 获取脚本模板
  * @param {string} tpl_dir 模板文件路径
  * @returns {string} 脚本模板内容
  */
-Drive.prototype.getScriptTpl = function (tpl_dir = '') {
-  var f = './script.tpl.js'.fullname(tpl_dir || this._getTplDir());
+Drive.prototype._getScriptTpl = function () {
+  var f = './script.tpl.js'.fullname(this._getTplDir());
   if (f.hasFile()) {
     return f.loadText();
   } else {
@@ -201,7 +195,7 @@ Drive.prototype.getScriptTpl = function (tpl_dir = '') {
 };
 
 /**
- * 移除脚本
+ * 移除加载的脚本模块
  */
 Drive.prototype._remove = function () {
   let file = this._getScriptFile();
@@ -220,7 +214,7 @@ Drive.prototype._remove = function () {
       }
     }
   } catch (err) {
-    this.log('error', `移除脚本失败: `, err);
+    this.log('error', `移除加载的脚本模块失败: `, err);
   }
 };
 
@@ -255,7 +249,6 @@ Drive.prototype.unloadCore = async function (remove) {
 
 /**
  * 获取脚本文件路径
- * @param {string} file 文件路径
  * @returns {string | null} 完整文件路径
  */
 Drive.prototype._getScriptFile = function () {
@@ -266,10 +259,67 @@ Drive.prototype._getScriptFile = function () {
   if (!filename.hasFile()) {
     this.newScript(filename);
   }
-
   return filename;
 };
 
+/**
+ * 新建脚本文件
+ * @param {string} file 目标文件路径
+ */
+Drive.prototype.newScript = function (file) {
+  this._createScriptFile(file, this.config);
+};
+
+/**
+ * 创建脚本
+ * @param {string} file 文件路径
+ * @param {object} model 模板数据
+ * @returns {boolean} 创建是否成功
+ */
+Drive.prototype._createScriptFile = function (file, model = {}) {
+  var tpl = this._getScriptTpl();
+  var content = $.tpl.render(tpl, model);
+  if (!content) {
+    return '';
+  }
+  let code = this.prettyCode(content, 'js');
+  file.saveText(code);
+  return file;
+};
+
+/**
+ * 获取脚本模板
+ * @param {string} tpl_dir 模板文件路径
+ * @returns {string} 脚本模板内容
+ */
+Drive.prototype._getScriptTpl = function () {
+  var f = './script.tpl.js'.fullname(this._getTplDir());
+  if (f.hasFile()) {
+    return f.loadText();
+  } else {
+    return `module.exports = {
+  /**
+   * 初始化
+   */
+  async _init() {
+    this.log('debug', \`初始化！\`);
+  },
+  /**
+   * 销毁
+   */
+  async _destroy(...args) {
+    this.log('debug', \`销毁！\`);
+  },
+  /**
+   * 主要逻辑
+   */
+  async main(...args) {
+    // 主要代码写在这
+  }
+}`;
+  }
+};
+
 /**
  * 加载脚本文件，支持热重载模式
  * 在开发模式(mode=3或4)下使用 $.require 进行热重载
@@ -333,25 +383,53 @@ Drive.prototype.newConfig = function (file) {
   this._createConfigFile(file);
 };
 
+/**
+ * 预试代码
+ * @param {string} code 要预试的代码
+ * @param {string} type 预试类型，可选"json"或"js"
+ * @returns {string} 预试后的代码字符串
+ */
+Drive.prototype.prettyCode = function (code, type = 'json5') {
+  let options = {
+    tabWidth: 2
+  };
+  switch (type) {
+    case 'json5':
+      options.parser = 'json5';
+      break;
+    case 'js':
+    case 'javascript':
+      options.parser = 'babel';
+      options.semi = true;           // 在语句末尾添加分号
+      options.singleQuote = true;     // 使用单引号
+      break;
+    default:
+      options.parser = 'json';
+      break;
+  }
+  let new_code = $.prettyCode(code, options);
+  return new_code;
+};
+
 /**
  * 创建配置文件
  * @param {string} file 目标文件路径
  * @returns {string} 配置文件内容
  */
 Drive.prototype._createConfigFile = function (file) {
-  var tpl = this.getConfigTpl();
+  var tpl = this._getConfigTpl();
   var content = $.tpl.render(tpl, this.config);
-  file.saveText(content);
+  let code = this.prettyCode(content, 'json');
+  file.saveText(code);
   return file;
 };
 
 /**
  * 获取配置模板
- * @param {string} tpl_dir 模板文件路径
  * @returns {string} 配置模板内容
  */
-Drive.prototype.getConfigTpl = function (tpl_dir = '') {
-  var f = './config.tpl.json'.fullname(tpl_dir || this._getTplDir());
+Drive.prototype._getConfigTpl = function () {
+  var f = './config.tpl.json'.fullname(this._getTplDir());
   if (f.hasFile()) {
     return f.loadText();
   } else {
@@ -370,8 +448,8 @@ Drive.prototype.getConfigTpl = function (tpl_dir = '') {
   "state": \${state || 1},
   // 模块排序，可选，默认100
   "sort": \${sort || 100},
-	// 是否结束，可选，默认false
-	"end": \${end || false}
+  // 是否结束，可选，默认false
+  "end": \${end || false}
 }`;
   }
 };
@@ -543,8 +621,8 @@ Drive.prototype.loadBefore = async function () {
  * 加载处理
  */
 Drive.prototype._loadCore = async function () {
-  this.loadConfig();
-  this.loadScript();
+  await this.loadConfig();
+  await this.loadScript();
 };
 
 /**
@@ -569,13 +647,13 @@ Drive.prototype.save = function () {
           // 如果不存在，添加新配置项
           config.push(this.config);
         }
-        file.saveText(JSON.stringify(config, null, 4));
+        file.saveText(JSON.stringify(config, null, 2));
         return;
       }
     }
 
     // 单对象格式配置，直接保存
-    file.saveText(JSON.stringify(this.config, null, 4));
+    file.saveText(JSON.stringify(this.config, null, 2));
   } catch (err) {
     this.log('error', `保存配置失败: `, err);
   }

package/index.js

@@ -45,6 +45,11 @@ class Manager extends Mod {
      * @type {string}
      */
     dir: './ai/manager'.fullname(),
+    /**
+     * 搜索模式 dir按目录搜索 | file按文件名搜索
+     * @type {string}
+     */
+    search_way: 'file',
     /**
      * 类型 script脚本 | json配置 | drive驱动
      * @type {string}
@@ -71,7 +76,7 @@ class Manager extends Mod {
      */
     sort_key: 'sort',
     /**
-     * 作用域（同时作为检索的后缀名）
+     * 作用域
      * @type {string}
      */
     scope: ($.val && $.val.scope) ? $.val.scope + '' : 'server'
@@ -366,6 +371,37 @@ Manager.prototype.loads = async function () {
   await Promise.all(promises);
 };
 
+/**
+ * 获取目录名
+ * @returns {string} 目录名
+ */
+Manager.prototype._getDirName = function () {
+  // 前缀，为当前类名
+  let prefix = this.constructor.name.toLowerCase();
+  // 后缀，为配置项名称
+  let name = this.config.name.toLowerCase();
+  // 目录名，为前缀加下划线加后缀
+  return prefix + '_' + name;
+};
+
+/**
+ * 获取所有目录下的文件
+ * @param {string} dir 目录路径
+ * @returns {Array} 文件路径数组
+ */
+Manager.prototype._getAllDirFiles = function (dir) {
+  let files = [];
+  let dirs = $.dir.getAll(dir, this._getDirName());
+  for (let i = 0; i < dirs.length; i++) {
+    let d = dirs[i];
+    let lt = $.file.getAll(d, this.config.filename);
+    if (lt.length > 0) {
+      files.push(...lt);
+    }
+  }
+  return files;
+};
+
 /**
  * 检索目录下所有文件
  * @private
@@ -383,6 +419,9 @@ Manager.prototype._findFiles = function (dir) {
     this.log('warn', '目录不存在', dir);
     return [];
   }
+  if (this.config.search_way === 'dir') {
+    return this._getAllDirFiles(dir);
+  }
   // 检索目录下所有文件
   return $.file.getAll(dir, this.config.filename);
 };
@@ -425,7 +464,7 @@ Manager.prototype.update = async function (dir, clear = true) {
   }
 
   // 查找所有json文件
-  let files = this._findFiles(dir);
+  let files = this._findFiles(dir || this.config.dir);
   // 批量注册模块
   this.batchRegister(files);
   // 更新配置信息
@@ -553,7 +592,7 @@ Manager.prototype._runSub = async function (mod, method, ...params) {
 
     return ret;
   } catch (err) {
-    $.log.error(`[${this.type}] 执行模块方法失败: `, err);
+    $.log.error(`执行模块方法失败: `, err);
     return null;
   }
 };

package/mod.js

@@ -142,10 +142,8 @@ Mod.prototype._onDestroy = function () {
  * @param {object} ctx 上下文对象
  */
 Mod.prototype._onLoad = async function (ctx) {
-  let status_last = 'created';
   // 1. 加载
   this.on('load:before', async (ctx) => {
-    status_last = this.getState();
     this.setState('loading');
     await this._loadCore(...ctx.params);
   });
@@ -222,13 +220,19 @@ Mod.prototype._onUnload = async function (ctx) {
 };
 
 /**
- * 初始化监听
+ * 初始化监听事件
+ * @param {object} ctx 上下文对象
  */
 Mod.prototype._onDestroy = async function (ctx) {
   let status_last = 'unloaded';
+  // 6. 销毁
+  this.on('destroy:before', async (ctx) => {
+    status_last = this.getState();
+    this.setState('destroying');
+  });
   this.on('destroy:check', (ctx) => {
     if (status_last !== 'unloaded') {
-      status_last = this.getState();
+      this.setState(status_last);
       ctx.error = new Error('销毁前必须先卸载');
     }
   });
@@ -354,7 +358,7 @@ Mod.prototype.main = async function (...args) {
     throw new Error(`方法${method}不存在`);
   }
   let params = args.slice(1);
-  return await this.methods[method](...params); 
+  return await this.methods[method](...params);
 };
 
 /**
@@ -371,7 +375,8 @@ Mod.prototype.setMethods = function (methods) {
 
 /**
  * 设置脚本方法
- * @param {object} methods 脚本方法对象
+ * @param {string} method 方法名
+ * @param {Function} func 方法函数
  */
 Mod.prototype.setMethod = function (method, func) {
   if (!func) throw new Error(`函数(func)不能为空`);
@@ -448,9 +453,9 @@ Mod.prototype._callBefore = async function (method, params) {
  * @returns {Promise<any>} 主方法结果
  */
 Mod.prototype._callMain = async function (method_func, params, result) {
-  let method_result = await method_func.call(this._mod || this, ...params);
+  let ret = await method_func.call(this._mod || this, ...params);
   // 修复：如果主方法返回undefined，则使用前置钩子的结果
-  return method_result !== undefined ? method_result : result;
+  return ret !== undefined ? ret : result;
 };
 
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mm_machine",
-  "version": "2.3.4",
+  "version": "2.3.6",
   "description": "A flexible Node.js plugin mechanism system for dynamic loading, management and execution of modules. Supports hot reload, lifecycle management, and modern JavaScript features.",
   "main": "index.js",
   "scripts": {
@@ -33,6 +33,7 @@
     "README.md"
   ],
   "dependencies": {
+    "@prettier/sync": "^0.6.1",
     "mm_config": "^2.3.0",
     "mm_hot_reload": "^1.3.0",
     "mm_tpl": "^2.4.9"