class MainActivity : AppCompatActivity() {
    
    private lateinit var binding: ActivityMainBinding
    private lateinit var digiboostActivityResultLauncher: ActivityResultLauncher<Intent>
    
    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        binding = ActivityMainBinding.inflate(layoutInflater)
        setContentView(binding.root)
        
        // Register for activity result before onCreate completes
        registerActivityForResult()
        
        // Set up button click listener
        setupSdkLaunch()
    }
    
    private fun setupSdkLaunch() {
        binding.btnGetStarted.setOnClickListener {
            // Use the token from Step 1
            val token = "YOUR_TOKEN_FROM_STEP_1"  // Replace with actual token
            val env = "PREPROD"  // or "PROD" for production
            
            openDigiboostActivity(env, token)
        }
    }
}

private fun openDigiboostActivity(env: String, token: String) {
    try {
        val intent = Intent(this, InitSdk::class.java).apply {
            putExtra("token", token)
            putExtra("env", env)
        }
        digiboostActivityResultLauncher.launch(intent)
    } catch (e: Exception) {
        Log.e("MainActivity", "Error launching Digiboost SDK: ${e.message}")
        // Handle error appropriately
    }
}

private fun registerActivityForResult() {
    digiboostActivityResultLauncher = registerForActivityResult(
        ActivityResultContracts.StartActivityForResult()
    ) { result ->
        when (result.resultCode) {
            RESULT_OK -> {
                val data = result.data
                if (data != null) {
                    val digiboostResponse = data.getStringExtra("signedResponse")
                    Log.d("MainActivity", "Digiboost Response: $digiboostResponse")
                    handleSuccessResponse(digiboostResponse)
                } else {
                    Log.w("MainActivity", "No data received from Digiboost SDK")
                }
            }
            RESULT_CANCELED -> {
                Log.i("MainActivity", "User cancelled Digiboost SDK")
                handleCancelledResponse()
            }
            else -> {
                Log.e("MainActivity", "Unexpected result code: ${result.resultCode}")
                handleErrorResponse("Unexpected result")
            }
        }
    }
}

private fun handleSuccessResponse(response: String?) {
    response?.let {
        // Process the signed response
        showResponse(it)
        // Parse and handle the response according to your app's needs
    }
}

private fun handleCancelledResponse() {
    // Handle user cancellation
    Toast.makeText(this, "Document verification cancelled", Toast.LENGTH_SHORT).show()
}

private fun handleErrorResponse(error: String) {
    // Handle error cases
    Toast.makeText(this, "Error: $error", Toast.LENGTH_SHORT).show()
}

private fun showResponse(response: String?) {
    // Display response in your UI
    // You can show it in a dialog, text view, or navigate to a results screen
    AlertDialog.Builder(this)
        .setTitle("Verification Complete")
        .setMessage("Response: $response")
        .setPositiveButton("OK", null)
        .show()
}

private fun downloadAadhaar(clientId: String) {
    val apiUrl = if (isProduction) {
        "https://kyc-api.surepass.app/api/v1/digilocker/download-aadhaar"
    } else {
        "https://sandbox.surepass.app/api/v1/digilocker/download-aadhaar"
    }
    
    val requestBody = JSONObject().apply {
        put("client_id", clientId)
    }
    
    val request = Request.Builder()
        .url(apiUrl)
        .addHeader("Content-Type", "application/json")
        .addHeader("Authorization", "Bearer $YOUR_API_TOKEN")
        .post(requestBody.toString().toRequestBody("application/json".toMediaType()))
        .build()
    
    client.newCall(request).enqueue(object : Callback {
        override fun onResponse(call: Call, response: Response) {
            if (response.isSuccessful) {
                val responseBody = response.body?.string()
                handleAadhaarDownloadSuccess(responseBody)
            } else {
                handleAadhaarDownloadError("API Error: ${response.code}")
            }
        }
        
        override fun onFailure(call: Call, e: IOException) {
            handleAadhaarDownloadError("Network Error: ${e.message}")
        }
    })
}

private fun handleAadhaarDownloadSuccess(responseJson: String?) {
    try {
        val jsonResponse = JSONObject(responseJson ?: "")
        val data = jsonResponse.getJSONObject("data")
        
        // Extract user information
        val name = data.getString("name")
        val aadhaarNumber = data.getString("aadhaar_number")
        val dateOfBirth = data.getString("date_of_birth")
        val gender = data.getString("gender")
        
        // Extract documents (base64 encoded)
        val aadhaarPdf = data.getString("aadhaar_pdf")
        val aadhaarXml = data.getString("aadhaar_xml")
        
        // Extract address
        val address = data.getJSONObject("address")
        val fullAddress = "${address.getString("house")}, ${address.getString("street")}, " +
                         "${address.getString("locality")}, ${address.getString("vtc")}, " +
                         "${address.getString("district")}, ${address.getString("state")} - ${address.getString("pincode")}"
        
        // Process the downloaded data
        Log.d("AadhaarDownload", "Name: $name, Aadhaar: $aadhaarNumber")
        
        // Save or display the documents
        saveAadhaarDocuments(aadhaarPdf, aadhaarXml)
        
    } catch (e: Exception) {
        Log.e("AadhaarDownload", "Error parsing response: ${e.message}")
    }
}

private fun saveAadhaarDocuments(pdfBase64: String, xmlBase64: String) {
    try {
        // Decode base64 and save PDF
        val pdfBytes = Base64.decode(pdfBase64, Base64.DEFAULT)
        val pdfFile = File(filesDir, "aadhaar_document.pdf")
        pdfFile.writeBytes(pdfBytes)
        
        // Decode base64 and save XML
        val xmlBytes = Base64.decode(xmlBase64, Base64.DEFAULT)
        val xmlFile = File(filesDir, "aadhaar_data.xml")
        xmlFile.writeBytes(xmlBytes)
        
        Log.d("AadhaarDownload", "Documents saved successfully")
        
    } catch (e: Exception) {
        Log.e("AadhaarDownload", "Error saving documents: ${e.message}")
    }
}

private fun handleAadhaarDownloadError(error: String) {
    Log.e("AadhaarDownload", "Download failed: $error")
    Toast.makeText(this, "Failed to download Aadhaar: $error", Toast.LENGTH_SHORT).show()
}