垃圾桶数据库

PhoneManager/Class/Database/TrashDatabase.swift

+//
+//  TrashDatabase.swift
+//  PhoneManager
+//
+//  Created by edy on 2025/5/12.
+//
+
+import Foundation
+import SQLite3
+
+class TrashDatabase {
+    
+    static let shared = TrashDatabase()
+    
+    private var db: OpaquePointer?
+    private let dbPath: String
+    
+    private init() {
+        let fileURL = try! FileManager.default
+            .url(for: .documentDirectory, in: .userDomainMask, appropriateFor: nil, create: false)
+            .appendingPathComponent("trash.sqlite")
+        
+        dbPath = fileURL.path
+        
+        if sqlite3_open(dbPath, &db) != SQLITE_OK {
+            print("无法打开数据库")
+            return
+        }
+        
+        createTable()
+    }
+    
+    private func createTable() {
+        let createTableString = """
+            CREATE TABLE IF NOT EXISTS trash(
+                localIdentifier TEXT PRIMARY KEY,
+                assetSize DOUBLE,
+                createDate DOUBLE,
+                mediaType INTEGER
+            );
+        """
+        
+        var createTableStatement: OpaquePointer?
+        
+        if sqlite3_prepare_v2(db, createTableString, -1, &createTableStatement, nil) == SQLITE_OK {
+            if sqlite3_step(createTableStatement) == SQLITE_DONE {
+                print("成功创建trash表")
+            } else {
+                print("创建表失败")
+            }
+        } else {
+            print("创建表语句准备失败")
+        }
+        
+        sqlite3_finalize(createTableStatement)
+    }
+    
+    // 插入数据（带去重检查）
+    func insert(localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int) -> Bool {
+        // 先检查是否已存在相同的 localIdentifier
+        if let _ = query(localIdentifier: localIdentifier) {
+            // 如果已存在，则执行更新操作
+            return update(localIdentifier: localIdentifier, assetSize: assetSize, createDate: createDate, mediaType: mediaType)
+        }
+        
+        // 如果不存在，执行插入操作
+        let insertStatementString = "INSERT INTO trash (localIdentifier, assetSize, createDate, mediaType) VALUES (?, ?, ?, ?);"
+        var insertStatement: OpaquePointer?
+        
+        if sqlite3_prepare_v2(db, insertStatementString, -1, &insertStatement, nil) == SQLITE_OK {
+            sqlite3_bind_text(insertStatement, 1, (localIdentifier as NSString).utf8String, -1, nil)
+            sqlite3_bind_double(insertStatement, 2, assetSize)
+            sqlite3_bind_double(insertStatement, 3, createDate.timeIntervalSince1970)
+            sqlite3_bind_int(insertStatement, 4, Int32(mediaType))
+            
+            if sqlite3_step(insertStatement) == SQLITE_DONE {
+                print("成功插入数据")
+                sqlite3_finalize(insertStatement)
+                return true
+            } else {
+                print("插入数据失败")
+            }
+        }
+        
+        sqlite3_finalize(insertStatement)
+        return false
+    }
+    
+    // 删除数据
+    func delete(localIdentifier: String) -> Bool {
+        let deleteStatementString = "DELETE FROM trash WHERE localIdentifier = ?;"
+        var deleteStatement: OpaquePointer?
+        
+        if sqlite3_prepare_v2(db, deleteStatementString, -1, &deleteStatement, nil) == SQLITE_OK {
+            sqlite3_bind_text(deleteStatement, 1, (localIdentifier as NSString).utf8String, -1, nil)
+            
+            if sqlite3_step(deleteStatement) == SQLITE_DONE {
+                print("成功删除数据")
+                sqlite3_finalize(deleteStatement)
+                return true
+            } else {
+                print("删除数据失败")
+            }
+        }
+        
+        sqlite3_finalize(deleteStatement)
+        return false
+    }
+    
+    // 更新数据
+    func update(localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int) -> Bool {
+        let updateStatementString = "UPDATE trash SET assetSize = ?, createDate = ?, mediaType = ? WHERE localIdentifier = ?;"
+        var updateStatement: OpaquePointer?
+        
+        if sqlite3_prepare_v2(db, updateStatementString, -1, &updateStatement, nil) == SQLITE_OK {
+            sqlite3_bind_double(updateStatement, 1, assetSize)
+            sqlite3_bind_double(updateStatement, 2, createDate.timeIntervalSince1970)
+            sqlite3_bind_int(updateStatement, 3, Int32(mediaType))
+            sqlite3_bind_text(updateStatement, 4, (localIdentifier as NSString).utf8String, -1, nil)
+            
+            if sqlite3_step(updateStatement) == SQLITE_DONE {
+                print("成功更新数据")
+                sqlite3_finalize(updateStatement)
+                return true
+            } else {
+                print("更新数据失败")
+            }
+        }
+        
+        sqlite3_finalize(updateStatement)
+        return false
+    }
+    
+    // 查询所有数据
+    func queryAll() -> [(localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int)] {
+        let queryStatementString = "SELECT * FROM trash;"
+        var queryStatement: OpaquePointer?
+        var result: [(localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int)] = []
+        
+        if sqlite3_prepare_v2(db, queryStatementString, -1, &queryStatement, nil) == SQLITE_OK {
+            while sqlite3_step(queryStatement) == SQLITE_ROW {
+                let localIdentifier = String(cString: sqlite3_column_text(queryStatement, 0))
+                let assetSize = sqlite3_column_double(queryStatement, 1)
+                let createDate = Date(timeIntervalSince1970: sqlite3_column_double(queryStatement, 2))
+                let mediaType = Int(sqlite3_column_int(queryStatement, 3))
+                
+                result.append((localIdentifier, assetSize, createDate, mediaType))
+            }
+        }
+        
+        sqlite3_finalize(queryStatement)
+        return result
+    }
+    
+    // 根据localIdentifier查询单条数据
+    func query(localIdentifier: String) -> (localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int)? {
+        let queryStatementString = "SELECT * FROM trash WHERE localIdentifier = ?;"
+        var queryStatement: OpaquePointer?
+        
+        if sqlite3_prepare_v2(db, queryStatementString, -1, &queryStatement, nil) == SQLITE_OK {
+            sqlite3_bind_text(queryStatement, 1, (localIdentifier as NSString).utf8String, -1, nil)
+            
+            if sqlite3_step(queryStatement) == SQLITE_ROW {
+                let localIdentifier = String(cString: sqlite3_column_text(queryStatement, 0))
+                let assetSize = sqlite3_column_double(queryStatement, 1)
+                let createDate = Date(timeIntervalSince1970: sqlite3_column_double(queryStatement, 2))
+                let mediaType = Int(sqlite3_column_int(queryStatement, 3))
+                
+                sqlite3_finalize(queryStatement)
+                return (localIdentifier, assetSize, createDate, mediaType)
+            }
+        }
+        
+        sqlite3_finalize(queryStatement)
+        return nil
+    }
+    
+    // 根据mediaType查询数据
+    func queryByMediaType(_ mediaType: Int) -> [(localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int)] {
+        let queryStatementString = "SELECT * FROM trash WHERE mediaType = ?;"
+        var queryStatement: OpaquePointer?
+        var result: [(localIdentifier: String, assetSize: Double, createDate: Date, mediaType: Int)] = []
+        
+        if sqlite3_prepare_v2(db, queryStatementString, -1, &queryStatement, nil) == SQLITE_OK {
+            sqlite3_bind_int(queryStatement, 1, Int32(mediaType))
+            
+            while sqlite3_step(queryStatement) == SQLITE_ROW {
+                let localIdentifier = String(cString: sqlite3_column_text(queryStatement, 0))
+                let assetSize = sqlite3_column_double(queryStatement, 1)
+                let createDate = Date(timeIntervalSince1970: sqlite3_column_double(queryStatement, 2))
+                let mediaType = Int(sqlite3_column_int(queryStatement, 3))
+                
+                result.append((localIdentifier, assetSize, createDate, mediaType))
+            }
+        }
+        
+        sqlite3_finalize(queryStatement)
+        return result
+    }
+
+    deinit {
+        sqlite3_close(db)
+    }
+}

PhoneManager/Class/Page/Trash/Model/TrashUIModel.swift

@@ -9,8 +9,23 @@ import Foundation
 
 enum TrashTypeEnum{
     case video,other,shot,chat
+    
+    var dbType:Int{
+        switch self {
+        case .video:
+            return 0
+        case .other:
+            return 1
+        case .shot:
+            return 2
+        case .chat:
+            return 3
+        }
+    }
 }
 
+
+
 struct TrashPageScrollModel{
     var offset:CGFloat = 0
     var page:Int = 1