mirror of
https://github.com/firebase/firebase-ios-sdk
synced 2026-11-10 22:45:59 +05:45
manjanac
e6b60202d4
* Restores ML Pods after M77. * Fix Package.swift * Refactor model info to be in-memory. * Add improved model download functionality w/ single set of download handlers per task. * Rename model download to better match URLSessionTask. * Rename model download to better match URLSessionTask. * Update unit test to read/write to user defaults. * Configure downloader with Firebase app. * Minor updates to model downloader and tests. * Singleton for model downloader instance, for default and custom app. * Manually add app lifecycle handling + TODOs for Firebase Components. * Replace options with app name as a property in ModelDownloader. * Update unit test to check model downloader instance creation + minor rename.
80 lines
2.9 KiB
Swift
80 lines
2.9 KiB
Swift
// Copyright 2020 Google LLC
|
|
//
|
|
// 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 Foundation
|
|
import FirebaseCore
|
|
|
|
/// Model info object with details about pending or downloaded model.
|
|
struct ModelInfo {
|
|
/// Model name.
|
|
let name: String
|
|
|
|
// TODO: revisit UserDefaultsBacked
|
|
/// Download URL for the model file, as returned by server.
|
|
let downloadURL: URL
|
|
|
|
/// Hash of the model, as returned by server.
|
|
let modelHash: String
|
|
|
|
/// Size of the model, as returned by server.
|
|
let size: Int
|
|
|
|
/// Local path of the model.
|
|
var path: String?
|
|
|
|
/// Initialize model info and create user default keys.
|
|
init(name: String, downloadURL: URL, modelHash: String, size: Int) {
|
|
self.name = name
|
|
self.downloadURL = downloadURL
|
|
self.modelHash = modelHash
|
|
self.size = size
|
|
}
|
|
|
|
/// Get user defaults key prefix.
|
|
private static func getUserDefaultsKeyPrefix(appName: String, modelName: String) -> String {
|
|
let bundleID = Bundle.main.bundleIdentifier ?? ""
|
|
return "\(bundleID).\(appName).\(modelName)"
|
|
}
|
|
|
|
// TODO: Move reading and writing to user defaults to a new file.
|
|
init?(fromDefaults defaults: UserDefaults, modelName: String, appName: String) {
|
|
let defaultsPrefix = ModelInfo.getUserDefaultsKeyPrefix(appName: appName, modelName: modelName)
|
|
guard let downloadURL = defaults
|
|
.value(forKey: "\(defaultsPrefix).model-download-url") as? String,
|
|
let url = URL(string: downloadURL),
|
|
let modelHash = defaults.value(forKey: "\(defaultsPrefix).model-hash") as? String,
|
|
let size = defaults.value(forKey: "\(defaultsPrefix).model-size") as? Int,
|
|
let path = defaults.value(forKey: "\(defaultsPrefix).model-path") as? String else {
|
|
return nil
|
|
}
|
|
name = modelName
|
|
self.downloadURL = url
|
|
self.modelHash = modelHash
|
|
self.size = size
|
|
self.path = path
|
|
}
|
|
|
|
func writeToDefaults(_ defaults: UserDefaults, appName: String) throws {
|
|
guard let modelPath = path else {
|
|
throw DownloadedModelError
|
|
.fileIOError(description: "Could not save model info to user defaults.")
|
|
}
|
|
let defaultsPrefix = ModelInfo.getUserDefaultsKeyPrefix(appName: appName, modelName: name)
|
|
defaults.setValue(downloadURL.absoluteString, forKey: "\(defaultsPrefix).model-download-url")
|
|
defaults.setValue(modelHash, forKey: "\(defaultsPrefix).model-hash")
|
|
defaults.setValue(size, forKey: "\(defaultsPrefix).model-size")
|
|
defaults.setValue(modelPath, forKey: "\(defaultsPrefix).model-path")
|
|
}
|
|
}
|