elpha-ios/elpha-ios/FLAnimatedImageView+LoadImageURL.swift

//
//  FLAnimatedImage+LoadImageURL.swift
//  elpha-ios
//
//  Created by Dwayne Harris on 11/12/18.
//  Copyright © 2018 Elpha. All rights reserved.
//

import Alamofire
import AlamofireImage
import FLAnimatedImage
import Foundation

class ImageCache {
    static var cache: NSCache<AnyObject, AnyObject> = NSCache<AnyObject, AnyObject>()
    
    static func set(string: String, data: Data) {
        ImageCache.cache.setObject(data as AnyObject, forKey: string as AnyObject)
    }
    
    static func set(url: URL, data: Data) {
        set(string: url.absoluteString, data: data)
    }
    
    static func set(request: URLRequest, data: Data) {
        set(url: request.url!, data: data)
    }
    
    static func get(string: String) -> Data? {
        if let data = ImageCache.cache.object(forKey: string as AnyObject) as? Data {
            return data
        } else {
            return nil
        }
    }
    
    static func get(url: URL) -> Data? {
        return get(string: url.absoluteString)
    }
    
    static func get(request: URLRequest) -> Data? {
        return get(url: request.url!)
    }
}

extension FLAnimatedImageView {
    func setPlaceholder() {
        self.contentMode = .center
        self.image = UIImage(named: "Help")
    }
    
    func loadImageURL(_ url: URL) {
        if let data = ImageCache.get(url: url) {
            if url.absoluteString.hasSuffix(".gif") {
                self.animatedImage = FLAnimatedImage(animatedGIFData: data)
            } else {
                self.image = UIImage(data: data)
            }
        } else {
            setPlaceholder()
            
            Alamofire.request(url).responseImage { response in
                if let data = response.data {
                    DispatchQueue.main.async {
                        ImageCache.set(url: url, data: data)
                        self.contentMode = .scaleAspectFill
                        
                        if url.absoluteString.hasSuffix(".gif") {
                            self.animatedImage = FLAnimatedImage(animatedGIFData: data)
                        } else {
                            self.image = UIImage(data: data)
                        }
                    }
                }
            }
        }
    }
}