迁移仓库

2026-02-24 09:55:08 +08:00
parent 2d5c056b82
commit df6210aaf7
30 changed files with 1121 additions and 81 deletions
--- a/ShenQi/GDWebPageHookService.m
+++ b/ShenQi/GDWebPageHookService.m
@@ -0,0 +1,80 @@
+//
+//  GDWebPageHookService.m
+//  DESThreeTest
+//
+//  Created by Qwerdf on 2025/10/24.
+//  Copyright © 2025 zhang alan. All rights reserved.
+//
+
+#import "GDWebPageHookService.h"
+
+@implementation GDWebPageHookService
+
+#pragma mark - 初始化
+- (instancetype)initWithSchemeTask:(id<WKURLSchemeTask>)task {
+    if (self = [super init]) {
+        _schemeTask = task;
+    }
+    return self;
+}
+
+#pragma mark - GDWebCustomURLSchemeProtocol
+ (BOOL)canService:(id<WKURLSchemeTask>)task {
+    // 处理需要钩子的请求（如包含 "/api/hook" 的接口）
+    NSString *urlString = task.request.URL.absoluteString;
+    return [urlString containsString:@"/api/hook"];
+}
+
+- (void)startLoading:(void (^)(void))completion {
+    self.completion = completion;
+    NSURLRequest *request = self.schemeTask.request;
+    
+    // 1. 发起原始网络请求
+    self.dataTask = [[NSURLSession sharedSession] dataTaskWithRequest:request completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
+        if (error) {
+            // 网络错误，直接返回给网页
+            [self.schemeTask didFailWithError:error];
+            if (completion) completion();
+            return;
+        }
+        
+        // 2. 钩子逻辑：修改响应数据（示例：给 API 响应添加自定义字段）
+        NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)response;
+        NSMutableDictionary *headers = [httpResponse.allHeaderFields mutableCopy];
+        // 允许网页接收修改后的数据（处理跨域）
+        headers[@"Access-Control-Allow-Origin"] = @"*";
+        
+        // 若为 JSON 响应，修改内容
+        NSData *modifiedData = data;
+        if ([httpResponse.MIMEType containsString:@"application/json"] && data) {
+            NSMutableDictionary *json = [NSJSONSerialization JSONObjectWithData:data options:NSJSONReadingMutableContainers error:nil];
+            if (json) {
+                json[@"_injected_by_native"] = @"true"; // 注入自定义字段
+                modifiedData = [NSJSONSerialization dataWithJSONObject:json options:0 error:nil];
+            }
+        }
+        
+        // 3. 构造修改后的响应并返回给网页
+        NSHTTPURLResponse *modifiedResponse = [[NSHTTPURLResponse alloc] initWithURL:httpResponse.URL
+                                                                           statusCode:httpResponse.statusCode
+                                                                          HTTPVersion:@"HTTP/1.1"
+                                                                         headerFields:headers];
+        [self.schemeTask didReceiveResponse:modifiedResponse];
+        [self.schemeTask didReceiveData:modifiedData];
+        [self.schemeTask didFinish];
+        
+        // 调用完成回调
+        if (completion) completion();
+    }];
+    [self.dataTask resume];
+}
+
+- (void)stopLoading {
+    // 停止网络请求
+    [self.dataTask cancel];
+    self.dataTask = nil;
+    self.schemeTask = nil;
+    if (self.completion) self.completion();
+}
+
+@end