// // HttpRequest.swift // PushDeer // // Created by HEXT on 2022/1/4. // import Foundation import Moya struct TokenAuthorizationPlugin: PluginType { func prepare(_ request: URLRequest, target: TargetType) -> URLRequest { if let url = request.url, var components = URLComponents(url: url, resolvingAgainstBaseURL: true), let queryItems = components.queryItems { let queryItems_new = queryItems.map { item -> URLQueryItem in if item.name == "token" { // 把请求参数中的 token 都换成最新的 return URLQueryItem(name: "token", value: AppState.shared.token) } return item } components.queryItems = queryItems_new var request_mutable = request request_mutable.url = components.url return request_mutable } return request } } @MainActor struct HttpRequest { static let provider = MoyaProvider(callbackQueue: DispatchQueue.main, plugins: [TokenAuthorizationPlugin()]) /// 统一处理接口请求, 并且封装成 Swift Concurrency 模式 (async / await) static func request(_ targetType: PushDeerApi, resultType: T.Type) async throws -> T { do { return try await _request(targetType, resultType: resultType) } catch { if (error as NSError).code == 80403 { // token 失效处理 let stoken = AppState.shared.getLocalSToken() // 取本地 stoken if isNotEmpty(stoken) { let token = try? await stokenLogin(stoken: stoken!).token // 用 stoken 登录换 token if isNotEmpty(token) { AppState.shared.token = token! // 更新 token do { return try await _request(targetType, resultType: resultType) // 用新 token 再次请求原接口 } catch { throw error // 再次请求报错 } } AppState.shared.deleteLocalSToken() // 到这来说明 stoken 已经失效, 删掉 stoken } // 退出登录 AppState.shared.token = "" // 到这来 说明 stoken 不存在 或 已经失效, 清空 token 使其切换到登录界面 } throw error // 抛出原请求的 error (stoken流程失败, 或者不是token失效的错误, 会到这儿来) } } static func _request(_ targetType: PushDeerApi, resultType: T.Type) async throws -> T { return try await withCheckedThrowingContinuation { continuation in provider.request(targetType) { result in switch result { case let .success(response): do { print(response) if response.statusCode != 200 { continuation.resume(throwing: NSError(domain: NSLocalizedString("接口报错", comment: "接口报错时提示"), code: response.statusCode, userInfo: [NSLocalizedDescriptionKey: NSLocalizedString("接口报错", comment: "接口报错时提示") + "(\(response.statusCode)"])) return } print((try? response.mapJSON()) ?? "返回值解析失败") let result = try JSONDecoder().decode(ApiResult.self, from: response.data) print(result) if let content = result.content, result.code == 0 { continuation.resume(returning: content) } else if result.code == 80403 { continuation.resume(throwing: NSError(domain: result.error ?? NSLocalizedString("登录过期", comment: "token失效时提示"), code: result.code, userInfo: [NSLocalizedDescriptionKey: result.error ?? NSLocalizedString("登录过期", comment: "token失效时提示")])) } else { continuation.resume(throwing: NSError(domain: result.error ?? NSLocalizedString("接口报错", comment: "接口报错时提示"), code: result.code, userInfo: [NSLocalizedDescriptionKey: result.error ?? NSLocalizedString("接口报错", comment: "接口报错时提示")])) } } catch { print(error) continuation.resume(throwing: error) } case let .failure(error): print(error) continuation.resume(throwing: error) } } } } static func fake() async throws -> TokenContent { return try await request(.fake, resultType: TokenContent.self) } static func login(idToken: String) async throws -> TokenContent { return try await request(.login(idToken: idToken), resultType: TokenContent.self) } static func wechatLogin(code: String) async throws -> TokenContent { return try await request(.wechatLogin(code: code), resultType: TokenContent.self) } /// 合并用户并将旧用户删除 /// | 参数 | 说明 | /// | - | - | /// | type | 字符串,必须为 apple 或者 wechat | /// | tokenorcode | type 为 apple时此字段为 idToken,否则为 微信code | static func mergeUser(type: String, tokenorcode: String) async throws -> ResultContent { return try await request(.mergeUser(token: AppState.shared.token, type: type, tokenorcode: tokenorcode), resultType: ResultContent.self) } static func getUserInfo() async throws -> UserInfoContent { return try await request(.getUserInfo(token: AppState.shared.token), resultType: UserInfoContent.self) } static func regDevice() async throws -> DeviceContent { return try await request(.regDevice( token: AppState.shared.token, name: UIDevice.current.name, device_id: AppState.shared.deviceToken, is_clip: AppState.shared.isAppClip ? 1 : 0 ), resultType: DeviceContent.self) } static func rmDevice(id: Int) async throws -> ActionContent { return try await request(.rmDevice(token: AppState.shared.token, id: id), resultType: ActionContent.self) } static func renameDevice(id: Int, name: String) async throws -> ActionContent { return try await request(.renameDevice(token: AppState.shared.token, id: id, name: name), resultType: ActionContent.self) } static func getDevices() async throws -> DeviceContent { return try await request(.getDevices(token: AppState.shared.token), resultType: DeviceContent.self) } static func loadDevices() { _Concurrency.Task { let result = try await getDevices() AppState.shared.devices = result.devices } } static func genKey() async throws -> KeyContent { return try await request(.genKey(token: AppState.shared.token), resultType: KeyContent.self) } static func regenKey(id: Int) async throws -> ActionContent { return try await request(.regenKey(token: AppState.shared.token, id: id), resultType: ActionContent.self) } static func renameKey(id: Int, name: String) async throws -> ActionContent { return try await request(.renameKey(token: AppState.shared.token, id: id, name: name), resultType: ActionContent.self) } static func rmKey(id: Int) async throws -> ActionContent { return try await request(.rmKey(token: AppState.shared.token, id: id), resultType: ActionContent.self) } static func getKeys() async throws -> KeyContent { return try await request(.getKeys(token: AppState.shared.token), resultType: KeyContent.self) } static func loadKeys() { _Concurrency.Task { let result = try await getKeys() AppState.shared.keys = result.keys } } static func push(pushkey: String, text: String, desp: String, type: String) async throws -> ResultContent { return try await request(.push(pushkey: pushkey, text: text, desp: desp, type: type), resultType: ResultContent.self) } static func getMessages() async throws -> MessageContent { return try await request(.getMessages(token: AppState.shared.token, limit: 100), resultType: MessageContent.self) } static func rmMessage(id: Int) async throws -> ActionContent { return try await request(.rmMessage(token: AppState.shared.token, id: id), resultType: ActionContent.self) } static func rmAllMessage() async throws -> ActionContent { return try await request(.rmAllMessage(token: AppState.shared.token), resultType: ActionContent.self) } static func stokenLogin(stoken: String) async throws -> TokenContent { return try await request(.stokenLogin(stoken: stoken), resultType: TokenContent.self) } static func stokenRegen() async throws -> STokenContent { return try await request(.stokenRegen(token: AppState.shared.token), resultType: STokenContent.self) } static func stokenRemove() async throws -> STokenContent { return try await request(.stokenRemove(token: AppState.shared.token), resultType: STokenContent.self) } }