114 lines
3.8 KiB
Swift
114 lines
3.8 KiB
Swift
//
|
|
// CloudTransferWorkers.swift
|
|
// suixinkan
|
|
//
|
|
|
|
import Foundation
|
|
import Photos
|
|
|
|
/// 云盘 OSS 上传协议，便于传输管理注入测试替身。
|
|
@MainActor
|
|
protocol CloudDriveUploading {
|
|
/// 上传云盘文件并返回远程 URL。
|
|
func uploadCloudDriveFile(
|
|
data: Data,
|
|
fileName: String,
|
|
fileType: Int,
|
|
scenicId: Int,
|
|
onProgress: @escaping (Int) -> Void
|
|
) async throws -> String
|
|
}
|
|
|
|
extension OSSUploadService: CloudDriveUploading {}
|
|
|
|
/// 云盘下载协议。
|
|
protocol CloudFileDownloading {
|
|
/// 下载远程文件并返回本地临时文件 URL。
|
|
func downloadFile(
|
|
remoteURL: String,
|
|
fileName: String,
|
|
onProgress: @escaping @Sendable (Int) -> Void
|
|
) async throws -> URL
|
|
}
|
|
|
|
/// URLSession 实现的云盘下载器。
|
|
struct CloudURLSessionDownloader: CloudFileDownloading {
|
|
func downloadFile(
|
|
remoteURL: String,
|
|
fileName: String,
|
|
onProgress: @escaping @Sendable (Int) -> Void
|
|
) async throws -> URL {
|
|
guard let url = URL(string: remoteURL) else {
|
|
throw CloudTransferError.invalidURL
|
|
}
|
|
onProgress(1)
|
|
let (temporaryURL, _) = try await URLSession.shared.download(from: url)
|
|
try Task.checkCancellation()
|
|
let directory = FileManager.default.temporaryDirectory.appendingPathComponent("cloud_drive_downloads", isDirectory: true)
|
|
try FileManager.default.createDirectory(at: directory, withIntermediateDirectories: true)
|
|
let destinationURL = directory.appendingPathComponent(CloudTransferFileName.sanitized(fileName))
|
|
if FileManager.default.fileExists(atPath: destinationURL.path) {
|
|
try FileManager.default.removeItem(at: destinationURL)
|
|
}
|
|
try FileManager.default.moveItem(at: temporaryURL, to: destinationURL)
|
|
onProgress(100)
|
|
return destinationURL
|
|
}
|
|
}
|
|
|
|
/// 云盘下载后的系统相册保存协议。
|
|
protocol CloudPhotoSaving {
|
|
/// 将图片或视频保存到系统相册。
|
|
func saveToPhotoLibrary(fileURL: URL, fileType: Int) async throws
|
|
}
|
|
|
|
/// Photos 框架实现的相册保存器。
|
|
struct CloudPhotoLibrarySaver: CloudPhotoSaving {
|
|
func saveToPhotoLibrary(fileURL: URL, fileType: Int) async throws {
|
|
let status = await PHPhotoLibrary.requestAuthorization(for: .addOnly)
|
|
guard status == .authorized || status == .limited else {
|
|
throw CloudTransferError.photoPermissionDenied
|
|
}
|
|
|
|
try await PHPhotoLibrary.shared().performChanges {
|
|
if fileType == 1 {
|
|
PHAssetChangeRequest.creationRequestForAssetFromVideo(atFileURL: fileURL)
|
|
} else {
|
|
PHAssetChangeRequest.creationRequestForAssetFromImage(atFileURL: fileURL)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// 云盘传输错误。
|
|
enum CloudTransferError: LocalizedError, Equatable {
|
|
case invalidURL
|
|
case localFileMissing
|
|
case photoPermissionDenied
|
|
|
|
var errorDescription: String? {
|
|
switch self {
|
|
case .invalidURL:
|
|
"文件地址不存在"
|
|
case .localFileMissing:
|
|
"本地文件不存在"
|
|
case .photoPermissionDenied:
|
|
"请允许访问相册后再下载"
|
|
}
|
|
}
|
|
}
|
|
|
|
/// 云盘传输文件名工具。
|
|
enum CloudTransferFileName {
|
|
/// 清理文件名中不适合落盘的字符。
|
|
static func sanitized(_ fileName: String) -> String {
|
|
let trimmed = fileName.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
var unsafe = CharacterSet.controlCharacters
|
|
unsafe.formUnion(CharacterSet(charactersIn: "/\\"))
|
|
let value = trimmed.unicodeScalars.reduce(into: "") { result, scalar in
|
|
result += unsafe.contains(scalar) ? "_" : String(scalar)
|
|
}
|
|
return value.isEmpty ? "cloud_file" : value
|
|
}
|
|
}
|