AppBuilderiOS/appbuilder-ios/NexilisLite/NexilisLite/Source/Download.swift

//
//  Download.swift
//  Runner
//
//  Created by Yayan Dwi on 24/04/20.
//  Copyright © 2020 The Chromium Authors. All rights reserved.
//

import Foundation
import Alamofire

public class Download {
    
    public init() {}
    
    var delegate : DownloadDelegate?
    
    public func getDelegate() -> DownloadDelegate? {
        return delegate
    }
    
    private var downloadBufferQueue = DispatchQueue(label: "DOWNLOAD_BUFFER", attributes: .concurrent)
    
    var DOWNLOAD_BUFFER = [Data?]()
    var DOWNLOAD_SESSION = [Session]()
    var DOWNLOAD_URL = "https://newuniverse.io/filepalio/image/"
    
    public func start(forKey: String, delegate: DownloadDelegate){
        self.delegate = delegate
        let download = Nexilis.getDownload(forKey: forKey)
        if download == nil {
            Nexilis.addDownload(forKey: forKey, download: self)
        }
        _ = Nexilis.write(message: CoreMessage_TMessageBank.getImageDownload(p_image_id: forKey))
    }
    
    var onDownloadProgress: ((String, Double) -> ())?
    
    public func start(forKey: String, completion: @escaping (String, Double)->()) {
        self.onDownloadProgress = completion
        let download = Nexilis.getDownload(forKey: forKey)
        if download == nil {
            Nexilis.addDownload(forKey: forKey, download: self)
        }
        _ = Nexilis.write(message: CoreMessage_TMessageBank.getImageDownload(p_image_id: forKey))
    }
    
    public func startHTTP(forKey: String, completion: @escaping (String, Double)->()) {
        _ = startHTTP(filename: forKey, baseURL: DOWNLOAD_URL, completion: completion)
    }
    
    public func startHTTP(filename: String, baseURL: String, completion: @escaping (String, Double)->()) -> DownloadRequest {
        var sep = ""
        if baseURL.last != "/" {
            sep = "/"
        }
        let fullURL = "\(baseURL)\(sep)\(filename)"
        do {
//            let destination = DownloadRequest.suggestedDownloadDestination(for: .documentDirectory, in: .userDomainMask, options: [.removePreviousFile, .createIntermediateDirectories])
            let downloadRequest = AF.download(fullURL)
            .downloadProgress(queue: downloadBufferQueue) { progress in
                completion(filename,progress.fractionCompleted*100)
            }
            .responseData { result in
                if let successResponse = result.value {
                    //print("Response success")
                    do {
                        let documentDir = try FileManager.default.url(for: .documentDirectory, in: .userDomainMask, appropriateFor: nil, create: true)
                        let url = documentDir.appendingPathComponent(filename)
                        //print("write file \(url.path)")
                        try successResponse.write(to: url)
                        completion(filename,100)
                    }
                    catch {}
                }
                else {
                    let statusCode = result.response?.statusCode
                    //print("Response fail: \(statusCode)")
                    completion(filename,0)
                }
            }
            return downloadRequest
        }
        catch {}
        
        
    }
    
    func put(part: Int, buffer: Data){
        downloadBufferQueue.async (flags: .barrier){
            self.DOWNLOAD_BUFFER.insert(buffer, at: part)
        }
    }
    
    func size() -> Int {
        var size = 0
        downloadBufferQueue.sync {
            for b in DOWNLOAD_BUFFER {
                size += b?.count ?? 0
            }
        }
        return size
    }
    
    func remove() -> Data {
        var result = Data()
        downloadBufferQueue.sync {
            for i in DOWNLOAD_BUFFER {
                if let b = i {
                    result.append(contentsOf: b)
                }
            }
        }
        return result
    }
}