refactor: move session traits and stucts into module

1 files changed, 2 insertions, 92 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 94ce0ac..0dcdf7b 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,102 +1,12 @@
-use std::{
-    collections::{hash_map::Entry, HashMap},
-    fmt::Display,
-    iter::IntoIterator,
-    time::{Duration, SystemTime, UNIX_EPOCH},
-};
-
-use serde::{Deserialize, Serialize};
-
 pub use config::Config;
 pub use history::History;
+pub use session::{Session, SessionSource};
 pub use ssh::KnownHosts;
 pub use tmux::Tmux;
 
 mod config;
 mod history;
+mod session;
 mod ssh;
 mod tmux;
 mod unix;
-
-pub trait SessionSource {
-    type Error;
-
-    type Iter: IntoIterator<Item = Session>;
-
-    fn sessions(&self) -> Result<Self::Iter, Self::Error>;
-
-    fn update(
-        &self,
-        mut hash_map: HashMap<String, Session>,
-    ) -> Result<HashMap<String, Session>, Self::Error> {
-        for session in self.sessions()? {
-            match hash_map.entry(session.name.to_owned()) {
-                Entry::Occupied(mut o) if &session > o.get() => {
-                    o.insert(session);
-                }
-                Entry::Vacant(v) => {
-                    v.insert(session);
-                }
-                _ => {}
-            }
-        }
-        Ok(hash_map)
-    }
-}
-
-#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Serialize, Deserialize)]
-pub enum State {
-    Discovered,
-    Connected,
-    Updated,
-}
-
-#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Serialize, Deserialize)]
-pub struct Session {
-    pub state: State,
-
-    #[serde(with = "epoch_timestamp")]
-    pub timestamp: Duration,
-
-    pub name: String,
-}
-
-mod epoch_timestamp {
-    use std::time::Duration;
-
-    use serde::{self, Deserialize, Deserializer, Serializer};
-
-    pub fn serialize<S>(duration: &Duration, serializer: S) -> Result<S::Ok, S::Error>
-    where
-        S: Serializer,
-    {
-        serializer.serialize_u64(duration.as_secs())
-    }
-
-    pub fn deserialize<'de, D>(d: D) -> Result<Duration, D::Error>
-    where
-        D: Deserializer<'de>,
-    {
-        u64::deserialize(d).map(Duration::from_secs)
-    }
-}
-
-impl Display for Session {
-    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
-        write!(f, "{}", self.name)
-    }
-}
-
-impl From<String> for Session {
-    fn from(name: String) -> Self {
-        let timestamp = SystemTime::now()
-            .duration_since(UNIX_EPOCH)
-            .expect("Current time is pre-epoch. (Time traveler?)");
-
-        Self {
-            state: State::Discovered,
-            timestamp,
-            name,
-        }
-    }
-}