libnoa 0.3.2

use anyhow::Context;
use std::{
    path::{Path, PathBuf},
    sync::Arc,
};

use redb::Database;

use crate::{
    config::RepoConfig,
    error::{NoaError, Result},
    log::FileAgentLog,
    object::RedbObjectStore,
    refs::RedbRefStore,
    snapshot::RedbSnapshotStore,
    workspace::WorkspaceManager,
};

/// Name of the `.noa` directory when standalone.
pub const NOA_DIR_NAME: &str = ".noa";
/// Sub-path inside `.git/` for parasitic mode.
pub const NOA_PARASITIC_PATH: &str = ".git/noa";
/// Filename of the embedded redb database inside `.noa/`.
pub const DB_NAME: &str = "noa.redb";
/// Subdirectory holding per-workspace JSONL agent logs.
pub const AGENT_LOGS_DIR: &str = "agent-logs";
/// Filename of the HEAD pointer (contains the active workspace name).
pub const HEAD_FILE: &str = "HEAD";
/// Filename of the ORIG_HEAD pointer (saved before merge operations).
pub const ORIG_HEAD_FILE: &str = "ORIG_HEAD";

/// Handle to an opened noa repository (the `.noa/` directory alongside `.git`).
///
/// Provides access to the object store, snapshot store, ref store, workspace
/// manager, and agent logs for a single repository.
pub struct Repository {
    pub root: PathBuf,
    pub noa_dir: PathBuf,
    pub db: Arc<Database>,
    pub config: RepoConfig,
}

impl Repository {
    /// Resolves the noa directory path for a given workspace root.
    ///
    /// In parasitic mode (`.git/` exists), returns `<root>/.git/noa`.
    /// Otherwise returns `<root>/.noa`.
    pub fn resolve_noa_dir(root: &Path) -> PathBuf {
        if root.join(".git").is_dir() {
            root.join(NOA_PARASITIC_PATH)
        } else {
            root.join(NOA_DIR_NAME)
        }
    }

    /// Returns true if the noa directory is parasitic (inside `.git/`).
    #[must_use]
    pub fn is_parasitic(&self) -> bool {
        self.noa_dir.starts_with(self.root.join(".git"))
    }
    /// Initializes a new `.noa/` repository at the given path.
    ///
    /// # Errors
    /// Returns an error if `.noa/` already exists or if filesystem operations fail.
    pub fn init(path: &Path) -> Result<Self> {
        Self::init_with_noa_remote(path, None)
    }

    /// Initializes with an optional noa remote URL (also configures `.gitattributes`).
    pub fn init_with_noa_remote(path: &Path, noa_remote: Option<&str>) -> Result<Self> {
        let config = RepoConfig {
            noa_remote: noa_remote.map(std::string::ToString::to_string),
            ..RepoConfig::default()
        };
        Self::init_inner(path, config, |path, cfg| {
            if let Some(url) = &cfg.noa_remote {
                manage_gitattributes(path, url)?;
            }
            Ok(())
        })
    }

    pub fn init_with_remotes(
        path: &Path,
        remotes: Vec<crate::config::RemoteConfig>,
    ) -> Result<Self> {
        let config = RepoConfig {
            name: "default".to_string(),
            remotes,
            noa_remote: None,
            sync: None,
            storage: Vec::new(),
        };
        Self::init_inner(path, config, |_path, _cfg| Ok(()))
    }

    fn init_inner<F>(path: &Path, config: RepoConfig, extra: F) -> Result<Self>
    where
        F: FnOnce(&Path, &RepoConfig) -> Result<()>,
    {
        let noa_dir = Self::resolve_noa_dir(path);

        if noa_dir.exists() {
            return Err(NoaError::RepositoryAlreadyExists {
                path: noa_dir.display().to_string(),
            }
            .into());
        }

        std::fs::create_dir_all(&noa_dir)?;
        std::fs::create_dir_all(noa_dir.join(AGENT_LOGS_DIR))?;

        config.save_to_dir(&noa_dir)?;

        std::fs::write(noa_dir.join(HEAD_FILE), "default\n")?;

        let db = Self::open_db(&noa_dir)?;
        Self::init_tables(&db)?;

        let db = Arc::new(db);
        Self::create_default_workspace(&db)?;

        let repo = Repository {
            root: path.to_path_buf(),
            noa_dir,
            db,
            config,
        };

        if !repo.is_parasitic() {
            manage_gitignore(path)?;
        }

        extra(path, &repo.config)?;

        Ok(repo)
    }

    /// Opens an existing `.noa/` repository at the given path.
    ///
    /// # Errors
    /// Returns an error if `.noa/` does not exist or is invalid.
    pub fn open(path: &Path) -> Result<Self> {
        let noa_dir = Self::resolve_noa_dir(path);

        if !noa_dir.exists() {
            return Err(NoaError::RepositoryNotFound {
                path: noa_dir.display().to_string(),
            }
            .into());
        }

        Self::validate(&noa_dir)?;

        let config = RepoConfig::load_from_dir(&noa_dir)?;
        let db = Self::open_db(&noa_dir)?;
        Self::init_tables(&db)?;

        Ok(Repository {
            root: path.to_path_buf(),
            noa_dir,
            db: Arc::new(db),
            config,
        })
    }

    /// Walks up from `from` to find the nearest `.noa/` repository root.
    ///
    /// # Errors
    /// Returns an error if no `.noa/` is found in any ancestor directory.
    pub fn find(from: &Path) -> Result<PathBuf> {
        let mut current = from.to_path_buf();
        loop {
            let standalone = current.join(NOA_DIR_NAME);
            let parasitic = current.join(NOA_PARASITIC_PATH);
            if standalone.exists() || parasitic.exists() {
                return Ok(current);
            }
            match current.parent() {
                Some(parent) => current = parent.to_path_buf(),
                None => {
                    return Err(NoaError::RepositoryNotFound {
                        path: from.display().to_string(),
                    }
                    .into())
                }
            }
        }
    }

    #[must_use]
    pub fn exists(path: &Path) -> bool {
        path.join(NOA_DIR_NAME).exists() || path.join(NOA_PARASITIC_PATH).exists()
    }

    fn validate(noa_dir: &Path) -> Result<()> {
        if !noa_dir.join(DB_NAME).exists() {
            anyhow::bail!("invalid repository: missing noa.redb");
        }
        if !noa_dir.join(AGENT_LOGS_DIR).exists() {
            anyhow::bail!("invalid repository: missing agent-logs/ directory");
        }
        if !noa_dir.join("config").exists() {
            anyhow::bail!("invalid repository: missing config file");
        }
        Ok(())
    }

    fn open_db(noa_dir: &Path) -> Result<Database> {
        let db_path = noa_dir.join(DB_NAME);
        let db = Database::builder().create(&db_path)?;
        Ok(db)
    }

    fn init_tables(db: &Database) -> Result<()> {
        let write_txn = db.begin_write()?;

        // Ensure all required tables exist (redb creates them lazily on open_table)
        write_txn.open_table::<&[u8], &[u8]>(redb::TableDefinition::new("blobs"))?;
        write_txn.open_table::<&[u8], &[u8]>(redb::TableDefinition::new("trees"))?;
        write_txn.open_table::<&str, &[u8]>(redb::TableDefinition::new("snapshots"))?;
        write_txn.open_table::<&str, &[u8]>(redb::TableDefinition::new("workspaces"))?;
        write_txn.open_table::<&str, &[u8]>(redb::TableDefinition::new("refs"))?;

        write_txn.commit()?;
        Ok(())
    }

    fn create_default_workspace(db: &Arc<Database>) -> Result<()> {
        let ws = crate::workspace::Workspace {
            name: "default".to_string(),
            head: crate::snapshot::empty_snapshot_id(),
            base: crate::snapshot::empty_snapshot_id(),
            agent_id: None,
            last_seq: 0,
            created_at: 0,
            updated_at: 0,
        };
        let data = rmp_serde::to_vec(&ws)?;
        let txn = db.begin_write()?;
        {
            let mut table =
                txn.open_table(redb::TableDefinition::<&str, &[u8]>::new("workspaces"))?;
            table.insert("default", data.as_slice())?;
        }
        txn.commit()?;
        Ok(())
    }

    pub fn read_head(&self) -> Result<String> {
        let head_path = self.noa_dir.join(HEAD_FILE);
        let content = std::fs::read_to_string(&head_path)?;
        Ok(content.trim().to_string())
    }

    pub fn write_head(&self, name: &str) -> Result<()> {
        let head_path = self.noa_dir.join(HEAD_FILE);
        std::fs::write(&head_path, format!("{name}\n"))?;
        Ok(())
    }

    pub fn read_orig_head(&self) -> Result<Option<String>> {
        let path = self.noa_dir.join(ORIG_HEAD_FILE);
        if path.exists() {
            let content = std::fs::read_to_string(&path)?;
            Ok(Some(content.trim().to_string()))
        } else {
            Ok(None)
        }
    }

    pub fn write_orig_head(&self, name: &str) -> Result<()> {
        let path = self.noa_dir.join(ORIG_HEAD_FILE);
        std::fs::write(&path, format!("{name}\n"))?;
        Ok(())
    }

    #[must_use]
    pub fn agent_logs_dir(&self) -> PathBuf {
        self.noa_dir.join(AGENT_LOGS_DIR)
    }

    #[must_use]
    pub fn agent_log_path(&self, workspace: &str) -> PathBuf {
        self.agent_logs_dir().join(format!("{workspace}.log"))
    }

    pub fn save_config(&mut self) -> Result<()> {
        self.config.save_to_dir(&self.noa_dir)
    }

    pub fn object_store(&self) -> Result<RedbObjectStore> {
        RedbObjectStore::new(Arc::clone(&self.db))
    }

    pub fn snapshot_store(&self) -> Result<RedbSnapshotStore> {
        RedbSnapshotStore::new(Arc::clone(&self.db))
    }

    pub fn ref_store(&self) -> Result<RedbRefStore> {
        RedbRefStore::new(Arc::clone(&self.db))
    }

    pub fn workspace_manager(&self) -> Result<WorkspaceManager> {
        WorkspaceManager::new(Arc::clone(&self.db))
    }

    pub fn agent_log(&self, workspace: &str) -> Result<FileAgentLog> {
        let path = self.agent_log_path(workspace);
        FileAgentLog::create(&path)
    }

    pub fn init_for_sync(path: &Path) -> Result<SyncInitResult> {
        let noa_dir = Self::resolve_noa_dir(path);

        let noa_initialized = if !noa_dir.exists() {
            Self::init_inner(path, RepoConfig::default(), |_p, _c| Ok(()))?;
            true
        } else {
            false
        };

        let current_branch = get_current_git_branch(path)?;

        Ok(SyncInitResult {
            repo_id: format!("{}:{}", "sync", path.display()),
            current_branch,
            noa_initialized,
            gitignore_updated: false,
        })
    }
}

pub struct SyncInitResult {
    pub repo_id: String,
    pub current_branch: String,
    pub noa_initialized: bool,
    pub gitignore_updated: bool,
}

pub fn get_current_git_branch(workspace_root: &Path) -> Result<String> {
    let output = std::process::Command::new("git")
        .args(["rev-parse", "--abbrev-ref", "HEAD"])
        .current_dir(workspace_root)
        .output()
        .with_context(|| "git rev-parse failed")?;

    if !output.status.success() {
        let stderr = String::from_utf8_lossy(&output.stderr);
        anyhow::bail!("failed to determine current git branch: {}", stderr.trim());
    }

    Ok(String::from_utf8_lossy(&output.stdout).trim().to_string())
}

pub fn manage_gitignore(root: &Path) -> Result<()> {
    let gitignore_path = root.join(".gitignore");

    if !gitignore_path.exists() {
        std::fs::write(
            &gitignore_path,
            "# Added by noa \u{2014} keep agent iteration data out of git\n.noa/\n",
        )?;
        return Ok(());
    }

    let content = std::fs::read_to_string(&gitignore_path)?;

    for line in content.lines() {
        if line.trim() == ".noa" || line.trim() == ".noa/" {
            return Ok(());
        }
    }

    let trimmed = content.trim_end_matches('\n');
    std::fs::write(&gitignore_path, format!("{trimmed}\n.noa/\n"))?;

    Ok(())
}

pub fn manage_gitattributes(root: &Path, noa_remote_url: &str) -> Result<()> {
    let noa_dir = Repository::resolve_noa_dir(root);
    let noa_rel = noa_dir
        .strip_prefix(root)
        .unwrap_or_else(|_| Path::new(".noa"));
    let gitattributes_path = root.join(".gitattributes");
    let attr_line = format!("{}/**   noa-remote={}", noa_rel.display(), noa_remote_url);

    if !gitattributes_path.exists() {
        let content = format!(
            "# Added by noa \u{2014} specifies where agent iteration data is hosted\n{attr_line}\n"
        );
        std::fs::write(&gitattributes_path, content)?;
        return Ok(());
    }

    let content = std::fs::read_to_string(&gitattributes_path)?;

    for line in content.lines() {
        if line.contains("noa-remote=") {
            return Ok(());
        }
    }

    std::fs::write(
        &gitattributes_path,
        format!(
            "{}\n# Added by noa \u{2014} specifies where agent iteration data is hosted\n{}\n",
            content.trim_end(),
            attr_line
        ),
    )?;

    Ok(())
}

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

    #[test]
    fn test_init_creates_structure() {
        let tmp = TempDir::new().unwrap();
        let repo = Repository::init(tmp.path()).unwrap();

        assert!(repo.noa_dir.exists());
        assert!(repo.noa_dir.join(DB_NAME).exists());
        assert!(repo.noa_dir.join(AGENT_LOGS_DIR).exists());
        assert!(repo.noa_dir.join(HEAD_FILE).exists());
        assert!(repo.noa_dir.join("config").exists());
        assert_eq!(repo.read_head().unwrap(), "default");
    }

    #[test]
    fn test_init_fails_if_exists() {
        let tmp = TempDir::new().unwrap();
        Repository::init(tmp.path()).unwrap();
        let result = Repository::init(tmp.path());
        assert!(result.is_err());
    }

    #[test]
    fn test_open_existing() {
        let tmp = TempDir::new().unwrap();
        Repository::init(tmp.path()).unwrap();
        let repo = Repository::open(tmp.path()).unwrap();
        assert_eq!(repo.read_head().unwrap(), "default");
    }

    #[test]
    fn test_open_fails_if_missing() {
        let tmp = TempDir::new().unwrap();
        let result = Repository::open(tmp.path());
        assert!(result.is_err());
    }

    #[test]
    fn test_find_repo() {
        let tmp = TempDir::new().unwrap();
        Repository::init(tmp.path()).unwrap();

        let subdir = tmp.path().join("src").join("deep");
        std::fs::create_dir_all(&subdir).unwrap();

        let found = Repository::find(&subdir).unwrap();
        assert_eq!(found, tmp.path());
    }

    #[test]
    fn test_exists() {
        let tmp = TempDir::new().unwrap();
        assert!(!Repository::exists(tmp.path()));
        Repository::init(tmp.path()).unwrap();
        assert!(Repository::exists(tmp.path()));
    }

    #[tokio::test]
    async fn test_stores_accessible() {
        use crate::object::ObjectStore;
        use crate::refs::RefStore;
        use crate::snapshot::SnapshotStore;

        let tmp = TempDir::new().unwrap();
        let repo = Repository::init(tmp.path()).unwrap();

        // The previous test only checked `.is_ok()` on each constructor,
        // which would still pass if a constructor opened a database handle
        // to the wrong table or a stale file descriptor. We now drive at
        // least one trivial read method on each store to prove the handle
        // is actually wired to the right schema.

        let object_store = repo.object_store().expect("object_store must construct");
        let snapshot_store = repo
            .snapshot_store()
            .expect("snapshot_store must construct");
        let ref_store = repo.ref_store().expect("ref_store must construct");
        let workspace_mgr = repo
            .workspace_manager()
            .expect("workspace_manager must construct");
        let agent_log = repo.agent_log("default").expect("agent_log must construct");

        // ObjectStore: a fresh repo must NOT have a made-up blob id.
        let phantom_blob_id = crate::object::BlobId("nonexistent-blob-id".to_string());
        let has_blob = object_store
            .has_blob(&phantom_blob_id)
            .await
            .expect("has_blob must succeed (returning false) on a fresh repo");
        assert!(!has_blob, "fresh repo must not have a phantom blob id");

        // SnapshotStore: a fresh repo must have zero snapshots.
        let all_snaps = snapshot_store
            .list_all()
            .await
            .expect("list_all must succeed");
        assert!(
            all_snaps.is_empty(),
            "fresh repo must have zero snapshots, got {}",
            all_snaps.len()
        );

        // RefStore: a fresh repo must have zero refs.
        let refs = ref_store.list().await.expect("ref list must succeed");
        assert!(
            refs.is_empty(),
            "fresh repo must have zero refs, got {}",
            refs.len()
        );

        // WorkspaceManager: Repository::init creates a `default` workspace,
        // so a fresh repo must have exactly 1 workspace named "default".
        let workspaces = workspace_mgr
            .list()
            .await
            .expect("workspace list must succeed");
        assert_eq!(
            workspaces.len(),
            1,
            "fresh repo must have exactly the 'default' workspace, got {:?}",
            workspaces.iter().map(|w| &w.name).collect::<Vec<_>>()
        );
        assert_eq!(
            workspaces[0].name, "default",
            "the seeded workspace must be named 'default'"
        );

        // AgentLog: must expose a sequence counter that starts at 0 (or
        // whatever the engine's documented initial value is) — the call
        // must not panic and must return a deterministic value.
        let _seq = crate::log::AgentLog::next_seq(&agent_log)
            .await
            .expect("next_seq must succeed on a fresh agent log");
    }

    #[test]
    fn test_init_creates_gitignore() {
        let tmp = TempDir::new().unwrap();
        Repository::init(tmp.path()).unwrap();
        let gitignore = tmp.path().join(".gitignore");
        assert!(gitignore.exists());
        let content = std::fs::read_to_string(&gitignore).unwrap();
        assert!(content.contains(".noa/"));
    }

    #[test]
    fn test_init_appends_to_existing_gitignore() {
        let tmp = TempDir::new().unwrap();
        std::fs::write(tmp.path().join(".gitignore"), "*.log\n").unwrap();
        Repository::init(tmp.path()).unwrap();
        let content = std::fs::read_to_string(tmp.path().join(".gitignore")).unwrap();
        assert!(content.contains("*.log"));
        assert!(content.contains(".noa/"));
    }

    #[test]
    fn test_init_does_not_duplicate_gitignore_entry() {
        let tmp = TempDir::new().unwrap();
        std::fs::write(tmp.path().join(".gitignore"), ".noa/\n").unwrap();
        let before = std::fs::read_to_string(tmp.path().join(".gitignore")).unwrap();
        Repository::init(tmp.path()).unwrap();
        let after = std::fs::read_to_string(tmp.path().join(".gitignore")).unwrap();
        assert_eq!(before, after);
    }

    #[test]
    fn test_init_with_noa_remote_creates_gitattributes() {
        let tmp = TempDir::new().unwrap();
        let repo =
            Repository::init_with_noa_remote(tmp.path(), Some("https://noa.example.com/repo"))
                .unwrap();

        let gitattributes = tmp.path().join(".gitattributes");
        assert!(gitattributes.exists());
        let content = std::fs::read_to_string(&gitattributes).unwrap();
        assert!(content.contains("noa-remote=https://noa.example.com/repo"));

        assert_eq!(
            repo.config.noa_remote,
            Some("https://noa.example.com/repo".to_string())
        );
    }

    #[test]
    fn test_init_without_noa_remote_no_gitattributes() {
        let tmp = TempDir::new().unwrap();
        Repository::init(tmp.path()).unwrap();
        assert!(!tmp.path().join(".gitattributes").exists());
    }

    #[test]
    fn test_init_with_noa_remote_appends_to_existing_gitattributes() {
        let tmp = TempDir::new().unwrap();
        std::fs::write(tmp.path().join(".gitattributes"), "*.bin binary\n").unwrap();
        Repository::init_with_noa_remote(tmp.path(), Some("https://noa.example.com/repo")).unwrap();

        let content = std::fs::read_to_string(tmp.path().join(".gitattributes")).unwrap();
        assert!(content.contains("*.bin binary"));
        assert!(content.contains("noa-remote="));
    }

    #[test]
    fn test_parasitic_mode_functional() {
        let tmp = TempDir::new().unwrap();
        let output = std::process::Command::new("git")
            .args(["init"])
            .current_dir(tmp.path())
            .output()
            .expect("git init failed");
        assert!(output.status.success(), "git init must succeed");
        assert!(tmp.path().join(".git").is_dir(), ".git/ must exist");

        {
            let repo = Repository::init(tmp.path()).unwrap();
            let expected = tmp.path().join(".git").join("noa");
            assert_eq!(
                repo.noa_dir, expected,
                "parasitic noa dir must be .git/noa/"
            );
            assert!(repo.is_parasitic(), "is_parasitic() must return true");
            assert!(
                tmp.path().join(".git/noa").exists(),
                ".git/noa/ must exist on disk"
            );
        }
        assert!(Repository::exists(tmp.path()), "exists() must return true");

        let opened = Repository::open(tmp.path());
        assert!(
            opened.is_ok(),
            "open() must succeed in parasitic mode, got: {:?}",
            opened.err()
        );
        assert!(opened.unwrap().is_parasitic());

        let subdir = tmp.path().join("src").join("deep");
        std::fs::create_dir_all(&subdir).unwrap();
        let found = Repository::find(&subdir);
        assert!(found.is_ok(), "find() from subdir must succeed");
        assert_eq!(found.unwrap(), tmp.path());

        let gitignore_path = tmp.path().join(".gitignore");
        if gitignore_path.exists() {
            let content = std::fs::read_to_string(&gitignore_path).unwrap();
            assert!(
                !content.contains(".noa/"),
                "parasitic mode must NOT add .noa/ to .gitignore, got: {content}"
            );
        }
    }

    #[test]
    fn test_standalone_mode_functional() {
        let tmp = TempDir::new().unwrap();
        assert!(
            !tmp.path().join(".git").exists(),
            "must start without .git/"
        );

        let repo = Repository::init(tmp.path()).unwrap();

        let expected = tmp.path().join(".noa");
        assert_eq!(repo.noa_dir, expected, "standalone noa dir must be .noa/");
        assert!(!repo.is_parasitic(), "is_parasitic() must return false");
        assert!(tmp.path().join(".noa").exists(), ".noa/ must exist on disk");

        let gitignore_path = tmp.path().join(".gitignore");
        assert!(gitignore_path.exists(), ".gitignore must be created");
        let content = std::fs::read_to_string(&gitignore_path).unwrap();
        assert!(
            content.contains(".noa/"),
            "standalone mode must add .noa/ to .gitignore"
        );
    }

    #[test]
    fn test_backward_compat_existing_standalone() {
        let tmp = TempDir::new().unwrap();
        {
            Repository::init(tmp.path()).unwrap();
        }
        let original_dir = tmp.path().join(".noa");
        assert!(original_dir.exists(), "pre-existing .noa/ must exist");

        let opened = Repository::open(tmp.path());
        assert!(
            opened.is_ok(),
            "open() must succeed on existing standalone repo"
        );
        let opened = opened.unwrap();
        assert!(!opened.is_parasitic());

        assert!(Repository::exists(tmp.path()), "exists() must return true");

        let subdir = tmp.path().join("sub");
        std::fs::create_dir_all(&subdir).unwrap();
        let found = Repository::find(&subdir);
        assert!(found.is_ok(), "find() must find existing standalone repo");
        assert_eq!(found.unwrap(), tmp.path());
    }
}