import Foundation
import UIKit
import AnimationCache
import Display
import RLottieBinding
import GZip
import WebPBinding

public func cacheLottieAnimation(data: Data, width: Int, height: Int, writer: AnimationCacheItemWriter) {
    writer.queue.async {
        let decompressedData = TGGUnzipData(data, 1 * 1024 * 1024) ?? data
        guard let animation = LottieInstance(data: decompressedData, fitzModifier: .none, colorReplacements: nil, cacheKey: "") else {
            writer.finish()
            return
        }
        
        let frameDuration = 1.0 / Double(animation.frameRate)
        for i in 0 ..< animation.frameCount {
            if writer.isCancelled {
                break
            }
            writer.add(with: { surface in
                animation.renderFrame(with: i, into: surface.argb, width: Int32(surface.width), height: Int32(surface.height), bytesPerRow: Int32(surface.bytesPerRow))
            }, proposedWidth: width, proposedHeight: height, duration: frameDuration)
        }
        
        writer.finish()
    }
}

public func cacheStillSticker(path: String, width: Int, height: Int, writer: AnimationCacheItemWriter) {
    writer.queue.async {
        if let data = try? Data(contentsOf: URL(fileURLWithPath: path)), let image = WebP.convert(fromWebP: data) {
            writer.add(with: { surface in
                let context = DrawingContext(size: CGSize(width: CGFloat(surface.width), height: CGFloat(surface.height)), scale: 1.0, opaque: false, clear: true, bytesPerRow: surface.bytesPerRow)
                context.withFlippedContext { c in
                    UIGraphicsPushContext(c)
                    c.draw(image.cgImage!, in: CGRect(origin: CGPoint(), size: context.size))
                    UIGraphicsPopContext()
                }
                memcpy(surface.argb, context.bytes, surface.height * surface.bytesPerRow)
            }, proposedWidth: width, proposedHeight: height, duration: 1.0)
        }
        
        writer.finish()
    }
}