PDF_Office/PDF Office/PDF Master/Class/AIInfo/KMAIRequestServerManager.swift

//
//  KMAIRequestServerManager.swift
//  PDF Reader Pro Edition
//
//  Created by wanjun on 2024/1/17.
//

import Cocoa
import AFNetworking

@objcMembers
class ResultWrapper: NSObject {
    var success: Bool = false
    var result: NSDictionary = [:]
//    var message: String = ""
//    var code : Int = 200
//    var error: NSError?
//    var state: KMAIRewritingState = .rewrite
    var content: String = ""

//    init(success: Bool, result: NSDictionary, message: String, code: Int, error: NSError? = nil, state: KMAIRewritingState) {
//        self.success = success
//        self.result = result
//        self.message = message
//        self.code = code
//        self.error = error
//        self.state = state
//    }
    init(success: Bool, content: String) {
        self.success = success
        self.content = content
        self.result = NSDictionary.init()
    }
    
    init(success: Bool, resultData: NSDictionary) {
        self.success = success
        self.content = ""
        self.result = resultData
    }
}

@objc enum KMAIRewritingState : Int {
    case rewrite = 0        // 重写
    case correctTypos       // 纠错
    case uploadTranslate    // 翻译上传文件
    case fileTranslate      // 翻译文件
    case textTranslate      // 翻译文本
    case extractSummaryFile //摘要
}

@objc enum KMAIFileUploadState : Int {
    case Translation = 0
    case ExtractAbstract
}

//typealias KMRequestServerComplete = (_ success: Bool, _ result: Result?) -> Void

@objcMembers class KMAIRequestServerManager: NSObject {
    static let defaultManager = KMAIRequestServerManager()
    typealias KMRequestServerComplete = (_ wrapper: ResultWrapper) -> Void
    private var downFileUrl: String = "" // 文件翻译
    var downFileName: String = "" // 文件翻译
    
    var fromLanguages: [String] = ["Automatic", "English", "Simplified Chinese", "Traditional Chinese", "Japanese", "Korean", "French", "Spanish", "Italian", "German", "Portuguese", "Russian", "Vietnamese", "Thai", "Arabic", "Greek", "Bulgarian", "Finnish", "Slovene", "Dutch", "Czech", "Swedish", "Polish", "Danish", "Romanian", "Hungarian"]
    var toLanguages: [String] = ["English", "Simplified Chinese", "Traditional Chinese", "Japanese", "Korean", "French", "Spanish", "Italian", "German", "Portuguese", "Russian", "Vietnamese", "Thai", "Arabic", "Greek", "Bulgarian", "Finnish", "Slovene", "Dutch", "Czech", "Swedish", "Polish", "Danish", "Romanian", "Hungarian"]
    
    // MARK: AI Action (public)
    
    func aiAction(content: String, state: KMAIRewritingState, from: String = "auto", to: String = "en", complete: @escaping KMRequestServerComplete) {
        if state == .rewrite || state == .correctTypos {  //重写 && 纠错
            aiRewriting(content: content, state: state, complete: complete)
        } else if state == .uploadTranslate || state == .fileTranslate {    // 上传文件 || 文件翻译
            aiTranslationFileUpload(file: content, complete: { [weak self] wrapper in
                let success: Bool = wrapper.success
                let fileKey = wrapper.content
                if success {
                    let fileData = wrapper.result
                    complete(wrapper)
//                    aiTranslationFileTranslateHandle(fileKey: fileKey, from: languageAbbreviation(from), to: languageAbbreviation(to), complete: complete)
                } else {
                    complete(wrapper)
                }
            })
        } else if state == .textTranslate { // 文本翻译
            aiTranslationTextTrans(q: content, from: languageAbbreviation(from), to: languageAbbreviation(to), complete: complete)
        } else if state == .extractSummaryFile {    // 提取摘要
            extractSummaryFile(file: content, complete: complete)
        }
    }
    
    // MARK: Private
    
    private func uuid() -> String {
        return GetHardwareUUID()!
    }
    
    private func platform() -> String {
        var platform = "DMG"
#if VERSION_DMG
        platform = "DMG"
#else
        platform = "AiStore"
#endif
        return platform
    }
    
    private func app_name() -> String {
        var bundleID: String!
        bundleID = Bundle.main.object(forInfoDictionaryKey: "CFBundleIdentifier") as! String
        return bundleID;
    }
    
    func downloadFile(filePath: String, downFileName: String, toPath: String, complete: @escaping KMRequestServerComplete) {
        self.downFileUrl = ""
        guard let fileURL = URL(string: filePath) else {
            let alert = NSAlert()
            alert.alertStyle = .critical
            alert.messageText = NSLocalizedString("Invalid file link", comment: "")
            alert.runModal()

            return
        }
        var destinationURL = FileManager.default.temporaryDirectory.appendingPathComponent(downFileName)
        if URL(fileURLWithPath: toPath) != nil {
            destinationURL = URL(fileURLWithPath: toPath)
        }
        if FileManager.default.fileExists(atPath: destinationURL.path) {
            do {
                try FileManager.default.removeItem(at: destinationURL)
                print("删除旧文件成功")
            } catch {
                print("删除旧文件失败：\(error)")
            }
        }
        let sessionConfiguration = URLSessionConfiguration.default
        let session = URLSession(configuration: sessionConfiguration)
        let downloadTask = session.downloadTask(with: fileURL) { (tempLocalURL, response, error) in
            if let error = error {
                let alert = NSAlert()
                alert.alertStyle = .critical
                alert.messageText = String(format: "%@:\(error)", NSLocalizedString("Download failed", comment: ""))
                alert.runModal()
                
                return
            }

            guard let tempLocalURL = tempLocalURL else {
                let alert = NSAlert()
                alert.alertStyle = .critical
                alert.messageText = NSLocalizedString("Invalid temporary directory", comment: "")
                alert.runModal()

                return
            }
            

            do {
                try FileManager.default.moveItem(at: tempLocalURL, to: destinationURL)
                
                let wrapper = ResultWrapper(success: true, content: "")
                wrapper.content = destinationURL.path
                complete(wrapper)
            } catch {
                let alert = NSAlert()
                alert.alertStyle = .critical
                alert.messageText = String(format: "%@:\(error)", NSLocalizedString("Failed to save file", comment: ""))
                alert.runModal()
            }
        }
        self.downFileUrl = destinationURL.path
        downloadTask.resume()
    }
    
    private func languageAbbreviation(_ language: String) -> String {
        if language == "Automatic" || language == "auto" {
            return "auto"
        } else if language == "English" || language == "en" {
            return "en"
        } else if language == "Simplified Chinese" || language == "zh" {
            return "zh"
        } else if language == "Traditional Chinese" || language == "cht" {
            return "cht"
        } else if language == "Japanese" || language == "jp" {
            return "jp"
        } else if language == "Korean" || language == "kor" {
            return "kor"
        } else if language == "French" || language == "fra" {
            return "fra"
        } else if language == "Spanish" || language == "spa" {
            return "spa"
        } else if language == "Italian" || language == "it" {
            return "it"
        } else if language == "German" || language == "de" {
            return "de"
        } else if language == "Portuguese" || language == "pt" {
            return "pt"
        } else if language == "Russian" || language == "ru" {
            return "ru"
        } else if language == "Vietnamese" || language == "vie" {
            return "vie"
        } else if language == "Thai" || language == "th" {
            return "th"
        } else if language == "Arabic" || language == "ara" {
            return "ara"
        } else if language == "Greek" || language == "el" {
            return "el"
        } else if language == "Bulgarian" || language == "bul" {
            return "bul"
        } else if language == "Finnish" || language == "fin" {
            return "fin"
        } else if language == "Slovene" || language == "slo" {
            return "slo"
        } else if language == "Dutch" || language == "nl" {
            return "nl"
        } else if language == "Czech" || language == "cs" {
            return "cs"
        } else if language == "Swedish" || language == "swe" {
            return "swe"
        } else if language == "Polish" || language == "pl" {
            return "pl"
        } else if language == "Danish" || language == "dan" {
            return "dan"
        } else if language == "Romanian" || language == "rom" {
            return "rom"
        } else if language == "Hungarian" || language == "hu" {
            return "hu"
        }
        return "auto"
    }
    
    // MARK: AI Action (private)
    
    /**
     @abstract 翻译上传
     @param file  文件路径
     @param complete 上传完成回调
     */
    private func aiTranslationFileUpload(file: String, complete: @escaping KMRequestServerComplete) {
        let infoDictionary = Bundle .main.infoDictionary!
        let urlString = AIInfoConfig().aiActionURL + "/api/AI/uploadTranslate"
        let fileData = FileManager.default.contents(atPath: file)
        let params: [String: Any] = ["file": fileData,
                                     "uuid": uuid(),
                                     "platform": platform(),
                                     "app_name": app_name(),
                                     "calculate":"1"]
        KMAIRequestServer.requestServer.uploadFile(urlString: urlString, params: params) { formData in
            let fileURL = URL(fileURLWithPath: file)
            try? formData.appendPart(withFileURL: fileURL, name: "file", fileName: fileURL.lastPathComponent, mimeType: "application/octet-stream")
        } requestSerializer: { requestSerializer in
            requestSerializer.setValue("Apifox/1.0.0 (https://www.apifox.cn)", forHTTPHeaderField: "User-Agent")
        } completion: { task, responseObject, error in
            if responseObject != nil {
                let data: NSDictionary = responseObject!["data"] as? NSDictionary ?? [:]
                var code: String = responseObject!["code"] as? String ?? "06005"
                if code == "06005" {
                    let tempCode: Int = responseObject!["code"] as? Int ?? 0
                    if tempCode == 501 {
                        code = "501"
                    }
                }
                
                let message: String = responseObject!["message"] as? String ?? "unknown error"
                let error = NSError(domain: message, code: Int(code)!)
//                let result = ResultWrapper(success: true, result: data, message: message, code: Int(code)!, error: error, state: .uploadTranslate)
//                if result.code == 200 {
//                    result.success = true
//                    complete(result)
//                } else {
//                    result.success = false
//                    complete(result)
//                }
                if Int(code)! == 200 {
                    let wrapper = ResultWrapper(success: true, resultData: data)
                    wrapper.content = data["fileKey"] as! String
                    complete(wrapper)
                } else {
                    let wrapper = ResultWrapper(success: false, resultData: data)
                    if Int(code)! == 501 {
                        wrapper.content = "501"
                    } else {
                        wrapper.content = message
                    }
                    complete(wrapper)
                }
            } else {
//                let error = NSError(domain: "unknown error", code: 404)
//                let result = ResultWrapper(success: false, result: [:], message: "unknown error", code: 404, error: error, state: .uploadTranslate)
//                complete(result)
                complete(ResultWrapper(success: false, content: "unknown error"))
            }
        }
    }
    
    /**
     @abstract 文件翻译
     @param fileKey  文件Key
     @param from 初始语言
     @param to 结束语言
     */
    func aiTranslationFileTranslateHandle(fileKey: String, from: String, to: String, complete: @escaping KMRequestServerComplete) {
        
        let infoDictionary = Bundle .main.infoDictionary!
        let urlString = AIInfoConfig().aiActionURL + "/api/AI/fileTranslate"

        let paraDict = ["fileKey": fileKey,
                        "from": languageAbbreviation(from),
                        "to": languageAbbreviation(to),
                        "uuid": uuid(),
                        "platform": platform(),
                        "app_name": app_name()]
        var postData  = try! JSONSerialization.data(withJSONObject: paraDict)

        
        var request = URLRequest(url: URL(string: urlString)!,timeoutInterval: Double.infinity)
        request.addValue("Apifox/1.0.0 (https://www.apifox.cn)", forHTTPHeaderField: "User-Agent")
        request.addValue("application/json", forHTTPHeaderField: "Content-Type")

        request.httpMethod = "POST"
        request.httpBody = postData

        let task = URLSession.shared.dataTask(with: request) { data, response, error in
           guard let data = data else {
               print(String(describing: error))
               return
           }
            let jsonObject = try? JSONSerialization.jsonObject(with: data, options: []) as? NSDictionary ?? [:]
            if jsonObject != nil {
                let data1: NSDictionary = jsonObject!["data"] as? NSDictionary ?? [:]
                let code: String = jsonObject!["code"] as? String ?? "06005"
                let message: String = jsonObject!["message"] as? String ?? "unknown error"
                let error = NSError(domain: message, code: Int(code)!)
//                let result = ResultWrapper(success: true, result: data1, message: message, code: Int(code)!, error: error, state: .fileTranslate)
//                if result.code == 200 {
//                    result.success = true
//                    complete(result)
//                } else {
//                    result.success = false
//                    complete(result)
//                }
                if Int(code)! == 200 {
                    let fileUrl = data1["fileUrl"]
                    let downFileUrl = data1["downFileUrl"]
                    let ossDownUrl = data1["ossDownUrl"]
                    let fileName = data1["fileName"]
                    let downFileName = data1["downFileName"]
                    let fromStr = data1["from"]
                    let toStr = data1["to"]
                    
                    self.downFileName = downFileName as! String
//                    self.downloadFile(filePath: ossDownUrl as! String, downFileName: downFileName as! String)
                    complete(ResultWrapper(success: true, content: ossDownUrl as! String))
                } else {
                    complete(ResultWrapper(success: false, content: message))
                }
            } else {
//                let error = NSError(domain: "unknown error", code: 404)
//                let result = ResultWrapper(success: false, result: [:], message: "unknown error", code: 404, error: error, state: .fileTranslate)
//                complete(result)
                complete(ResultWrapper(success: false, content: "unknown error"))
            }
        }
        task.resume()
    }

    /**
     @abstract 内容翻译
     @param q  选中
     @param from 初始语言
     @param to 结束语言
     */
    private func aiTranslationTextTrans(q: String, from: String, to: String, complete: @escaping KMRequestServerComplete) {
        
        let infoDictionary = Bundle .main.infoDictionary!
        let urlString = AIInfoConfig().aiActionURL + "/api/AI/textTranslate"
        
        var escapedString = q
        escapedString = String(data: escapedString.data(using: .utf8)!, encoding: .utf8)!
        let paraDict = ["text":escapedString,
                        "from":from,
                        "to":to,
                        "uuid":uuid(),
                        "platform": platform(),
                        "app_name": app_name()]
        var postData  = try! JSONSerialization.data(withJSONObject: paraDict)

        var request = URLRequest(url: URL(string: urlString)!,timeoutInterval: Double.infinity)
        request.addValue("Apifox/1.0.0 (https://www.apifox.cn)", forHTTPHeaderField: "User-Agent")
        request.addValue("application/json", forHTTPHeaderField: "Content-Type")

        request.httpMethod = "POST"
        request.httpBody = postData

        let task = URLSession.shared.dataTask(with: request) { data, response, error in
           guard let data = data else {
               print(String(describing: error))
               return
           }
            let jsonObject = try? JSONSerialization.jsonObject(with: data, options: []) as? NSDictionary ?? [:]
            if jsonObject != nil {
                let data1: NSDictionary = jsonObject!["data"] as? NSDictionary ?? [:]
                let code: String = jsonObject!["code"] as? String ?? "06005"
                let message: String = jsonObject!["message"] as? String ?? "unknown error"
                let error = NSError(domain: message, code: Int(code)!)
                if Int(code)! == 200 {
                    var resultStr = data1["dst"] as! String
                    complete(ResultWrapper(success: true, content: resultStr))
                } else {
                    complete(ResultWrapper(success: false, content: message))
                }
            } else {
                complete(ResultWrapper(success: false, content: "unknown error"))
            }
        }
        task.resume()
    }
    
    /**
     @abstract重写 & 纠错
     @param content  内容
     @param state  类型
     */
    private func aiRewriting(content: String, state: KMAIRewritingState, complete: @escaping KMRequestServerComplete) {
        
        var urlString = AIInfoConfig().aiActionURL + "/api/AI/rewrite"
        if state == .correctTypos {
            urlString = AIInfoConfig().aiActionURL + "/api/AI/correctTypos"
        }
        let params: [String: Any] = ["content": content,
                                     "uuid": uuid(),
                                     "platform": platform(),
                                     "app_name": app_name()]
        
        KMAIRequestServer.requestServer.aiRewriting(urlString: urlString, params: params) { formData in
            
        } requestSerializer: { requestSerializer in
            requestSerializer.setValue("Apifox/1.0.0 (https://www.apifox.cn)", forHTTPHeaderField: "User-Agent")
        } completion: { task, responseObject, error in
            if responseObject != nil {
                let data: NSDictionary = responseObject!["data"] as? NSDictionary ?? [:]
                let code: String = responseObject!["code"] as? String ?? "06005"
                let message: String = responseObject!["message"] as? String ?? "unknown error"
                let error = NSError(domain: message, code: Int(code)!)
//                let result = ResultWrapper(success: true, result: data, message: message, code: Int(code)!, error: error, state: state)
//                if result.code == 200 {
//                    result.success = true
//                    complete(result)
//                } else {
//                    result.success = false
//                    complete(result)
//                }
                if Int(code)! == 200 {
                    complete(ResultWrapper(success: true, content: data["content"] as! String))
                } else {
                    complete(ResultWrapper(success: false, content: message))
                }
            } else {
//                let error = NSError(domain: "unknown error", code: 404)
//                let result = ResultWrapper(success: false, result: [:], message: "unknown error", code: 404, error: error, state: state)
//                complete(result)
                complete(ResultWrapper(success: false, content: "unknown error"))
            }
        }
    }
    
    /**
     @abstract 提取摘要
     @param file  文件路径
     */
    private func extractSummaryFile(file: String, complete: @escaping KMRequestServerComplete) {
        
        let infoDictionary = Bundle .main.infoDictionary!
        let urlString = AIInfoConfig().aiActionURL + "/api/AI/extractSummaryFile"
        let fileData = FileManager.default.contents(atPath: file)
        let params: [String: Any] = ["file": fileData,
                                     "uuid": uuid(),
                                     "platform": platform(),
                                     "app_name": app_name()]
        KMAIRequestServer.requestServer.uploadFile(urlString: urlString, params: params) { formData in
            let fileURL = URL(fileURLWithPath: file)
            try? formData.appendPart(withFileURL: fileURL, name: "file", fileName: fileURL.lastPathComponent, mimeType: "application/octet-stream")
        } requestSerializer: { requestSerializer in
            requestSerializer.setValue("Apifox/1.0.0 (https://www.apifox.cn)", forHTTPHeaderField: "User-Agent")
        } completion: { task, responseObject, error in
            if responseObject != nil {
                let data: NSDictionary = responseObject!["data"] as? NSDictionary ?? [:]
                let code: Int = responseObject!["code"] as? Int ?? 06005
                let message: String = responseObject!["message"] as? String ?? "unknown error"
                let error = NSError(domain: message, code: code)
//                let result = ResultWrapper(success: true, result: data, message: message, code: Int(code)!, error: error, state: .extractSummaryFile)
//                if result.code == 200 {
//                    result.success = true
//                    complete(result)
//                } else {
//                    result.success = false
//                    complete(result)
//                }
                if code == 200 {
                    complete(ResultWrapper(success: true, content: data["summary"] as! String))
                } else {
                    complete(ResultWrapper(success: false, content: message))
                }
            } else {
                let error = NSError(domain: "unknown error", code: 404)
//                let result = ResultWrapper(success: false, result: [:], message: "unknown error", code: 404, error: error, state: .extractSummaryFile)
//                complete(result)
                complete(ResultWrapper(success: false, content: "unknown error"))
            }
        }
    }

}

//typealias KMHttpRequestServerComplete = (_ task: URLSessionDataTask?, _ responseObject: AnyObject?, _ error: NSError?) -> Void
class KMAIRequestServer {
    var sessionManager: AFHTTPSessionManager!
    static let requestServer = KMAIRequestServer()
    var task: URLSessionTask?
    
    init() {
        let configuration: URLSessionConfiguration = URLSessionConfiguration.default
        sessionManager = AFHTTPSessionManager.init(sessionConfiguration: configuration)
        sessionManager.securityPolicy = AFSecurityPolicy.default()
    }
    
    /**
     上传文件
        - 翻译
        - 摘要提取
     */
    func uploadFile(urlString: String,
                        params: Dictionary<String, Any>?,
                       body: ((_ formData:AFMultipartFormData) -> Void)?,
          requestSerializer: ((_ requestSerializer:AFHTTPRequestSerializer) -> Void)?,
                     completion: KMHttpRequestServerComplete?) -> Void {
        sessionManager.requestSerializer.setValue("multipart/form-data", forHTTPHeaderField: "Content-Type")
        sessionManager.requestSerializer.timeoutInterval = 60
        sessionManager.responseSerializer = AFJSONResponseSerializer()
        sessionManager.responseSerializer.acceptableContentTypes = ["application/json","text/html","text/json","text/javascript","text/plain","image/gif"]
        if (requestSerializer != nil) {
            requestSerializer!(sessionManager.requestSerializer);
        }
        if (body != nil) {
            self.task = sessionManager.post(urlString, parameters: params, headers: nil) { formData in
                body!(formData)
            } progress: { progress in
                
            } success: { task, responseObject in
                let responseObject = responseObject as AnyObject
                if (completion != nil) {
                    completion!(task,responseObject,nil);
                }
            } failure: { task, error in
                if (completion != nil) {
                    completion!(task, nil, error as NSError);
                }
            }
        } else {
            self.task = sessionManager.post(urlString, parameters: params, headers: nil) { progress in
            
            } success: { task, responseObject in
                if (completion != nil) {
                    let responseObject = responseObject as AnyObject
                    completion!(task,responseObject,nil);
                }
            } failure: { task, error in
                if (completion != nil) {
                    completion!(task, nil, error as NSError);
                }
            }
        }
    }
    
    /**
     文件翻译
     */
    func aiFileTranslateHandle(urlString: String,
                                  params: Dictionary<String, Any>?,
                                    body: ((_ formData:AFMultipartFormData) -> Void)?,
                       requestSerializer: ((_ requestSerializer:AFHTTPRequestSerializer) -> Void)?,
                              completion: KMHttpRequestServerComplete?) -> Void {
        sessionManager.requestSerializer = AFJSONRequestSerializer()
        sessionManager.responseSerializer = AFJSONResponseSerializer()
        sessionManager.responseSerializer.acceptableContentTypes = ["application/json","text/html","text/json","text/javascript","text/plain","image/gif"]
        if (requestSerializer != nil) {
            requestSerializer!(sessionManager.requestSerializer);
        }
        if (body != nil) {
            self.task = sessionManager.post(urlString, parameters: params, headers: nil) { formData in
                body!(formData)
            } progress: { progress in
                
            } success: { task, responseObject in
                let responseObject = responseObject as AnyObject
                if (completion != nil) {
                    completion!(task,responseObject,nil);
                }
            } failure: { task, error in
                if (completion != nil) {
                    completion!(task, nil, error as NSError);
                }
            }
        } else {
            self.task = sessionManager.post(urlString, parameters: params, headers: nil) { progress in
            
            } success: { task, responseObject in
                if (completion != nil) {
                    let responseObject = responseObject as AnyObject
                    completion!(task,responseObject,nil);
                }
            } failure: { task, error in
                if (completion != nil) {
                    completion!(task, nil, error as NSError);
                }
            }
        }
    }
    
    /**
     重写
     */
    func aiRewriting(urlString: String,
                        params: Dictionary<String, Any>?,
                          body: ((_ formData:AFMultipartFormData) -> Void)?,
             requestSerializer: ((_ requestSerializer:AFHTTPRequestSerializer) -> Void)?,
                    completion: KMHttpRequestServerComplete?) -> Void {
        sessionManager.requestSerializer.setValue("multipart/form-data", forHTTPHeaderField: "Content-Type")
        sessionManager.requestSerializer.timeoutInterval = 60
        sessionManager.responseSerializer = AFJSONResponseSerializer()
        sessionManager.responseSerializer.acceptableContentTypes = ["application/json","text/html","text/json","text/javascript","text/plain","image/gif"]
        if (requestSerializer != nil) {
            requestSerializer!(sessionManager.requestSerializer);
        }
        if (body != nil) {
            self.task = sessionManager.post(urlString, parameters: params, headers: nil) { formData in
                body!(formData)
            } progress: { progress in
                
            } success: { task, responseObject in
                let responseObject = responseObject as AnyObject
                if (completion != nil) {
                    completion!(task,responseObject,nil);
                }
            } failure: { task, error in
                if (completion != nil) {
                    completion!(task, nil, error as NSError);
                }
            }
        } else {
            self.task = sessionManager.post(urlString, parameters: params, headers: nil) { progress in
            
            } success: { task, responseObject in
                if (completion != nil) {
                    let responseObject = responseObject as AnyObject
                    completion!(task,responseObject,nil);
                }
            } failure: { task, error in
                if (completion != nil) {
                    completion!(task, nil, error as NSError);
                }
            }
        }
    }
}