prek 0.3.11

use std::fmt::Write;
use std::fmt::{Display, Formatter};
use std::ops::AddAssign;
use std::path::Path;
use std::time::{Duration, SystemTime};

use anyhow::Result;
use owo_colors::OwoColorize;
use rustc_hash::FxHashMap;
use rustc_hash::FxHashSet;
use strum::IntoEnumIterator;
use tracing::{debug, trace, warn};

use crate::cli::ExitStatus;
use crate::cli::cache_size::{dir_size_bytes, human_readable_bytes};
use crate::config::{self, Error as ConfigError, Repo as ConfigRepo, load_config};
use crate::hook::{HOOK_MARKER, HookEnvKey, HookSpec, InstallInfo, Repo as HookRepo};
use crate::printer::Printer;
use crate::store::{CacheBucket, REPO_MARKER, Store, ToolBucket};

#[derive(Debug, Copy, Clone, Eq, PartialEq)]
enum RemovalKind {
    Repos,
    HookEnvs,
    Tools,
    CacheEntries,
    PatchFiles,
}

impl RemovalKind {
    fn display(self, count: usize) -> &'static str {
        if count > 1 {
            match self {
                RemovalKind::Repos => "repos",
                RemovalKind::HookEnvs => "hook envs",
                RemovalKind::Tools => "tools",
                RemovalKind::CacheEntries => "cache entries",
                RemovalKind::PatchFiles => "patch files",
            }
        } else {
            match self {
                RemovalKind::Repos => "repo",
                RemovalKind::HookEnvs => "hook env",
                RemovalKind::Tools => "tool",
                RemovalKind::CacheEntries => "cache entry",
                RemovalKind::PatchFiles => "patch file",
            }
        }
    }
}

const STALE_PATCH_RETENTION: Duration = Duration::from_hours(30 * 24);

#[derive(Debug, Clone)]
struct RemovalItem {
    label: String,
    abs_path: String,
    lines: Vec<String>,
}

impl RemovalItem {
    fn new(label: String, abs_path: String) -> Self {
        Self {
            label,
            abs_path,
            lines: Vec::new(),
        }
    }
}

#[derive(Debug, Clone)]
struct Removal {
    kind: RemovalKind,
    count: usize,
    bytes: u64,
    items: Vec<RemovalItem>,
}

impl Removal {
    fn new(kind: RemovalKind) -> Self {
        Self {
            kind,
            count: 0,
            bytes: 0,
            items: Vec::new(),
        }
    }
}

impl Display for Removal {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{} {}",
            self.count.cyan().bold(),
            self.kind.display(self.count)
        )
    }
}

impl AddAssign for Removal {
    fn add_assign(&mut self, rhs: Self) {
        debug_assert_eq!(self.kind, rhs.kind);

        self.count += rhs.count;
        self.bytes = self.bytes.saturating_add(rhs.bytes);
        self.items.extend(rhs.items);
    }
}

#[derive(Debug, Default)]
struct RemovalSummary {
    parts: Vec<String>,
    count: usize,
    bytes: u64,
}

impl RemovalSummary {
    fn is_empty(&self) -> bool {
        self.parts.is_empty()
    }

    fn joined(&self) -> String {
        self.parts.join(", ")
    }

    fn total_bytes(&self) -> u64 {
        self.bytes
    }
}

impl AddAssign<&Removal> for RemovalSummary {
    fn add_assign(&mut self, rhs: &Removal) {
        if rhs.count > 0 {
            self.parts.push(rhs.to_string());
        }
        self.count += rhs.count;
        self.bytes = self.bytes.saturating_add(rhs.bytes);
    }
}

pub(crate) async fn cache_gc(
    store: &Store,
    dry_run: bool,
    verbose: bool,
    printer: Printer,
) -> Result<ExitStatus> {
    let _lock = store.lock_async().await?;

    let tracked_configs = store.tracked_configs()?;
    if tracked_configs.is_empty() {
        writeln!(printer.stdout(), "{}", "Nothing to clean".bold())?;
        return Ok(ExitStatus::Success);
    }

    let mut kept_configs: FxHashSet<&Path> = FxHashSet::default();
    let mut used_repo_keys: FxHashSet<String> = FxHashSet::default();
    let mut used_hook_env_dirs: FxHashSet<String> = FxHashSet::default();
    let mut used_tools: FxHashSet<ToolBucket> = FxHashSet::default();
    let mut used_tool_versions: FxHashMap<ToolBucket, FxHashSet<String>> = FxHashMap::default();
    let mut used_cache: FxHashSet<CacheBucket> = FxHashSet::default();
    let mut used_env_keys: Vec<HookEnvKey> = Vec::new();

    // Always keep Prek's own cache.
    used_cache.insert(CacheBucket::Prek);

    let installed = store.installed_hooks().await;

    for config_path in &tracked_configs {
        let config = match load_config(config_path) {
            Ok(config) => {
                trace!(path = %config_path.display(), "Found tracked config");
                config
            }
            Err(err) => match err {
                ConfigError::Io(e) if e.kind() == std::io::ErrorKind::NotFound => {
                    debug!(path = %config_path.display(), "Tracked config does not exist, dropping");
                    continue;
                }
                err => {
                    warn!(path = %config_path.display(), %err, "Failed to parse config, skipping for GC");
                    kept_configs.insert(config_path);
                    continue;
                }
            },
        };
        kept_configs.insert(config_path);

        used_env_keys.extend(hook_env_keys_from_config(store, &config));

        // Mark repos referenced by this config (if present in store).
        // We do this via config parsing (no clone), so GC won't keep repos for missing configs.
        for repo in &config.repos {
            if let ConfigRepo::Remote(remote) = repo {
                let key = store
                    .repo_path(remote)
                    .file_name()
                    .and_then(|name| name.to_str())
                    .map(str::to_string);
                if let Some(key) = key {
                    used_repo_keys.insert(key);
                }
            }
        }
    }

    // Mark tools/caches from hook languages.
    for key in &used_env_keys {
        used_tools.extend(key.language.tool_buckets());
        used_cache.extend(key.language.cache_buckets());
    }

    // Mark hook environments by matching already-installed env metadata.
    // While doing this, try to derive the specific tool *version* directories in use from
    // `InstallInfo.toolchain` (which is persisted in `.prek-hook.json`).
    for info in &installed {
        if used_env_keys.iter().any(|k| k.matches_install_info(info)) {
            if let Some(dir) = info
                .env_path
                .file_name()
                .and_then(|s| s.to_str())
                .map(str::to_string)
            {
                used_hook_env_dirs.insert(dir);
            }

            mark_tool_versions_from_install_info(store, info, &mut used_tool_versions);
        }
    }

    // Update tracking file to drop configs that no longer exist.
    if !dry_run && kept_configs.len() != tracked_configs.len() {
        let kept_configs = kept_configs.into_iter().map(Path::to_path_buf).collect();
        store.update_tracked_configs(&kept_configs)?;
    }

    // Sweep repos/<hash>
    let removed_repos = sweep_dir_by_name(
        RemovalKind::Repos,
        &store.repos_dir(),
        &used_repo_keys,
        dry_run,
        verbose,
    )?;

    // Sweep hooks/<hash>
    let removed_hooks = sweep_dir_by_name(
        RemovalKind::HookEnvs,
        &store.hooks_dir(),
        &used_hook_env_dirs,
        dry_run,
        verbose,
    )?;

    // Sweep tools/<bucket>
    let tools_root = store.tools_dir();
    let used_tool_names: FxHashSet<String> = used_tools.iter().map(ToString::to_string).collect();
    let removed_tool_buckets = sweep_dir_by_name(
        RemovalKind::Tools,
        &tools_root,
        &used_tool_names,
        dry_run,
        verbose,
    )?;

    // Sweep tools/<bucket>/<version>
    let removed_tool_versions = sweep_tool_versions(store, &used_tool_versions, dry_run, verbose)?;

    let mut removed_tools = removed_tool_buckets;
    removed_tools += removed_tool_versions;

    // Sweep cache/<bucket>
    let cache_root = store.cache_dir();
    let used_cache_names: FxHashSet<String> = used_cache.iter().map(ToString::to_string).collect();
    let removed_cache = sweep_dir_by_name(
        RemovalKind::CacheEntries,
        &cache_root,
        &used_cache_names,
        dry_run,
        verbose,
    )?;

    // Seep scratch/, as it is only temporary data.
    if !dry_run {
        let _ = fs_err::remove_dir_all(store.scratch_path());
    }
    // Keep recent recovery patches, but clear out stale ones that are unlikely to be useful.
    let removed_patches = sweep_stale_patch_files(&store.patches_dir(), dry_run, verbose)?;

    let mut removed = RemovalSummary::default();
    removed += &removed_repos;
    removed += &removed_hooks;
    removed += &removed_tools;
    removed += &removed_cache;
    removed += &removed_patches;

    let removed_total_bytes = removed.total_bytes();
    let (removed_bytes, removed_unit) = human_readable_bytes(removed_total_bytes);

    let verb = if dry_run { "Would remove" } else { "Removed" };
    if removed.is_empty() {
        writeln!(printer.stdout(), "{}", "Nothing to clean".bold())?;
    } else {
        writeln!(
            printer.stdout(),
            "{verb} {} ({})",
            removed.joined(),
            format!("{removed_bytes:.1}{removed_unit}").cyan().bold(),
        )?;

        if verbose {
            print_removed_details(printer, verb, &removed_repos)?;
            print_removed_details(printer, verb, &removed_hooks)?;
            print_removed_details(printer, verb, &removed_tools)?;
            print_removed_details(printer, verb, &removed_cache)?;
            print_removed_details(printer, verb, &removed_patches)?;
        }
    }

    Ok(ExitStatus::Success)
}

fn print_removed_details(printer: Printer, verb: &str, removal: &Removal) -> Result<()> {
    if removal.count == 0 {
        return Ok(());
    }

    writeln!(
        printer.stdout(),
        "\n{}:",
        format!("{verb} {removal}").bold()
    )?;

    let mut items = removal.items.clone();
    items.sort_unstable_by(|a, b| a.label.cmp(&b.label));
    for item in items {
        writeln!(printer.stdout(), "{} {}", "-".dimmed(), item.label.bold())?;
        writeln!(
            printer.stdout(),
            "  {}: {}",
            "path".bold().dimmed(),
            item.abs_path
        )?;

        for line in item.lines {
            writeln!(printer.stdout(), "  {line}")?;
        }
    }

    Ok(())
}

fn hook_env_keys_from_config(store: &Store, config: &config::Config) -> Vec<HookEnvKey> {
    let mut keys = Vec::new();

    for repo_config in &config.repos {
        match repo_config {
            ConfigRepo::Remote(repo_config) => {
                let repo_path = store.repo_path(repo_config);
                if !repo_path.is_dir() {
                    continue;
                }

                let repo = match HookRepo::remote(
                    repo_config.repo.clone(),
                    repo_config.rev.clone(),
                    repo_path,
                ) {
                    Ok(repo) => repo,
                    Err(err) => {
                        warn!(repo = %repo_config.repo, %err, "Failed to load repo manifest, skipping");
                        continue;
                    }
                };

                let remote_dep = repo_config.to_string();

                for hook_config in &repo_config.hooks {
                    let Some(manifest_hook) = repo.get_hook(&hook_config.id) else {
                        continue;
                    };

                    let mut hook_spec = manifest_hook.clone();
                    hook_spec.apply_remote_hook_overrides(hook_config);

                    match HookEnvKey::from_hook_spec(config, hook_spec, Some(&remote_dep)) {
                        Ok(Some(key)) => keys.push(key),
                        Ok(None) => {}
                        Err(err) => {
                            warn!(hook = %hook_config.id, repo = %remote_dep, %err, "Failed to compute hook env key, skipping");
                        }
                    }
                }
            }
            ConfigRepo::Local(repo_config) => {
                for hook in &repo_config.hooks {
                    let hook_spec = HookSpec::from(hook.clone());
                    match HookEnvKey::from_hook_spec(config, hook_spec, None) {
                        Ok(Some(key)) => keys.push(key),
                        Ok(None) => {}
                        Err(err) => {
                            warn!(hook = %hook.id, %err, "Failed to compute hook env key, skipping");
                        }
                    }
                }
            }
            _ => {} // Meta repos and builtin repos do not have hook envs.
        }
    }

    keys
}

fn mark_tool_versions_from_install_info(
    store: &Store,
    info: &InstallInfo,
    used_tool_versions: &mut FxHashMap<ToolBucket, FxHashSet<String>>,
) {
    // NOTE: `InstallInfo.toolchain` is typically the executable path (e.g.
    // tools/go/1.24.0/bin/go). We keep the first path component under the tool bucket.
    // If we can't recognize it, we do nothing (and GC will keep all versions).
    for bucket in info.language.tool_buckets() {
        let bucket_root = store.tools_path(*bucket);
        if let Some(version) = tool_version_dir_name(&bucket_root, &info.toolchain) {
            used_tool_versions
                .entry(*bucket)
                .or_default()
                .insert(version);
        }
    }
}

fn tool_version_dir_name(bucket_root: &Path, toolchain: &Path) -> Option<String> {
    let rel = toolchain.strip_prefix(bucket_root).ok()?;
    let version = rel.components().next()?.as_os_str().to_str()?;
    if version.is_empty() {
        return None;
    }
    Some(version.to_string())
}

fn sweep_tool_versions(
    store: &Store,
    used_tool_versions: &FxHashMap<ToolBucket, FxHashSet<String>>,
    dry_run: bool,
    verbose: bool,
) -> Result<Removal> {
    let mut total = Removal::new(RemovalKind::Tools);

    for bucket in ToolBucket::iter() {
        let bucket_root = store.tools_path(bucket);
        let keep_versions = used_tool_versions.get(&bucket);
        let removed =
            sweep_tool_bucket_versions(bucket, &bucket_root, keep_versions, dry_run, verbose)?;
        total += removed;
    }

    Ok(total)
}

fn sweep_tool_bucket_versions(
    bucket: ToolBucket,
    bucket_root: &Path,
    keep_versions: Option<&FxHashSet<String>>,
    dry_run: bool,
    collect_names: bool,
) -> Result<Removal> {
    let mut removal = Removal::new(RemovalKind::Tools);

    let entries = match fs_err::read_dir(bucket_root) {
        Ok(entries) => entries,
        Err(err) if err.kind() == std::io::ErrorKind::NotFound => {
            return Ok(Removal::new(RemovalKind::Tools));
        }
        Err(err) => return Err(err.into()),
    };

    for entry in entries {
        let entry = match entry {
            Ok(entry) => entry,
            Err(err) => {
                warn!(%err, root = %bucket_root.display(), "Failed to read tool bucket entry");
                continue;
            }
        };
        let path = entry.path();
        // Don't remove files (uv, and rustup are files inside tools/).
        if !path.is_dir() {
            continue;
        }

        let Some(version_name) = path.file_name().and_then(|n| n.to_str()) else {
            continue;
        };
        // Skip hidden/system dirs.
        if version_name.starts_with('.') {
            continue;
        }
        if keep_versions.is_some_and(|keep| keep.contains(version_name)) {
            continue;
        }

        let entry_bytes = dir_size_bytes(&path);

        let item = if collect_names {
            Some(RemovalItem::new(
                format!("{bucket}/{version_name}"),
                path.to_string_lossy().to_string(),
            ))
        } else {
            None
        };

        if dry_run {
            removal.count += 1;
            removal.bytes = removal.bytes.saturating_add(entry_bytes);
            if let Some(item) = item {
                removal.items.push(item);
            }
            continue;
        }

        if let Err(err) = fs_err::remove_dir_all(&path) {
            warn!(%err, path = %path.display(), "Failed to remove unused tool version");
        } else {
            removal.count += 1;
            removal.bytes = removal.bytes.saturating_add(entry_bytes);
            if let Some(item) = item {
                removal.items.push(item);
            }
        }
    }

    Ok(removal)
}

fn sweep_dir_by_name(
    kind: RemovalKind,
    root: &Path,
    keep_names: &FxHashSet<String>,
    dry_run: bool,
    collect_names: bool,
) -> Result<Removal> {
    let mut removal = Removal::new(kind);
    let entries = match fs_err::read_dir(root) {
        Ok(entries) => entries,
        Err(err) if err.kind() == std::io::ErrorKind::NotFound => return Ok(Removal::new(kind)),
        Err(err) => return Err(err.into()),
    };

    for entry in entries {
        let entry = match entry {
            Ok(entry) => entry,
            Err(err) => {
                warn!(%err, root = %root.display(), "Failed to read store entry");
                continue;
            }
        };
        let path = entry.path();
        if !path.is_dir() {
            continue;
        }

        let Some(name) = path.file_name().and_then(|n| n.to_str()) else {
            continue;
        };
        // Skip hidden/system dirs.
        if name.starts_with('.') {
            continue;
        }
        if keep_names.contains(name) {
            continue;
        }

        let entry_bytes = dir_size_bytes(&path);

        let item = if collect_names {
            let repo_marker = (kind == RemovalKind::Repos)
                .then(|| read_repo_marker(&path))
                .flatten();
            let hook_marker = (kind == RemovalKind::HookEnvs)
                .then(|| read_hook_marker(&path))
                .flatten();

            let mut item = RemovalItem::new(name.to_string(), path.to_string_lossy().to_string());

            if let Some(label) = label_for_entry(kind, repo_marker.as_ref(), hook_marker.as_ref()) {
                item.label = label;
            }

            item.lines = detail_lines_for_entry(kind, repo_marker.as_ref(), hook_marker.as_ref());
            Some(item)
        } else {
            None
        };

        if dry_run {
            removal.count += 1;
            removal.bytes = removal.bytes.saturating_add(entry_bytes);
            if collect_names && let Some(item) = item {
                removal.items.push(item);
            }
            continue;
        }

        // Best-effort cleanup.
        if let Err(err) = fs_err::remove_dir_all(&path) {
            warn!(%err, path = %path.display(), "Failed to remove unused cache entry");
        } else {
            removal.count += 1;
            removal.bytes = removal.bytes.saturating_add(entry_bytes);
            if collect_names {
                if let Some(item) = item {
                    removal.items.push(item);
                }
            }
        }
    }

    Ok(removal)
}

fn sweep_stale_patch_files(root: &Path, dry_run: bool, collect_names: bool) -> Result<Removal> {
    let mut removal = Removal::new(RemovalKind::PatchFiles);
    let entries = match fs_err::read_dir(root) {
        Ok(entries) => entries,
        Err(err) if err.kind() == std::io::ErrorKind::NotFound => {
            return Ok(Removal::new(RemovalKind::PatchFiles));
        }
        Err(err) => return Err(err.into()),
    };

    let cutoff = SystemTime::now()
        .checked_sub(STALE_PATCH_RETENTION)
        .unwrap_or(SystemTime::UNIX_EPOCH);

    for entry in entries {
        let Ok(entry) = entry else {
            continue;
        };

        if !entry.file_type().is_ok_and(|file_type| file_type.is_file()) {
            continue;
        }

        let path = entry.path();
        let Some(name) = path.file_name().and_then(|n| n.to_str()) else {
            continue;
        };
        if name.starts_with('.') || path.extension().and_then(|ext| ext.to_str()) != Some("patch") {
            continue;
        }

        let metadata = match entry.metadata() {
            Ok(metadata) => metadata,
            Err(err) => {
                warn!(%err, path = %path.display(), "Failed to read patch metadata");
                continue;
            }
        };
        let modified = match metadata.modified() {
            Ok(modified) => modified,
            Err(err) => {
                warn!(%err, path = %path.display(), "Failed to read patch modified time");
                continue;
            }
        };
        if modified > cutoff {
            continue;
        }

        let entry_bytes = metadata.len();
        let item = collect_names
            .then(|| RemovalItem::new(name.to_string(), path.to_string_lossy().to_string()));

        if dry_run {
            removal.count += 1;
            removal.bytes = removal.bytes.saturating_add(entry_bytes);
            if let Some(item) = item {
                removal.items.push(item);
            }
            continue;
        }

        if let Err(err) = fs_err::remove_file(&path) {
            warn!(%err, path = %path.display(), "Failed to remove old patch file");
        } else {
            removal.count += 1;
            removal.bytes = removal.bytes.saturating_add(entry_bytes);
            if let Some(item) = item {
                removal.items.push(item);
            }
        }
    }

    Ok(removal)
}

fn label_for_entry(
    kind: RemovalKind,
    repo_marker: Option<&RepoMarker>,
    hook_marker: Option<&InstallInfo>,
) -> Option<String> {
    match kind {
        RemovalKind::Repos => repo_marker.map(|repo| format!("{}@{}", repo.repo, repo.rev)),
        RemovalKind::HookEnvs => hook_marker.map(|info| {
            // Keep this short; more info goes in detail lines.
            format!("{} env", info.language.as_ref())
        }),
        _ => None,
    }
}

fn detail_lines_for_entry(
    kind: RemovalKind,
    _repo_marker: Option<&RepoMarker>,
    hook_marker: Option<&InstallInfo>,
) -> Vec<String> {
    const MAX_VALUE_CHARS: usize = 140;

    match kind {
        RemovalKind::Repos => vec![],
        RemovalKind::HookEnvs => {
            let Some(info) = hook_marker else {
                return Vec::new();
            };

            let mut lines = Vec::new();
            lines.push(format!(
                "{}: {} ({})",
                "language".dimmed().bold(),
                info.language.as_ref(),
                info.language_version
            ));

            let (repo_dep, deps) = split_repo_dependency(&info.dependencies);
            if let Some(repo_dep) = repo_dep {
                lines.push(format!(
                    "{}: {}",
                    "repo".dimmed().bold(),
                    truncate_end(&repo_dep, MAX_VALUE_CHARS)
                ));
            }
            if !deps.is_empty() {
                let deps_str = format_dependency_list(&deps, 6, MAX_VALUE_CHARS);
                lines.push(format!("{}: {deps_str}", "deps".dimmed().bold()));
            }
            lines
        }
        _ => Vec::new(),
    }
}

#[derive(Debug, serde::Deserialize)]
struct RepoMarker {
    repo: String,
    rev: String,
}

fn read_repo_marker(root: &Path) -> Option<RepoMarker> {
    // NOTE: `Store::clone_repo` serializes `RemoteRepo`, but with some fields skipped during
    // serialization (e.g. `hooks`). That means deserializing back into `RemoteRepo` can fail.
    // For GC display, we only need `repo` + `rev`.
    let content = fs_err::read_to_string(root.join(REPO_MARKER)).ok()?;
    serde_json::from_str(&content).ok()
}

fn read_hook_marker(root: &Path) -> Option<InstallInfo> {
    let content = fs_err::read_to_string(root.join(HOOK_MARKER)).ok()?;
    serde_json::from_str(&content).ok()
}

fn truncate_end(s: &str, max_chars: usize) -> String {
    if s.chars().count() <= max_chars {
        return s.to_string();
    }
    let mut out = s
        .chars()
        .take(max_chars.saturating_sub(1))
        .collect::<String>();
    out.push('…');
    out
}

fn split_repo_dependency(deps: &FxHashSet<String>) -> (Option<String>, Vec<String>) {
    // Best-effort: the remote repo dependency is typically `repo@rev`.
    // Prefer URL-like values to avoid accidentally treating PEP508 deps as repo identifiers.
    let mut repo_dep: Option<String> = None;
    let mut rest = Vec::new();

    for dep in deps {
        if repo_dep.is_none()
            && dep.contains('@')
            && (dep.contains("://")
                || dep.starts_with('/')
                || dep.starts_with("..")
                || dep.starts_with('.'))
        {
            repo_dep = Some(dep.clone());
        } else {
            rest.push(dep.clone());
        }
    }

    rest.sort_unstable();
    (repo_dep, rest)
}

fn format_dependency_list(deps: &[String], max_items: usize, max_chars: usize) -> String {
    if deps.is_empty() {
        return String::new();
    }

    let shown: Vec<&str> = deps.iter().take(max_items).map(String::as_str).collect();
    let extra = deps.len().saturating_sub(shown.len());
    let mut rendered = shown.join(", ");
    if extra > 0 {
        let _ = write!(&mut rendered, ", … (+{extra} more)");
    }
    truncate_end(&rendered, max_chars)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn truncate_end_returns_input_when_short_enough() {
        assert_eq!(truncate_end("abc", 3), "abc");
        assert_eq!(truncate_end("abc", 10), "abc");
    }

    #[test]
    fn truncate_end_truncates_and_appends_ellipsis() {
        assert_eq!(truncate_end("abcd", 3), "ab…");
        assert_eq!(truncate_end("abcdef", 5), "abcd…");
    }

    #[test]
    fn truncate_end_counts_chars_not_bytes() {
        // 3 unicode scalar values.
        assert_eq!(truncate_end("ééé", 3), "ééé");
        assert_eq!(truncate_end("ééé", 2), "é…");
    }

    #[test]
    fn split_repo_dependency_prefers_url_like_repo_at_rev() {
        let mut deps = FxHashSet::default();
        deps.insert("requests==2.32.0".to_string());
        deps.insert("black==24.1.0".to_string());
        deps.insert("https://github.com/pre-commit/pre-commit-hooks@v1.0.0".to_string());

        let (repo_dep, rest) = split_repo_dependency(&deps);

        assert_eq!(
            repo_dep.as_deref(),
            Some("https://github.com/pre-commit/pre-commit-hooks@v1.0.0")
        );
        assert_eq!(rest, vec!["black==24.1.0", "requests==2.32.0"]);
    }

    #[test]
    fn split_repo_dependency_returns_none_when_no_repo_like_dep() {
        let mut deps = FxHashSet::default();
        deps.insert("requests==2.32.0".to_string());
        deps.insert("black==24.1.0".to_string());

        let (repo_dep, rest) = split_repo_dependency(&deps);
        assert!(repo_dep.is_none());
        assert_eq!(rest, vec!["black==24.1.0", "requests==2.32.0"]);
    }

    #[test]
    fn format_dependency_list_includes_more_suffix() {
        let deps = vec!["a".to_string(), "b".to_string(), "c".to_string()];
        assert_eq!(format_dependency_list(&deps, 2, 200), "a, b, … (+1 more)");
    }

    #[test]
    fn format_dependency_list_truncates_rendered_string() {
        let deps = vec!["abcdef".to_string()];
        assert_eq!(format_dependency_list(&deps, 6, 5), "abcd…");
    }
}