@pixui-dev/pxw 0.1.20 → 0.1.21

package/config/pfbs.js

@@ -1,245 +1,245 @@
-let fs = require('fs');
-let path = require('path');
-let cp = require('child_process');
-const { isHeritageClause } = require('typescript');
-
-function isBufferHeaderMatch(buffer, pattern, n = pattern.length) {
-	// 确保输入是 Buffer 类型
-	if (!Buffer.isBuffer(buffer) || !Buffer.isBuffer(pattern)) {
-		throw new TypeError('Both buffer and pattern must be of type Buffer');
-	}
-
-	// 如果 n 超过 buffer 或 pattern 的长度，返回 false
-	if (n > buffer.length || n > pattern.length) {
-		return false;
-	}
-
-	// 比较前 n 个字节
-	for (let i = 0; i < n; i++) {
-		if (buffer[i] !== pattern[i]) {
-			return false;
-		}
-	}
-
-	return true;
-}
-
-let suffix = '';
-if (process.platform == 'win32') suffix = '.exe';
-if (process.platform == 'linux') suffix = '-ubuntu';
-let pfbsLatest = path.resolve(process.cwd(), `.cache/pfbs${suffix}`);
-let pfbsConvert = async (ver, isJS, input, fn) => {
-	let pfbs = pfbsLatest;
-	if (ver[0] == '0') pfbs = pfbsLatest.replace('pfbs', 'pfbs-0.4');
-	console.log('use pfbs', pfbs);
-	if (!fs.existsSync(pfbs)) {
-		let url = process.env.PFBS_DOWNLOAD_SITE || 'http://9.134.150.252:8090/dev/tools/pfbs/';
-		url += path.basename(pfbs);
-		try {
-			let buffer;
-			if (typeof fetch !== 'undefined') {
-				let res = await fetch(url);
-				buffer = await res.arrayBuffer();
-			} 
-			else if (typeof XMLHttpRequest !== 'undefined') {
-				buffer = await new Promise((resolve, reject) => {
-					const xhr = new XMLHttpRequest();
-					xhr.open('GET', url, true);
-					xhr.responseType = 'arraybuffer';
-					
-					xhr.onload = () => {
-						if (xhr.status >= 200 && xhr.status < 300) {
-							resolve(xhr.response);
-						} else {
-							reject(new Error(`XHR failed with status ${xhr.status}`));
-						}
-					};
-					
-					xhr.onerror = () => reject(new Error('Network Error'));
-					xhr.send();
-				});
-			}else if (typeof require !== 'undefined' && require('http')) {
-				const http = require('http');
-				console.log('http: ', url)
-				buffer = await new Promise((resolve, reject) => {
-					http.get(url, (res) => {
-						if (res.statusCode >= 300 && res.statusCode < 400 && res.headers.location) {
-							return resolve(this._fetchBuffer(res.headers.location));
-						}
-		
-						const chunks = [];
-						res.on('data', (chunk) => chunks.push(chunk));
-						res.on('end', () => resolve(Buffer.concat(chunks)));
-					}).on('error', reject);
-				});
-			}else {
-				throw new Error('No supported HTTP client available');
-			}
-
-			if (!fs.existsSync(path.dirname(pfbs))) fs.mkdirSync(path.dirname(pfbs));
-			fs.writeFileSync(pfbs, new Uint8Array(buffer));
-			fs.chmodSync(pfbs, '755');
-		} catch (e) {
-			console.error('Request failed:', e);
-			fn(input);
-			return;
-		}
-	}
-	let buffers = [];
-	let addBuffer = (chunk, encoding) => {
-		if (chunk === undefined) return;
-		if (typeof chunk === 'string') {
-			chunk = Buffer.from(chunk, encoding);
-		}
-		buffers.push(chunk);
-	};
-	let opts = isJS ? ['--js'] : [];
-	let child = cp.spawn(pfbs, ['--src', 'stdin', ...opts], {
-		windowsHide: true,
-	});
-	child.stdout.on('data', (data) => {
-		// console.log('get data from fbs', data.byteLength);
-		addBuffer(data);
-	});
-	child.stdout.on('close', () => {
-		let output = Buffer.concat(buffers);
-		// console.log('get data from fbs finish!!', output.length);
-		fn(output);
-	});
-	child.stderr.on('data', (data) => {
-		console.log('get err from pfbs', data.toString());
-	});
-	child.stdin.write(input);
-	child.stdin.end();
-};
-
-module.exports.getMiddleware = () => {
-	let log = () => {};
-	log = console.log;
-	//拦截输出，当agent为PixUI且输出类型为html时，转换为fbs二进制数据再返回
-	return (req, res, next) => {
-		//去掉缓存标记，因为很可能源文件没变，但转换后的表示变了
-		delete req.headers['if-none-match'];
-		let ua = req.get('User-Agent');
-		let ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
-		let disablePFBS = req.headers['x-disable-http-pfbs'] == 'true';
-		log('req', req.url, ua, ip, `disablePFBS:${disablePFBS}`);
-		// log( JSON.stringify(req.headers));
-
-		let _write = res.write;
-		let _end = res.end;
-		let _send = res.send;
-
-		let buffers = [];
-		let addBuffer = (chunk, encoding) => {
-			if (chunk === undefined) return;
-			if (typeof chunk === 'string') {
-				chunk = Buffer.from(chunk, encoding);
-			}
-			buffers.push(chunk);
-		};
-
-		let guessMime = (url) => {
-			if (url.endsWith('.js')) return 'text/javascript';
-			if (url.endsWith('.html')) return 'text/html';
-			return 'text/plain';
-		};
-
-		let mime = '';
-		let _setHeader = res.setHeader;
-		res.setHeader = function (k, v) {
-			console.log('hook set', k, v);
-			if (k == 'Content-Length') return; //不要显式设置长度，pfbs转换后可能会变小，导致客户端空等
-			if (k == 'Content-Type') mime = v;
-			_setHeader.call(res, k, v);
-		};
-
-		//处理直接裸写的情况，一般见于发送静态文件
-		res.writeLog = 0;
-		res.write = function (chunk, encoding) {
-			let type = res.get('Content-Type');
-			if (++res.writeLog <= 10) {
-				log('hook write...', type, chunk.length);
-				if (res.writeLog == 10) {
-					log('hook write is too many, ignore next log');
-				}
-			}
-			addBuffer(chunk, encoding);
-		};
-		const pattern1 = Buffer.from([0x50, 0x49, 0x58, 0x32, 0x43, 0x42, 0x4a, 0x51]);
-		const pattern2 = Buffer.from([0x10, 0, 0, 0, 0x50, 0x49, 0x58, 0x30]);
-		res.end = function (chunk, encoding) {
-			res.write = _write;
-			res.end = _end;
-			let type = res.get('Content-Type') || mime;
-			if (chunk) addBuffer(chunk, encoding);
-			let data = Buffer.concat(buffers);
-
-			let isAlreadyPfbs = isBufferHeaderMatch(data, pattern1) || isBufferHeaderMatch(data, pattern2);
-
-			log('hook end...', type, data.length, req.url, 'isAlreadyPfbs', isAlreadyPfbs);
-
-			let isPixUI = /PixUI/.test(ua);
-			let isHTML = type.indexOf('text/html') >= 0;
-			let isJS = type.indexOf('javascript') >= 0;
-			let isCSS = type.indexOf('text/css') >= 0;
-			if (isHTML && !isAlreadyPfbs) {
-				let c1 = data.toString('utf-8');
-				let c2 = c1.replace(
-					'</body>',
-					`<script>
-(function() {
-	var ws = new WebSocket('ws://' + window.location.host+ '/notify?reloadMode=RELOAD_ON_SAVE&referrer=' + encodeURIComponent(window.location.pathname))
-	ws.onmessage = function (msg) {
-		if (msg.data === 'reload') {
-			window.location.reload();
-		}
-	};
-})();
-			</script></body>`,
-				);
-				console.log('insert hotreload to html', data.length, c1.length, c2.length);
-				// console.log(c1);
-				data = c2;
-			}
-			if (isPixUI && !isAlreadyPfbs && !disablePFBS && (isHTML || isJS)) {
-				let ver = /(\d)\.(\d)\.(\d+)/.exec(ua); // 0.2.*以上
-				pfbsConvert(ver.splice(1), isJS, data, (out) => {
-					console.log('pfbsconvert>>', req.url, data.length, out.length);
-					res.write = _write;
-					res.end = _end;
-					_end.call(this, out);
-				});
-			}
-			else {
-				console.log('directly call end...', data.length, type, _end);
-				_end.call(this, data);
-			}
-		};
-
-		// 处理通过send发送的情况，如webpack合成的html
-		res.send = function (c) {
-			let isPixUI = /PixUI/.test(ua);
-			let type = res.get('Content-Type') || 'text/plain';
-			let isHTML = type.indexOf('text/html') >= 0;
-			let isJS = type.indexOf('javascript') >= 0;
-			let isCSS = type.indexOf('text/css') >= 0;
-			if (isPixUI && !disablePFBS && (isHTML || isJS)) {
-				log('hook send...', c.length, type);
-				let ver = /(\d)\.(\d)\.(\d+)/.exec(ua); // 0.2.*以上
-				pfbsConvert(ver.splice(1), isJS, c, (out) => {
-					res.send = _send;
-					res.end = _end;
-					res.setHeader = _setHeader;
-					res.setHeader('Content-Type', 'application/octet-stream');
-					_send.call(res, out);
-				});
-			}
-			else {
-				console.log('directly call send...', c.length, type, _send);
-				_send.call(this, c);
-			}
-		};
-		next();
-	};
-};
+let fs = require('fs');
+let path = require('path');
+let cp = require('child_process');
+const { isHeritageClause } = require('typescript');
+
+function isBufferHeaderMatch(buffer, pattern, n = pattern.length) {
+	// 确保输入是 Buffer 类型
+	if (!Buffer.isBuffer(buffer) || !Buffer.isBuffer(pattern)) {
+		throw new TypeError('Both buffer and pattern must be of type Buffer');
+	}
+
+	// 如果 n 超过 buffer 或 pattern 的长度，返回 false
+	if (n > buffer.length || n > pattern.length) {
+		return false;
+	}
+
+	// 比较前 n 个字节
+	for (let i = 0; i < n; i++) {
+		if (buffer[i] !== pattern[i]) {
+			return false;
+		}
+	}
+
+	return true;
+}
+
+let suffix = '';
+if (process.platform == 'win32') suffix = '.exe';
+if (process.platform == 'linux') suffix = '-ubuntu';
+let pfbsLatest = path.resolve(process.cwd(), `.cache/pfbs${suffix}`);
+let pfbsConvert = async (ver, isJS, input, fn) => {
+	let pfbs = pfbsLatest;
+	if (ver[0] == '0') pfbs = pfbsLatest.replace('pfbs', 'pfbs-0.4');
+	console.log('use pfbs', pfbs);
+	if (!fs.existsSync(pfbs)) {
+		let url = process.env.PFBS_DOWNLOAD_SITE || 'http://9.134.150.252:8090/dev/tools/pfbs/';
+		url += path.basename(pfbs);
+		try {
+			let buffer;
+			if (typeof fetch !== 'undefined') {
+				let res = await fetch(url);
+				buffer = await res.arrayBuffer();
+			} 
+			else if (typeof XMLHttpRequest !== 'undefined') {
+				buffer = await new Promise((resolve, reject) => {
+					const xhr = new XMLHttpRequest();
+					xhr.open('GET', url, true);
+					xhr.responseType = 'arraybuffer';
+					
+					xhr.onload = () => {
+						if (xhr.status >= 200 && xhr.status < 300) {
+							resolve(xhr.response);
+						} else {
+							reject(new Error(`XHR failed with status ${xhr.status}`));
+						}
+					};
+					
+					xhr.onerror = () => reject(new Error('Network Error'));
+					xhr.send();
+				});
+			}else if (typeof require !== 'undefined' && require('http')) {
+				const http = require('http');
+				console.log('http: ', url)
+				buffer = await new Promise((resolve, reject) => {
+					http.get(url, (res) => {
+						if (res.statusCode >= 300 && res.statusCode < 400 && res.headers.location) {
+							return resolve(this._fetchBuffer(res.headers.location));
+						}
+		
+						const chunks = [];
+						res.on('data', (chunk) => chunks.push(chunk));
+						res.on('end', () => resolve(Buffer.concat(chunks)));
+					}).on('error', reject);
+				});
+			}else {
+				throw new Error('No supported HTTP client available');
+			}
+
+			if (!fs.existsSync(path.dirname(pfbs))) fs.mkdirSync(path.dirname(pfbs));
+			fs.writeFileSync(pfbs, new Uint8Array(buffer));
+			fs.chmodSync(pfbs, '755');
+		} catch (e) {
+			console.error('Request failed:', e);
+			fn(input);
+			return;
+		}
+	}
+	let buffers = [];
+	let addBuffer = (chunk, encoding) => {
+		if (chunk === undefined) return;
+		if (typeof chunk === 'string') {
+			chunk = Buffer.from(chunk, encoding);
+		}
+		buffers.push(chunk);
+	};
+	let opts = isJS ? ['--js'] : [];
+	let child = cp.spawn(pfbs, ['--src', 'stdin', ...opts], {
+		windowsHide: true,
+	});
+	child.stdout.on('data', (data) => {
+		// console.log('get data from fbs', data.byteLength);
+		addBuffer(data);
+	});
+	child.stdout.on('close', () => {
+		let output = Buffer.concat(buffers);
+		// console.log('get data from fbs finish!!', output.length);
+		fn(output);
+	});
+	child.stderr.on('data', (data) => {
+		console.log('get err from pfbs', data.toString());
+	});
+	child.stdin.write(input);
+	child.stdin.end();
+};
+
+module.exports.getMiddleware = () => {
+	let log = () => {};
+	log = console.log;
+	//拦截输出，当agent为PixUI且输出类型为html时，转换为fbs二进制数据再返回
+	return (req, res, next) => {
+		//去掉缓存标记，因为很可能源文件没变，但转换后的表示变了
+		delete req.headers['if-none-match'];
+		let ua = req.get('User-Agent');
+		let ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
+		let disablePFBS = req.headers['x-disable-http-pfbs'] == 'true';
+		log('req', req.url, ua, ip, `disablePFBS:${disablePFBS}`);
+		// log( JSON.stringify(req.headers));
+
+		let _write = res.write;
+		let _end = res.end;
+		let _send = res.send;
+
+		let buffers = [];
+		let addBuffer = (chunk, encoding) => {
+			if (chunk === undefined) return;
+			if (typeof chunk === 'string') {
+				chunk = Buffer.from(chunk, encoding);
+			}
+			buffers.push(chunk);
+		};
+
+		let guessMime = (url) => {
+			if (url.endsWith('.js')) return 'text/javascript';
+			if (url.endsWith('.html')) return 'text/html';
+			return 'text/plain';
+		};
+
+		let mime = '';
+		let _setHeader = res.setHeader;
+		res.setHeader = function (k, v) {
+			console.log('hook set', k, v);
+			if (k == 'Content-Length') return; //不要显式设置长度，pfbs转换后可能会变小，导致客户端空等
+			if (k == 'Content-Type') mime = v;
+			_setHeader.call(res, k, v);
+		};
+
+		//处理直接裸写的情况，一般见于发送静态文件
+		res.writeLog = 0;
+		res.write = function (chunk, encoding) {
+			let type = res.get('Content-Type');
+			if (++res.writeLog <= 10) {
+				log('hook write...', type, chunk.length);
+				if (res.writeLog == 10) {
+					log('hook write is too many, ignore next log');
+				}
+			}
+			addBuffer(chunk, encoding);
+		};
+		const pattern1 = Buffer.from([0x50, 0x49, 0x58, 0x32, 0x43, 0x42, 0x4a, 0x51]);
+		const pattern2 = Buffer.from([0x10, 0, 0, 0, 0x50, 0x49, 0x58, 0x30]);
+		res.end = function (chunk, encoding) {
+			res.write = _write;
+			res.end = _end;
+			let type = res.get('Content-Type') || mime;
+			if (chunk) addBuffer(chunk, encoding);
+			let data = Buffer.concat(buffers);
+
+			let isAlreadyPfbs = isBufferHeaderMatch(data, pattern1) || isBufferHeaderMatch(data, pattern2);
+
+			log('hook end...', type, data.length, req.url, 'isAlreadyPfbs', isAlreadyPfbs);
+
+			let isPixUI = /PixUI/.test(ua);
+			let isHTML = type.indexOf('text/html') >= 0;
+			let isJS = type.indexOf('javascript') >= 0;
+			let isCSS = type.indexOf('text/css') >= 0;
+			if (isHTML && !isAlreadyPfbs) {
+				let c1 = data.toString('utf-8');
+				let c2 = c1.replace(
+					'</body>',
+					`<script>
+(function() {
+	var ws = new WebSocket('ws://' + window.location.host+ '/notify?reloadMode=RELOAD_ON_SAVE&referrer=' + encodeURIComponent(window.location.pathname))
+	ws.onmessage = function (msg) {
+		if (msg.data === 'reload') {
+			window.location.reload();
+		}
+	};
+})();
+			</script></body>`,
+				);
+				console.log('insert hotreload to html', data.length, c1.length, c2.length);
+				// console.log(c1);
+				data = c2;
+			}
+			if (isPixUI && !isAlreadyPfbs && !disablePFBS && (isHTML || isJS)) {
+				let ver = /(\d)\.(\d)\.(\d+)/.exec(ua); // 0.2.*以上
+				pfbsConvert(ver.splice(1), isJS, data, (out) => {
+					console.log('pfbsconvert>>', req.url, data.length, out.length);
+					res.write = _write;
+					res.end = _end;
+					_end.call(this, out);
+				});
+			}
+			else {
+				console.log('directly call end...', data.length, type, _end);
+				_end.call(this, data);
+			}
+		};
+
+		// 处理通过send发送的情况，如webpack合成的html
+		res.send = function (c) {
+			let isPixUI = /PixUI/.test(ua);
+			let type = res.get('Content-Type') || 'text/plain';
+			let isHTML = type.indexOf('text/html') >= 0;
+			let isJS = type.indexOf('javascript') >= 0;
+			let isCSS = type.indexOf('text/css') >= 0;
+			if (isPixUI && !disablePFBS && (isHTML || isJS)) {
+				log('hook send...', c.length, type);
+				let ver = /(\d)\.(\d)\.(\d+)/.exec(ua); // 0.2.*以上
+				pfbsConvert(ver.splice(1), isJS, c, (out) => {
+					res.send = _send;
+					res.end = _end;
+					res.setHeader = _setHeader;
+					res.setHeader('Content-Type', 'application/octet-stream');
+					_send.call(res, out);
+				});
+			}
+			else {
+				console.log('directly call send...', c.length, type, _send);
+				_send.call(this, c);
+			}
+		};
+		next();
+	};
+};