bft-crdt-experiment/side-node/src/node.rs

use bft_json_crdt::json_crdt::BaseCrdt;
use fastcrypto::ed25519::Ed25519KeyPair;

use crate::{crdt::TransactionList, websocket::WebSocketClient};

pub(crate) struct SideNode {
    crdt: BaseCrdt<TransactionList>,
    keys: fastcrypto::ed25519::Ed25519KeyPair,
    websocket_client: WebSocketClient,
}

impl SideNode {
    pub(crate) fn new(
        websocket_client: WebSocketClient,
        crdt: BaseCrdt<TransactionList>,
        keys: Ed25519KeyPair,
    ) -> Self {
        Self {
            crdt,
            keys,
            websocket_client,
        }
    }

    pub(crate) async fn start(&mut self) {
        self.websocket_client.start().await;
    }

    fn add_transaction_local(
        &mut self,
        transaction: serde_json::Value,
    ) -> bft_json_crdt::json_crdt::SignedOp {
        let last = self
            .crdt
            .doc
            .list
            .ops
            .last()
            .expect("couldn't find last op");
        let signed_op = self
            .crdt
            .doc
            .list
            .insert(last.id, transaction)
            .sign(&self.keys);
        signed_op
    }
}