feedback

atuin-client/src/record/sqlite_store.rs

@@ -152,14 +152,14 @@ impl Store for SqliteStore {
         self.idx(host, tag, 0).await
     }
 
-    async fn len(&self, host: HostId, tag: &str) -> Result<Option<u64>> {
+    async fn len(&self, host: HostId, tag: &str) -> Result<u64> {
         let last = self.last(host, tag).await?;
 
         if let Some(last) = last {
-            return Ok(Some(last.idx + 1));
+            return Ok(last.idx + 1);
         }
 
-        return Ok(None);
+        return Ok(0);
     }
 
     async fn next(
@@ -336,7 +336,7 @@ mod tests {
             .await
             .expect("failed to get store len");
 
-        assert_eq!(len, Some(1), "expected length of 1 after insert");
+        assert_eq!(len, 1, "expected length of 1 after insert");
     }
 
     #[tokio::test]
@@ -355,8 +355,8 @@ mod tests {
         let first_len = db.len(first.host.id, first.tag.as_str()).await.unwrap();
         let second_len = db.len(second.host.id, second.tag.as_str()).await.unwrap();
 
-        assert_eq!(first_len, Some(1), "expected length of 1 after insert");
-        assert_eq!(second_len, Some(1), "expected length of 1 after insert");
+        assert_eq!(first_len, 1, "expected length of 1 after insert");
+        assert_eq!(second_len, 1, "expected length of 1 after insert");
     }
 
     #[tokio::test]
@@ -373,7 +373,7 @@ mod tests {
 
         assert_eq!(
             db.len(tail.host.id, tail.tag.as_str()).await.unwrap(),
-            Some(100),
+            100,
             "failed to insert 100 records"
         );
     }
@@ -396,7 +396,7 @@ mod tests {
 
         assert_eq!(
             db.len(tail.host.id, tail.tag.as_str()).await.unwrap(),
-            Some(10000),
+            10000,
             "failed to insert 10k records"
         );
     }

atuin-client/src/record/store.rs

@@ -2,7 +2,6 @@ use async_trait::async_trait;
 use eyre::Result;
 
 use atuin_common::record::{EncryptedData, HostId, Record, RecordId, RecordIdx, RecordStatus};
-
 /// A record store stores records
 /// In more detail - we tend to need to process this into _another_ format to actually query it.
 /// As is, the record store is intended as the source of truth for arbitratry data, which could
@@ -21,12 +20,12 @@ pub trait Store {
     ) -> Result<()>;
 
     async fn get(&self, id: RecordId) -> Result<Record<EncryptedData>>;
-    async fn len(&self, host: HostId, tag: &str) -> Result<Option<u64>>;
+    async fn len(&self, host: HostId, tag: &str) -> Result<u64>;
 
     async fn last(&self, host: HostId, tag: &str) -> Result<Option<Record<EncryptedData>>>;
     async fn first(&self, host: HostId, tag: &str) -> Result<Option<Record<EncryptedData>>>;
 
-    /// Get the record that follows this record
+    /// Get the next `limit` records, after and including the given index
     async fn next(
         &self,
         host: HostId,

atuin-common/src/record.rs

@@ -14,7 +14,7 @@ pub struct EncryptedData {
     pub content_encryption_key: String,
 }
 
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, PartialOrd, Ord, Eq)]
 pub struct Diff {
     pub host: HostId,
     pub tag: String,
@@ -194,14 +194,7 @@ impl RecordStatus {
         }
 
         // Stability is a nice property to have
-        ret.sort_by(|a, b| {
-            (a.host, a.tag.clone(), a.local, a.remote).cmp(&(
-                b.host,
-                b.tag.clone(),
-                b.local,
-                b.remote,
-            ))
-        });
+        ret.sort();
         ret
     }
 }