mirror of
https://github.com/logos-blockchain/logos-blockchain-block-explorer-template.git
synced 2026-02-22 05:53:07 +00:00
79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
from typing import List, Self
|
|
|
|
from pydantic import Field
|
|
|
|
from core.models import NbeSerializer
|
|
from models.transactions.transaction import Transaction
|
|
from node.api.serializers.fields import BytesFromIntArray
|
|
from node.api.serializers.proof import (
|
|
OperationProofSerializer,
|
|
OperationProofSerializerField,
|
|
ZkSignatureComponentsSerializer,
|
|
)
|
|
from node.api.serializers.transaction import TransactionSerializer
|
|
from utils.protocols import FromRandom
|
|
from utils.random import random_bytes
|
|
|
|
|
|
class SignedTransactionSerializer(NbeSerializer, FromRandom):
|
|
transaction: TransactionSerializer = Field(alias="mantle_tx", description="Transaction.")
|
|
operations_proofs: List[OperationProofSerializerField] = Field(
|
|
alias="ops_proofs", description="List of OperationProof. Order should match `Self::transaction::operations`."
|
|
)
|
|
ledger_transaction_proof: ZkSignatureComponentsSerializer = Field(
|
|
alias="ledger_tx_proof", description="ZK proof with pi_a, pi_b, pi_c."
|
|
)
|
|
|
|
def into_transaction(self) -> Transaction:
|
|
operations_contents = self.transaction.operations_contents
|
|
if len(operations_contents) != len(self.operations_proofs):
|
|
raise ValueError(
|
|
f"Number of operations ({len(operations_contents)}) does not match number of operation proofs ({len(self.operations_proofs)})."
|
|
)
|
|
|
|
operations = [
|
|
{
|
|
"content": content.into_operation_content(),
|
|
"proof": proof.into_operation_proof(),
|
|
}
|
|
for content, proof in zip(operations_contents, self.operations_proofs)
|
|
]
|
|
|
|
ledger_transaction = self.transaction.ledger_transaction
|
|
outputs = [output.into_note() for output in ledger_transaction.outputs]
|
|
|
|
# Combine pi_a, pi_b, pi_c into single proof bytes
|
|
proof_bytes = (
|
|
self.ledger_transaction_proof.pi_a +
|
|
self.ledger_transaction_proof.pi_b +
|
|
self.ledger_transaction_proof.pi_c
|
|
)
|
|
|
|
return Transaction.model_validate(
|
|
{
|
|
"hash": self.transaction.hash,
|
|
"operations": operations,
|
|
"inputs": ledger_transaction.inputs,
|
|
"outputs": outputs,
|
|
"proof": proof_bytes,
|
|
"execution_gas_price": self.transaction.execution_gas_price,
|
|
"storage_gas_price": self.transaction.storage_gas_price,
|
|
}
|
|
)
|
|
|
|
@classmethod
|
|
def from_random(cls) -> Self:
|
|
transaction = TransactionSerializer.from_random()
|
|
n = len(transaction.operations_contents)
|
|
operations_proofs = [OperationProofSerializer.from_random() for _ in range(n)]
|
|
return cls.model_validate(
|
|
{
|
|
"mantle_tx": transaction,
|
|
"ops_proofs": operations_proofs,
|
|
"ledger_tx_proof": {
|
|
"pi_a": list(random_bytes(32)),
|
|
"pi_b": list(random_bytes(64)),
|
|
"pi_c": list(random_bytes(32)),
|
|
},
|
|
}
|
|
) |