117 lines
4.0 KiB
Swift
117 lines
4.0 KiB
Swift
//
|
|
// Copyright 2022 New Vector Ltd
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
import Kingfisher
|
|
import UIKit
|
|
|
|
struct MediaProvider: MediaProviderProtocol {
|
|
private let mediaLoader: MediaLoaderProtocol
|
|
private let imageCache: Kingfisher.ImageCache
|
|
private let backgroundTaskService: BackgroundTaskServiceProtocol?
|
|
|
|
init(mediaLoader: MediaLoaderProtocol,
|
|
imageCache: Kingfisher.ImageCache,
|
|
backgroundTaskService: BackgroundTaskServiceProtocol?) {
|
|
self.mediaLoader = mediaLoader
|
|
self.imageCache = imageCache
|
|
self.backgroundTaskService = backgroundTaskService
|
|
}
|
|
|
|
// MARK: Images
|
|
|
|
func imageFromSource(_ source: MediaSourceProxy?, size: CGSize?) -> UIImage? {
|
|
guard let url = source?.url else {
|
|
return nil
|
|
}
|
|
let cacheKey = cacheKeyForURL(url, size: size)
|
|
return imageCache.retrieveImageInMemoryCache(forKey: cacheKey, options: nil)
|
|
}
|
|
|
|
func loadImageFromSource(_ source: MediaSourceProxy, size: CGSize?) async -> Result<UIImage, MediaProviderError> {
|
|
if let image = imageFromSource(source, size: size) {
|
|
return .success(image)
|
|
}
|
|
|
|
let loadImageBgTask = await backgroundTaskService?.startBackgroundTask(withName: "LoadImage: \(source.url.hashValue)")
|
|
defer {
|
|
loadImageBgTask?.stop()
|
|
}
|
|
|
|
let cacheKey = cacheKeyForURL(source.url, size: size)
|
|
|
|
if case let .success(cacheResult) = await imageCache.retrieveImage(forKey: cacheKey),
|
|
let image = cacheResult.image {
|
|
return .success(image)
|
|
}
|
|
|
|
do {
|
|
let imageData: Data
|
|
if let size {
|
|
imageData = try await mediaLoader.loadMediaThumbnailForSource(source, width: UInt(size.width), height: UInt(size.height))
|
|
} else {
|
|
imageData = try await mediaLoader.loadMediaContentForSource(source)
|
|
}
|
|
|
|
guard let image = UIImage(data: imageData) else {
|
|
MXLog.error("Invalid image data")
|
|
return .failure(.invalidImageData)
|
|
}
|
|
|
|
imageCache.store(image, forKey: cacheKey)
|
|
|
|
return .success(image)
|
|
} catch {
|
|
MXLog.error("Failed retrieving image with error: \(error)")
|
|
return .failure(.failedRetrievingImage)
|
|
}
|
|
}
|
|
|
|
// MARK: Files
|
|
|
|
func loadFileFromSource(_ source: MediaSourceProxy) async -> Result<MediaFileHandleProxy, MediaProviderError> {
|
|
let loadFileBgTask = await backgroundTaskService?.startBackgroundTask(withName: "LoadFile: \(source.url.hashValue)")
|
|
defer { loadFileBgTask?.stop() }
|
|
|
|
do {
|
|
let file = try await mediaLoader.loadMediaFileForSource(source)
|
|
return .success(file)
|
|
} catch {
|
|
MXLog.error("Failed retrieving file with error: \(error)")
|
|
return .failure(.failedRetrievingFile)
|
|
}
|
|
}
|
|
|
|
// MARK: - Private
|
|
|
|
private func cacheKeyForURL(_ url: URL, size: CGSize?) -> String {
|
|
if let size {
|
|
return "\(url.absoluteString){\(size.width),\(size.height)}"
|
|
} else {
|
|
return url.absoluteString
|
|
}
|
|
}
|
|
}
|
|
|
|
private extension ImageCache {
|
|
func retrieveImage(forKey key: String) async -> Result<ImageCacheResult, KingfisherError> {
|
|
await withCheckedContinuation { continuation in
|
|
retrieveImage(forKey: key) { result in
|
|
continuation.resume(returning: result)
|
|
}
|
|
}
|
|
}
|
|
}
|