another.im-ios/AnotherIM/AppData/Services/Database.swift

import Combine
import Foundation
import GRDB
import SwiftUI

// MARK: - Models protocol
typealias DBStorable = Codable & FetchableRecord & Identifiable & PersistableRecord & TableRecord

// MARK: - Database init
final class Database {
    static let shared = Database()
    let dbQueue: DatabaseQueue
    private var dbPath: String

    private init() {
        do {
            // Create db folder if not exists
            let fileManager = FileManager.default
            let appSupportURL = try fileManager.url(
                for: .applicationSupportDirectory, in: .userDomainMask,
                appropriateFor: nil, create: true
            )
            let directoryURL = appSupportURL.appendingPathComponent("anotherim", isDirectory: true)
            try fileManager.createDirectory(at: directoryURL, withIntermediateDirectories: true)

            // Open or create the database
            let databaseURL = directoryURL.appendingPathComponent("db.sqlite")
            dbQueue = try DatabaseQueue(path: databaseURL.path, configuration: Database.config)
            dbPath = databaseURL.path

            // Some debug info
            #if DEBUG
                print("Database path: \(databaseURL.path)")
            #endif

            // Apply migrations
            try Database.migrator.migrate(dbQueue)
        } catch {
            fatalError("Database initialization failed: \(error)")
        }
    }
}

// MARK: - Config
private extension Database {
    static let config: Configuration = {
        var config = Configuration()
        #if DEBUG
            // verbose and debugging in DEBUG builds only.
            config.publicStatementArguments = true
            config.prepareDatabase { db in
                db.trace { print("SQL> \($0)\n") }
            }
        #endif
        return config
    }()
}

// MARK: - flush all data for debug
#if DEBUG
    extension Database {
        func flushAllData() {
            do {
                try dbQueue.write { db in
                    // Fetch all table names
                    let tables = try String.fetchAll(db, sql: """
                        SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%';
                    """)

                    // Generate and execute DELETE statements for each table
                    for table in tables {
                        try db.execute(sql: "DELETE FROM \(table);")
                    }
                }
            } catch {
                print("Error flushing all data: \(error)")
            }
        }
    }
#endif