Initial commit

AIGrammar/lib/ColorToString.swift

@@ -0,0 +1,8 @@
+//
+//  ColorToString.swift
+//  AIGrammar
+//
+//  Created by oscar on 2024/3/28.
+//
+
+import Foundation

AIGrammar/lib/CommFunc.swift

@@ -0,0 +1,8 @@
+//
+//  CommFunc.swift
+//  AIGrammar
+//
+//  Created by oscar on 2024/8/5.
+//
+
+import Foundation

AIGrammar/lib/IapManager.swift

@@ -0,0 +1,8 @@
+//
+//  Iap.swift
+//  AIGrammar
+//
+//  Created by oscar on 2024/7/3.
+//
+
+import Foundation

AIGrammar/lib/InitAPP.swift

@@ -0,0 +1,10 @@
+//
+//  InitAPP.swift
+//  AIGrammar
+//
+//  Created by oscar on 2024/6/26.
+//
+
+import Foundation
+
+

AIGrammar/lib/LogManager.swift

@@ -0,0 +1,8 @@
+//
+//  LogManager.swift
+//  AIGrammar
+//
+//  Created by oscar on 2024/7/3.
+//
+
+import Foundation

AIGrammar/lib/NetworkManager.swift

@@ -0,0 +1,233 @@
+//
+//  request.swift
+//  AIGrammar
+//
+//  Created by oscar on 2024/6/4.
+//
+
+import Foundation
+import Alamofire
+import SwiftJWT
+import SwiftyBeaver
+
+struct MyClaims: Claims {
+    var deviceID: String
+    var gid: Int
+    var exp1: Int
+}
+
+struct APIResponse<T: Decodable>: Decodable {
+    let ret: Int
+    let message: String
+    let data: T?
+}
+
+struct Translation: Identifiable, Codable, Equatable {
+    var id = UUID()
+    var input: String
+    var translation: String
+}
+
+struct TranslationResponse: Codable {
+    let translation: String
+}
+
+struct WordDetails {
+    var word: String = ""
+    var explanations: [String] = []
+    var phrases: [String] = []
+    var synonyms: [String] = []
+}
+
+struct WordDetailsResponse: Codable {
+    let word: String
+    let explain: [String]?
+    let phrase: [String]?
+    let sync: [String]?
+
+    // Mapping to WordDetails for use in UI
+    func toWordDetails() -> WordDetails {
+        return WordDetails(
+            word: word,
+            explanations: explain ?? [],
+            phrases: phrase ?? [],
+            synonyms: sync ?? []
+        )
+    }
+}
+
+struct VIPStatusResponse: Codable {
+    let id: Int
+    let userid: String
+    let username: String
+    let vip: Int
+}
+
+
+struct NetworkManager {
+    static let shared = NetworkManager()
+    private let jwtSecret = globalEnvironment.jwtSecret
+    
+    // 语法检查功能
+    func checkGrammar(inputText: String, completion: @escaping ([GrammarRes]?, Error?) -> Void) {
+        let url = globalEnvironment.grammarURL
+        let headers: HTTPHeaders = createAuthorizationHeader()
+        
+        let parameters: [String: Any] = [
+            "input": inputText,
+            "lang": "eng"
+        ]
+        AF.request(url, method: .post, parameters: parameters,  encoding: URLEncoding.httpBody, headers: headers)
+            .responseDecodable(of: [GrammarRes].self) { response in
+            switch response.result {
+            case .success(let results):
+                completion(results, nil)
+            case .failure(let error):
+                completion(nil, error)
+            }
+        }
+    }
+    
+    // 单词的词典功能
+    func fetchWordDetails(inputText: String, lang: String = "eng", completion: @escaping (Result<WordDetails, Error>) -> Void) {
+        
+        guard !inputText.isEmpty else { return }
+        let parameters: [String: Any] = ["input": inputText, "lang": lang]
+        let url = globalEnvironment.dictURL
+        let headers: HTTPHeaders = createAuthorizationHeader()
+        
+        AF.request(url, method: .post, parameters: parameters, encoding: URLEncoding.httpBody, headers: headers)
+            .responseDecodable(of: WordDetailsResponse.self) { response in
+                switch response.result {
+                case .success(let detailsResponse):
+                    print("Success: Received data for \(detailsResponse.word)")
+                    let details = detailsResponse.toWordDetails()  // Convert here
+                    completion(.success(details))
+                case .failure(let error):
+                    print("Error: \(error)")
+                    print("Response Data: \(String(data: response.data ?? Data(), encoding: .utf8) ?? "No data")")
+                    
+                    completion(.failure(error))
+                }
+            }
+    }
+    
+    // 翻译功能
+    func translate(inputText: String, lang: String = "chs", completion: @escaping (Result<Translation, Error>) -> Void) {
+        
+        guard !inputText.isEmpty else { return }
+        let url = globalEnvironment.translateURL
+        let parameters: [String: Any] = ["input": inputText, "lang": lang]
+        let headers: HTTPHeaders = createAuthorizationHeader()
+
+        AF.request(url, method: .post, parameters: parameters, encoding: URLEncoding.httpBody, headers: headers)
+            .responseDecodable(of: TranslationResponse.self) { response in
+                switch response.result {
+                case .success(let translationResponse):
+                    let newTranslation = Translation(input: inputText, translation: translationResponse.translation)
+                    completion(.success(newTranslation))
+                case .failure(let error):
+                    completion(.failure(error))
+                }
+            }
+    }
+    
+    // 翻译的点赞、点踩功能
+    func sendFeedback(input: String, output: String, isPositive: Bool) {
+        let url = globalEnvironment.feedbackURL
+        let parameters: [String: Any] = [
+            "product": "trans",
+            "input": input,
+            "output": output,
+            "res": isPositive ? "good" : "bad"
+        ]
+        
+        let headers: HTTPHeaders = createAuthorizationHeader()
+        AF.request(url, method: .post, parameters: parameters, encoding: URLEncoding.httpBody, headers: headers).response { response in
+            switch response.result {
+            case .success(let data):
+                print("Feedback sent successfully: \(String(describing: data))")
+            case .failure(let error):
+                print("Error sending feedback: \(error)")
+            }
+        }
+                                         
+    }
+    
+    // 查询VIP状态
+    func getUserProfile(completion: @escaping (Result<VIPStatusResponse, Error>) -> Void) {
+        let url = globalEnvironment.userURL
+        
+        let parameters: [String: Any] = [:]
+        let headers: HTTPHeaders = createAuthorizationHeader()
+        
+        performRequest(
+            endpoint: url,
+            parameters: parameters,
+            method: .post,
+            encoding: URLEncoding.httpBody,
+            headers: headers,  // 示例：使用JWT token
+            completion: { (result: Result<VIPStatusResponse, Error>) in
+                switch result {
+                case .success(let vipData):
+                    logger.info("VIP Status: \(vipData.vip)")
+                    completion(.success(vipData))
+                case .failure(let error):
+                    logger.error("Error: \(error.localizedDescription)")
+                    completion(.failure(error))
+                }
+            }
+        )
+        
+    }
+        
+    func performRequest<T: Decodable>(
+        endpoint: String,
+        parameters: Parameters,  // 明确使用 Alamofire 的 Parameters 类型
+        method: HTTPMethod = .post,
+        encoding: URLEncoding = .httpBody,
+        headers: HTTPHeaders? = nil,
+        completion: @escaping (Result<T, Error>) -> Void
+    ) {
+        let url = endpoint
+        let defaultHeaders: HTTPHeaders = [.contentType("application/x-www-form-urlencoded")]
+        let combinedHeaders = headers ?? defaultHeaders
+
+        // 确保参数是正确的类型
+        AF.request(url, method: .post, parameters: parameters, encoding: encoding, headers: combinedHeaders).responseDecodable(of: APIResponse<T>.self) { response in
+            switch response.result {
+            case .success(let apiResponse):
+                if apiResponse.ret == 0, let data = apiResponse.data {
+                    completion(.success(data))
+                } else {
+                    let error = NSError(domain: "", code: apiResponse.ret, userInfo: [NSLocalizedDescriptionKey: apiResponse.message])
+                    completion(.failure(error))
+                }
+            case .failure(let error):
+                completion(.failure(error))
+            }
+        }
+    }
+    
+    private func createAuthorizationHeader() -> HTTPHeaders {
+        // Generate JWT and return headers
+        guard let jwtToken = generateJWT(deviceID: globalEnvironment.deviceID, gID: globalEnvironment.GID, jwtSecret: jwtSecret) else {
+            return [:]
+        }
+        return ["Authorization": "Bearer \(jwtToken)"]
+    }
+    
+    private func generateJWT(deviceID: String, gID: Int, jwtSecret: String) -> String? {
+        let claims = MyClaims(deviceID: deviceID, gid: gID, exp1: Int(Date().timeIntervalSince1970) + 86400)
+        var jwt = JWT(claims: claims)
+        do {
+            let jwtSigner = JWTSigner.hs256(key: Data(jwtSecret.utf8))
+            let signedJWT = try jwt.sign(using: jwtSigner)
+            return signedJWT
+        } catch {
+            logger.error("JWT encoding failed with error: \(error)")
+            return nil
+        }
+    }
+}
+