mm_os 1.3.3 → 1.3.5

package/middleware/cors/middleware.json

@@ -0,0 +1,7 @@
+{
+	"name": "web_cors",
+	"title": "web跨域请求",
+	"description": "用于跨域请求限制",
+	"version": "1.0",
+	"sort": 4
+}

package/middleware/mqtt_base/index.js

@@ -0,0 +1,10 @@
+const MQTT = require('mm_mqtt');
+
+/**
+ * 中间件
+ * @param {Object} server 服务
+ * @param {Object} config 配置参数
+ */
+module.exports = function(server, config) {
+	return server;
+};

package/middleware/mqtt_base/middleware.json

@@ -0,0 +1,8 @@
+{
+	"mode": "mqtt",
+	"name": "mqtt_base",
+	"title": "示例中间件",
+	"description": "描述该中间件的作用",
+	"version": "1.0",
+	"sort": 3
+}

package/{waf.js → middleware/waf/index.js}

@@ -36,19 +36,29 @@ function waf_check(url) {
 	return null;
 }
 
+function getClientIP(req) {
+	return req.headers['x-forwarded-for'] || req.headers['X-Forwarded-For'] ||
+		req.connection.remoteAddress ||
+		req.socket.remoteAddress ||
+		req.connection.socket.remoteAddress;
+};
+
 /**
  * 应用
  * @param {Object} server 服务
+ * @param {Object} config 配置参数
  */
-module.exports = function(server) {
+module.exports = function(server, config) {
 	/* WAF（web防火墙） */
 	server.use(async (ctx, next) => {
 		var url = ctx.url;
 		var danger = waf_check(url);
 		if (danger) {
-			$.log.warn('检测到攻击', "规则:", danger);
+			var ip = getClientIP(ctx.req);
+			$.log.warn(`检测到来自IP ${ip} 的攻击`, "规则:", danger.toString());
 		} else {
 			await next();
 		}
 	});
+	return server;
 };

package/middleware/waf/middleware.json

@@ -0,0 +1,9 @@
+{
+	"name": "web_waf",
+	"title": "web防火墙",
+	"description": "用于防止sql注入、脚本注入等",
+	"version": "1.0",
+	"type": "web",
+	"process_type": "common_before",
+	"sort": 3
+}

package/middleware/web_base/index.js

@@ -0,0 +1,67 @@
+const xmlParser = require("mm_xml");
+const session = require('mm_session');
+const compress = require('koa-compress');
+const koaBody = require('koa-body');
+const send = require('koa-send');
+
+/**
+ * 应用
+ * @param {Object} server 服务
+ * @param {Object} config 配置参数
+ */
+module.exports = function(server, config) {
+	/**
+	 * 发送文件
+	 */
+	server.use(async(ctx, next) => {
+		ctx.send = async function(src){
+			await send(ctx, src);
+		}
+		await next();
+	});
+	
+	// 设置session 保存时长2小时
+	server.use(session({
+		maxAge: 7200
+	}));
+
+	// 使用压缩
+	if (config.compress) {
+		server.use(
+			compress({
+				filter: function(content_type) {
+					// 只有在请求的content-type中有gzip类型，我们才会考虑压缩，因为zlib是压缩成gzip类型的
+					return /text/i.test(content_type);
+				},
+				// 阀值，当数据超过1kb的时候，可以压缩
+				threshold: 1024,
+				// zlib是node的压缩模块
+				flush: require('zlib').Z_SYNC_FLUSH
+			})
+		);
+	}
+	
+	// 解析 text/xml
+	server.use(xmlParser());
+	
+	var func = koaBody({
+		jsonLimit: config.jsonLimit || '20mb',
+		multipart: true,
+		formidable: {
+			// 设置上传文件大小最大限制，默认20M
+			maxFileSize: 2000 * 1024 * 1024
+		}
+	});
+	
+	// 解析 application/json、application/x-www-form-urlencoded、text/plain
+	// 接收主体
+	server.use(async(ctx, next) => {
+		if(!ctx.request.body){
+			await func(ctx, next);
+		}
+		else {
+			await next();
+		}
+	});
+	return server;
+};

package/middleware/web_base/middleware.json

@@ -0,0 +1,9 @@
+{
+	"name": "web_base",
+	"title": "web基本功能",
+	"description": "用于网站常规功能",
+	"version": "1.0",
+	"type": "web",
+	"process_type": "common_before",
+	"sort": 12
+}

package/middleware/web_event/index.js

@@ -0,0 +1,411 @@
+/**
+ * 拓展类
+ * @class
+ */
+class Event {
+	/**
+	 * 构造函数
+	 * @constructor
+	 * @param {Object} config 配置参数
+	 */
+	constructor(server, config) {
+		/**
+		 * 事件集合
+		 */
+		this.events = {
+			/**
+			 * 请求前
+			 */
+			before: [],
+			/**
+			 * 验证
+			 */
+			check: [],
+			/**
+			 * 主要
+			 */
+			main: [],
+			/**
+			 * 渲染
+			 */
+			render: [],
+			/**
+			 * 请求后
+			 */
+			after: []
+		};
+
+		/**
+		 * 路由集合
+		 */
+		this.routes = {
+			post: [],
+			get: []
+		};
+
+		$.push(server, this, true);
+
+		if (config.event && $.event_admin) {
+			// 创建一个API事件
+			var apis = $.event_admin('api', 'API事件');
+			apis.update();
+			for (var k in this.events) {
+				this.events[k] = apis["list_" + k];
+			}
+		};
+
+		/**
+		 * 路由前后事件
+		 */
+		server.use(async (ctx, next) => {
+			ctx.db = {};
+			var db = ctx.db;
+			db.ret = null;
+			if (ctx.path !== "/favicon.ico") {
+				var ret = await this.run_event(ctx, this.events.before, db);
+				if (!ret) {
+					await next();
+					ret = await this.run_event(ctx, this.events.after, db) || db.ret;
+					if (ret) {
+						ctx.body = ret;
+						ctx.status = ctx.status == 404 ? 200 : ctx.status;
+					}
+				} else {
+					ctx.body = ret;
+					ctx.status = ctx.status == 404 ? 200 : ctx.status;
+					next();
+				}
+			} else {
+				next();
+			}
+		});
+
+		return server;
+	}
+}
+
+/**
+ * 执行
+ * @param {Object} ctx 请求上下文
+ * @param {Function} next 跳过当前
+ * @param {Array} list 路由列表
+ * @param {Object} db 数据管理器
+ */
+Event.prototype.run_event = async function(ctx, list, db) {
+	try {
+		var path = ctx.path.toLocaleLowerCase();
+		var method = ctx.method;
+		for (var i = 0; i < list.length; i++) {
+			var o = list[i];
+			var cg = o.config;
+			if (cg.method == "ALL" || cg.method == method) {
+				if (cg.state === 1 && path.has(cg.target)) {
+					var ret = await o.run(ctx, db);
+					if (ret) {
+						db.ret = ret;
+						if (cg.end) {
+							break;
+						}
+					}
+				}
+			}
+		}
+	} catch (e) {
+		console.log(e);
+	}
+	return db.ret;
+};
+
+/**
+ * 执行
+ * @param {Object} ctx 请求上下文
+ * @param {Object} db 数据库管理器
+ * @param {Array} list 路由列表
+ */
+Event.prototype.run_route = async function(ctx, db, list) {
+	try {
+		if (list) {
+			var path = ctx.path.toLocaleLowerCase();
+			for (var i = 0; i < list.length; i++) {
+				var o = list[i];
+				var cg = o.config;
+				if (cg.state === 1 && path.has(cg.path)) {
+					var ret = await o.run(ctx, db);
+					if (ret) {
+						db.ret = ret;
+					}
+					if (cg.end) {
+						break;
+					}
+				}
+			}
+		}
+	} catch (e) {
+		console.log(e);
+	}
+	return db.ret;
+};
+
+/**
+ * 新建事件模型
+ * @param {Object} obj 附加值
+ * @return {Object} 返回模型
+ */
+Event.prototype.event_model = function(obj, run) {
+	return {
+		// 配置参数
+		config: Object.assign({
+			// 接受的请求方式
+			method: "ALL",
+			// 目标
+			target: "",
+			// 排序
+			sort: 100,
+			// 结束
+			end: true,
+		}, obj),
+		// 回调函数
+		run: run || function(ctx, db) {
+			return null;
+		}
+	};
+};
+
+/**
+ * 新建路由模型
+ * @param {Object} obj 附加值
+ * @return {Object} 返回模型
+ */
+Event.prototype.route_model = function(obj, run) {
+	return {
+		// 配置参数
+		config: Object.assign({
+			// 接受的请求方式
+			method: "GET",
+			// 请求路径
+			path: "",
+			// 排序
+			sort: 100,
+			// 结束
+			end: true,
+			// 缓存时长, 单位:分钟
+			cache: 0,
+			// 缓存方式 client客户端缓存, server服务端缓存
+			cache_method: "client server"
+		}, obj),
+		// 回调函数
+		run: run || function(ctx, db) {
+			return null;
+		}
+	};
+};
+
+/**
+ * 添加post路由
+ * @param {String} path 路由路径
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.post = function(path, run, sort = 100) {
+	this.routes["post"].push(this.route_model({
+		method: "POST",
+		path,
+		sort
+	}, run));
+	this.route_sort('post');
+};
+
+/**
+ * 添加get路由
+ * @param {String} path 路由路径
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.get = function(path, run, sort = 100) {
+	this.routes["get"].push(this.route_model({
+		method: "GET",
+		path,
+		sort
+	}, run));
+	this.route_sort('get');
+};
+
+/**
+ * 请求前
+ * @param {String} target 目标
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.before = function(target, run, sort = 100) {
+	this.events["before"].push(this.event_model({
+		end: false,
+		target,
+		sort
+	}, run));
+	this.event_sort('before');
+};
+
+/**
+ * 请求后
+ * @param {String} target 目标
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.check = function(target, run, sort = 100) {
+	this.events["check"].push(this.event_model({
+		target,
+		sort
+	}, run));
+	this.event_sort('check');
+};
+
+/**
+ * 主要
+ * @param {String} target 目标
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.main = function(target, run, sort = 100) {
+	this.events["main"].push(this.event_model({
+		end: false,
+		target,
+		sort
+	}, run));
+	this.event_sort('main');
+};
+
+/**
+ * 渲染
+ * @param {String} target 目标
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.render = function(target, run, sort = 100) {
+	this.events["render"].push(this.event_model({
+		target,
+		sort
+	}, run));
+	this.event_sort('render');
+};
+
+/**
+ * 请求后
+ * @param {String} target 目标
+ * @param {Function} func 回调函数
+ * @param {Number} sort 排列顺序
+ */
+Event.prototype.after = function(target, run, sort = 100) {
+	this.events["after"].push(this.event_model({
+		end: false,
+		target,
+		sort
+	}, run));
+	this.event_sort('after');
+};
+
+
+/**
+ * 路由排序方式
+ * @param {Object} obj_a 对象A
+ * @param {Object} obj_b 对象B
+ */
+Event.prototype.route_sort_way = function(obj_a, obj_b) {
+	var a = obj_a.config;
+	var b = obj_b.config;
+	var n = a.sort - b.sort;
+	if (n == 0) {
+		return b.path.length - a.path.length;
+	} else {
+		return n;
+	}
+};
+
+/**
+ * 路由排序
+ * @param {String} key 排序类型
+ */
+Event.prototype.route_sort = function(key) {
+	if (key) {
+		if (this.routes[key]) {
+			this.routes[key].sort(this.route_sort_way);
+		}
+	} else {
+		this.routes["post"].sort(this.route_sort_way);
+		this.routes["get"].sort(this.route_sort_way);
+	}
+};
+
+/**
+ * 事件排序方式
+ * @param {Object} obj_a 对象A
+ * @param {Object} obj_b 对象B
+ */
+Event.prototype.event_sort_way = function(obj_a, obj_b) {
+	var a = obj_a.config;
+	var b = obj_b.config;
+	var n = a.sort - b.sort;
+	if (n == 0) {
+		return b.target.length - a.target.length;
+	} else {
+		return n;
+	}
+};
+
+/**
+ * 事件排序
+ * @param {String} key 排序类型
+ */
+Event.prototype.event_sort = function(key) {
+	if (key) {
+		if (this.events[key]) {
+			this.events[key].sort(this.event_sort_way);
+		}
+	} else {
+		this.events["before"].sort(this.event_sort_way);
+		this.events["check"].sort(this.event_sort_way);
+		this.events["main"].sort(this.event_sort_way);
+		this.events["render"].sort(this.event_sort_way);
+		this.events["after"].sort(this.event_sort_way);
+	}
+};
+
+/**
+ * 对象转移
+ * @param {Array} lt 排序类型
+ * @param {Array} list 排序类型
+ * @param {String} paths 路径集合
+ * @param {String} key 主键
+ */
+Event.prototype.shift = function(lt, list, paths, key = "target") {
+	for (var i = list.length - 1; i >= 0; i--) {
+		var o = list[i];
+		var p = o.config[key];
+		for (var n = 0; n < paths.length; n++) {
+			if (p.has(paths[n])) {
+				lt.push(o);
+				list.splice(i, 1);
+			};
+		}
+	}
+};
+
+/**
+ * 排序
+ * @param {String} key 排序类型
+ */
+Event.prototype.sort = function() {
+	this.event_sort();
+	this.route_sort();
+};
+
+/**
+ * 应用
+ * @param {Object} server 服务
+ * @param {Object} config 配置参数
+ */
+module.exports = function(server, config) {
+	"./app".fullname().addDir();
+
+	server = new Event(server, config);
+	return server;
+};

package/middleware/web_event/middleware.json

@@ -0,0 +1,10 @@
+{
+	"name": "web_event",
+	"title": "web路由事件",
+	"description": "用于拓展事件API",
+	"version": "1.0",
+	"type": "web",
+	"process_type": "common_before",
+	"after": true,
+	"sort": 15
+}

package/middleware/web_proxy/index.js

@@ -0,0 +1,24 @@
+const {
+	proxy,
+	proxyTo,
+	isMatch
+} = require('mm_koa_proxy');
+
+/**
+ * 应用
+ * @param {Object} server 服务
+ * @param {Object} config 配置参数
+ */
+module.exports = function(server, config) {
+	if (config && config.proxy) {
+		var options = config.proxy;
+		if (options.targets) {
+			server.use(proxy(options, function(op, ctx, next) {
+				if (ctx.session && ctx.session.user) {
+					ctx.request.header['user_id'] = ctx.session.user.user_id;
+				}
+			}));
+		}
+	}
+	return server;
+};

package/middleware/web_proxy/middleware.json

@@ -0,0 +1,9 @@
+{
+	"name": "web_proxy",
+	"title": "web代理服务",
+	"description": "用于将特定路由转发到别的服务端上做处理",
+	"version": "1.0",
+	"type": "web",
+	"process_type": "worker",
+	"sort": 6
+}

package/middleware/web_router/index.js

@@ -0,0 +1,33 @@
+/**
+ * 应用
+ * @param {Object} server 服务
+ * @param {Object} config 配置参数
+ */
+module.exports = function(server, config) {
+	"./app".fullname().addDir();
+	
+	// 追加主程序
+	server.main("/*", async (ctx, db) => {
+		var md = ctx.method.toLocaleLowerCase();
+		var list = server.routes[md];
+		return await server.run_route(ctx, db, list);
+	});
+	
+	// 使用路由(主要)
+	server.use(async (ctx, next) => {
+		var db = ctx.db;
+		if (ctx.path !== "/favicon.ico") {
+			db.ret = await server.run_event(ctx, server.events.check, db);
+			if (!db.ret) {
+				db.ret = await server.run_event(ctx, server.events.main, db);
+			}
+			db.ret = await server.run_event(ctx, server.events.render, db) || db.ret;
+			if (db.ret) {
+				ctx.body = db.ret;
+			}
+		}
+		await next();
+	});
+	
+	return server;
+};

package/middleware/web_router/middleware.json

@@ -0,0 +1,10 @@
+{
+	"name": "web_router",
+	"title": "web路由拓展",
+	"description": "用于拓展web路由的基本功能和事件API",
+	"version": "1.0",
+	"type": "web",
+	"process_type": "common_after",
+	"after": true,
+	"sort": 21
+}

package/middleware/web_socket/index.js

@@ -0,0 +1,21 @@
+const websocket = require('koa-websocket');
+
+/**
+ * 应用
+ * @param {Object} server 服务
+ */
+module.exports = function(server, config) {
+	// 引入web socket通讯
+	server = websocket(server);
+	if (config.socket && $.Socket) {
+		// console.log("已启动websocket服务！");
+		//使用 websocket 服务
+		$.socket = new $.Socket();
+		$.socket.update();
+		server.ws.use((ctx, next) => {
+			$.socket.run(ctx, next);
+		});
+	}
+
+	return server;
+};

package/middleware/web_socket/middleware.json

@@ -0,0 +1,9 @@
+{
+	"name": "web_scoket",
+	"title": "webscoket通讯",
+	"description": "用于即时通讯",
+	"version": "1.0",
+	"type": "web",
+	"process_type": "master",
+	"sort": 18
+}

package/middleware/web_static/index.js

@@ -0,0 +1,26 @@
+const statics = require('mm_statics');
+
+/**
+ * 应用
+ * @param {Object} server 服务
+ * @param {Object} config 配置参数
+ */
+module.exports = function(server, config) {
+	// 处理静态文件
+	if (config.static) {
+		server.use(statics(
+			config.static_path.fullname(), {
+				maxAge: 60 * 60 * 24 * 7,
+				gzip: true,
+				brotli: true
+			}));
+		// 使用多路径静态文件处理器
+		if ($.Static) {
+			const Static = $.Static;
+			$.static = new Static();
+			$.static.update();
+			server.use($.static.run);
+		}
+	}
+	return server;
+};