From 955046c959d810fd6230fc1b4a0f613103d83d2e Mon Sep 17 00:00:00 2001
From: Venket <venketaramana.balachandran@tigerlab.com>
Date: Sat, 29 Jun 2024 16:01:50 +0800
Subject: [PATCH] [rust] implement list_distros match_condition method in rust

---
 crates/hyfetch/build.rs             | 60 +++++++++++++++++++++++++++++
 crates/hyfetch/src/neofetch_util.rs |  6 +++
 2 files changed, 66 insertions(+)

diff --git a/crates/hyfetch/build.rs b/crates/hyfetch/build.rs
index b67fd0a5..71a9156d 100644
--- a/crates/hyfetch/build.rs
+++ b/crates/hyfetch/build.rs
@@ -100,6 +100,66 @@ impl Distro {
 ",
     );
 
+    buf.push_str(
+        "
+impl Distro {
+    pub fn detect(name: &str) -> Option<Self> {
+",
+    );
+    for (variant, distro) in &variants {
+        let distro_pattern = &distro.pattern;
+        let matches: Vec<&str> = distro_pattern.split('|').map(|s| s.trim()).collect();
+        let mut condition = Vec::new();
+
+        for m in matches {
+            let stripped = m.trim_matches(|c| c == '*' || c == '\'' || c == '"').to_lowercase();
+
+            if stripped.contains('*') || stripped.contains('"') {
+                println!("TODO: Cannot properly parse: {}", m);
+            }
+
+            // Exact matches
+            if m.trim_matches('*') == m {
+                condition.push(format!("name == r#\"{}\"#", stripped));
+                continue;
+            }
+
+            // Both sides are *
+            if m.starts_with('*') && m.ends_with('*') {
+                condition.push(format!("(name.starts_with(r#\"{}\"#) || name.ends_with(r#\"{}\"#))", stripped, stripped));
+                continue;
+            }
+
+            // Ends with *
+            if m.ends_with('*') {
+                condition.push(format!("name.starts_with(r#\"{}\"#)", stripped));
+                continue;
+            }
+
+            // Starts with *
+            if m.starts_with('*') {
+                condition.push(format!("name.ends_with(r#\"{}\"#)", stripped));
+                continue;
+            }
+        }
+
+        let condition = condition.join(" || ");
+    
+        buf.push_str(&format!("
+        if {condition} {{
+            return Some(Self::{variant});
+        }}"
+        ));
+    };
+    buf.push_str(&format!("
+        None
+"
+    ));
+
+    buf.push_str("
+    }
+}");
+
     fs::write(out_path.join("distros.rs"), buf).expect("couldn't write distros.rs");
 }
 
diff --git a/crates/hyfetch/src/neofetch_util.rs b/crates/hyfetch/src/neofetch_util.rs
index 85b21c13..214d2279 100644
--- a/crates/hyfetch/src/neofetch_util.rs
+++ b/crates/hyfetch/src/neofetch_util.rs
@@ -9,6 +9,8 @@ use std::{env, fmt};
 use anyhow::{anyhow, Context, Result};
 use tracing::debug;
 
+use crate::distros::Distro;
+
 /// Gets the absolute path of the neofetch command.
 pub fn get_command_path() -> Result<PathBuf> {
     if let Ok(workspace_dir) = env::var("CARGO_WORKSPACE_DIR") {
@@ -61,6 +63,10 @@ where
     };
     debug!(%distro, "distro name");
 
+    if let Some(distro) = Distro::detect(&distro) {
+        return Ok(distro.ascii_art().to_owned());
+    }
+
     todo!()
 }