logos-blockchain-specs/da/dispersal.py
2025-06-04 15:44:15 +02:00

46 lines
1.3 KiB
Python

from dataclasses import dataclass
from typing import List, Generator
from da.common import NodeId, Column
from da.encoder import EncodedData
from da.verifier import DAShare
@dataclass
class DispersalSettings:
nodes_ids: List[NodeId]
threshold: int
class Dispersal:
def __init__(self, settings: DispersalSettings):
self.settings = settings
# sort over public keys
self.settings.nodes_ids.sort()
def _prepare_data(self, encoded_data: EncodedData) -> Generator[DAShare, None, None]:
columns = encoded_data.extended_matrix.columns
row_commitments = encoded_data.row_commitments
column_proofs = encoded_data.combined_column_proofs
blobs_data = zip(columns, column_proofs)
for column_idx, (column, proof) in enumerate(blobs_data):
blob = DAShare(
Column(column),
column_idx,
proof,
row_commitments
)
yield blob
def _send_and_await_response(self, node: NodeId, blob: DAShare) -> bool:
pass
def disperse(self, encoded_data: EncodedData):
blob_data = zip(
self.settings.nodes_ids,
self._prepare_data(encoded_data)
)
for node, blob in blob_data:
self._send_and_await_response(node, blob)