StartActivityForResult but with parameter

StartActivityForResultWithParam.kt

/*
    Example usage:

    captureImageActivity.launch(IntentWithParam(Intent(MediaStore.ACTION_IMAGE_CAPTURE), SOME_PARAM))

    private val captureImageActivity = registerForActivityResult(StartActivityForResultWithParam()) { result ->
        if (result.resultCode == Activity.RESULT_OK) {
            val SOME_PARAM = result.param

            ...
        }
    }
*/

data class IntentWithParam(
    val intent: Intent,
    val param: Any?,
)

data class ActivityResultWithParam (
    val resultCode: Int,
    val data: Intent?,
    val param: Any?,
)

class StartActivityForResultWithParam : ActivityResultContract<IntentWithParam, ActivityResultWithParam>() {
    var param: Any? = null

    override fun createIntent(context: Context, input: IntentWithParam): Intent {
        param = input.param
        return input.intent
    }

    override fun parseResult(
        resultCode: Int,
        intent: Intent?
    ): ActivityResultWithParam =  ActivityResultWithParam(resultCode = resultCode, data = intent, param = param)
}