Amir Abbas Mousavian
300 lines
15 KiB
Swift
300 lines
15 KiB
Swift
//
|
|
// OneDriveHelper.swift
|
|
// FileProvider
|
|
//
|
|
// Created by Amir Abbas Mousavian.
|
|
// Copyright © 2017 Mousavian. Distributed under MIT license.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
/// Error returned by OneDrive server when trying to access or do operations on a file or folder.
|
|
public struct FileProviderOneDriveError: FileProviderHTTPError {
|
|
public let code: FileProviderHTTPErrorCode
|
|
public let path: String
|
|
public let errorDescription: String?
|
|
}
|
|
|
|
/// Containts path, url and attributes of a OneDrive file or resource.
|
|
public final class OneDriveFileObject: FileObject {
|
|
internal init(baseURL: URL?, name: String, path: String) {
|
|
var rpath = path
|
|
if path.hasPrefix("/") {
|
|
rpath.remove(at: rpath.startIndex)
|
|
}
|
|
let url = URL(string: rpath, relativeTo: baseURL) ?? URL(string: path)!
|
|
super.init(url: url, name: name, path: path)
|
|
}
|
|
|
|
internal convenience init? (baseURL: URL?, drive: String, jsonStr: String) {
|
|
guard let json = jsonToDictionary(jsonStr) else { return nil }
|
|
self.init(baseURL: baseURL, drive: drive, json: json)
|
|
}
|
|
|
|
internal convenience init? (baseURL: URL?, drive: String, json: [String: AnyObject]) {
|
|
guard let name = json["name"] as? String else { return nil }
|
|
guard let path = (json["parentReference"] as? NSDictionary)?["path"] as? String else { return nil }
|
|
let lPath = path.replacingOccurrences(of: "/drive/\(drive):", with: "/", options: .anchored, range: nil)
|
|
self.init(baseURL: baseURL, name: name, path: lPath)
|
|
self.size = (json["size"] as? NSNumber)?.int64Value ?? -1
|
|
self.modifiedDate = resolve(dateString: json["lastModifiedDateTime"] as? String ?? "")
|
|
self.creationDate = resolve(dateString: json["createdDateTime"] as? String ?? "")
|
|
self.type = (json["folder"] as? String) != nil ? .directory : .regular
|
|
self.id = json["id"] as? String
|
|
self.entryTag = json["eTag"] as? String
|
|
}
|
|
|
|
/// The document identifier is a value assigned by the OneDrive to a file.
|
|
/// This value is used to identify the document regardless of where it is moved on a volume.
|
|
/// The identifier persists across system restarts.
|
|
open internal(set) var id: String? {
|
|
get {
|
|
return allValues[.documentIdentifierKey] as? String
|
|
}
|
|
set {
|
|
allValues[.documentIdentifierKey] = newValue
|
|
}
|
|
}
|
|
|
|
/// MIME type of file contents returned by OneDrive server.
|
|
open internal(set) var contentType: String {
|
|
get {
|
|
return allValues[.mimeType] as? String ?? ""
|
|
}
|
|
set {
|
|
allValues[.mimeType] = newValue
|
|
}
|
|
}
|
|
|
|
/// HTTP E-Tag, can be used to mark changed files.
|
|
open internal(set) var entryTag: String? {
|
|
get {
|
|
return allValues[.entryTag] as? String
|
|
}
|
|
set {
|
|
allValues[.entryTag] = newValue
|
|
}
|
|
}
|
|
}
|
|
|
|
// codebeat:disable[ARITY]
|
|
internal extension OneDriveFileProvider {
|
|
func list(_ path: String, cursor: String? = nil, prevContents: [OneDriveFileObject] = [], completionHandler: @escaping ((_ contents: [FileObject], _ cursor: String?, _ error: Error?) -> Void)) {
|
|
let url: URL
|
|
if let cursor = cursor {
|
|
url = URL(string: cursor)!
|
|
} else {
|
|
url = URL(string: escaped(path: path), relativeTo: driveURL)!
|
|
}
|
|
var request = URLRequest(url: url)
|
|
request.httpMethod = "GET"
|
|
request.setValue("Bearer \(credential?.password ?? "")", forHTTPHeaderField: "Authorization")
|
|
let task = session.dataTask(with: request, completionHandler: { (data, response, error) in
|
|
var responseError: FileProviderOneDriveError?
|
|
var files = prevContents
|
|
if let code = (response as? HTTPURLResponse)?.statusCode , code >= 300, let rCode = FileProviderHTTPErrorCode(rawValue: code) {
|
|
responseError = FileProviderOneDriveError(code: rCode, path: path, errorDescription: String(data: data ?? Data(), encoding: .utf8))
|
|
}
|
|
if let data = data, let jsonStr = String(data: data, encoding: .utf8) {
|
|
let json = jsonToDictionary(jsonStr)
|
|
if let entries = json?["value"] as? [AnyObject] , entries.count > 0 {
|
|
for entry in entries {
|
|
if let entry = entry as? [String: AnyObject], let file = OneDriveFileObject(baseURL: self.baseURL, drive: self.drive, json: entry) {
|
|
files.append(file)
|
|
}
|
|
}
|
|
let ncursor = json?["@odata.nextLink"] as? String
|
|
let hasmore = ncursor != nil
|
|
if hasmore {
|
|
self.list(path, cursor: ncursor, prevContents: files, completionHandler: completionHandler)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
completionHandler(files, nil, responseError ?? error)
|
|
})
|
|
task.taskDescription = FileOperationType.fetch(path: path).json
|
|
task.resume()
|
|
}
|
|
|
|
func upload_simple(_ targetPath: String, data: Data, modifiedDate: Date = Date(), overwrite: Bool, operation: FileOperationType, completionHandler: SimpleCompletionHandler) -> OperationHandle? {
|
|
if data.count > 100 * 1024 * 1024 {
|
|
let error = FileProviderOneDriveError(code: .payloadTooLarge, path: targetPath, errorDescription: nil)
|
|
completionHandler?(error)
|
|
self.delegateNotify(.create(path: targetPath), error: error)
|
|
return nil
|
|
}
|
|
let queryStr = overwrite ? "" : "?@name.conflictBehavior=fail"
|
|
let url = URL(string: escaped(path: targetPath) + ":/content" + queryStr, relativeTo: driveURL)!
|
|
var request = URLRequest(url: url)
|
|
request.httpMethod = "PUT"
|
|
request.setValue("Bearer \(credential?.password ?? "")", forHTTPHeaderField: "Authorization")
|
|
request.setValue("application/octet-stream", forHTTPHeaderField: "Content-Type")
|
|
request.httpBody = data
|
|
let task = session.uploadTask(with: request, from: data, completionHandler: { (data, response, error) in
|
|
var responseError: FileProviderOneDriveError?
|
|
if let code = (response as? HTTPURLResponse)?.statusCode , code >= 300, let rCode = FileProviderHTTPErrorCode(rawValue: code) {
|
|
responseError = FileProviderOneDriveError(code: rCode, path: targetPath, errorDescription: String(data: data ?? Data(), encoding: .utf8))
|
|
}
|
|
completionHandler?(responseError ?? error)
|
|
self.delegateNotify(.create(path: targetPath), error: responseError ?? error)
|
|
})
|
|
task.taskDescription = operation.json
|
|
task.resume()
|
|
return RemoteOperationHandle(operationType: operation, tasks: [task])
|
|
}
|
|
|
|
func upload_simple(_ targetPath: String, localFile: URL, modifiedDate: Date = Date(), overwrite: Bool, operation: FileOperationType, completionHandler: SimpleCompletionHandler) -> OperationHandle? {
|
|
let size = (try? localFile.resourceValues(forKeys: [.fileSizeKey]))?.fileSize ?? -1
|
|
if size > 100 * 1024 * 1024 {
|
|
let error = FileProviderOneDriveError(code: .payloadTooLarge, path: targetPath, errorDescription: nil)
|
|
completionHandler?(error)
|
|
self.delegateNotify(.create(path: targetPath), error: error)
|
|
return nil
|
|
}
|
|
let queryStr = overwrite ? "" : "?@name.conflictBehavior=fail"
|
|
let url = URL(string: escaped(path: targetPath) + ":/content" + queryStr, relativeTo: driveURL)!
|
|
var request = URLRequest(url: url)
|
|
request.httpMethod = "PUT"
|
|
request.setValue("Bearer \(credential?.password ?? "")", forHTTPHeaderField: "Authorization")
|
|
request.setValue("application/octet-stream", forHTTPHeaderField: "Content-Type")
|
|
let task = session.uploadTask(with: request, fromFile: localFile, completionHandler: { (data, response, error) in
|
|
var responseError: FileProviderOneDriveError?
|
|
if let code = (response as? HTTPURLResponse)?.statusCode , code >= 300, let rCode = FileProviderHTTPErrorCode(rawValue: code) {
|
|
responseError = FileProviderOneDriveError(code: rCode, path: targetPath, errorDescription: String(data: data ?? Data(), encoding: .utf8))
|
|
}
|
|
completionHandler?(responseError ?? error)
|
|
self.delegateNotify(.create(path: targetPath), error: responseError ?? error)
|
|
})
|
|
task.taskDescription = operation.json
|
|
task.resume()
|
|
return RemoteOperationHandle(operationType: operation, tasks: [task])
|
|
}
|
|
|
|
func search(_ startPath: String = "", query: String, next: String? = nil, foundItem:@escaping ((_ file: OneDriveFileObject) -> Void), completionHandler: @escaping ((_ error: Error?) -> Void)) {
|
|
let url: URL
|
|
if let next = next {
|
|
url = URL(string: next)!
|
|
} else if self.escaped(path: startPath) == "" {
|
|
url = URL(string: "/drive/\(drive)/view.search?q=\(query)", relativeTo: baseURL)!
|
|
} else {
|
|
url = URL(string: "\(escaped(path: startPath))/view.search?q=\(query)", relativeTo: driveURL)!
|
|
}
|
|
var request = URLRequest(url: url)
|
|
request.httpMethod = "GET"
|
|
request.setValue("Bearer \(credential?.password ?? "")", forHTTPHeaderField: "Authorization")
|
|
|
|
let task = session.dataTask(with: request, completionHandler: { (data, response, error) in
|
|
var responseError: FileProviderOneDriveError?
|
|
if let code = (response as? HTTPURLResponse)?.statusCode , code >= 300, let rCode = FileProviderHTTPErrorCode(rawValue: code) {
|
|
responseError = FileProviderOneDriveError(code: rCode, path: startPath, errorDescription: String(data: data ?? Data(), encoding: .utf8))
|
|
}
|
|
if let data = data, let jsonStr = String(data: data, encoding: .utf8) {
|
|
let json = jsonToDictionary(jsonStr)
|
|
if let entries = json?["value"] as? [AnyObject] , entries.count > 0 {
|
|
for entry in entries {
|
|
if let entry = entry as? [String: AnyObject], let file = OneDriveFileObject(baseURL: self.baseURL, drive: self.drive, json: entry) {
|
|
foundItem(file)
|
|
}
|
|
}
|
|
let next = json?["@odata.nextLink"] as? String
|
|
let hasmore = next != nil
|
|
if hasmore, let next = next {
|
|
self.search(startPath, query: query, next: next, foundItem: foundItem, completionHandler: completionHandler)
|
|
} else {
|
|
completionHandler(responseError ?? error)
|
|
}
|
|
return
|
|
}
|
|
}
|
|
completionHandler(responseError ?? error)
|
|
})
|
|
task.resume()
|
|
}
|
|
}
|
|
// codebeat:enable[ARITY]
|
|
|
|
internal extension OneDriveFileProvider {
|
|
static let dateFormatter = DateFormatter()
|
|
static let decimalFormatter = NumberFormatter()
|
|
|
|
func mapMediaInfo(_ json: [String: Any]) -> (dictionary: [String: Any], keys: [String]) {
|
|
|
|
func spaceCamelCase(_ text: String) -> String {
|
|
var newString: String = ""
|
|
|
|
let upperCase = CharacterSet.uppercaseLetters
|
|
for scalar in text.unicodeScalars {
|
|
if upperCase.contains(scalar) {
|
|
newString.append(" ")
|
|
}
|
|
let character = Character(scalar)
|
|
newString.append(character)
|
|
}
|
|
|
|
return newString.capitalized
|
|
}
|
|
|
|
var dic = [String: Any]()
|
|
var keys = [String]()
|
|
|
|
func add(key: String, value: Any?) {
|
|
if let value = value {
|
|
keys.append(key)
|
|
dic[key] = value
|
|
}
|
|
}
|
|
|
|
if let parent = json["image"] as? [String: Any] ?? json["video"] as? [String: Any], let height = parent["height"] as? UInt64, let width = parent["width"] as? UInt64 {
|
|
add(key: "Dimensions", value: "\(width)x\(height)")
|
|
}
|
|
if let location = json["location"] as? [String: Any], let latitude = location["latitude"] as? Double, let longitude = location["longitude"] as? Double {
|
|
OneDriveFileProvider.decimalFormatter.numberStyle = .decimal
|
|
OneDriveFileProvider.decimalFormatter.maximumFractionDigits = 5
|
|
let latStr = OneDriveFileProvider.decimalFormatter.string(from: NSNumber(value: latitude))
|
|
let longStr = OneDriveFileProvider.decimalFormatter.string(from: NSNumber(value: longitude))
|
|
add(key: "Location", value: "\(latStr), \(longStr)")
|
|
}
|
|
if let parent = json["image"] as? [String: Any] ?? json["video"] as? [String: Any], let duration = parent["duration"] as? UInt64 {
|
|
add(key: "Duration", value: OneDriveFileProvider.formatshort(interval: TimeInterval(duration) / 1000))
|
|
}
|
|
if let timeTakenStr = json["takenDateTime"] as? String, let timeTaken = resolve(dateString: timeTakenStr) {
|
|
OneDriveFileProvider.dateFormatter.dateFormat = "yyyy-MM-dd HH:mm:ss"
|
|
add(key: "Date taken", value: OneDriveFileProvider.dateFormatter.string(from: timeTaken))
|
|
}
|
|
|
|
if let photo = json["photo"] as? [String: Any] {
|
|
add(key: "Device make", value: photo["cameraMake"] as? String)
|
|
add(key: "Device model", value: photo["cameraModel"] as? String)
|
|
add(key: "focalLength", value: photo["focalLength"] as? Double)
|
|
add(key: "fNumber", value: photo["fNumber"] as? Double)
|
|
if let expNom = photo["exposureNumerator"] as? Double, let expDen = photo["exposureDenominator"] as? Double {
|
|
add(key: "Exposure time", value: "\(Int(expNom))/\(Int(expDen))")
|
|
}
|
|
add(key: "ISO speed", value: photo["iso"] as? Int64)
|
|
}
|
|
|
|
if let audio = json["audio"] as? [String: Any] {
|
|
for (key, value) in audio {
|
|
if key == "bitrate" || key == "isVariableBitrate" { continue }
|
|
let casedKey = spaceCamelCase(key)
|
|
add(key: casedKey, value: value)
|
|
}
|
|
}
|
|
|
|
add(key: "Bitrate", value: (json["video"] as? NSDictionary)?["bitrate"] as? Int)
|
|
|
|
return (dic, keys)
|
|
}
|
|
|
|
func delegateNotify(_ operation: FileOperationType, error: Error?) {
|
|
DispatchQueue.main.async(execute: {
|
|
if error == nil {
|
|
self.delegate?.fileproviderSucceed(self, operation: operation)
|
|
} else {
|
|
self.delegate?.fileproviderFailed(self, operation: operation)
|
|
}
|
|
})
|
|
}
|
|
}
|