Implement TryFrom<proto::SelectionSet> on SelectionSet

More prep work for changing the selection set representation.
This commit is contained in:
Nathan Sobo 2021-10-22 12:46:02 -06:00
parent 401bdf0ba1
commit 3ae5ba09fd
2 changed files with 60 additions and 43 deletions

View File

@ -20,7 +20,7 @@ use rpc::proto;
pub use selection::*; pub use selection::*;
use std::{ use std::{
cmp, cmp,
convert::{TryFrom, TryInto}, convert::TryFrom,
iter::Iterator, iter::Iterator,
ops::Range, ops::Range,
str, str,
@ -486,20 +486,8 @@ impl Buffer {
.selections .selections
.into_iter() .into_iter()
.map(|set| { .map(|set| {
let set_id = clock::Lamport { let set = SelectionSet::try_from(set)?;
replica_id: set.replica_id as ReplicaId, Result::<_, anyhow::Error>::Ok((set.id, set))
value: set.lamport_timestamp,
};
let selections: Vec<Selection> = set
.selections
.into_iter()
.map(TryFrom::try_from)
.collect::<Result<_, _>>()?;
let set = SelectionSet {
selections: Arc::from(selections),
active: set.is_active,
};
Result::<_, anyhow::Error>::Ok((set_id, set))
}) })
.collect::<Result<_, _>>()?; .collect::<Result<_, _>>()?;
Ok(buffer) Ok(buffer)
@ -859,6 +847,7 @@ impl Buffer {
self.selections.insert( self.selections.insert(
set_id, set_id,
SelectionSet { SelectionSet {
id: set_id,
selections, selections,
active: false, active: false,
}, },
@ -1290,18 +1279,19 @@ impl Buffer {
pub fn add_selection_set(&mut self, selections: impl Into<Arc<[Selection]>>) -> Operation { pub fn add_selection_set(&mut self, selections: impl Into<Arc<[Selection]>>) -> Operation {
let selections = selections.into(); let selections = selections.into();
let lamport_timestamp = self.lamport_clock.tick(); let set_id = self.lamport_clock.tick();
self.selections.insert( self.selections.insert(
lamport_timestamp, set_id,
SelectionSet { SelectionSet {
id: set_id,
selections: selections.clone(), selections: selections.clone(),
active: false, active: false,
}, },
); );
Operation::UpdateSelections { Operation::UpdateSelections {
set_id: lamport_timestamp, set_id,
selections, selections,
lamport_timestamp, lamport_timestamp: set_id,
} }
} }
@ -2390,26 +2380,6 @@ impl TryFrom<proto::Anchor> for Anchor {
} }
} }
impl TryFrom<proto::Selection> for Selection {
type Error = anyhow::Error;
fn try_from(selection: proto::Selection) -> Result<Self, Self::Error> {
Ok(Selection {
id: selection.id as usize,
start: selection
.start
.ok_or_else(|| anyhow!("missing selection start"))?
.try_into()?,
end: selection
.end
.ok_or_else(|| anyhow!("missing selection end"))?
.try_into()?,
reversed: selection.reversed,
goal: SelectionGoal::None,
})
}
}
pub trait ToOffset { pub trait ToOffset {
fn to_offset<'a>(&self, content: impl Into<Content<'a>>) -> usize; fn to_offset<'a>(&self, content: impl Into<Content<'a>>) -> usize;
} }

View File

@ -1,7 +1,13 @@
use rpc::proto;
use crate::{Anchor, Buffer, Point, ToOffset as _, ToPoint as _}; use crate::{Anchor, Buffer, Point, ToOffset as _, ToPoint as _};
use std::{cmp::Ordering, mem, ops::Range, sync::Arc}; use anyhow::anyhow;
use rpc::proto;
use std::{
cmp::Ordering,
convert::{TryFrom, TryInto},
mem,
ops::Range,
sync::Arc,
};
pub type SelectionSetId = clock::Lamport; pub type SelectionSetId = clock::Lamport;
pub type SelectionsVersion = usize; pub type SelectionsVersion = usize;
@ -24,8 +30,9 @@ pub struct Selection {
#[derive(Clone, Debug, Eq, PartialEq)] #[derive(Clone, Debug, Eq, PartialEq)]
pub struct SelectionSet { pub struct SelectionSet {
pub selections: Arc<[Selection]>, pub id: SelectionSetId,
pub active: bool, pub active: bool,
pub selections: Arc<[Selection]>,
} }
impl Selection { impl Selection {
@ -92,3 +99,43 @@ impl<'a> Into<proto::Selection> for &'a Selection {
} }
} }
} }
impl TryFrom<proto::Selection> for Selection {
type Error = anyhow::Error;
fn try_from(selection: proto::Selection) -> Result<Self, Self::Error> {
Ok(Selection {
id: selection.id as usize,
start: selection
.start
.ok_or_else(|| anyhow!("missing selection start"))?
.try_into()?,
end: selection
.end
.ok_or_else(|| anyhow!("missing selection end"))?
.try_into()?,
reversed: selection.reversed,
goal: SelectionGoal::None,
})
}
}
impl TryFrom<proto::SelectionSet> for SelectionSet {
type Error = anyhow::Error;
fn try_from(set: proto::SelectionSet) -> Result<Self, Self::Error> {
Ok(Self {
id: clock::Lamport {
replica_id: set.replica_id as u16,
value: set.lamport_timestamp,
},
active: set.is_active,
selections: Arc::from(
set.selections
.into_iter()
.map(TryInto::try_into)
.collect::<Result<Vec<Selection>, _>>()?,
),
})
}
}