first run semi-pass

Author: yashksaini-coder

src/app.rs

@@ -1,7 +1,34 @@
-// TODO: TO create a app for the project
+use crate::scanner::VenvInfo;
 
-pub fn create_app() -> Result<(), Box<dyn Error>> {
+pub struct App {
+    pub venvs: Vec<VenvInfo>,
+    pub selected: usize, // index of selected venv
+}
+
+impl App {
+    pub fn new(venvs: Vec<VenvInfo>) -> Self {
+        Self {
+            venvs,
+            selected: 0,
+        }
+    }
+
+    /// Move selection up
+    pub fn previous(&mut self) {
+        if self.selected > 0 {
+            self.selected -= 1;
+        }
+    }
 
-    let app = App::new();
+    /// Move selection down
+    pub fn next(&mut self) {
+        if self.selected + 1 < self.venvs.len() {
+            self.selected += 1;
+        }
+    }
 
+    /// Get the currently selected venv, if any
+    pub fn selected_venv(&self) -> Option<&VenvInfo> {
+        self.venvs.get(self.selected)
+    }
 }

src/main.rs

@@ -1,4 +1,25 @@
+mod scanner;
+mod utils;
+mod app;
+mod ui;
+
+use std::env;
+use app::App;
+use scanner::scan_for_venvs;
+use ui::draw_ui;
+
 fn main() {
-    println!("Hello, world!");
-    
+    let cwd = env::current_dir().unwrap();
+    let venvs = scan_for_venvs(&cwd);
+
+    let mut app = App::new(venvs);
+
+    // Simulate some navigation
+    draw_ui(&app);
+    println!("\n-- Moving down --");
+    app.next();
+    draw_ui(&app);
+    println!("\n-- Moving up --");
+    app.previous();
+    draw_ui(&app);
 }

src/scanner.rs

@@ -1,12 +1,68 @@
-// TODO: TO create a scanner that will scan the project for virtual environment of python using sys.prefix
+use std::fs;
+use std::path::{Path, PathBuf};
+use chrono::{DateTime, Local};
+use walkdir::WalkDir;
 
-pub fn scan_project(project_path: &str) -> Result<(), Box<dyn Error>> {
+#[derive(Debug)]
+pub struct VenvInfo {
+    pub path: PathBuf,
+    pub size: u64,
+    pub last_modified: Option<DateTime<Local>>,
+}
+
+impl VenvInfo {
+    pub fn new(path: PathBuf) -> Option<Self> {
+        // Check if this is really a virtualenv
+        let pyvenv_cfg = path.join("pyvenv.cfg");
+        if !pyvenv_cfg.exists() {
+            return None;
+        }
+
+        // Calculate size
+        let size = get_dir_size(&path).unwrap_or(0);
+
+        // Get last modified time
+        let metadata = fs::metadata(&path).ok()?;
+        let modified = metadata.modified().ok()
+            .and_then(|t| DateTime::<Local>::from(t).into());
+
+        Some(Self {
+            path,
+            size,
+            last_modified: modified,
+        })
+    }
+}
+
+/// Scan a directory recursively for potential Python virtualenv folders
+pub fn scan_for_venvs(root: &Path) -> Vec<VenvInfo> {
+    let mut results = Vec::new();
+
+    for entry in WalkDir::new(root).into_iter().filter_map(Result::ok) {
+        if !entry.file_type().is_dir() {
+            continue;
+        }
 
-    let project_path = Path::new(project_path);
-    let project_path = project_path.canonicalize()?;
+        let name = entry.file_name().to_string_lossy().to_lowercase();
+        if name == "venv" || name == ".venv" || name == "env" {
+            if let Some(venv) = VenvInfo::new(entry.path().to_path_buf()) {
+                results.push(venv);
+            }
+        }
+    }
+
+    results
+}
 
-    let project_path = project_path.to_str().unwrap();
+/// Recursively calculate directory size in bytes
+fn get_dir_size(path: &Path) -> Result<u64, std::io::Error> {
+    let mut size = 0;
 
-    let project_path = project_path.to_string();
+    for entry in WalkDir::new(path).into_iter().filter_map(Result::ok) {
+        if entry.file_type().is_file() {
+            size += entry.metadata()?.len();
+        }
+    }
 
+    Ok(size)
 }

src/ui.rs

@@ -1,7 +1,17 @@
-// TODO: TO create a ui for the project
+use crate::app::App;
 
-pub fn create_ui() -> Result<(), Box<dyn Error>> {
+pub fn draw_ui(app: &App) {
+    println!("Found {} virtual environments:\n", app.venvs.len());
 
-    let ui = Ui::new();
+    for (i, venv) in app.venvs.iter().enumerate() {
+        let marker = if i == app.selected { ">" } else { " " };
+        let size_mb = venv.size / 1024 / 1024;
 
+        println!(
+            "{} [{} MB] {}",
+            marker,
+            size_mb,
+            venv.path.display()
+        );
+    }
 }

src/utils.rs

@@ -1,7 +1,7 @@
-// TODO: TO create a utils for the project
+use std::fs;
+use std::path::Path;
 
-pub fn get_project_path() -> Result<(), Box<dyn Error>> {
-
-    let project_path = Path::new(project_path);
-    let project_path = project_path.canonicalize()?;
+/// Deletes a directory recursively
+pub fn delete_venv(path: &Path) -> Result<(), std::io::Error> {
+    fs::remove_dir_all(path)
 }