import JxlCoder
// Decompress data
let uiImage: UIImage = try JXLCoder.decode(data: Data()) // or any max CGSize of image
// Compress
let data: Data = try JXLCoder.encode(data: UIImage())

// Decoding
let decoder = try! JXLAnimatedDecoder(data: animationJxlData)
let framesCount = Int(decoder.numberOfFrames)
print("frames count \(framesCount)")
let duration = decoder.frameDuration(currentFrame)
let frame: UIImage = try! decoder.get(frame: currentFrame)

// Encoding
let animEncoder = try! JXLAnimatedEncoder(width: frameToAnimate.size.width,
                                         height: frameToAnimate.size.height)
try! animEncoder.add(frame: frameToAnimate, duration: 150) // duration is in ms
try! animEncoder.add(frame: frameToAnimate, duration: 150)
try! animEncoder.add(frame: frameToAnimate, duration: 150)
try! animEncoder.add(frame: frameToAnimate, duration: 150)
try! animEncoder.add(frame: frameToAnimate, duration: 150)
// etc and then finish the encoding
let animationJxlData = try! animEncoder.finish()

let transcoded = try! JXLCoder.transcode(jpegData: Data())
let jpegData: Data = try! JXLCoder.inverse(jxlData: Data())

let encoded: Data = try! JpegLiEncoder.encode(image: UIImage())

public final class JxlNukePlugin: Nuke.ImageDecoding {
    public func decode(_ data: Data) throws -> Nuke.ImageContainer {
        guard try JXLCoder.isJXL(data: data) else { throw JXLNukePluginDecodeError() }
        let image = try JXLCoder.decode(data: data)
        return ImageContainer(image: image)
    }

    public init() {
    }

    public func decodePartiallyDownloadedData(_ data: Data) -> ImageContainer? {
        return nil
    }
}

public struct JXLNukePluginDecodeError: LocalizedError, CustomNSError {
    public var errorDescription: String? {
        "JXL file cannot be decoded"
    }

    public var errorUserInfo: [String : Any] {
        [NSLocalizedDescriptionKey: "JXL file cannot be decoded"]
    }
}

// MARK: - check JXL format data.
extension JxlNukePlugin {

    public static func enable() {
        Nuke.ImageDecoderRegistry.shared.register { (context) -> ImageDecoding? in
            JxlNukePlugin.enable(context: context)
        }
    }

    public static func enable(context: Nuke.ImageDecodingContext) -> Nuke.ImageDecoding? {
        return try? JXLCoder.isJXL(data: context.data) ? JxlNukePlugin() : nil
    }

}

#if canImport(JxlCoder)
import JxlCoder
#endif
import SDWebImage

public class JxlSDWebImageCoder: NSObject, SDImageCoder {
    public override init() {
    }

    public func canDecode(from data: Data?) -> Bool {
        guard let data else {
            return false
        }
        return (try? JXLCoder.isJXL(data: data)) ?? false
    }

    public func decodedImage(with data: Data?, options: [SDImageCoderOption : Any]? = nil) -> UIImage? {
        guard let data else {
            return nil
        }
        return try? JXLCoder.decode(data: data)
    }

    public func canEncode(to format: SDImageFormat) -> Bool {
        true
    }

    public func encodedData(with image: UIImage?, format: SDImageFormat, options: [SDImageCoderOption : Any]? = nil) -> Data? {
        guard let image else {
            return nil
        }
        return try? JXLCoder.encode(image: image)
    }
}

// don't forget to register the plugin after
SDImageCodersManager.shared.addCoder(JxlSDWebImageCoder())
// IMPORTANT: if you will use the animated Jpeg XL image you have to use other plugin
SDImageCodersManager.shared.addCoder(JxlAnimatedSDWebImageCoder())