|
| 1 | +package to.bitkit.dev |
| 2 | + |
| 3 | +import android.content.ContentProvider |
| 4 | +import android.content.ContentValues |
| 5 | +import android.net.Uri |
| 6 | +import android.os.Binder |
| 7 | +import android.os.Bundle |
| 8 | +import android.os.Process |
| 9 | +import androidx.core.os.bundleOf |
| 10 | +import dagger.hilt.EntryPoint |
| 11 | +import dagger.hilt.InstallIn |
| 12 | +import dagger.hilt.android.EntryPointAccessors |
| 13 | +import dagger.hilt.components.SingletonComponent |
| 14 | +import kotlinx.serialization.Serializable |
| 15 | +import kotlinx.serialization.json.Json |
| 16 | +import to.bitkit.async.ServiceQueue |
| 17 | +import to.bitkit.repositories.LightningRepo |
| 18 | +import to.bitkit.utils.Logger |
| 19 | + |
| 20 | +private const val TAG = "DevToolsProvider" |
| 21 | + |
| 22 | +class DevToolsProvider : ContentProvider() { |
| 23 | + |
| 24 | + @EntryPoint |
| 25 | + @InstallIn(SingletonComponent::class) |
| 26 | + interface Dependencies { |
| 27 | + fun lightningRepo(): LightningRepo |
| 28 | + } |
| 29 | + |
| 30 | + private val deps: Dependencies by lazy { |
| 31 | + EntryPointAccessors.fromApplication(requireNotNull(context) { "DevToolsProvider context is null" }, Dependencies::class.java) |
| 32 | + } |
| 33 | + |
| 34 | + override fun call(method: String, arg: String?, extras: Bundle?): Bundle { |
| 35 | + check(Binder.getCallingUid() == Process.SHELL_UID) { "Only ADB shell callers are allowed" } |
| 36 | + return runCatching { |
| 37 | + val command = requireNotNull(DevCommand.parse(method, arg)) { "Unknown command: '$method'" } |
| 38 | + ServiceQueue.LDK.blocking { command.execute(deps) } |
| 39 | + }.getOrElse { |
| 40 | + Logger.error("Failed to execute command '$method'", it, context = TAG) |
| 41 | + DevResult.Error(it.message) |
| 42 | + }.toBundle() |
| 43 | + } |
| 44 | + |
| 45 | + override fun onCreate() = true |
| 46 | + override fun getType(uri: Uri): String? = null |
| 47 | + override fun insert(uri: Uri, values: ContentValues?): Uri? = null |
| 48 | + override fun delete(uri: Uri, sel: String?, args: Array<String>?) = 0 |
| 49 | + override fun update(uri: Uri, values: ContentValues?, sel: String?, args: Array<String>?) = 0 |
| 50 | + override fun query(uri: Uri, proj: Array<String>?, sel: String?, args: Array<String>?, sort: String?) = null |
| 51 | +} |
| 52 | + |
| 53 | +private sealed interface DevCommand { |
| 54 | + |
| 55 | + companion object { |
| 56 | + fun parse(method: String, arg: String?): DevCommand? = when (method) { |
| 57 | + CreateInvoice.METHOD -> CreateInvoice.parse(arg) |
| 58 | + else -> null |
| 59 | + } |
| 60 | + } |
| 61 | + |
| 62 | + suspend fun execute(deps: DevToolsProvider.Dependencies): DevResult |
| 63 | + |
| 64 | + data class CreateInvoice(val args: Args) : DevCommand { |
| 65 | + companion object { |
| 66 | + const val METHOD = "createInvoice" |
| 67 | + fun parse(arg: String?) = CreateInvoice(arg.deserialize<Args>()) |
| 68 | + } |
| 69 | + |
| 70 | + @Serializable |
| 71 | + data class Args(val amount: ULong? = null, val description: String = "dev-invoice") |
| 72 | + |
| 73 | + override suspend fun execute(deps: DevToolsProvider.Dependencies) = |
| 74 | + deps.lightningRepo().createInvoice(args.amount, args.description).fold( |
| 75 | + onSuccess = { DevResult.Invoice(it) }, |
| 76 | + onFailure = { DevResult.Error(it.message) }, |
| 77 | + ) |
| 78 | + } |
| 79 | +} |
| 80 | + |
| 81 | +@Serializable |
| 82 | +private sealed interface DevResult { |
| 83 | + |
| 84 | + companion object { |
| 85 | + private const val KEY_RESULT = "result" |
| 86 | + } |
| 87 | + |
| 88 | + @Serializable data class Invoice(val bolt11: String) : DevResult |
| 89 | + |
| 90 | + @Serializable data class Error(val message: String? = null) : DevResult |
| 91 | + |
| 92 | + fun toBundle() = bundleOf(KEY_RESULT to Json.encodeToString(this)) |
| 93 | +} |
| 94 | + |
| 95 | +private inline fun <reified T> String?.deserialize(): T = |
| 96 | + if (isNullOrBlank()) Json.decodeFromString("{}") else Json.decodeFromString(this) |
0 commit comments