mirror of https://github.com/oxen-io/session-ios
Matthew Chen
8 years ago
parent
94f02c0d1d
commit
2282733fa9
@ -0,0 +1,217 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
import Foundation
|
||||
|
||||
@objc
|
||||
protocol DataSource {
|
||||
// This method should not be called unless necessary as it
|
||||
// be expensive.
|
||||
func data() -> Data
|
||||
|
||||
func dataUrl(fileExtension: String) -> URL?
|
||||
func dataPath(fileExtension: String) -> String?
|
||||
func dataPathIfOnDisk() -> String?
|
||||
func dataLength() -> Int
|
||||
}
|
||||
|
||||
@objc
|
||||
class DataSourceValue: NSObject, DataSource {
|
||||
static let TAG = "[DataSourceValue]"
|
||||
|
||||
private let value: Data
|
||||
|
||||
private var path: String?
|
||||
|
||||
// MARK: Constructor
|
||||
|
||||
internal required init(_ value: Data) {
|
||||
self.value = value
|
||||
super.init()
|
||||
}
|
||||
|
||||
func data() -> Data {
|
||||
return value
|
||||
}
|
||||
|
||||
func dataUrl(fileExtension: String) -> URL? {
|
||||
guard let path = dataPath(fileExtension:fileExtension) else {
|
||||
return nil
|
||||
}
|
||||
return URL(fileURLWithPath: path)
|
||||
}
|
||||
|
||||
func dataPath(fileExtension: String) -> String? {
|
||||
if let path = path {
|
||||
return path
|
||||
}
|
||||
|
||||
let directory = NSTemporaryDirectory()
|
||||
let fileName = NSUUID().uuidString + "." + fileExtension
|
||||
let filePath = (directory as NSString).appendingPathComponent(fileName)
|
||||
do {
|
||||
try value.write(to: URL(fileURLWithPath:filePath))
|
||||
path = filePath
|
||||
} catch {
|
||||
owsFail("\(DataSourceValue.TAG) Could not write data to disk: \(fileExtension)")
|
||||
}
|
||||
return filePath
|
||||
}
|
||||
|
||||
func dataPathIfOnDisk() -> String? {
|
||||
if let path = path {
|
||||
return path
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func dataLength() -> Int {
|
||||
return value.count
|
||||
}
|
||||
|
||||
class func empty() -> DataSource {
|
||||
return DataSourceValue(Data())
|
||||
}
|
||||
}
|
||||
|
||||
@objc
|
||||
class DataSourcePath: NSObject, DataSource {
|
||||
static let TAG = "[DataSourcePath]"
|
||||
|
||||
private let path: String
|
||||
|
||||
private var cachedData: Data?
|
||||
|
||||
private var cachedLength: Int?
|
||||
|
||||
// MARK: Constructor
|
||||
|
||||
internal required init(_ path: String) {
|
||||
self.path = path
|
||||
super.init()
|
||||
}
|
||||
|
||||
func data() -> Data {
|
||||
if let cachedData = cachedData {
|
||||
return cachedData
|
||||
}
|
||||
Logger.error("\(DataSourcePath.TAG) reading data: \(path)")
|
||||
do {
|
||||
try cachedData = NSData(contentsOfFile:path) as Data
|
||||
} catch {
|
||||
owsFail("\(DataSourcePath.TAG) Could not read data from disk: \(path)")
|
||||
cachedData = Data()
|
||||
}
|
||||
return cachedData!
|
||||
}
|
||||
|
||||
func dataUrl(fileExtension: String) -> URL? {
|
||||
return URL(fileURLWithPath: path)
|
||||
}
|
||||
|
||||
func dataPath(fileExtension: String) -> String? {
|
||||
return path
|
||||
}
|
||||
|
||||
func dataPathIfOnDisk() -> String? {
|
||||
return path
|
||||
}
|
||||
|
||||
func dataLength() -> Int {
|
||||
if let cachedLength = cachedLength {
|
||||
return cachedLength
|
||||
}
|
||||
|
||||
do {
|
||||
let fileAttributes = try FileManager.default.attributesOfItem(atPath: path)
|
||||
let fileSize = fileAttributes[FileAttributeKey.size] as! UInt64
|
||||
cachedLength = Int(fileSize)
|
||||
} catch {
|
||||
owsFail("\(DataSourcePath.TAG) Could not read data length from disk: \(path)")
|
||||
cachedLength = 0
|
||||
}
|
||||
|
||||
return cachedLength!
|
||||
}
|
||||
}
|
||||
|
||||
@objc
|
||||
class DataSourceUrl: NSObject, DataSource {
|
||||
static let TAG = "[DataSourceUrl]"
|
||||
|
||||
private let url: URL
|
||||
|
||||
private var cachedData: Data?
|
||||
|
||||
private var cachedLength: Int?
|
||||
|
||||
// MARK: Constructor
|
||||
|
||||
internal required init(_ url: URL) {
|
||||
if !url.isFileURL {
|
||||
owsFail("\(DataSourceUrl.TAG) URL is not a file URL: \(url)")
|
||||
}
|
||||
self.url = url
|
||||
super.init()
|
||||
}
|
||||
|
||||
func data() -> Data {
|
||||
if let cachedData = cachedData {
|
||||
return cachedData
|
||||
}
|
||||
guard url.isFileURL else {
|
||||
owsFail("\(DataSourceUrl.TAG) URL is not a file URL: \(url)")
|
||||
return Data()
|
||||
}
|
||||
Logger.error("\(DataSourceUrl.TAG) reading data: \(url)")
|
||||
do {
|
||||
try cachedData = Data(contentsOf:url)
|
||||
} catch {
|
||||
owsFail("\(DataSourceUrl.TAG) Could not read data from disk: \(url)")
|
||||
cachedData = Data()
|
||||
}
|
||||
return cachedData!
|
||||
}
|
||||
|
||||
func dataUrl(fileExtension: String) -> URL? {
|
||||
return url
|
||||
}
|
||||
|
||||
func dataPath(fileExtension: String) -> String? {
|
||||
guard url.isFileURL else {
|
||||
owsFail("\(DataSourceUrl.TAG) URL is not a file URL: \(url)")
|
||||
return nil
|
||||
}
|
||||
return url.path
|
||||
}
|
||||
|
||||
func dataPathIfOnDisk() -> String? {
|
||||
guard url.isFileURL else {
|
||||
owsFail("\(DataSourceUrl.TAG) URL is not a file URL: \(url)")
|
||||
return nil
|
||||
}
|
||||
return url.path
|
||||
}
|
||||
|
||||
func dataLength() -> Int {
|
||||
if let cachedLength = cachedLength {
|
||||
return cachedLength
|
||||
}
|
||||
guard url.isFileURL else {
|
||||
owsFail("\(DataSourceUrl.TAG) URL is not a file URL: \(url)")
|
||||
return 0
|
||||
}
|
||||
|
||||
do {
|
||||
let fileAttributes = try FileManager.default.attributesOfItem(atPath: url.path)
|
||||
let fileSize = fileAttributes[FileAttributeKey.size] as! UInt64
|
||||
cachedLength = Int(fileSize)
|
||||
} catch {
|
||||
owsFail("\(DataSourceUrl.TAG) Could not read data length from disk: \(url)")
|
||||
cachedLength = 0
|
||||
}
|
||||
|
||||
return cachedLength!
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue