mnemo-rs 1.6.23

use anyhow::{Context, Result};
use rusqlite::{Connection, OpenFlags};
use std::path::Path;
use std::time::{SystemTime, UNIX_EPOCH};

use crate::migrations;

/// Données nécessaires pour insérer une nouvelle commande.
#[derive(Debug, Clone, Default)]
pub struct NewCommand {
    pub command: String,
    pub cwd: Option<String>,
    pub shell: Option<String>,
    pub hostname: Option<String>,
    pub exit_code: Option<i64>,
    pub created_at: String,
    /// Racine du dépôt Git (`git rev-parse --show-toplevel`), si applicable.
    pub git_root: Option<String>,
    /// Branche Git courante, si applicable.
    pub git_branch: Option<String>,
    /// URL du remote `origin`, si applicable.
    pub git_remote: Option<String>,
    /// Identifiant de session shell, si fourni (`MNEMO_SESSION_ID`).
    pub session_id: Option<String>,
}

/// Commande lue depuis la base.
///
/// Certains champs (id, shell, hostname, exit_code) font partie du modèle de
/// données mais ne sont pas tous affichés par le MVP de la TUI.
#[allow(dead_code)]
#[derive(Debug, Clone)]
pub struct CommandRecord {
    pub id: i64,
    pub command: String,
    pub cwd: Option<String>,
    pub shell: Option<String>,
    pub hostname: Option<String>,
    pub exit_code: Option<i64>,
    pub created_at: String,
    pub git_root: Option<String>,
    pub git_branch: Option<String>,
    pub git_remote: Option<String>,
    pub session_id: Option<String>,
}

/// Filtre optionnel appliqué à la recherche (contexte Git).
#[derive(Debug, Clone, Default)]
pub struct SearchFilter {
    /// Filtre sur le projet : nom du dossier racine Git ou chemin `git_root`.
    pub project: Option<String>,
    /// Filtre sur la branche Git.
    pub branch: Option<String>,
}

impl SearchFilter {
    /// Vrai si aucun critère n'est défini.
    #[allow(dead_code)]
    pub fn is_empty(&self) -> bool {
        self.project.is_none() && self.branch.is_none()
    }
}

/// Filtre de requête avancé (`mnemo search`, `mnemo stats --since`).
///
/// Tous les critères sont **combinables** (ET logique). Les champs temporels
/// `since`/`before` sont des bornes au format `YYYY-MM-DD HH:MM:SS` (ou un
/// préfixe `YYYY-MM-DD`), comparées lexicographiquement à `created_at`, ce qui
/// équivaut à une comparaison chronologique grâce au format ISO trié.
#[derive(Debug, Clone, Default)]
pub struct QueryFilter {
    /// Projet : chemin `git_root` complet ou nom du dossier racine.
    pub project: Option<String>,
    /// Branche Git exacte.
    pub branch: Option<String>,
    /// Répertoire de travail exact.
    pub cwd: Option<String>,
    /// Shell exact (ex : `bash`).
    pub shell: Option<String>,
    /// Code de sortie exact.
    pub exit_code: Option<i64>,
    /// N'inclure que les échecs (`exit_code` présent et ≠ 0).
    pub failed: bool,
    /// Borne inférieure incluse : `created_at >= since`.
    pub since: Option<String>,
    /// Borne supérieure exclue : `created_at < before`.
    pub before: Option<String>,
}

impl QueryFilter {
    /// Construit le fragment `WHERE` et les paramètres liés associés.
    ///
    /// La clause est toujours valide (au minimum `1=1`), et chaque critère est
    /// passé en paramètre lié (jamais interpolé) pour éviter toute injection.
    fn build_where(&self) -> (String, Vec<Box<dyn rusqlite::ToSql>>) {
        let mut clauses: Vec<String> = Vec::new();
        let mut params: Vec<Box<dyn rusqlite::ToSql>> = Vec::new();

        if let Some(branch) = &self.branch {
            clauses.push("git_branch = ?".to_string());
            params.push(Box::new(branch.clone()));
        }
        if let Some(project) = &self.project {
            clauses.push("(git_root = ? OR git_root LIKE ?)".to_string());
            params.push(Box::new(project.clone()));
            params.push(Box::new(format!("%/{project}")));
        }
        if let Some(cwd) = &self.cwd {
            clauses.push("cwd = ?".to_string());
            params.push(Box::new(cwd.clone()));
        }
        if let Some(shell) = &self.shell {
            clauses.push("shell = ?".to_string());
            params.push(Box::new(shell.clone()));
        }
        if let Some(code) = self.exit_code {
            clauses.push("exit_code = ?".to_string());
            params.push(Box::new(code));
        }
        if self.failed {
            clauses.push("exit_code IS NOT NULL AND exit_code != 0".to_string());
        }
        if let Some(since) = &self.since {
            clauses.push("created_at >= ?".to_string());
            params.push(Box::new(since.clone()));
        }
        if let Some(before) = &self.before {
            clauses.push("created_at < ?".to_string());
            params.push(Box::new(before.clone()));
        }

        let where_sql = if clauses.is_empty() {
            "1 = 1".to_string()
        } else {
            clauses.join(" AND ")
        };
        (where_sql, params)
    }
}

/// Charge les commandes correspondant à un [`QueryFilter`] avancé, des plus
/// récentes aux plus anciennes. `limit` borne le nombre de lignes (`None` =
/// toutes), utile pour `mnemo stats` qui agrège l'intégralité.
pub fn fetch_query(
    conn: &Connection,
    filter: &QueryFilter,
    limit: Option<usize>,
) -> Result<Vec<CommandRecord>> {
    let (where_sql, params) = filter.build_where();
    let limit_sql = match limit {
        Some(n) => format!("LIMIT {}", n as i64),
        None => String::new(),
    };
    let sql = format!(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands
         WHERE {where_sql}
         ORDER BY created_at DESC, id DESC
         {limit_sql}"
    );
    let mut stmt = conn.prepare(&sql)?;
    let rows = stmt.query_map(rusqlite::params_from_iter(params.iter()), row_to_record)?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Ouvre (ou crée) la base SQLite sur disque et initialise le schéma.
pub fn open(path: &Path) -> Result<Connection> {
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)
            .with_context(|| format!("création du dossier {}", parent.display()))?;
        crate::config::harden_dir(parent);
    }
    let conn = Connection::open(path)
        .with_context(|| format!("ouverture de la base {}", path.display()))?;
    migrations::apply(&conn)?;
    // La base contient l'historique shell : permissions privées.
    crate::config::harden_file(path);
    Ok(conn)
}

/// Ouvre la base et renvoie aussi le résultat des migrations appliquées.
/// Utilisé par `mnemo migrate` pour rendre compte de la transition de schéma.
pub fn open_and_migrate(path: &Path) -> Result<(Connection, migrations::Outcome)> {
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)
            .with_context(|| format!("création du dossier {}", parent.display()))?;
        crate::config::harden_dir(parent);
    }
    let conn = Connection::open(path)
        .with_context(|| format!("ouverture de la base {}", path.display()))?;
    let outcome = migrations::apply(&conn)?;
    crate::config::harden_file(path);
    Ok((conn, outcome))
}

/// Base SQLite en mémoire (utilisée pour les tests).
#[cfg(test)]
pub fn open_in_memory() -> Result<Connection> {
    let conn = Connection::open_in_memory()?;
    migrations::apply(&conn)?;
    Ok(conn)
}

/// Ouvre une base existante en lecture seule, SANS créer ni modifier le schéma.
/// Utilisé par `mnemo doctor` pour ne jamais altérer le système.
pub fn open_readonly(path: &Path) -> Result<Connection> {
    let conn = Connection::open_with_flags(path, OpenFlags::SQLITE_OPEN_READ_ONLY)
        .with_context(|| format!("ouverture en lecture seule de {}", path.display()))?;
    Ok(conn)
}

/// Indique si une table donnée existe dans la base.
pub fn table_exists(conn: &Connection, name: &str) -> Result<bool> {
    let n: i64 = conn.query_row(
        "SELECT COUNT(*) FROM sqlite_master WHERE type = 'table' AND name = ?1",
        [name],
        |row| row.get(0),
    )?;
    Ok(n > 0)
}

/// Hash FNV-1a 64 bits, déterministe, utilisé pour le dédoublonnage.
///
/// Le hash combine la commande et le répertoire courant : une même commande
/// dans deux répertoires différents n'est donc pas considérée comme doublon.
pub fn compute_hash(command: &str, cwd: Option<&str>) -> String {
    const FNV_OFFSET: u64 = 0xcbf2_9ce4_8422_2325;
    const FNV_PRIME: u64 = 0x0000_0100_0000_01b3;

    let mut hash = FNV_OFFSET;
    for b in command.bytes() {
        hash ^= b as u64;
        hash = hash.wrapping_mul(FNV_PRIME);
    }
    // Séparateur explicite entre commande et cwd.
    hash ^= 0x1f;
    hash = hash.wrapping_mul(FNV_PRIME);
    if let Some(cwd) = cwd {
        for b in cwd.bytes() {
            hash ^= b as u64;
            hash = hash.wrapping_mul(FNV_PRIME);
        }
    }
    format!("{hash:016x}")
}

/// Insère une commande. Retourne `true` si elle a été insérée, `false` si
/// c'était un doublon (même hash déjà présent).
pub fn insert_command(conn: &Connection, cmd: &NewCommand) -> Result<bool> {
    let hash = compute_hash(&cmd.command, cmd.cwd.as_deref());
    let changed = conn.execute(
        "INSERT OR IGNORE INTO commands
            (command, cwd, shell, hostname, exit_code, created_at, hash,
             git_root, git_branch, git_remote, session_id)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11)",
        rusqlite::params![
            cmd.command,
            cmd.cwd,
            cmd.shell,
            cmd.hostname,
            cmd.exit_code,
            cmd.created_at,
            hash,
            cmd.git_root,
            cmd.git_branch,
            cmd.git_remote,
            cmd.session_id,
        ],
    )?;
    Ok(changed > 0)
}

/// Charge les commandes les plus récentes (limite paramétrable).
#[allow(dead_code)]
pub fn fetch_all(conn: &Connection, limit: usize) -> Result<Vec<CommandRecord>> {
    fetch_filtered(conn, &SearchFilter::default(), limit)
}

/// Charge les commandes les plus récentes en appliquant un filtre de contexte
/// Git optionnel (projet / branche). Le filtrage fuzzy sur le texte de la
/// commande reste à la charge de l'appelant.
pub fn fetch_filtered(
    conn: &Connection,
    filter: &SearchFilter,
    limit: usize,
) -> Result<Vec<CommandRecord>> {
    // `project` correspond soit au chemin complet `git_root`, soit au nom du
    // dossier racine (dernier segment du chemin).
    let project_suffix = filter.project.as_ref().map(|p| format!("%/{p}"));
    let mut stmt = conn.prepare(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands
         WHERE (?1 IS NULL OR git_branch = ?1)
           AND (?2 IS NULL OR git_root = ?2 OR git_root LIKE ?3)
         ORDER BY created_at DESC, id DESC
         LIMIT ?4",
    )?;
    let rows = stmt.query_map(
        rusqlite::params![filter.branch, filter.project, project_suffix, limit as i64],
        row_to_record,
    )?;

    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Charge toutes les commandes correspondant au filtre Git (sans limite), pour
/// le calcul des statistiques. Un filtre vide renvoie l'intégralité de la base.
pub fn all_commands(conn: &Connection, filter: &SearchFilter) -> Result<Vec<CommandRecord>> {
    let project_suffix = filter.project.as_ref().map(|p| format!("%/{p}"));
    let mut stmt = conn.prepare(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands
         WHERE (?1 IS NULL OR git_branch = ?1)
           AND (?2 IS NULL OR git_root = ?2 OR git_root LIKE ?3)
         ORDER BY created_at DESC, id DESC",
    )?;
    let rows = stmt.query_map(
        rusqlite::params![filter.branch, filter.project, project_suffix],
        row_to_record,
    )?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Convertit une ligne SQL en [`CommandRecord`].
fn row_to_record(row: &rusqlite::Row) -> rusqlite::Result<CommandRecord> {
    Ok(CommandRecord {
        id: row.get(0)?,
        command: row.get(1)?,
        cwd: row.get(2)?,
        shell: row.get(3)?,
        hostname: row.get(4)?,
        exit_code: row.get(5)?,
        created_at: row.get(6)?,
        git_root: row.get(7)?,
        git_branch: row.get(8)?,
        git_remote: row.get(9)?,
        session_id: row.get(10)?,
    })
}

/// Résumé d'une session de travail, agrégé depuis les commandes partageant un
/// même `session_id`.
#[derive(Debug, Clone)]
pub struct SessionSummary {
    pub session_id: String,
    /// Nombre de commandes rattachées à la session.
    pub count: i64,
    /// Horodatage de la première commande (`YYYY-MM-DD HH:MM:SS`).
    pub started_at: String,
    /// Horodatage de la dernière commande.
    pub ended_at: String,
    /// Racine Git de la commande la plus récente de la session, si disponible.
    pub git_root: Option<String>,
}

/// Liste les sessions connues, de la plus récente activité à la plus ancienne.
///
/// Seules les commandes portant un `session_id` non vide sont prises en compte ;
/// les commandes importées ou enregistrées sans identifiant de session sont
/// ignorées (elles ne constituent pas une session). Grâce au comportement des
/// colonnes nues de SQLite avec `MAX(created_at)`, `git_root` provient de la
/// commande la plus récente de chaque session.
pub fn session_summaries(conn: &Connection, limit: Option<usize>) -> Result<Vec<SessionSummary>> {
    let limit_sql = match limit {
        Some(n) => format!("LIMIT {}", n as i64),
        None => String::new(),
    };
    let sql = format!(
        "SELECT session_id, COUNT(*) AS n,
                MIN(created_at) AS started, MAX(created_at) AS ended,
                git_root
         FROM commands
         WHERE session_id IS NOT NULL AND TRIM(session_id) <> ''
         GROUP BY session_id
         ORDER BY ended DESC, session_id DESC
         {limit_sql}"
    );
    let mut stmt = conn.prepare(&sql)?;
    let rows = stmt.query_map([], |row| {
        Ok(SessionSummary {
            session_id: row.get(0)?,
            count: row.get(1)?,
            started_at: row.get(2)?,
            ended_at: row.get(3)?,
            git_root: row.get(4)?,
        })
    })?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Charge les commandes d'une session, dans l'ordre chronologique croissant.
/// `limit` borne le nombre de commandes (`None` = toutes).
pub fn session_commands(
    conn: &Connection,
    session_id: &str,
    limit: Option<usize>,
) -> Result<Vec<CommandRecord>> {
    let limit_sql = match limit {
        Some(n) => format!("LIMIT {}", n as i64),
        None => String::new(),
    };
    let sql = format!(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands
         WHERE session_id = ?1
         ORDER BY created_at ASC, id ASC
         {limit_sql}"
    );
    let mut stmt = conn.prepare(&sql)?;
    let rows = stmt.query_map(rusqlite::params![session_id], row_to_record)?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Identifiant de la session la plus récemment active, le cas échéant.
pub fn latest_session_id(conn: &Connection) -> Result<Option<String>> {
    let mut stmt = conn.prepare(
        "SELECT session_id
         FROM commands
         WHERE session_id IS NOT NULL AND TRIM(session_id) <> ''
         ORDER BY created_at DESC, id DESC
         LIMIT 1",
    )?;
    let mut rows = stmt.query_map([], |row| row.get::<_, String>(0))?;
    match rows.next() {
        Some(r) => Ok(Some(r?)),
        None => Ok(None),
    }
}

/// Résumé agrégé de l'activité d'un projet (regroupement par `git_root`).
#[derive(Debug, Clone)]
pub struct ProjectSummary {
    /// Racine Git du projet.
    pub root: String,
    /// Nombre total de commandes enregistrées.
    pub command_count: i64,
    /// Nombre de sessions distinctes (commandes portant un `session_id`).
    pub session_count: i64,
    /// Horodatage de la première commande connue.
    pub first_activity: String,
    /// Horodatage de la dernière commande connue.
    pub last_activity: String,
    /// Branches Git rencontrées, triées et dédupliquées.
    pub branches: Vec<String>,
    /// Un remote Git connu du projet, si disponible.
    pub remote: Option<String>,
}

/// Découpe la concaténation `GROUP_CONCAT(DISTINCT git_branch)` en liste de
/// branches triée, sans doublon ni valeur vide.
fn split_branches(raw: Option<String>) -> Vec<String> {
    let mut branches: Vec<String> = raw
        .unwrap_or_default()
        .split(',')
        .map(|b| b.trim().to_string())
        .filter(|b| !b.is_empty())
        .collect();
    branches.sort();
    branches.dedup();
    branches
}

/// Liste les projets connus de l'historique, du plus récemment actif au plus
/// ancien. Seuls les `git_root` non nuls sont remontés. `limit` borne le nombre
/// de projets (`None` = tous).
pub fn project_summaries(conn: &Connection, limit: Option<usize>) -> Result<Vec<ProjectSummary>> {
    let limit_sql = match limit {
        Some(n) => format!("LIMIT {}", n as i64),
        None => String::new(),
    };
    let sql = format!(
        "SELECT git_root,
                COUNT(*) AS n,
                COUNT(DISTINCT CASE
                    WHEN session_id IS NOT NULL AND TRIM(session_id) <> ''
                    THEN session_id END) AS sessions,
                MIN(created_at) AS first_at,
                MAX(created_at) AS last_at,
                GROUP_CONCAT(DISTINCT git_branch) AS branches,
                MAX(git_remote) AS remote
         FROM commands
         WHERE git_root IS NOT NULL AND git_root <> ''
         GROUP BY git_root
         ORDER BY last_at DESC, git_root ASC
         {limit_sql}"
    );
    let mut stmt = conn.prepare(&sql)?;
    let rows = stmt.query_map([], project_summary_row)?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Résumé d'un projet identifié par sa racine Git exacte, ou `None` si la racine
/// est absente de l'historique.
pub fn project_summary(conn: &Connection, root: &str) -> Result<Option<ProjectSummary>> {
    let mut stmt = conn.prepare(
        "SELECT git_root,
                COUNT(*) AS n,
                COUNT(DISTINCT CASE
                    WHEN session_id IS NOT NULL AND TRIM(session_id) <> ''
                    THEN session_id END) AS sessions,
                MIN(created_at) AS first_at,
                MAX(created_at) AS last_at,
                GROUP_CONCAT(DISTINCT git_branch) AS branches,
                MAX(git_remote) AS remote
         FROM commands
         WHERE git_root = ?1
         GROUP BY git_root",
    )?;
    let mut rows = stmt.query_map(rusqlite::params![root], project_summary_row)?;
    match rows.next() {
        Some(r) => Ok(Some(r?)),
        None => Ok(None),
    }
}

/// Convertit une ligne agrégée en [`ProjectSummary`].
fn project_summary_row(row: &rusqlite::Row) -> rusqlite::Result<ProjectSummary> {
    Ok(ProjectSummary {
        root: row.get(0)?,
        command_count: row.get(1)?,
        session_count: row.get(2)?,
        first_activity: row.get(3)?,
        last_activity: row.get(4)?,
        branches: split_branches(row.get(5)?),
        remote: row.get(6)?,
    })
}

/// Liste les racines Git connues correspondant à `needle` : correspondance
/// exacte sur `git_root`, ou suffixe `%/needle` (nom court du projet). Sert à
/// résoudre l'argument de `mnemo project show|report`.
pub fn match_project_roots(conn: &Connection, needle: &str) -> Result<Vec<String>> {
    let suffix = format!("%/{needle}");
    let mut stmt = conn.prepare(
        "SELECT DISTINCT git_root
         FROM commands
         WHERE git_root IS NOT NULL AND git_root <> ''
           AND (git_root = ?1 OR git_root LIKE ?2)
         ORDER BY git_root ASC",
    )?;
    let rows = stmt.query_map(rusqlite::params![needle, suffix], |row| {
        row.get::<_, String>(0)
    })?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Charge les commandes d'un projet (racine Git exacte), de la plus récente à la
/// plus ancienne, en respectant un éventuel intervalle temporel et un filtre
/// d'échecs. `limit` borne le nombre de résultats (`None` = tous).
///
/// En lecture seule : aucune commande n'est exécutée, le contenu (y compris les
/// commandes déjà redactées) est restitué tel quel.
pub fn project_records(
    conn: &Connection,
    root: &str,
    since: Option<&str>,
    before: Option<&str>,
    failed_only: bool,
    limit: Option<usize>,
) -> Result<Vec<CommandRecord>> {
    let limit_sql = match limit {
        Some(n) => format!("LIMIT {}", n as i64),
        None => String::new(),
    };
    let sql = format!(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands
         WHERE git_root = ?1
           AND (?2 IS NULL OR created_at >= ?2)
           AND (?3 IS NULL OR created_at < ?3)
           AND (?4 = 0 OR (exit_code IS NOT NULL AND exit_code <> 0))
         ORDER BY created_at DESC, id DESC
         {limit_sql}"
    );
    let mut stmt = conn.prepare(&sql)?;
    let rows = stmt.query_map(
        rusqlite::params![root, since, before, failed_only as i64],
        row_to_record,
    )?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Nombre total de commandes stockées.
pub fn count(conn: &Connection) -> Result<i64> {
    let n = conn.query_row("SELECT COUNT(*) FROM commands", [], |row| row.get(0))?;
    Ok(n)
}

/// Récupère une commande par son identifiant, ou `None` si absente.
pub fn get_command(conn: &Connection, id: i64) -> Result<Option<CommandRecord>> {
    let mut stmt = conn.prepare(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands WHERE id = ?1",
    )?;
    let mut rows = stmt.query_map([id], row_to_record)?;
    match rows.next() {
        Some(r) => Ok(Some(r?)),
        None => Ok(None),
    }
}

/// Supprime une commande par identifiant, dans une transaction. Renvoie le
/// nombre de lignes effectivement supprimées (0 si l'ID n'existait pas).
pub fn delete_command(conn: &Connection, id: i64) -> Result<usize> {
    let tx = conn.unchecked_transaction()?;
    let n = tx.execute("DELETE FROM commands WHERE id = ?1", [id])?;
    tx.commit()?;
    Ok(n)
}

/// Applique des redactions de texte de commande en une seule transaction.
///
/// Seul le champ `command` est mis à jour ; tous les autres champs
/// (`created_at`, `cwd`, `exit_code`, contexte Git, `session_id`…) sont laissés
/// intacts. Requêtes paramétrées. Renvoie le nombre de lignes effectivement
/// modifiées. Atomique : en cas d'erreur, la transaction n'est pas validée.
pub fn apply_redactions(conn: &Connection, items: &[(i64, String)]) -> Result<usize> {
    let tx = conn.unchecked_transaction()?;
    let mut changed = 0usize;
    {
        let mut stmt = tx.prepare("UPDATE commands SET command = ?1 WHERE id = ?2")?;
        for (id, command) in items {
            changed += stmt.execute(rusqlite::params![command, id])?;
        }
    }
    tx.commit()?;
    Ok(changed)
}

/// Compte les commandes plus anciennes que `cutoff` (format `YYYY-MM-DD
/// HH:MM:SS`), en respectant le filtre de contexte Git.
pub fn count_older_than(conn: &Connection, cutoff: &str, filter: &SearchFilter) -> Result<i64> {
    let project_suffix = filter.project.as_ref().map(|p| format!("%/{p}"));
    let n = conn.query_row(
        "SELECT COUNT(*) FROM commands
         WHERE created_at < ?1
           AND (?2 IS NULL OR git_branch = ?2)
           AND (?3 IS NULL OR git_root = ?3 OR git_root LIKE ?4)",
        rusqlite::params![cutoff, filter.branch, filter.project, project_suffix],
        |row| row.get(0),
    )?;
    Ok(n)
}

/// Charge un échantillon de commandes plus anciennes que `cutoff` (les plus
/// récentes d'abord), pour prévisualiser un `prune`.
pub fn fetch_older_than(
    conn: &Connection,
    cutoff: &str,
    filter: &SearchFilter,
    limit: usize,
) -> Result<Vec<CommandRecord>> {
    let project_suffix = filter.project.as_ref().map(|p| format!("%/{p}"));
    let mut stmt = conn.prepare(
        "SELECT id, command, cwd, shell, hostname, exit_code, created_at,
                git_root, git_branch, git_remote, session_id
         FROM commands
         WHERE created_at < ?1
           AND (?2 IS NULL OR git_branch = ?2)
           AND (?3 IS NULL OR git_root = ?3 OR git_root LIKE ?4)
         ORDER BY created_at DESC, id DESC
         LIMIT ?5",
    )?;
    let rows = stmt.query_map(
        rusqlite::params![
            cutoff,
            filter.branch,
            filter.project,
            project_suffix,
            limit as i64
        ],
        row_to_record,
    )?;
    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

/// Supprime les commandes plus anciennes que `cutoff` (en respectant le filtre
/// Git), dans une transaction. Renvoie le nombre de lignes supprimées.
pub fn delete_older_than(conn: &Connection, cutoff: &str, filter: &SearchFilter) -> Result<usize> {
    let project_suffix = filter.project.as_ref().map(|p| format!("%/{p}"));
    let tx = conn.unchecked_transaction()?;
    let n = tx.execute(
        "DELETE FROM commands
         WHERE created_at < ?1
           AND (?2 IS NULL OR git_branch = ?2)
           AND (?3 IS NULL OR git_root = ?3 OR git_root LIKE ?4)",
        rusqlite::params![cutoff, filter.branch, filter.project, project_suffix],
    )?;
    tx.commit()?;
    Ok(n)
}

/// Horodatage courant au format `YYYY-MM-DD HH:MM:SS` (UTC).
pub fn now_timestamp() -> String {
    let secs = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|d| d.as_secs())
        .unwrap_or(0);
    format_timestamp(secs)
}

/// Convertit un timestamp Unix (secondes UTC) en `YYYY-MM-DD HH:MM:SS`.
pub fn format_timestamp(secs: u64) -> String {
    let days = (secs / 86_400) as i64;
    let rem = secs % 86_400;
    let hour = rem / 3600;
    let min = (rem % 3600) / 60;
    let sec = rem % 60;
    let (y, m, d) = civil_from_days(days);
    format!("{y:04}-{m:02}-{d:02} {hour:02}:{min:02}:{sec:02}")
}

/// Algorithme de Howard Hinnant : jours depuis l'époque -> (année, mois, jour).
fn civil_from_days(z: i64) -> (i64, u32, u32) {
    let z = z + 719_468;
    let era = if z >= 0 { z } else { z - 146_096 } / 146_097;
    let doe = (z - era * 146_097) as u64; // [0, 146096]
    let yoe = (doe - doe / 1460 + doe / 36524 - doe / 146_096) / 365; // [0, 399]
    let y = yoe as i64 + era * 400;
    let doy = doe - (365 * yoe + yoe / 4 - yoe / 100); // [0, 365]
    let mp = (5 * doy + 2) / 153; // [0, 11]
    let d = (doy - (153 * mp + 2) / 5 + 1) as u32; // [1, 31]
    let m = if mp < 10 { mp + 3 } else { mp - 9 } as u32; // [1, 12]
    let y = if m <= 2 { y + 1 } else { y };
    (y, m, d)
}

/// Valide une date au format `AAAA-MM-JJ` (sans heure).
pub fn is_valid_date(s: &str) -> bool {
    let b = s.as_bytes();
    if b.len() != 10 {
        return false;
    }
    for (i, c) in b.iter().enumerate() {
        let ok = match i {
            4 | 7 => *c == b'-',
            _ => c.is_ascii_digit(),
        };
        if !ok {
            return false;
        }
    }
    let month: u32 = s[5..7].parse().unwrap_or(0);
    let day: u32 = s[8..10].parse().unwrap_or(0);
    (1..=12).contains(&month) && (1..=31).contains(&day)
}

/// Secondes Unix courantes (UTC).
fn now_secs() -> u64 {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|d| d.as_secs())
        .unwrap_or(0)
}

/// Résout une borne inférieure `--since` : durée (`24h`, `7d`, `2w`, `3m`, `1y`)
/// ou date `AAAA-MM-JJ`. Renvoie `None` si la spec est invalide (**jamais de
/// panique**) : l'appelant peut alors ignorer le filtre proprement.
pub fn resolve_since(spec: &str) -> Option<String> {
    let spec = spec.trim();
    if let Ok(secs) = crate::prune::parse_duration(spec) {
        return Some(format_timestamp(now_secs().saturating_sub(secs)));
    }
    if is_valid_date(spec) {
        return Some(format!("{spec} 00:00:00"));
    }
    None
}

/// Résout une borne supérieure `--before` : date `AAAA-MM-JJ` (exclue) ou durée.
/// Renvoie `None` si la spec est invalide (pas de panique).
pub fn resolve_before(spec: &str) -> Option<String> {
    let spec = spec.trim();
    if is_valid_date(spec) {
        // `created_at < "AAAA-MM-JJ"` exclut tout ce jour : « avant cette date ».
        return Some(spec.to_string());
    }
    if let Ok(secs) = crate::prune::parse_duration(spec) {
        return Some(format_timestamp(now_secs().saturating_sub(secs)));
    }
    None
}

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

    #[test]
    fn hash_stable_et_distingue_le_cwd() {
        let a = compute_hash("ls -la", Some("/home"));
        let b = compute_hash("ls -la", Some("/home"));
        let c = compute_hash("ls -la", Some("/tmp"));
        assert_eq!(a, b);
        assert_ne!(a, c);
    }

    #[test]
    fn insertion_et_dedoublonnage() {
        let conn = open_in_memory().unwrap();
        let cmd = NewCommand {
            command: "echo hi".into(),
            cwd: Some("/tmp".into()),
            shell: Some("bash".into()),
            hostname: Some("host".into()),
            exit_code: Some(0),
            created_at: now_timestamp(),
            ..Default::default()
        };
        assert!(insert_command(&conn, &cmd).unwrap());
        // Même hash -> doublon ignoré.
        assert!(!insert_command(&conn, &cmd).unwrap());
        assert_eq!(count(&conn).unwrap(), 1);
    }

    #[test]
    fn fetch_renvoie_les_commandes() {
        let conn = open_in_memory().unwrap();
        for c in ["a", "b", "c"] {
            insert_command(
                &conn,
                &NewCommand {
                    command: c.into(),
                    cwd: None,
                    shell: None,
                    hostname: None,
                    exit_code: None,
                    created_at: now_timestamp(),
                    ..Default::default()
                },
            )
            .unwrap();
        }
        let all = fetch_all(&conn, 100).unwrap();
        assert_eq!(all.len(), 3);
    }

    #[test]
    fn format_timestamp_connu() {
        // 1609459200 = 2021-01-01 00:00:00 UTC
        assert_eq!(format_timestamp(1_609_459_200), "2021-01-01 00:00:00");
        // 0 = 1970-01-01 00:00:00 UTC
        assert_eq!(format_timestamp(0), "1970-01-01 00:00:00");
    }

    #[test]
    fn fetch_filtered_par_projet_et_branche() {
        let conn = open_in_memory().unwrap();
        let insert = |command: &str, root: &str, branch: &str| {
            insert_command(
                &conn,
                &NewCommand {
                    command: command.into(),
                    cwd: Some(root.into()),
                    created_at: now_timestamp(),
                    git_root: Some(root.into()),
                    git_branch: Some(branch.into()),
                    ..Default::default()
                },
            )
            .unwrap();
        };
        insert("cargo build", "/home/u/proj/mnemo", "main");
        insert("cargo test", "/home/u/proj/mnemo", "dev");
        insert("ls", "/home/u/proj/autre", "main");

        // Filtre par nom de projet (dernier segment du chemin).
        let by_name = fetch_filtered(
            &conn,
            &SearchFilter {
                project: Some("mnemo".into()),
                branch: None,
            },
            100,
        )
        .unwrap();
        assert_eq!(by_name.len(), 2);
        assert!(by_name
            .iter()
            .all(|r| r.git_root.as_deref() == Some("/home/u/proj/mnemo")));

        // Filtre par chemin git_root complet.
        let by_path = fetch_filtered(
            &conn,
            &SearchFilter {
                project: Some("/home/u/proj/autre".into()),
                branch: None,
            },
            100,
        )
        .unwrap();
        assert_eq!(by_path.len(), 1);

        // Filtre par branche.
        let by_branch = fetch_filtered(
            &conn,
            &SearchFilter {
                project: None,
                branch: Some("main".into()),
            },
            100,
        )
        .unwrap();
        assert_eq!(by_branch.len(), 2);

        // Combinaison projet + branche.
        let both = fetch_filtered(
            &conn,
            &SearchFilter {
                project: Some("mnemo".into()),
                branch: Some("dev".into()),
            },
            100,
        )
        .unwrap();
        assert_eq!(both.len(), 1);
        assert_eq!(both[0].command, "cargo test");
    }

    /// Insère une commande en forçant `created_at` (pour tester les bornes).
    fn insert_at(conn: &Connection, command: &str, shell: &str, exit: Option<i64>, when: &str) {
        insert_command(
            conn,
            &NewCommand {
                command: command.into(),
                cwd: Some("/tmp".into()),
                shell: Some(shell.into()),
                hostname: Some("host".into()),
                exit_code: exit,
                created_at: when.into(),
                git_root: None,
                git_branch: None,
                git_remote: None,
                session_id: None,
            },
        )
        .unwrap();
    }

    #[test]
    fn query_filter_combine_les_criteres() {
        let conn = open_in_memory().unwrap();
        insert_at(&conn, "ok-bash", "bash", Some(0), "2026-01-01 10:00:00");
        insert_at(&conn, "ko-bash", "bash", Some(1), "2026-03-01 10:00:00");
        insert_at(&conn, "ok-zsh", "zsh", Some(0), "2026-06-01 10:00:00");

        // Échecs uniquement.
        let failed = fetch_query(
            &conn,
            &QueryFilter {
                failed: true,
                ..Default::default()
            },
            None,
        )
        .unwrap();
        assert_eq!(failed.len(), 1);
        assert_eq!(failed[0].command, "ko-bash");

        // Code de sortie exact.
        let ok = fetch_query(
            &conn,
            &QueryFilter {
                exit_code: Some(0),
                ..Default::default()
            },
            None,
        )
        .unwrap();
        assert_eq!(ok.len(), 2);

        // Shell + borne temporelle (avant le 2026-04-01).
        let bash_before = fetch_query(
            &conn,
            &QueryFilter {
                shell: Some("bash".into()),
                before: Some("2026-04-01".into()),
                ..Default::default()
            },
            None,
        )
        .unwrap();
        assert_eq!(bash_before.len(), 2);

        // Borne since incluse.
        let since = fetch_query(
            &conn,
            &QueryFilter {
                since: Some("2026-03-01 00:00:00".into()),
                ..Default::default()
            },
            None,
        )
        .unwrap();
        assert_eq!(since.len(), 2);

        // Limite respectée.
        let limited = fetch_query(&conn, &QueryFilter::default(), Some(1)).unwrap();
        assert_eq!(limited.len(), 1);
    }
}