//
// Copyright 2025 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//

import Foundation
import GRDB

struct GroupMessageProcessorJobStore {
    func allEnqueuedGroupIds(tx: DBReadTransaction) throws -> [Data] {
        let sql = """
            SELECT DISTINCT \(GroupMessageProcessorJob.CodingKeys.groupId.rawValue)
            FROM \(GroupMessageProcessorJob.databaseTableName)
            """
        do {
            return try (Data?).fetchAll(tx.database, sql: sql).compacted()
        } catch {
            throw error.grdbErrorForLogging
        }
    }

    func nextJob(forGroupId groupId: Data, tx: DBReadTransaction) throws -> GroupMessageProcessorJob? {
        do {
            let sql = """
                SELECT *
                FROM \(GroupMessageProcessorJob.databaseTableName)
                WHERE \(GroupMessageProcessorJob.CodingKeys.groupId.rawValue) = ?
                ORDER BY \(GroupMessageProcessorJob.CodingKeys.id.rawValue)
                """
            return try GroupMessageProcessorJob.fetchOne(tx.database, sql: sql, arguments: [groupId])
        } catch {
            throw error.grdbErrorForLogging
        }
    }

    func newestJobId(tx: DBReadTransaction) throws -> Int64? {
        do {
            let sql = """
                SELECT \(GroupMessageProcessorJob.CodingKeys.id.rawValue)
                FROM \(GroupMessageProcessorJob.databaseTableName)
                ORDER BY \(GroupMessageProcessorJob.CodingKeys.id.rawValue) DESC
                """
            return try Int64.fetchOne(tx.database, sql: sql)
        } catch {
            throw error.grdbErrorForLogging
        }
    }

    public func existsJob(forGroupId groupId: Data, tx: DBReadTransaction) throws -> Bool {
        let sql = """
            SELECT 1 FROM \(GroupMessageProcessorJob.databaseTableName)
            WHERE \(GroupMessageProcessorJob.CodingKeys.groupId.rawValue) = ?
            """
        do {
            return try Bool.fetchOne(tx.database, sql: sql, arguments: [groupId]) ?? false
        } catch {
            throw error.grdbErrorForLogging
        }
    }

    public func removeJob(withRowId rowId: Int64, tx: DBWriteTransaction) throws {
        do {
            _ = try GroupMessageProcessorJob.deleteOne(tx.database, key: rowId)
        } catch {
            throw error.grdbErrorForLogging
        }
    }
}