132 lines
4.2 KiB
Swift
132 lines
4.2 KiB
Swift
import Foundation
|
|
import UIKit
|
|
|
|
class GIFFileManager {
|
|
static let shared = GIFFileManager()
|
|
|
|
private init() {
|
|
createGIFsDirectoryIfNeeded()
|
|
}
|
|
|
|
// MARK: - File Storage
|
|
|
|
private var documentsDirectory: URL {
|
|
let paths = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)
|
|
return paths[0]
|
|
}
|
|
|
|
private var gifsDirectory: URL {
|
|
return documentsDirectory.appendingPathComponent("SavedGIFs", isDirectory: true)
|
|
}
|
|
|
|
private func createGIFsDirectoryIfNeeded() {
|
|
let fileManager = FileManager.default
|
|
|
|
if !fileManager.fileExists(atPath: gifsDirectory.path) {
|
|
do {
|
|
try fileManager.createDirectory(at: gifsDirectory, withIntermediateDirectories: true)
|
|
} catch {
|
|
print("Error creating GIFs directory: \(error)")
|
|
}
|
|
}
|
|
}
|
|
|
|
func storeGIF(data: Data, fromURL urlString: String) -> String? {
|
|
// Create a unique filename based on the URL hash and timestamp
|
|
let urlHash = urlString.hashValue
|
|
let timestamp = Int(Date().timeIntervalSince1970)
|
|
let filename = "gif_\(urlHash)_\(timestamp).gif"
|
|
|
|
let fileURL = gifsDirectory.appendingPathComponent(filename)
|
|
|
|
do {
|
|
try data.write(to: fileURL)
|
|
return fileURL.path
|
|
} catch {
|
|
print("Error saving GIF to disk: \(error)")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func loadGIFData(from localPath: String) -> Data? {
|
|
let url = URL(fileURLWithPath: localPath)
|
|
|
|
do {
|
|
let data = try Data(contentsOf: url)
|
|
return data
|
|
} catch {
|
|
print("Error loading GIF from disk: \(error)")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func deleteGIF(at localPath: String) -> Bool {
|
|
let fileManager = FileManager.default
|
|
|
|
do {
|
|
try fileManager.removeItem(atPath: localPath)
|
|
return true
|
|
} catch {
|
|
print("Error deleting GIF from disk: \(error)")
|
|
return false
|
|
}
|
|
}
|
|
|
|
func fileExists(at localPath: String) -> Bool {
|
|
return FileManager.default.fileExists(atPath: localPath)
|
|
}
|
|
|
|
func getAllGIFsSize() -> Int64 {
|
|
let fileManager = FileManager.default
|
|
let enumerator = fileManager.enumerator(at: gifsDirectory, includingPropertiesForKeys: [.fileSizeKey])
|
|
|
|
var totalSize: Int64 = 0
|
|
|
|
while let fileURL = enumerator?.nextObject() as? URL {
|
|
do {
|
|
let attributes = try fileURL.resourceValues(forKeys: [.fileSizeKey])
|
|
if let fileSize = attributes.fileSize {
|
|
totalSize += Int64(fileSize)
|
|
}
|
|
} catch {
|
|
print("Error calculating file size: \(error)")
|
|
}
|
|
}
|
|
|
|
return totalSize
|
|
}
|
|
|
|
// Clean up old GIFs if storage exceeds 100 MB
|
|
func performStorageCleanupIfNeeded() {
|
|
let maxStorageSize: Int64 = 100 * 1024 * 1024 // 100 MB
|
|
|
|
if getAllGIFsSize() > maxStorageSize {
|
|
cleanupOldGIFs()
|
|
}
|
|
}
|
|
|
|
private func cleanupOldGIFs() {
|
|
let fileManager = FileManager.default
|
|
|
|
do {
|
|
// Get all files and their creation dates
|
|
let fileURLs = try fileManager.contentsOfDirectory(at: gifsDirectory, includingPropertiesForKeys: [.creationDateKey])
|
|
|
|
// Sort by creation date
|
|
let sortedFiles = try fileURLs.sorted {
|
|
let date1 = try $0.resourceValues(forKeys: [.creationDateKey]).creationDate ?? Date.distantPast
|
|
let date2 = try $1.resourceValues(forKeys: [.creationDateKey]).creationDate ?? Date.distantPast
|
|
return date1 < date2
|
|
}
|
|
|
|
// Delete the oldest 30% of files
|
|
let filesToDelete = Int(Double(sortedFiles.count) * 0.3)
|
|
|
|
for i in 0..<min(filesToDelete, sortedFiles.count) {
|
|
try fileManager.removeItem(at: sortedFiles[i])
|
|
}
|
|
} catch {
|
|
print("Error cleaning up old GIFs: \(error)")
|
|
}
|
|
}
|
|
} |