233 lines
9.1 KiB
Swift
Executable File
233 lines
9.1 KiB
Swift
Executable File
//
|
|
// MethodHandler.swift
|
|
// Runner
|
|
//
|
|
// Created by GFWFighter on 10/23/23.
|
|
//
|
|
|
|
import Flutter
|
|
import Combine
|
|
import Libcore
|
|
|
|
public class KRMethodHandler: NSObject, FlutterPlugin {
|
|
|
|
private var cancelBag: Set<AnyCancellable> = []
|
|
// 通常在类的属性中定义
|
|
private var cancellables = Set<AnyCancellable>()
|
|
public static let name = "\(Bundle.main.serviceIdentifier)/method"
|
|
|
|
public static func register(with registrar: FlutterPluginRegistrar) {
|
|
let channel = FlutterMethodChannel(name: Self.name, binaryMessenger: registrar.messenger())
|
|
let instance = KRMethodHandler()
|
|
registrar.addMethodCallDelegate(instance, channel: channel)
|
|
instance.channel = channel
|
|
}
|
|
|
|
private var channel: FlutterMethodChannel?
|
|
|
|
public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) {
|
|
@Sendable func kr_mainResult(_ res: Any?) async -> Void {
|
|
await MainActor.run {
|
|
result(res)
|
|
}
|
|
}
|
|
|
|
switch call.method {
|
|
case "parse_config":
|
|
guard
|
|
let args = call.arguments as? [String:Any?],
|
|
let path = args["path"] as? String,
|
|
let tempPath = args["tempPath"] as? String,
|
|
let debug = (args["debug"] as? NSNumber)?.boolValue
|
|
else {
|
|
result(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
var error: NSError?
|
|
MobileParse(path, tempPath, debug, &error)
|
|
if let error {
|
|
result(FlutterError(code: String(error.code), message: error.description, details: nil))
|
|
return
|
|
}
|
|
result("")
|
|
case "change_hiddify_options":
|
|
guard let options = call.arguments as? String else {
|
|
result(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
VPNConfig.shared.configOptions = options
|
|
result(true)
|
|
case "setup":
|
|
Task {
|
|
do {
|
|
try await VPNManager.shared.setup()
|
|
} catch {
|
|
await kr_mainResult(FlutterError(code: "SETUP", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
await kr_mainResult(true)
|
|
}
|
|
case "start":
|
|
Task {
|
|
guard
|
|
let args = call.arguments as? [String:Any?],
|
|
let path = args["path"] as? String
|
|
else {
|
|
await kr_mainResult(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
VPNConfig.shared.activeConfigPath = path
|
|
|
|
|
|
var error: NSError?
|
|
let config = MobileBuildConfig(path, VPNConfig.shared.configOptions, &error)
|
|
print(config);
|
|
if let error {
|
|
await kr_mainResult(FlutterError(code: String(error.code), message: error.description, details: nil))
|
|
return
|
|
}
|
|
do {
|
|
try await VPNManager.shared.setup()
|
|
try await VPNManager.shared.connect(with: config, disableMemoryLimit: VPNConfig.shared.disableMemoryLimit)
|
|
} catch {
|
|
await kr_mainResult(FlutterError(code: "SETUP_CONNECTION", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
await kr_mainResult(true)
|
|
}
|
|
case "restart":
|
|
|
|
Task { [unowned self] in
|
|
guard
|
|
let args = call.arguments as? [String:Any?],
|
|
let path = args["path"] as? String
|
|
else {
|
|
await kr_mainResult(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
VPNConfig.shared.activeConfigPath = path
|
|
|
|
|
|
VPNManager.shared.disconnect()
|
|
|
|
do {
|
|
try await kr_waitForStop().value
|
|
} catch {
|
|
await kr_mainResult(FlutterError(code: "SETUP_CONNECTION", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
|
|
|
|
var error: NSError?
|
|
let config = MobileBuildConfig(path, VPNConfig.shared.configOptions, &error)
|
|
|
|
|
|
if let error {
|
|
await kr_mainResult(FlutterError(code: "BUILD_CONFIG", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
do {
|
|
try await VPNManager.shared.setup()
|
|
try await VPNManager.shared.connect(with: config, disableMemoryLimit: VPNConfig.shared.disableMemoryLimit)
|
|
} catch {
|
|
await kr_mainResult(FlutterError(code: "SETUP_CONNECTION", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
await kr_mainResult(true)
|
|
}
|
|
case "stop":
|
|
VPNManager.shared.disconnect()
|
|
result(true)
|
|
case "reset":
|
|
VPNManager.shared.reset()
|
|
result(true)
|
|
case "url_test":
|
|
guard
|
|
let args = call.arguments as? [String:Any?]
|
|
else {
|
|
result(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
let group = args["groupTag"] as? String
|
|
FileManager.default.changeCurrentDirectoryPath(FilePath.sharedDirectory.path)
|
|
do {
|
|
try LibboxNewStandaloneCommandClient()?.urlTest(group)
|
|
} catch {
|
|
result(FlutterError(code: "URL_TEST", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
result(true)
|
|
case "select_outbound":
|
|
guard
|
|
let args = call.arguments as? [String:Any?],
|
|
let group = args["groupTag"] as? String,
|
|
let outbound = args["outboundTag"] as? String
|
|
else {
|
|
result(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
FileManager.default.changeCurrentDirectoryPath(FilePath.sharedDirectory.path)
|
|
do {
|
|
try LibboxNewStandaloneCommandClient()?.selectOutbound(group, outboundTag: outbound)
|
|
} catch {
|
|
result(FlutterError(code: "SELECT_OUTBOUND", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
result(true)
|
|
case "generate_config":
|
|
guard
|
|
let args = call.arguments as? [String:Any?],
|
|
let path = args["path"] as? String
|
|
else {
|
|
result(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
var error: NSError?
|
|
let config = MobileBuildConfig(path, VPNConfig.shared.configOptions, &error)
|
|
if let error {
|
|
result(FlutterError(code: "BUILD_CONFIG", message: error.localizedDescription, details: nil))
|
|
return
|
|
}
|
|
result(config)
|
|
case "generate_warp_config":
|
|
guard let args = call.arguments as? [String: Any],
|
|
let licenseKey = args["license-key"] as? String,
|
|
let accountId = args["previous-account-id"] as? String,
|
|
let accessToken = args["previous-access-token"] as? String else {
|
|
result(FlutterError(code: "INVALID_ARGS", message: nil, details: nil))
|
|
return
|
|
}
|
|
let warpConfig = MobileGenerateWarpConfig(licenseKey, accountId, accessToken, nil)
|
|
result(warpConfig)
|
|
default:
|
|
result(FlutterMethodNotImplemented)
|
|
}
|
|
}
|
|
|
|
private func kr_waitForStop(timeout: TimeInterval = 1) -> Future<Void, Error> {
|
|
Future { promise in
|
|
print("开始等待 VPN 停止...")
|
|
|
|
let timeoutTimer = Timer.scheduledTimer(withTimeInterval: timeout, repeats: false) { _ in
|
|
promise(.failure(NSError(domain: "VPNError", code: -1, userInfo: [NSLocalizedDescriptionKey: "等待 VPN 停止超时"])))
|
|
}
|
|
|
|
VPNManager.shared.$state
|
|
.handleEvents(receiveSubscription: { _ in
|
|
print("订阅状态变化")
|
|
}, receiveOutput: { state in
|
|
print("当前 VPN 状态: \(state)")
|
|
})
|
|
.filter { $0 == .disconnected }
|
|
.first()
|
|
.delay(for: 0.5, scheduler: DispatchQueue.main)
|
|
.sink { _ in
|
|
timeoutTimer.invalidate()
|
|
print("VPN 已停止")
|
|
promise(.success(()))
|
|
}
|
|
.store(in: &self.cancellables)
|
|
}
|
|
}
|
|
}
|