FileTooLargeException

suspend fun uploadDocument(
    kycService: KYCService,
    documentBytes: ByteArray,
    jwt: String
): CustomerFileResponse {
    try {
        return kycService.postCustomerFile(documentBytes, jwt)
    } catch (e: FileTooLargeException) {
        println("File too large: ${e.fileSize} bytes")
        println("Please compress or resize the file")

        // Attempt to compress image
        val compressedBytes = compressImage(documentBytes)
        println("Compressed to: ${compressedBytes.size} bytes")

        // Retry with compressed file
        return kycService.postCustomerFile(compressedBytes, jwt)
    }
}

fun validateFileSize(fileBytes: ByteArray, maxSizeMB: Int = 5): Boolean {
    val maxSizeBytes = maxSizeMB * 1024 * 1024
    if (fileBytes.size > maxSizeBytes) {
        println("File too large: ${fileBytes.size} bytes (max: $maxSizeBytes)")
        return false
    }
    return true
}

suspend fun uploadWithValidation(
    kycService: KYCService,
    fileBytes: ByteArray,
    jwt: String
) {
    if (!validateFileSize(fileBytes, maxSizeMB = 5)) {
        throw FileTooLargeException(fileBytes.size.toLong())
    }

    kycService.postCustomerFile(fileBytes, jwt)
}

suspend fun uploadImage(
    kycService: KYCService,
    imageBytes: ByteArray,
    jwt: String,
    maxSizeBytes: Long = 5_000_000 // 5 MB
): CustomerFileResponse {
    var bytes = imageBytes

    // Compress if too large
    if (bytes.size > maxSizeBytes) {
        println("Image too large (${bytes.size} bytes), compressing...")
        bytes = compressImage(bytes, quality = 0.8)
        println("Compressed to ${bytes.size} bytes")
    }

    return try {
        kycService.postCustomerFile(bytes, jwt)
    } catch (e: FileTooLargeException) {
        // If still too large after compression, try lower quality
        println("Still too large, reducing quality further...")
        bytes = compressImage(imageBytes, quality = 0.5)
        kycService.postCustomerFile(bytes, jwt)
    }
}

fun showFileSizeError(exception: FileTooLargeException, filename: String) {
    val sizeMB = (exception.fileSize ?: 0) / (1024.0 * 1024.0)
    showError(
        title = "File Too Large",
        message = buildString {
            append("The file '$filename' is too large ")
            if (exception.fileSize != null) {
                val sizeFormatted = (sizeMB * 100).toInt() / 100.0
                append("($sizeFormatted MB).\n\n")
            } else {
                append(".\n\n")
            }
            append("Please try:\n")
            append("• Compressing the file\n")
            append("• Reducing image resolution\n")
            append("• Converting to a smaller format (JPEG)")
        }
    )
}