snapshot of https://github.com/ActivityWatch/aw-server-rust.git (master, 2026-09-16; blobs > 131072B omitted)

aw-datastore/src/lib.rs

@@ -0,0 +1,61 @@
+#[macro_use]
+extern crate log;
+
+use std::fmt;
+
+#[macro_export]
+macro_rules! json_map {
+    { $( $key:literal : $value:expr),* } => {{
+        use serde_json::Value;
+        use serde_json::map::Map;
+        #[allow(unused_mut)]
+        let mut map : Map<String, Value> = Map::new();
+        $(
+          map.insert( $key.to_string(), json!($value) );
+        )*
+        map
+    }};
+}
+
+mod datastore;
+mod export;
+mod legacy_import;
+mod privacy_filter;
+mod worker;
+
+pub use self::datastore::DatastoreInstance;
+pub use self::worker::Datastore;
+
+#[derive(Clone)]
+pub enum DatastoreMethod {
+    Memory(),
+    File(String),
+    /// Encrypted SQLite file using SQLCipher. Only available with the
+    /// `encryption` or `encryption-vendored` feature flags.
+    #[cfg(any(feature = "encryption", feature = "encryption-vendored"))]
+    FileEncrypted(String, zeroize::Zeroizing<String>), // (path, key)
+}
+
+impl fmt::Debug for DatastoreMethod {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            DatastoreMethod::Memory() => write!(f, "Memory()"),
+            DatastoreMethod::File(p) => write!(f, "File({p:?})"),
+            #[cfg(any(feature = "encryption", feature = "encryption-vendored"))]
+            DatastoreMethod::FileEncrypted(p, _) => write!(f, "FileEncrypted({p:?}, <redacted>)"),
+        }
+    }
+}
+
+/* TODO: Implement this as a proper error */
+#[derive(Debug, Clone)]
+pub enum DatastoreError {
+    NoSuchBucket(String),
+    BucketAlreadyExists(String),
+    NoSuchKey(String),
+    MpscError,
+    InternalError(String),
+    // Errors specific to when migrate is disabled
+    Uninitialized(String),
+    OldDbVersion(String),
+}