io.provenance.client.grpc.BaseReq.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pb-grpc-client-common-kotlin Show documentation
A GRPC client for communicating with the Provenance Blockchain
The newest version!
package io.provenance.client.grpc

import com.google.protobuf.Any
import com.google.protobuf.ByteString
import cosmos.auth.v1beta1.Auth
import cosmos.crypto.secp256k1.Keys
import cosmos.tx.signing.v1beta1.Signing.SignMode
import cosmos.tx.v1beta1.TxOuterClass.AuthInfo
import cosmos.tx.v1beta1.TxOuterClass.Fee
import cosmos.tx.v1beta1.TxOuterClass.ModeInfo
import cosmos.tx.v1beta1.TxOuterClass.ModeInfo.Single
import cosmos.tx.v1beta1.TxOuterClass.SignDoc
import cosmos.tx.v1beta1.TxOuterClass.SignerInfo
import cosmos.tx.v1beta1.TxOuterClass.TxBody
import io.provenance.client.common.gas.GasEstimate
import io.provenance.client.common.extensions.discreteSum

interface Signer {
    fun address(): String
    fun pubKey(): Keys.PubKey
    fun sign(data: ByteArray): ByteArray
}

typealias CustomUnpackAccount = (Any.() -> Auth.BaseAccount)
typealias CustomizeSignerInfo = (SignerInfo.() -> SignerInfo)
data class BaseReqSigner(
    val signer: Signer,
    val sequenceOffset: Int = 0,
    val account: Auth.BaseAccount? = null,
    private val customizeSignerInfo: CustomizeSignerInfo? = null,
    val unpackAccount: CustomUnpackAccount? = null,
) {

    fun buildSignerInfo(): SignerInfo =
        SignerInfo.newBuilder()
            .setPublicKey(Any.pack(this.signer.pubKey(), ""))
            .setModeInfo(
                ModeInfo.newBuilder()
                    .setSingle(Single.newBuilder().setModeValue(SignMode.SIGN_MODE_DIRECT_VALUE))
            )
            .setSequence(this.account!!.sequence + this.sequenceOffset)
            .build().let {
                customizeSignerInfo?.invoke(it) ?: it
            }

    fun withCustomizeSignerInfo(customizeSignerInfo: CustomizeSignerInfo) = copy(customizeSignerInfo = customizeSignerInfo)
    fun withUnpackAccount(unpackAccount: CustomUnpackAccount) = copy(unpackAccount = unpackAccount)
}

data class BaseReq(
    val signers: List,
    val body: TxBody,
    val chainId: String,
    val gasAdjustment: Double? = null,
    val feeGranter: String? = null,
    val feePayer: String? = null,
) {

    fun buildAuthInfo(gasEstimate: GasEstimate = GasEstimate(0)): AuthInfo =
        AuthInfo.newBuilder()
            .setFee(
                Fee.newBuilder()
                    .addAllAmount(gasEstimate.feesCalculated.discreteSum())
                    .setGasLimit(gasEstimate.limit)
                    .also {
                        if (feeGranter != null) {
                            it.granter = feeGranter
                        }
                        if (feePayer != null) {
                            it.payer = feePayer
                        }
                    }
            )
            .addAllSignerInfos(
                signers.map { it.buildSignerInfo() }
            )
            .build()

    fun buildSignDoc(
        signer: BaseReqSigner,
        authInfoBytes: ByteString,
        bodyBytes: ByteString,
    ) =
        SignDoc.newBuilder()
            .setBodyBytes(bodyBytes)
            .setAuthInfoBytes(authInfoBytes)
            .setChainId(chainId)
            .setAccountNumber(signer.account!!.accountNumber)
            .build()

    fun buildSignDocBytesList(authInfoBytes: ByteString, bodyBytes: ByteString): List =
        signers.map {
            buildSignDoc(it, authInfoBytes, bodyBytes).toByteArray()
        }
}