PDF_Office/PDF Office/PDF Master/Class/Common/Tools/HTTP/KMHTTP.swift

//
//  KMHTTP.swift
//  PDF Reader Pro
//
//  Created by User-Tangchao on 2024/10/23.
//

import Cocoa

public class KMRequestResultModel: NSObject {
    public var code: Int = 0
    public var msg: String?
    public var data: Any?
    
    convenience init(dict: [String : Any]) {
        self.init()
        
        self.code = dict["code"] as? Int ?? 0
        self.msg = dict["msg"] as? String
        self.data = dict["data"]
    }
    
    public func isSuccess() -> Bool {
        return self.code == 200
    }
}

class KMHTTP: NSObject {
    private static let baseUrl = "https://api-server.compdf.com/server/"
    private static let boundary = "kBoundary"
    
    // MARK: - Base
    
    public class func GET(urlString: String, parameter: [String : String]? = nil, headers: [String : String]? = nil, callback:@escaping ((Bool, [String : Any]?, String?)->Void)) {
        var _urlString = "\(self.baseUrl)"+urlString
        if let data = parameter, !data.isEmpty {
            _urlString.append("?")
            var i = 0
            for (key, value) in data {
                _urlString.append("\(key)=\(value)")
                if (data.count > 1 && i != data.count-1) {
                    _urlString.append("&")
                }
                i += 1
            }
        }
        
        let url: URL = URL(string: _urlString)!
         
        let session = URLSession.shared
        var request = URLRequest(url: url)
        request.httpMethod = "GET"
        request.setValue("application/json; charset=UTF-8", forHTTPHeaderField: "Content-Type")
        if let _headers = headers {
            for (key, value) in _headers {
                request.setValue(value, forHTTPHeaderField: key)
            }
        }
        request.timeoutInterval = 60.0
        session.configuration.timeoutIntervalForRequest = 30.0
        
        let task: URLSessionDataTask = session.dataTask(with: request) { data , response, error in
            DispatchQueue.main.async {
                if let _ = error {
                    callback(false, nil, error.debugDescription)
                    return
                }
                guard let _data = data else {
                    callback(false, nil, error.debugDescription)
                    return
                }
                if let result = self.JsonDataParse(data: _data) {
                    let resultMap = KMRequestResultModel(dict: result)
                    var dataDict: [String : Any] = [:]
                    if let data = resultMap.data as? [String : Any] {
                        dataDict = data
                    } else {
                        if let dataArray = resultMap.data as? [Any] {
                            var i = 0
                            for dict in dataArray {
                                dataDict["\(i)"] = dict
                                i += 1
                            }
                        }
                    }

                    callback(resultMap.isSuccess(), dataDict, error.debugDescription)
                    return
                }
                callback(false, nil, error.debugDescription)
            }
        }
        task.resume()
    }
    
    public class func POST(urlString: String, parameter: [String : Any]?, headers: [String : String]?, callback:@escaping ((Bool, KMRequestResultModel? , String?)->Void)) {
        let url: URL = URL(string: urlString)!
         
        let session = URLSession.shared
        var request = URLRequest(url: url)
        request.httpMethod = "POST"
        request.setValue("application/json; charset=UTF-8", forHTTPHeaderField: "Content-Type")
        request.setValue("application/json; charset=UTF-8", forHTTPHeaderField: "Accept")
        if let _headers = headers {
            for (key, value) in _headers {
                if let con = request.allHTTPHeaderFields?.keys.contains(key), con {
                    request.addValue(value, forHTTPHeaderField: key)
                } else {
                    request.setValue(value, forHTTPHeaderField: key)
                }
            }
        }
        
        request.timeoutInterval = 60.0
        session.configuration.timeoutIntervalForRequest = 30.0
        
        if let data = parameter {
            let body = try?JSONSerialization.data(withJSONObject: data, options: [])
            request.httpBody = body
        }
        
        let task: URLSessionDataTask = session.dataTask(with: request) { data , response, error in
            DispatchQueue.main.async {
                if let _ = error {
                    callback(false, nil, error.debugDescription)
                    return
                }
                guard let _data = data else {
                    callback(false, nil, error.debugDescription)
                    return
                }
                if let result = self.JsonDataParse(data: _data) {
                    let resultMap = KMRequestResultModel(dict: result)
                    callback(resultMap.isSuccess(), resultMap, error.debugDescription)
                    return
                }
//
                callback(false, nil, error.debugDescription)
            }
        }
        task.resume()
    }
    
    public class func UploadFile(urlString: String, parameter: [String : Any]? = nil, headers: [String : String]? = nil, filepath: String, callback:@escaping ((Bool, [String : Any]?, String?)->Void)) {
        let url = URL(string: self.baseUrl + urlString)
        var request = URLRequest(url: url!)
        request.httpMethod = "POST"
        if let _headers = headers {
            for (key, value) in _headers {
                request.setValue(value, forHTTPHeaderField: key)
            }
        }
        
        request.setValue("multipart/form-data; boundary=\(self.boundary)", forHTTPHeaderField: "Content-Type")
        
        var bodyString = ""
        var imageFilepath = ""
        if let data = parameter {
            for (key,value) in data {
                bodyString.append("--\(self.boundary)\r\n")
                if (value is String) {
                    if (key == "filepath") {
                        imageFilepath = value as! String
                    } else {
                        bodyString.append("Content-disposition: form-data; name=\"\(key)\"")
                        bodyString.append("\r\n\r\n")
                        bodyString.append(value as! String)
                        bodyString.append("\r\n")
                    }
                } else {
                    // no things
                }
            }
        }
        
        let fileUrl = URL(fileURLWithPath: filepath)
        bodyString.append("--\(self.boundary)\r\n")
        bodyString.append("Content-disposition: form-data; name=\"file\"; filename=\"\(fileUrl.lastPathComponent)\"")
        bodyString.append("\r\n")
        bodyString.append("Content-Type: application/octet-stream")
        bodyString.append("\r\n\r\n")
        
        var mdata = Data()
        mdata.append(bodyString.data(using: .utf8)!)
        if let data = try?Data(contentsOf: fileUrl) {
            mdata.append(data)
        }
        
        mdata.append("\r\n".data(using: .utf8)!)
        
        if (!imageFilepath.isEmpty) {
            var dataString = ""
            
            let fileUrl = URL(fileURLWithPath: imageFilepath)
            dataString.append("--\(self.boundary)\r\n")
            dataString.append("Content-disposition: form-data; name=\"image\"; filename=\"\(fileUrl.lastPathComponent)\"")
            dataString.append("\r\n")
            dataString.append("Content-Type: application/octet-stream")
            dataString.append("\r\n\r\n")
//            var mdata = Data()
            mdata.append(dataString.data(using: .utf8)!)
            if let data = try?Data(contentsOf: fileUrl) {
                mdata.append(data)
            }
            
            mdata.append("\r\n".data(using: .utf8)!)
        }
        
        mdata.append("--\(self.boundary)--".data(using: .utf8)!)
        
        let session = URLSession.shared
//        request.timeoutInterval = 120

        request.httpBody = mdata
        let task = session.dataTask(with: request) {data , response, error in
            DispatchQueue.main.async {
                if let _ = error {
                    callback(false, nil, error.debugDescription)
                    return
                }
                guard let _data = data else {
                    callback(false, nil, error.debugDescription)
                    return
                }
                if let result = self.JsonDataParse(data: _data) {
                    let resultMap = KMRequestResultModel(dict: result)
                    callback(resultMap.isSuccess(), resultMap.data as? [String : Any], error.debugDescription)
                    return
                }
                
                callback(false, nil, error.debugDescription)
            }
        }
        task.resume()
    }
    
    // MARK: - Private Methods
    
    private class func JsonDataParse(data: Data) -> Dictionary<String,Any>? {
        let result = try?JSONSerialization.jsonObject(with: data, options: .mutableContainers)
        return result as? Dictionary<String, Any>
    }
}