Merge pull request #14 from abhixdd/feat/ux-cli-and-tests

feat: enhance UX, refine download path logic, and expand test suite

Author: abhixdd

README.md

@@ -13,7 +13,7 @@
 - **No more clone-and-delete**: Grab exactly what you need, when you need it.
 - **Easy on the eyes**: A clean terminal interface that makes browsing feel smooth.
 - **Works where you are**: Installs quickly via NPM, Cargo, or PIP.
-- **Find things fast**: Quickly search and navigate through any repo's folders.
+- **Find things fast**: Quickly search and navigate through any repo's folders with fuzzy search.
 - **Handles the big stuff**: Built-in support for GitHub LFS (Large File Storage).
 - **Batch mode**: Select a bunch of files and folders to download them all at once.
 
@@ -49,9 +49,21 @@ ghgrab
 Or, if you already have a link, just paste it in:
 
 ```bash
+# Browse a repository
 ghgrab https://github.com/rust-lang/rust
+
+# Download to current directory directly
+ghgrab https://github.com/rust-lang/rust --cwd --no-folder
 ```
 
+### CLI Flags
+
+| Flag | Description |
+|------|-------------|
+| `--cwd` | Forces download to the current working directory. |
+| `--no-folder` | Downloads files directly without creating a subfolder for the repo. |
+| `--token <TOKEN>`| Use a specific GitHub token for this run (doesn't save to settings). |
+
 ### Configuration
 
 To manage your settings:
@@ -79,7 +91,10 @@ We've kept it pretty standard, but here's a quick cheat sheet:
 |-----|--------|
 | `Enter` / `l` / `Right` | Enter directory or Submit URL |
 | `Backspace` / `h` / `Left` | Go back to previous folder |
-| `Esc` | **Return Home** (file list) or **Quit** (home screen) |
+| `Delete` | Clear URL input (Home page) |
+| `Tab` | Auto-fill `https://github.com/` (Home page) |
+| `/` | Start Searching (File list) |
+| `Esc` | **Exit Search** or **Return Home** (file list) or **Quit** (home screen) |
 | `q` / `Q` | **Quit** (from file list) |
 | `Ctrl+q` | **Force Quit** (anywhere) |
 | `Space` | Toggle selection for the current item |

bin/ghgrab.js

@@ -3,7 +3,6 @@ const { spawn } = require('child_process');
 const path = require('path');
 const fs = require('fs');
 
-// Binary lives next to this file in the bin/ directory
 const binaryName = 'ghgrab' + (process.platform === 'win32' ? '.exe' : '');
 const binPath = path.join(__dirname, binaryName);
 

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 name = "ghgrab"
 version = "1.0.0"
 description = "Download specific files and folders from GitHub repositories without cloning"
-authors = [{name = "Abhinav A"}]
+authors = [{name = "abhixdd"}]
 license = {text = "MIT"}
 readme = "README.md"
 requires-python = ">=3.7"

src/download.rs

@@ -26,10 +26,6 @@ impl Downloader {
         let mut errors = Vec::new();
 
         for item in items {
-            if !item.selected {
-                continue;
-            }
-
             let dest_path = self.base_path.join(&item.name);
 
             let result = if item.is_file() {

src/github.rs

@@ -52,6 +52,31 @@ impl GitHubUrl {
         })
     }
 
+    pub fn get_local_git_remote() -> Option<String> {
+        use std::process::Command;
+        let output = Command::new("git")
+            .args(["remote", "get-url", "origin"])
+            .output()
+            .ok()?;
+
+        if output.status.success() {
+            let url = String::from_utf8_lossy(&output.stdout).trim().to_string();
+            if !url.is_empty() {
+                if url.starts_with("git@github.com:") {
+                    let path = url
+                        .trim_start_matches("git@github.com:")
+                        .trim_end_matches(".git");
+                    return Some(format!("https://github.com/{}", path));
+                }
+                if url.contains("github.com") && url.ends_with(".git") {
+                    return Some(url.trim_end_matches(".git").to_string());
+                }
+                return Some(url);
+            }
+        }
+        None
+    }
+
     pub fn api_url(&self) -> String {
         let base = format!(
             "https://api.github.com/repos/{}/{}/contents",

src/main.rs

@@ -1,6 +1,7 @@
 use anyhow::Result;
 use clap::{Parser, Subcommand};
 use ghgrab::config::Config;
+
 use ghgrab::ui;
 
 #[derive(Parser)]
@@ -10,6 +11,15 @@ struct Cli {
     command: Option<Commands>,
 
     url: Option<String>,
+
+    #[arg(long, help = "Download files to current directory")]
+    cwd: bool,
+
+    #[arg(long, help = "Download files directly into target without repo folder")]
+    no_folder: bool,
+
+    #[arg(long, help = "One-time GitHub token (not stored)")]
+    token: Option<String>,
 }
 
 #[derive(Subcommand)]
@@ -109,7 +119,13 @@ async fn main() -> Result<()> {
         },
         None => {
             let config = Config::load().unwrap_or_default();
-            ui::run_tui(cli.url, config.github_token, config.download_path).await?;
+
+            let url = cli.url;
+
+            let download_path = config.download_path;
+
+            let token = cli.token.or(config.github_token);
+            ui::run_tui(url, token, download_path, cli.cwd, cli.no_folder).await?;
         }
     }
 

src/ui/components/browser.rs

@@ -15,35 +15,33 @@ pub struct BrowserState<'a> {
     pub current_url: Option<&'a GitHubUrl>,
     pub cursor: usize,
     pub scroll_offset: usize,
-    #[allow(dead_code)]
     pub status_msg: &'a str,
     pub is_downloading: bool,
     pub ascii_mode: bool,
     pub folder_sizes: &'a HashMap<String, u64>,
+    pub is_searching: bool,
+    pub search_query: &'a str,
 }
 
 pub fn render(f: &mut Frame, area: Rect, state: &BrowserState) {
-    let chunks = if state.is_downloading {
-        Layout::default()
-            .direction(Direction::Vertical)
-            .constraints([
-                Constraint::Length(3), // Breadcrumb
-                Constraint::Min(10),   // File list
-                Constraint::Length(2), // Download status
-                Constraint::Length(2),
-            ])
-            .split(area)
-    } else {
-        Layout::default()
-            .direction(Direction::Vertical)
-            .constraints([
-                Constraint::Length(3), // Breadcrumb
-                Constraint::Min(10),   // File list
-                Constraint::Length(1), // Spacer
-                Constraint::Length(2),
-            ])
-            .split(area)
-    };
+    let chunks = Layout::default()
+        .direction(Direction::Vertical)
+        .constraints([
+            Constraint::Length(3), // Breadcrumb
+            Constraint::Min(10),   // File list
+            if state.is_downloading {
+                Constraint::Length(2)
+            } else {
+                Constraint::Length(0)
+            },
+            if state.is_searching {
+                Constraint::Length(3)
+            } else {
+                Constraint::Length(0)
+            },
+            Constraint::Length(2), // Help
+        ])
+        .split(area);
 
     let breadcrumb_text = if let Some(url) = state.current_url {
         format!(" {}/{} : {}", url.owner, url.repo, url.path)
@@ -161,20 +159,26 @@ pub fn render(f: &mut Frame, area: Rect, state: &BrowserState) {
                     .unwrap_or_else(|| format!("{:>12}", ""))
             };
 
-            let display_name = if item.name.len() > 35 {
-                if let Some(dot_pos) = item.name.rfind('.') {
-                    let ext = &item.name[dot_pos..];
-                    let name_part = &item.name[..dot_pos];
+            let source_name = if state.is_searching {
+                &item.path
+            } else {
+                &item.name
+            };
+
+            let display_name = if source_name.len() > 35 {
+                if let Some(dot_pos) = source_name.rfind('.') {
+                    let ext = &source_name[dot_pos..];
+                    let name_part = &source_name[..dot_pos];
                     if name_part.len() > 30 {
                         format!("{}.....{}", &name_part[..30], ext)
                     } else {
-                        item.name.clone()
+                        source_name.clone()
                     }
                 } else {
-                    format!("{}.....", &item.name[..35])
+                    format!("{}.....", &source_name[..35])
                 }
             } else {
-                item.name.clone()
+                source_name.clone()
             };
 
             let name_with_icon = format!("{}{}", icon, display_name);
@@ -244,6 +248,18 @@ pub fn render(f: &mut Frame, area: Rect, state: &BrowserState) {
         f.render_widget(status, chunks[2]);
     }
 
+    // Search Bar
+    if state.is_searching {
+        let search_text = state.search_query.to_string();
+        let search_bar = Paragraph::new(search_text).block(
+            Block::default()
+                .borders(Borders::ALL)
+                .title(" Search ")
+                .border_style(Style::default().fg(SUCCESS_COLOR)),
+        );
+        f.render_widget(search_bar, chunks[3]);
+    }
+
     let help_spans = vec![
         Span::styled("  ", Style::default()),
         Span::styled(
@@ -329,5 +345,5 @@ pub fn render(f: &mut Frame, area: Rect, state: &BrowserState) {
     let help = Paragraph::new(Line::from(help_spans))
         .alignment(ratatui::layout::Alignment::Center)
         .style(Style::default().bg(BG_COLOR));
-    f.render_widget(help, chunks[3]);
+    f.render_widget(help, chunks[4]);
 }

src/ui/components/input.rs

@@ -8,7 +8,14 @@ use ratatui::{
 
 use crate::ui::theme::*;
 
-pub fn render(f: &mut Frame, area: Rect, input_text: &str, status_msg: &str, cursor_visible: bool) {
+pub fn render(
+    f: &mut Frame,
+    area: Rect,
+    input_text: &str,
+    url_cursor: usize,
+    status_msg: &str,
+    cursor_visible: bool,
+) {
     let vertical_layout = Layout::default()
         .direction(Direction::Vertical)
         .constraints([
@@ -96,13 +103,41 @@ pub fn render(f: &mut Frame, area: Rect, input_text: &str, status_msg: &str, cur
         ])
         .split(vertical_layout[4]);
 
-    let display_text = if cursor_visible {
-        format!("{}_", input_text)
+    let display_content = if input_text.is_empty() {
+        if cursor_visible {
+            Line::from(vec![
+                Span::styled("_", Style::default().fg(FG_COLOR)),
+                Span::styled(
+                    " (Press Tab to auto-fill GitHub URL)",
+                    Style::default()
+                        .fg(BORDER_COLOR)
+                        .add_modifier(Modifier::ITALIC),
+                ),
+            ])
+        } else {
+            Line::from(vec![
+                Span::styled(" ", Style::default().fg(FG_COLOR)),
+                Span::styled(
+                    " (Press Tab to auto-fill GitHub URL)",
+                    Style::default()
+                        .fg(BORDER_COLOR)
+                        .add_modifier(Modifier::ITALIC),
+                ),
+            ])
+        }
+    } else if cursor_visible {
+        let mut s = input_text.to_string();
+        if url_cursor >= s.len() {
+            s.push('_');
+        } else {
+            s.replace_range(url_cursor..url_cursor + 1, "_");
+        }
+        Line::from(Span::raw(s))
     } else {
-        format!("{} ", input_text)
+        Line::from(Span::raw(input_text.to_string()))
     };
 
-    let input = Paragraph::new(display_text)
+    let input = Paragraph::new(display_content)
         .block(
             Block::default()
                 .borders(Borders::ALL)