type QueryClient = {
    query: (query: string, values: unknown[]) => Promise<{ rowCount?: number | null }>
}

type TenantImportData = {
    tenantId: number
    tables: Record<string, Record<string, any>[]>
}

export const restoreImportedTenantNumberRanges = async (
    client: QueryClient,
    exportData: TenantImportData
) => {
    const tenantRow = (exportData.tables.tenants || []).find(
        (row) => Number(row.id) === Number(exportData.tenantId)
    )

    if (!tenantRow || tenantRow.numberRanges === null || typeof tenantRow.numberRanges === "undefined") {
        return 0
    }

    const result = await client.query(
        `update "tenants" set "numberRanges" = $1::jsonb where "id" = $2`,
        [JSON.stringify(tenantRow.numberRanges), exportData.tenantId]
    )

    return result.rowCount || 0
}