mirror of
https://github.com/wanghongenpin/proxypin.git
synced 2026-06-03 17:25:48 +08:00
desktop request map
This commit is contained in:
@@ -15,58 +15,67 @@
|
||||
*/
|
||||
|
||||
import 'dart:convert';
|
||||
import 'dart:io';
|
||||
|
||||
import 'package:flutter_js/flutter_js.dart';
|
||||
import 'package:proxypin/network/components/interceptor.dart';
|
||||
import 'package:proxypin/network/http/http.dart';
|
||||
import 'package:proxypin/network/util/file_read.dart';
|
||||
|
||||
import 'js/script_engine.dart';
|
||||
import 'manager/request_map_manager.dart';
|
||||
import 'manager/script_manager.dart';
|
||||
|
||||
/// RequestRewriteComponent is a component that can rewrite the request before sending it to the server.
|
||||
/// @author Hongen Wang
|
||||
class RequestMapInterceptor extends Interceptor {
|
||||
static RequestMapInterceptor instance = RequestMapInterceptor._();
|
||||
static JavascriptRuntime? flutterJs;
|
||||
static Map<dynamic, dynamic> scriptSession = {};
|
||||
|
||||
final managerInstance = RequestMapManager.instance;
|
||||
|
||||
RequestMapInterceptor._();
|
||||
|
||||
///脚本上下文
|
||||
Map<String, dynamic> scriptContext(RequestMapRule rule) {
|
||||
return {'scriptName': rule.name, 'os': Platform.operatingSystem, 'session': scriptSession};
|
||||
}
|
||||
|
||||
@override
|
||||
Future<HttpResponse?> execute(HttpRequest request) async {
|
||||
final manager = await managerInstance;
|
||||
if (!manager.enabled) {
|
||||
return null;
|
||||
}
|
||||
RequestMapRule? mapRule = manager.findMatch(request.requestUrl);
|
||||
if (mapRule == null) {
|
||||
return null;
|
||||
}
|
||||
var item = await manager.getMapItem(mapRule);
|
||||
if (item == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return null;
|
||||
HttpResponse? response;
|
||||
if (mapRule.type == RequestMapType.local) {
|
||||
// 本地映射
|
||||
response = await mapLocalResponse(mapRule, item);
|
||||
} else if (mapRule.type == RequestMapType.script && item.script != null) {
|
||||
response = await executeScript(request, mapRule, item.script!);
|
||||
}
|
||||
|
||||
response?.request = request;
|
||||
request.response = response;
|
||||
return response;
|
||||
}
|
||||
|
||||
/// 重写响应
|
||||
Future<void> mapLocalResponse(String url, HttpResponse response) async {
|
||||
final manager = await RequestMapManager.instance;
|
||||
var mapRule = manager.findMatch(url);
|
||||
if (mapRule == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (mapRule.type == RequestMapType.script) {
|
||||
// var rewriteItems = await manager.getMapItem(rewriteRule);
|
||||
// await _replaceResponse(response, item);
|
||||
}
|
||||
}
|
||||
|
||||
//替换相应
|
||||
Future<HttpResponse> _replaceResponse(RequestMapRule rule, RequestMapItem item) async {
|
||||
// if (rule.type == RequestMapType.script) {
|
||||
// response.status = HttpStatus.valueOf(item.statusCode!);
|
||||
// return;
|
||||
// }
|
||||
|
||||
Future<HttpResponse> mapLocalResponse(RequestMapRule rule, RequestMapItem item) async {
|
||||
HttpResponse response = HttpResponse(HttpStatus.valueOf(item.statusCode ?? 200));
|
||||
item.headers?.forEach((key, value) {
|
||||
response.headers.set(key, value);
|
||||
});
|
||||
|
||||
if (item.bodyType == MapBodyType.file.name) {
|
||||
if (item.bodyFile == null) return response;
|
||||
response.body = await FileRead.readFile(item.bodyFile!);
|
||||
@@ -76,4 +85,26 @@ class RequestMapInterceptor extends Interceptor {
|
||||
}
|
||||
return response;
|
||||
}
|
||||
|
||||
/// script执行
|
||||
Future<HttpResponse?> executeScript(HttpRequest request, RequestMapRule rule, String script) async {
|
||||
flutterJs ??= await JavaScriptEngine.getJavaScript(consoleLog: ScriptManager.consoleLog);
|
||||
var context = jsonEncode(scriptContext(rule));
|
||||
var jsRequest = jsonEncode(await JavaScriptEngine.convertJsRequest(request));
|
||||
|
||||
var jsResult = await flutterJs!.evaluateAsync(
|
||||
"""var request = $jsRequest, context = $context; request['scriptContext'] = context; $script\n onRequest(context, request)""");
|
||||
// print("response: ${jsResult.isPromise} ${jsResult.isError} ${jsResult.rawResult}");
|
||||
var result = await JavaScriptEngine.jsResultResolve(flutterJs!, jsResult);
|
||||
if (result == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
if (result['scriptContext']?['session'] != null) {
|
||||
scriptSession = result['scriptContext']['session'];
|
||||
}
|
||||
HttpResponse response = HttpResponse(HttpStatus.valueOf(200));
|
||||
response = JavaScriptEngine.convertHttpResponse(response, result);
|
||||
return response;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user