From f869c7f52d8fd1f1ef61e218bbec4d0dac27673d Mon Sep 17 00:00:00 2001
From: Max Audron <audron@cocaine.farm>
Date: Tue, 7 Jun 2022 12:28:18 +0200
Subject: init

---
 src/local/repostate.rs | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 src/local/repostate.rs

(limited to 'src/local/repostate.rs')

diff --git a/src/local/repostate.rs b/src/local/repostate.rs
new file mode 100644
index 0000000..ea3c5a6
--- /dev/null
+++ b/src/local/repostate.rs
@@ -0,0 +1,36 @@
+use super::Repo;
+
+#[derive(Clone, Debug)]
+pub enum RepoState {
+    Local,
+    Remote,
+    Synced,
+    Unknown,
+}
+
+impl std::fmt::Display for RepoState {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        use ansi_term::Colour::{Blue, Green, Red, Yellow};
+
+        match self {
+            RepoState::Local => f.write_str(&Yellow.paint("LOCAL  ").to_string()),
+            RepoState::Remote => f.write_str(&Blue.paint("REMOTE ").to_string()),
+            RepoState::Synced => f.write_str(&Green.paint("SYNCED ").to_string()),
+            RepoState::Unknown => f.write_str(&Red.paint("UNKNOWN").to_string()),
+        }
+    }
+}
+
+impl From<&Repo> for RepoState {
+    fn from(repo: &Repo) -> Self {
+        if repo.repo.is_some() && repo.forge.is_some() {
+            RepoState::Synced
+        } else if repo.repo.is_some() {
+            RepoState::Local
+        } else if repo.forge.is_some() {
+            RepoState::Remote
+        } else {
+            RepoState::Unknown
+        }
+    }
+}
-- 
cgit v1.2.3