turso/sync/engine/src/protocol_io.rs

use crate::{
    types::{DatabaseRowMutation, DatabaseRowTransformResult},
    Result,
};

pub trait DataPollResult<T> {
    fn data(&self) -> &[T];
}

pub trait DataCompletion<T> {
    type DataPollResult: DataPollResult<T>;
    fn status(&self) -> Result<Option<u16>>;
    fn poll_data(&self) -> Result<Option<Self::DataPollResult>>;
    fn is_done(&self) -> Result<bool>;
}

pub trait ProtocolIO {
    type DataCompletionBytes: DataCompletion<u8>;
    type DataCompletionTransform: DataCompletion<DatabaseRowTransformResult>;
    fn full_read(&self, path: &str) -> Result<Self::DataCompletionBytes>;
    fn full_write(&self, path: &str, content: Vec<u8>) -> Result<Self::DataCompletionBytes>;
    fn transform(
        &self,
        mutations: Vec<DatabaseRowMutation>,
    ) -> Result<Self::DataCompletionTransform>;
    fn http(
        &self,
        method: &str,
        path: &str,
        body: Option<Vec<u8>>,
        headers: &[(&str, &str)],
    ) -> Result<Self::DataCompletionBytes>;
}