Make log_index and timestamp public

dkellner · Aug 7, 2021 · 4f66a72 · 4f66a72
1 parent e58b75d
commit 4f66a72
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 21 deletions.
diff --git a/src/internal.rs b/src/internal.rs
@@ -27,7 +27,9 @@ impl<A: Author, T> Chronofold<A, T> {
                 if let Some((_, idx)) = self
                     .iter_log_indices_causal_range(reference..)
                     .filter(|(_, i)| self.references.get(i) == Some(reference))
-                    .filter(|(c, i)| matches!(c, Change::Delete) || self.timestamp(i).unwrap() > id)
+                    .filter(|(c, i)| {
+                        matches!(c, Change::Delete) || self.timestamp(*i).unwrap() > id
+                    })
                     .last()
                 {
                     self.iter_subtree(idx).last()
@@ -43,24 +45,6 @@ impl<A: Author, T> Chronofold<A, T> {
         }
     }
 
-    pub(crate) fn log_index(&self, timestamp: &Timestamp<A>) -> Option<LogIndex> {
-        for i in (timestamp.0).0..self.log.len() {
-            if self.timestamp(&LogIndex(i)).unwrap() == *timestamp {
-                return Some(LogIndex(i));
-            }
-        }
-        None
-    }
-
-    pub(crate) fn timestamp(&self, index: &LogIndex) -> Option<Timestamp<A>> {
-        if let (Some(shift), Some(author)) = (self.index_shifts.get(index), self.authors.get(index))
-        {
-            Some(Timestamp(index - shift, *author))
-        } else {
-            None
-        }
-    }
-
     pub(crate) fn apply_change(
         &mut self,
         id: Timestamp<A>,

diff --git a/src/iter.rs b/src/iter.rs
@@ -182,11 +182,11 @@ where
         let idx = LogIndex(self.idx_iter.next()?);
         let id = self
             .cfold
-            .timestamp(&idx)
+            .timestamp(idx)
             .expect("timestamps of already applied ops have to exist");
         let reference = self.cfold.references.get(&idx).map(|r| {
             self.cfold
-                .timestamp(&r)
+                .timestamp(r)
                 .expect("references of already applied ops have to exist")
         });
         let payload = match &self.cfold.log[idx.0] {

diff --git a/src/lib.rs b/src/lib.rs
@@ -196,6 +196,25 @@ impl<A: Author, T> Chronofold<A, T> {
         Session::new(author, self)
     }
 
+    pub fn log_index(&self, timestamp: &Timestamp<A>) -> Option<LogIndex> {
+        for i in (timestamp.0).0..self.log.len() {
+            if self.timestamp(LogIndex(i)).unwrap() == *timestamp {
+                return Some(LogIndex(i));
+            }
+        }
+        None
+    }
+
+    pub fn timestamp(&self, index: LogIndex) -> Option<Timestamp<A>> {
+        if let (Some(shift), Some(author)) =
+            (self.index_shifts.get(&index), self.authors.get(&index))
+        {
+            Some(Timestamp(&index - shift, *author))
+        } else {
+            None
+        }
+    }
+
     /// Applies an op to the chronofold.
     pub fn apply<V>(&mut self, op: Op<A, V>) -> Result<(), ChronofoldError<A, V>>
     where