mirror of
https://github.com/hexedtech/codemp-vscode.git
synced 2024-11-25 08:54:48 +01:00
feat: added reference client in lib
This commit is contained in:
parent
2472164350
commit
b12b6dc68f
3 changed files with 112 additions and 1 deletions
110
src/lib/client.rs
Normal file
110
src/lib/client.rs
Normal file
|
@ -0,0 +1,110 @@
|
|||
/// TODO better name for this file
|
||||
|
||||
use std::sync::{Arc, Mutex};
|
||||
use tracing::{error, warn};
|
||||
use uuid::Uuid;
|
||||
|
||||
use crate::{
|
||||
opfactory::OperationFactory,
|
||||
proto::{buffer_client::BufferClient, BufferPayload, OperationRequest, RawOp},
|
||||
tonic::{transport::Channel, Status, Streaming},
|
||||
};
|
||||
|
||||
type FactoryHandle = Arc<Mutex<OperationFactory>>;
|
||||
|
||||
impl From::<BufferClient<Channel>> for CodempClient {
|
||||
fn from(x: BufferClient<Channel>) -> CodempClient {
|
||||
CodempClient {
|
||||
id: Uuid::new_v4(),
|
||||
client:x,
|
||||
factory: Arc::new(Mutex::new(OperationFactory::new(None)))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct CodempClient {
|
||||
id: Uuid,
|
||||
client: BufferClient<Channel>,
|
||||
factory: FactoryHandle, // TODO less jank solution than Arc<Mutex>
|
||||
}
|
||||
|
||||
impl CodempClient {
|
||||
pub async fn create(&mut self, path: String, content: Option<String>) -> Result<bool, Status> {
|
||||
Ok(
|
||||
self.client.create(
|
||||
BufferPayload {
|
||||
path,
|
||||
content,
|
||||
user: self.id.to_string(),
|
||||
}
|
||||
)
|
||||
.await?
|
||||
.into_inner()
|
||||
.accepted
|
||||
)
|
||||
}
|
||||
|
||||
pub async fn insert(&mut self, path: String, txt: String, pos: u64) -> Result<bool, Status> {
|
||||
let res = { self.factory.lock().unwrap().insert(&txt, pos) };
|
||||
match res {
|
||||
Ok(op) => {
|
||||
Ok(
|
||||
self.client.edit(
|
||||
OperationRequest {
|
||||
path,
|
||||
hash: "".into(),
|
||||
opseq: serde_json::to_string(&op).unwrap(),
|
||||
user: self.id.to_string(),
|
||||
}
|
||||
)
|
||||
.await?
|
||||
.into_inner()
|
||||
.accepted
|
||||
)
|
||||
},
|
||||
Err(e) => Err(Status::internal(format!("invalid operation: {}", e))),
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn attach<F : Fn(String) -> () + Send + 'static>(&mut self, path: String, callback: F) -> Result<(), Status> {
|
||||
let stream = self.client.attach(
|
||||
BufferPayload {
|
||||
path,
|
||||
content: None,
|
||||
user: self.id.to_string(),
|
||||
}
|
||||
)
|
||||
.await?
|
||||
.into_inner();
|
||||
|
||||
let factory = self.factory.clone();
|
||||
tokio::spawn(async move { Self::worker(stream, factory, callback).await } );
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn worker<F : Fn(String) -> ()>(mut stream: Streaming<RawOp>, factory: FactoryHandle, callback: F) {
|
||||
loop {
|
||||
match stream.message().await {
|
||||
Ok(v) => match v {
|
||||
Some(operation) => {
|
||||
let op = serde_json::from_str(&operation.opseq).unwrap();
|
||||
let res = { factory.lock().unwrap().process(op) };
|
||||
match res {
|
||||
Ok(x) => callback(x),
|
||||
Err(e) => break error!("desynched: {}", e),
|
||||
}
|
||||
}
|
||||
None => break warn!("stream closed"),
|
||||
},
|
||||
Err(e) => break error!("error receiving change: {}", e),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn content(&self) -> String {
|
||||
let factory = self.factory.lock().unwrap();
|
||||
factory.content()
|
||||
}
|
||||
}
|
|
@ -1,5 +1,6 @@
|
|||
pub mod proto;
|
||||
pub mod opfactory;
|
||||
pub mod client;
|
||||
|
||||
pub use tonic;
|
||||
pub use tokio;
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
use operational_transform::{OperationSeq, OTError};
|
||||
use tracing::{debug, info};
|
||||
use tracing::info;
|
||||
|
||||
|
||||
#[derive(Clone)]
|
||||
|
|
Loading…
Reference in a new issue