lemma 0.8.18

#[cfg(not(target_arch = "wasm32"))]
use std::path::Path;
use std::sync::Arc;
#[cfg(not(target_arch = "wasm32"))]
use std::time::Duration;

#[cfg(not(target_arch = "wasm32"))]
use tokio::sync::Notify;
use tokio::sync::RwLock;
use tower_lsp::jsonrpc::Result;
use tower_lsp::lsp_types::*;
use tower_lsp::{Client, LanguageServer};

use super::diagnostics;
use super::registry::Registry;
use super::semantic_tokens;
use super::workspace::WorkspaceModel;
#[cfg(not(target_arch = "wasm32"))]
use lemma::Context;
use lemma::{DataValue, SpecRef};

async fn publish_workspace_diagnostics(client: &Client, workspace: &WorkspaceModel) {
    let file_diagnostics = workspace.validate_workspace();
    for file_diag in file_diagnostics {
        let lsp_diagnostics = diagnostics::errors_to_diagnostics(
            &file_diag.errors,
            &file_diag.text,
            &file_diag.attribute,
        );
        client
            .publish_diagnostics(file_diag.url, lsp_diagnostics, None)
            .await;
    }
}

/// Shared mutable state accessed by both the LSP handlers and the debounce background task.
struct SharedState {
    workspace: RwLock<WorkspaceModel>,
    #[cfg(not(target_arch = "wasm32"))]
    debounce_notify: Notify,
    /// The workspace root URI, set during `initialize`.
    root_uri: RwLock<Option<Url>>,
}

/// The Lemma Language Server.
///
/// Implements the LSP protocol for Lemma files:
/// - Diagnostics (parse errors + planning errors) published on file open/change
/// - Registry and local `lemma_deps/` document links from parsed [`SpecRef`] spans
pub struct LemmaLanguageServer {
    client: Client,
    state: Arc<SharedState>,
    registry: Arc<dyn Registry>,
}

impl LemmaLanguageServer {
    pub fn new(client: Client, registry: Box<dyn Registry>) -> Self {
        Self {
            client,
            state: Arc::new(SharedState {
                workspace: RwLock::new(WorkspaceModel::new()),
                #[cfg(not(target_arch = "wasm32"))]
                debounce_notify: Notify::new(),
                root_uri: RwLock::new(None),
            }),
            registry: Arc::from(registry),
        }
    }

    /// Signal the debounce task that a workspace re-validation is needed.
    #[cfg(not(target_arch = "wasm32"))]
    fn request_workspace_validation(&self) {
        self.state.debounce_notify.notify_one();
    }

    /// Run full workspace validation inline and publish all diagnostics.
    ///
    /// On WASM there is no background debounce task (it requires `Send` futures),
    /// so we validate synchronously inside `did_open`/`did_change` instead.
    /// This is fine for the playground: a single file, no registry, fast validation.
    #[cfg(target_arch = "wasm32")]
    async fn publish_full_diagnostics(&self) {
        let workspace = self.state.workspace.read().await;
        publish_workspace_diagnostics(&self.client, &workspace).await;
    }

    /// Discover all `.lemma` files under a directory and add them to the workspace.
    /// No-op on WASM (no filesystem); the single spec is provided via didOpen.
    #[cfg(not(target_arch = "wasm32"))]
    async fn discover_workspace_files(&self, root_path: &Path) {
        let lemma_files = find_lemma_files(root_path);
        let mut workspace = self.state.workspace.write().await;
        workspace.set_workspace_root(root_path.to_path_buf());

        for file_path in lemma_files {
            if let Ok(content) = std::fs::read_to_string(&file_path) {
                if let Ok(url) = Url::from_file_path(&file_path) {
                    workspace.update_file(url, content);
                }
            }
        }
    }

    /// Spawn the background debounce task.
    ///
    /// Waits for a quiet period after edits, then runs full workspace validation
    /// (parse + planning). Fetched registry bundles under `<workspace>/lemma_deps/` are loaded
    /// like the CLI; unresolved `@` references surface as planning errors.
    ///
    /// Not available on WASM — `tokio::spawn` requires `Send` futures, but on WASM
    /// the registry trait uses `?Send` futures.
    #[cfg(not(target_arch = "wasm32"))]
    fn spawn_debounce_task(&self) {
        let state = Arc::clone(&self.state);
        let client = self.client.clone();

        tokio::spawn(async move {
            loop {
                state.debounce_notify.notified().await;

                loop {
                    let timeout_result = tokio::time::timeout(
                        Duration::from_millis(250),
                        state.debounce_notify.notified(),
                    )
                    .await;
                    match timeout_result {
                        Ok(()) => continue,
                        Err(_) => break,
                    }
                }

                let workspace = state.workspace.read().await;
                publish_workspace_diagnostics(&client, &workspace).await;
            }
        });
    }
}

#[tower_lsp::async_trait]
impl LanguageServer for LemmaLanguageServer {
    async fn initialize(&self, params: InitializeParams) -> Result<InitializeResult> {
        // Store the workspace root for file discovery during initialized().
        if let Some(root_uri) = params.root_uri {
            let mut root = self.state.root_uri.write().await;
            *root = Some(root_uri);
        }

        Ok(InitializeResult {
            capabilities: ServerCapabilities {
                text_document_sync: Some(TextDocumentSyncCapability::Kind(
                    TextDocumentSyncKind::FULL,
                )),
                document_link_provider: Some(DocumentLinkOptions {
                    resolve_provider: Some(false),
                    work_done_progress_options: WorkDoneProgressOptions {
                        work_done_progress: Some(false),
                    },
                }),
                hover_provider: Some(HoverProviderCapability::Simple(true)),
                document_formatting_provider: Some(OneOf::Left(true)),
                semantic_tokens_provider: Some(
                    SemanticTokensServerCapabilities::SemanticTokensOptions(
                        SemanticTokensOptions {
                            legend: SemanticTokensLegend {
                                token_types: semantic_tokens::TOKEN_TYPES.to_vec(),
                                token_modifiers: semantic_tokens::TOKEN_MODIFIERS.to_vec(),
                            },
                            full: Some(SemanticTokensFullOptions::Bool(true)),
                            range: None,
                            ..SemanticTokensOptions::default()
                        },
                    ),
                ),
                ..ServerCapabilities::default()
            },
            server_info: Some(ServerInfo {
                name: "lsp".to_string(),
                version: Some(env!("CARGO_PKG_VERSION").to_string()),
            }),
        })
    }

    async fn initialized(&self, _params: InitializedParams) {
        // Spawn the debounce background task (native only; requires Send futures).
        #[cfg(not(target_arch = "wasm32"))]
        self.spawn_debounce_task();

        // Discover workspace `.lemma` files from the workspace root, if available (native only).
        #[cfg(not(target_arch = "wasm32"))]
        {
            let root_uri = {
                let root = self.state.root_uri.read().await;
                root.clone()
            };
            if let Some(root_uri) = root_uri {
                if let Ok(root_path) = root_uri.to_file_path() {
                    self.discover_workspace_files(&root_path).await;
                    let workspace = self.state.workspace.read().await;
                    publish_workspace_diagnostics(&self.client, &workspace).await;
                }
            }
        }

        self.client
            .log_message(MessageType::INFO, "Lemma LSP server initialized")
            .await;
    }

    async fn shutdown(&self) -> Result<()> {
        Ok(())
    }

    async fn did_open(&self, params: DidOpenTextDocumentParams) {
        let uri = params.text_document.uri;
        let text = params.text_document.text;

        {
            let mut workspace = self.state.workspace.write().await;
            workspace.update_file(uri.clone(), text);
        }

        #[cfg(not(target_arch = "wasm32"))]
        self.request_workspace_validation();
        #[cfg(target_arch = "wasm32")]
        self.publish_full_diagnostics().await;
    }

    async fn did_change(&self, params: DidChangeTextDocumentParams) {
        let uri = params.text_document.uri;

        // With FULL sync, the last content change contains the entire spec.
        if let Some(change) = params.content_changes.into_iter().last() {
            {
                let mut workspace = self.state.workspace.write().await;
                workspace.update_file(uri.clone(), change.text);
            }
        }
        #[cfg(not(target_arch = "wasm32"))]
        self.request_workspace_validation();
        #[cfg(target_arch = "wasm32")]
        self.publish_full_diagnostics().await;
    }

    async fn did_close(&self, params: DidCloseTextDocumentParams) {
        let uri = params.text_document.uri;

        {
            let mut workspace = self.state.workspace.write().await;
            workspace.remove_file(&uri);
        }

        // Clear diagnostics for the closed file.
        self.client.publish_diagnostics(uri, Vec::new(), None).await;

        // Signal re-validation since removing a file may affect other files' diagnostics.
        #[cfg(not(target_arch = "wasm32"))]
        self.request_workspace_validation();
        #[cfg(target_arch = "wasm32")]
        self.publish_full_diagnostics().await;
    }

    async fn formatting(&self, params: DocumentFormattingParams) -> Result<Option<Vec<TextEdit>>> {
        let uri = params.text_document.uri;

        let (text, attribute) = {
            let workspace = self.state.workspace.read().await;
            match workspace.get_file_text_and_attribute(&uri) {
                Some((text, attribute)) => (text.to_string(), attribute.to_string()),
                None => return Ok(None),
            }
        };

        // Only format if the file parses successfully — don't mangle broken code.
        match lemma::format_source(
            &text,
            lemma::SourceType::Path(std::sync::Arc::new(std::path::PathBuf::from(&attribute))),
        ) {
            Ok(formatted) if formatted == text => Ok(None), // No changes needed
            Ok(formatted) => {
                let line_count = text.lines().count() as u32;
                // Replace the entire spec with the formatted text.
                let edit = TextEdit {
                    range: Range {
                        start: Position::new(0, 0),
                        end: Position::new(line_count, 0),
                    },
                    new_text: formatted,
                };
                Ok(Some(vec![edit]))
            }
            Err(_) => Ok(None), // Parse error — don't format
        }
    }

    async fn document_link(&self, params: DocumentLinkParams) -> Result<Option<Vec<DocumentLink>>> {
        let uri = params.text_document.uri;
        let workspace = self.state.workspace.read().await;
        let Some(text) = workspace.get_file_text(&uri).map(|t| t.to_string()) else {
            return Ok(None);
        };
        let Some(parse_result) = workspace.parse_success_for_url(&uri) else {
            return Ok(None);
        };

        #[cfg(target_arch = "wasm32")]
        {
            let _ = (text, parse_result);
            Ok(None)
        }

        #[cfg(not(target_arch = "wasm32"))]
        {
            let Some(root) = workspace.workspace_root().cloned() else {
                return Ok(None);
            };
            let engine = workspace.engine_with_workspace();
            let ctx = engine.specs();
            let text = text.as_str();
            let root = root.as_path();

            let links: Vec<DocumentLink> = parse_result
                .repositories
                .values()
                .flatten()
                .flat_map(|consumer| {
                    consumer.data.iter().filter_map(move |data| {
                        let DataValue::Import(spec_ref) = &data.value else {
                            return None;
                        };
                        build_uses_document_link(
                            spec_ref,
                            &consumer.effective_from,
                            text,
                            root,
                            ctx,
                        )
                    })
                })
                .collect();

            Ok((!links.is_empty()).then_some(links))
        }
    }

    async fn hover(&self, params: HoverParams) -> Result<Option<Hover>> {
        let uri = params.text_document_position_params.text_document.uri;
        let position = params.text_document_position_params.position;

        let workspace = self.state.workspace.read().await;
        let Some(text) = workspace.get_file_text(&uri).map(|t| t.to_string()) else {
            return Ok(None);
        };
        let Some(parse_result) = workspace.parse_success_for_url(&uri) else {
            return Ok(None);
        };

        for specs in parse_result.repositories.values() {
            for consumer in specs {
                for data in &consumer.data {
                    let DataValue::Import(spec_ref) = &data.value else {
                        continue;
                    };
                    let Some(repo_qual) = spec_ref.repository.as_ref() else {
                        continue;
                    };
                    if !repo_qual.is_registry() {
                        continue;
                    }
                    let qualifier_name = repo_qual.name.as_str();
                    let Some(hit_range) = spec_ref_hit_range(spec_ref, &text, position) else {
                        continue;
                    };
                    let Some(repo_url) = self.registry.url_for_id(qualifier_name, None) else {
                        return Ok(None);
                    };
                    let markdown = format!("[Open `{qualifier_name}` in LemmaBase]({repo_url})");
                    return Ok(Some(Hover {
                        contents: HoverContents::Markup(MarkupContent {
                            kind: MarkupKind::Markdown,
                            value: markdown,
                        }),
                        range: Some(hit_range),
                    }));
                }
            }
        }

        Ok(None)
    }

    async fn semantic_tokens_full(
        &self,
        params: SemanticTokensParams,
    ) -> Result<Option<SemanticTokensResult>> {
        let uri = params.text_document.uri;

        let text = {
            let workspace = self.state.workspace.read().await;
            workspace.get_file_text(&uri).map(|t| t.to_string())
        };

        match text {
            Some(text) => {
                let tokens = semantic_tokens::tokenize(&text);
                Ok(Some(SemanticTokensResult::Tokens(SemanticTokens {
                    result_id: None,
                    data: tokens,
                })))
            }
            None => Ok(None),
        }
    }
}

/// True when `position` falls inside `range` using LSP's half-open interval
/// (`range.start <= position < range.end`). Position comparison is lexicographic
/// over (line, character in UTF-16 code units), matching how `span_to_range`
/// produces ranges.
fn position_within_range(position: Position, range: &Range) -> bool {
    let start_before_or_at = position.line > range.start.line
        || (position.line == range.start.line && position.character >= range.start.character);
    let end_after = position.line < range.end.line
        || (position.line == range.end.line && position.character < range.end.character);
    start_before_or_at && end_after
}

/// Return the LSP `Range` of the [`SpecRef`] span that `position` is currently
/// inside (qualifier span or target span), or `None` if `position` is not on
/// either span. Used by the hover handler to scope the popup to the hovered
/// portion of the reference.
fn spec_ref_hit_range(spec_ref: &SpecRef, text: &str, position: Position) -> Option<Range> {
    let qualifier_range = spec_ref
        .repository_span
        .as_ref()
        .map(|s| diagnostics::span_to_range(text, s.start, s.end));
    let target_range = spec_ref
        .target_span
        .as_ref()
        .map(|s| diagnostics::span_to_range(text, s.start, s.end));
    qualifier_range
        .filter(|range| position_within_range(position, range))
        .or_else(|| target_range.filter(|range| position_within_range(position, range)))
}

/// Return one LSP `Range` covering the full `uses` reference from the qualifier
/// (e.g. `@lemma/std` or `lemma`) through the target span (`finance 2026-01-01`,
/// `units`). Used by `document_link` so the entire reference is a single clickable
/// region, instead of two separate hot spots on `repository_span` and
/// `target_span`. Falls back to whichever span is present if one is missing.
#[cfg(not(target_arch = "wasm32"))]
fn full_ref_range(spec_ref: &SpecRef, text: &str) -> Option<Range> {
    let start = spec_ref
        .repository_span
        .as_ref()
        .or(spec_ref.target_span.as_ref())
        .map(|s| s.start)?;
    let end = spec_ref
        .target_span
        .as_ref()
        .or(spec_ref.repository_span.as_ref())
        .map(|s| s.end)?;
    Some(diagnostics::span_to_range(text, start, end))
}

/// Build a single `DocumentLink` covering the entire `uses` reference (qualifier +
/// target span) and pointing at the local on-disk file backing the reference:
/// - Registry refs (`@user/repo`): `<workspace>/lemma_deps/<qualifier>.lemma#L<line>` or the
///   actual `SourceType::Path` of the resolved spec if the workspace already has it.
/// - Embedded stdlib (`uses lemma ...`): `<workspace>/lemma_deps/lemma.std#L<line>`, lazily
///   materialised on first call.
///
/// Returns `None` when the reference is not a registry/embedded-stdlib qualifier, when
/// no span information is available, when the qualifier isn't loaded into `ctx`, when
/// the spec name can't be resolved at the consumer's effective slice, or when the
/// destination path can't be expressed as a `file://` URL. The single combined range
/// guarantees that VS Code renders one clickable region for the full `uses` token group
/// instead of separate hot spots on the qualifier and the spec name.
#[cfg(not(target_arch = "wasm32"))]
fn build_uses_document_link(
    spec_ref: &SpecRef,
    consumer_effective: &lemma::EffectiveDate,
    text: &str,
    workspace_root: &Path,
    ctx: &Context,
) -> Option<DocumentLink> {
    let repo_qual = spec_ref.repository.as_ref()?;
    let qualifier_name = repo_qual.name.as_str();
    let is_embedded_stdlib = qualifier_name == lemma::EMBEDDED_STDLIB_REPOSITORY;
    if !repo_qual.is_registry() && !is_embedded_stdlib {
        return None;
    }
    let full_range = full_ref_range(spec_ref, text)?;
    let repo_arc = ctx.find_repository(qualifier_name)?;
    let instant = spec_ref.at(consumer_effective);
    let spec_set = ctx.spec_set(&repo_arc, spec_ref.name.as_str())?;
    let resolved = spec_set.spec_at(&instant)?;
    let dep_path = if is_embedded_stdlib {
        materialize_embedded_stdlib_view(workspace_root)?
    } else {
        match resolved.source_type.as_ref() {
            Some(lemma::SourceType::Path(p)) => p.as_ref().clone(),
            _ => lemma::deps::dependency_cache_file(workspace_root, qualifier_name),
        }
    };
    let mut file_url = Url::from_file_path(&dep_path).ok()?;
    file_url.set_fragment(Some(&format!("L{}", resolved.start_line)));
    Some(DocumentLink {
        range: full_range,
        target: Some(file_url),
        tooltip: Some(format!(
            "Open {} (line {})",
            dep_path.display(),
            resolved.start_line
        )),
        data: None,
    })
}

/// Path under `<workspace>/lemma_deps/` where the LSP writes a view-only copy of the embedded
/// units standard library for editor navigation. The `.std` extension keeps the file out of
/// every `.lemma` discovery pass (CLI loaders, watchers, LSP workspace scan).
#[cfg(not(target_arch = "wasm32"))]
fn embedded_stdlib_view_path(workspace_root: &Path) -> std::path::PathBuf {
    lemma::deps::lemma_deps_dir(workspace_root).join("lemma.std")
}

/// Lazily write the embedded units standard library to `<workspace>/lemma_deps/lemma.std`.
///
/// Called from `document_link` only when a `uses lemma ...` reference is present in the
/// active file, so the file appears in `lemma_deps/` the first time a user wants to navigate
/// into the standard library and never otherwise. Returns the destination path on success
/// (or when the file already exists with the current content); returns `None` if the
/// filesystem write failed -- callers then skip emitting the link rather than producing
/// a link to a non-existent target.
#[cfg(not(target_arch = "wasm32"))]
fn materialize_embedded_stdlib_view(workspace_root: &Path) -> Option<std::path::PathBuf> {
    let destination = embedded_stdlib_view_path(workspace_root);
    let expected = lemma::UNITS_LEMMA;
    match std::fs::read_to_string(&destination) {
        Ok(current) if current == expected => return Some(destination),
        Ok(_) | Err(_) => {}
    }
    if let Some(parent) = destination.parent() {
        std::fs::create_dir_all(parent).ok()?;
    }
    std::fs::write(&destination, expected).ok()?;
    Some(destination)
}

/// Recursively find all `.lemma` files under a directory. Not used on WASM.
#[cfg(not(target_arch = "wasm32"))]
fn find_lemma_files(root: &Path) -> Vec<std::path::PathBuf> {
    let mut results = Vec::new();
    find_lemma_files_recursive(root, &mut results);
    results
}

#[cfg(not(target_arch = "wasm32"))]
fn find_lemma_files_recursive(directory: &Path, results: &mut Vec<std::path::PathBuf>) {
    let entries = match std::fs::read_dir(directory) {
        Ok(entries) => entries,
        Err(_) => return,
    };

    for entry in entries {
        let entry = match entry {
            Ok(entry) => entry,
            Err(_) => continue,
        };

        let path = entry.path();

        if path.is_dir() {
            let dir_name = path.file_name().and_then(|name| name.to_str());
            match dir_name {
                Some(name) if name.starts_with('.') => {}
                _ => find_lemma_files_recursive(&path, results),
            }
        } else if path.extension().and_then(|ext| ext.to_str()) == Some("lemma") {
            results.push(path);
        }
    }
}

#[cfg(all(test, not(target_arch = "wasm32")))]
mod tests {
    use super::*;
    use lemma::DataValue;
    use lemma::LemmaBase;
    use lemma::EMBEDDED_STDLIB_REPOSITORY;

    #[test]
    fn position_within_range_treats_end_as_exclusive() {
        let range = Range {
            start: Position {
                line: 1,
                character: 5,
            },
            end: Position {
                line: 1,
                character: 10,
            },
        };
        assert!(position_within_range(
            Position {
                line: 1,
                character: 5
            },
            &range
        ));
        assert!(position_within_range(
            Position {
                line: 1,
                character: 9
            },
            &range
        ));
        assert!(!position_within_range(
            Position {
                line: 1,
                character: 10
            },
            &range
        ));
        assert!(!position_within_range(
            Position {
                line: 1,
                character: 4
            },
            &range
        ));
        assert!(!position_within_range(
            Position {
                line: 0,
                character: 7
            },
            &range
        ));
        assert!(!position_within_range(
            Position {
                line: 2,
                character: 0
            },
            &range
        ));
    }

    #[test]
    fn position_within_range_spans_multiple_lines() {
        let range = Range {
            start: Position {
                line: 1,
                character: 5,
            },
            end: Position {
                line: 3,
                character: 2,
            },
        };
        assert!(position_within_range(
            Position {
                line: 1,
                character: 5
            },
            &range
        ));
        assert!(position_within_range(
            Position {
                line: 2,
                character: 100
            },
            &range
        ));
        assert!(position_within_range(
            Position {
                line: 3,
                character: 0
            },
            &range
        ));
        assert!(!position_within_range(
            Position {
                line: 3,
                character: 2
            },
            &range
        ));
        assert!(!position_within_range(
            Position {
                line: 1,
                character: 4
            },
            &range
        ));
    }

    /// Fresh empty temp directory unique to the current test process (nextest runs
    /// each test in its own process, so pid alone is unique).
    fn test_workspace() -> std::path::PathBuf {
        let path = std::env::temp_dir().join(format!("lemma_lsp_test_{}", std::process::id()));
        let _ = std::fs::remove_dir_all(&path);
        std::fs::create_dir_all(&path).expect("create test workspace");
        path
    }

    #[test]
    fn embedded_stdlib_view_path_lives_under_lemma_deps() {
        let workspace = test_workspace();
        let path = embedded_stdlib_view_path(&workspace);
        assert_eq!(path.file_name().and_then(|n| n.to_str()), Some("lemma.std"));
        assert_eq!(
            path.parent()
                .and_then(|p| p.file_name())
                .and_then(|n| n.to_str()),
            Some(lemma::deps::LEMMA_DEPS_DIR_NAME),
        );
        assert!(
            path.extension().and_then(|e| e.to_str()) != Some("lemma"),
            "view path must not use the .lemma extension or it would be picked up by workspace discovery",
        );
    }

    #[test]
    fn materialize_writes_units_lemma_when_missing() {
        let workspace = test_workspace();
        let materialized = materialize_embedded_stdlib_view(&workspace)
            .expect("first materialization must succeed");
        assert_eq!(materialized, embedded_stdlib_view_path(&workspace));
        let written = std::fs::read_to_string(&materialized).expect("written file readable");
        assert_eq!(written, lemma::UNITS_LEMMA);
    }

    #[test]
    fn materialize_is_idempotent_when_content_already_matches() {
        let workspace = test_workspace();
        let first = materialize_embedded_stdlib_view(&workspace)
            .expect("first materialization must succeed");
        let metadata_before = std::fs::metadata(&first).expect("metadata before");
        std::thread::sleep(std::time::Duration::from_millis(20));
        let second = materialize_embedded_stdlib_view(&workspace)
            .expect("second materialization must succeed");
        assert_eq!(first, second);
        let metadata_after = std::fs::metadata(&second).expect("metadata after");
        assert_eq!(
            metadata_before.modified().expect("mtime before"),
            metadata_after.modified().expect("mtime after"),
            "matching content must not be re-written",
        );
    }

    #[test]
    fn materialize_overwrites_stale_content() {
        let workspace = test_workspace();
        let destination = embedded_stdlib_view_path(&workspace);
        std::fs::create_dir_all(destination.parent().expect("deps dir parent"))
            .expect("create lemma_deps dir");
        std::fs::write(&destination, "outdated stdlib snapshot").expect("seed stale content");
        let materialized =
            materialize_embedded_stdlib_view(&workspace).expect("materialize must succeed");
        let written = std::fs::read_to_string(&materialized).expect("written file readable");
        assert_eq!(written, lemma::UNITS_LEMMA);
    }

    /// Mirrors the link-resolution body of `document_link` for the embedded stdlib path,
    /// so we can exercise it without constructing a `tower_lsp::Client`. Asserts both the
    /// lazy write and that we can derive a valid `file://` URL at the resolved `spec units`
    /// line number.
    #[test]
    fn uses_lemma_units_produces_link_to_lemma_deps_lemma_std() {
        let workspace = test_workspace();
        let workspace_root = workspace.as_path();
        let source = "spec consumer\nuses lemma units\n";

        let parse_result = lemma::parse(
            source,
            lemma::SourceType::Path(Arc::new(workspace_root.join("consumer.lemma"))),
            &lemma::ResourceLimits::default(),
        )
        .expect("parse consumer");

        let mut engine = lemma::Engine::new();
        for (parsed_repo, specs) in &parse_result.repositories {
            for spec in specs {
                engine
                    .specs_mut()
                    .insert_spec(Arc::clone(parsed_repo), Arc::new(spec.clone()))
                    .expect("insert workspace spec");
            }
        }
        let ctx = engine.specs();

        let consumer_spec = parse_result
            .repositories
            .values()
            .flatten()
            .next()
            .expect("at least one parsed spec");
        let spec_ref = consumer_spec
            .data
            .iter()
            .find_map(|d| match &d.value {
                DataValue::Import(sr) => Some(sr),
                _ => None,
            })
            .expect("consumer must contain a uses import");

        let repo_qual = spec_ref.repository.as_ref().expect("qualified import");
        assert_eq!(repo_qual.name, EMBEDDED_STDLIB_REPOSITORY);
        assert!(
            !repo_qual.is_registry(),
            "lemma is the reserved embedded stdlib repository, not a registry id",
        );

        let materialized = materialize_embedded_stdlib_view(workspace_root)
            .expect("lazy materialization must succeed");
        assert_eq!(materialized, embedded_stdlib_view_path(workspace_root));
        assert!(
            materialized.exists(),
            "materialized file must exist on disk",
        );

        let repository_link_target =
            Url::from_file_path(&materialized).expect("file URL for repository span");
        assert_eq!(
            repository_link_target.scheme(),
            "file",
            "repository link must be a file:// URL",
        );

        let repo_arc = ctx
            .find_repository(EMBEDDED_STDLIB_REPOSITORY)
            .expect("embedded stdlib must be in context");
        let consumer_eff = &consumer_spec.effective_from;
        let instant = spec_ref.at(consumer_eff);
        let spec_set = ctx
            .spec_set(&repo_arc, spec_ref.name.as_str())
            .expect("spec set for units");
        let resolved = spec_set.spec_at(&instant).expect("resolve spec units");
        assert!(
            resolved.start_line >= 1,
            "resolved start_line must reflect UNITS_LEMMA layout, got {}",
            resolved.start_line,
        );

        let mut target_link = Url::from_file_path(&materialized).expect("file URL for target span");
        target_link.set_fragment(Some(&format!("L{}", resolved.start_line)));
        assert_eq!(
            target_link.fragment(),
            Some(format!("L{}", resolved.start_line).as_str()),
        );

        let qualifier_span = spec_ref
            .repository_span
            .as_ref()
            .expect("embedded stdlib uses must carry a repository span");
        let target_span = spec_ref
            .target_span
            .as_ref()
            .expect("embedded stdlib uses must carry a target span");
        let full = full_ref_range(spec_ref, source).expect("full_ref_range must succeed");
        let expected_full =
            diagnostics::span_to_range(source, qualifier_span.start, target_span.end);
        assert_eq!(
            full, expected_full,
            "DocumentLink range must cover `lemma units` as one region (qualifier start to target end)",
        );
    }

    /// Mirrors the markdown-building body of `hover` for a registry `uses` reference
    /// so we can assert the popup shape without spinning up a `tower_lsp::Client`.
    /// Verifies exactly one repository-level LemmaBase link and the absence of any
    /// spec-level LemmaBase link or local `lemma_deps/` link.
    #[test]
    fn hover_on_registry_ref_emits_only_repository_lemmabase_link() {
        let workspace = test_workspace();
        let workspace_root = workspace.as_path();
        let source = "spec consumer\nuses @lemma/std finance\n";

        let parse_result = lemma::parse(
            source,
            lemma::SourceType::Path(Arc::new(workspace_root.join("consumer.lemma"))),
            &lemma::ResourceLimits::default(),
        )
        .expect("parse consumer");

        let consumer_spec = parse_result
            .repositories
            .values()
            .flatten()
            .next()
            .expect("at least one parsed spec");
        let spec_ref = consumer_spec
            .data
            .iter()
            .find_map(|d| match &d.value {
                DataValue::Import(sr) => Some(sr),
                _ => None,
            })
            .expect("consumer must contain a uses import");
        let repo_qual = spec_ref.repository.as_ref().expect("qualified import");
        assert!(repo_qual.is_registry(), "@lemma/std must be a registry id");
        let qualifier_name = repo_qual.name.as_str();
        assert_eq!(qualifier_name, "@lemma/std");

        let qualifier_span = spec_ref
            .repository_span
            .as_ref()
            .expect("registry uses must carry a repository span");
        let qualifier_range =
            diagnostics::span_to_range(source, qualifier_span.start, qualifier_span.end);
        let inside_qualifier = qualifier_range.start;
        let hit = spec_ref_hit_range(spec_ref, source, inside_qualifier)
            .expect("position inside qualifier must hit the SpecRef");
        assert_eq!(hit, qualifier_range);

        let target_span = spec_ref
            .target_span
            .as_ref()
            .expect("registry uses must carry a target span");
        let target_range = diagnostics::span_to_range(source, target_span.start, target_span.end);
        let inside_target = target_range.start;
        let hit_target = spec_ref_hit_range(spec_ref, source, inside_target)
            .expect("position inside target span must hit the SpecRef");
        assert_eq!(hit_target, target_range);

        let outside = Position {
            line: 0,
            character: 0,
        };
        assert!(
            spec_ref_hit_range(spec_ref, source, outside).is_none(),
            "positions outside both spans must not produce a hit",
        );

        let registry = LemmaBase::new();
        let repo_url = registry
            .url_for_id(qualifier_name, None)
            .expect("LemmaBase must yield a repository URL");
        let markdown = format!("[Open `{qualifier_name}` in LemmaBase]({repo_url})");

        assert_eq!(
            markdown,
            format!("[Open `@lemma/std` in LemmaBase]({repo_url})"),
        );
        assert_eq!(
            markdown.matches("](").count(),
            1,
            "hover popup must contain exactly one markdown link, got: {markdown}",
        );
        let spec_identifier = format!("{}/{}", qualifier_name, spec_ref.name);
        assert!(
            !markdown.contains(&spec_identifier),
            "spec-level LemmaBase link must be gone, but markdown contains `{spec_identifier}`: {markdown}",
        );
        assert!(
            !markdown.contains("Open local"),
            "local file hover link must be gone: {markdown}",
        );
        assert!(
            !markdown.contains("file://"),
            "hover popup must not embed a file:// URL: {markdown}",
        );

        let full = full_ref_range(spec_ref, source).expect("full_ref_range must succeed");
        let expected_full =
            diagnostics::span_to_range(source, qualifier_span.start, target_span.end);
        assert_eq!(
            full, expected_full,
            "DocumentLink range must cover `@lemma/std finance` as one region",
        );
        assert!(
            full.start.line < full.end.line
                || (full.start.line == full.end.line && full.start.character < full.end.character),
            "full ref range must be non-empty: {full:?}",
        );
        assert_eq!(
            full.start, qualifier_range.start,
            "full ref range must start where the qualifier starts",
        );
    }

    /// End-to-end exercise of the `document_link` body for the canonical user scenario:
    /// a consumer file with `uses @lemma/std finance 2026-01-01` plus a matching
    /// `lemma_deps/@lemma/std.lemma` dependency file. Asserts `build_uses_document_link`
    /// emits exactly one `DocumentLink` whose range spans from the start of `@lemma/std`
    /// to the end of `finance 2026-01-01` (one clickable region) and whose target is
    /// the on-disk path of the dependency file with a `#L<line>` fragment.
    #[test]
    fn registry_uses_emits_single_unified_document_link() {
        let root = test_workspace();
        let dep_path = lemma::deps::dependency_cache_file(&root, "@lemma/std");
        std::fs::create_dir_all(dep_path.parent().expect("dep parent")).expect("create dep dir");
        std::fs::write(&dep_path, "spec finance\ndata z: 1\n").expect("write dep");
        let consumer_path = root.join("consumer.lemma");
        let consumer_source = "spec demo\nuses @lemma/std finance 2026-01-01\n".to_string();
        std::fs::write(&consumer_path, &consumer_source).expect("write consumer");

        let mut workspace_model = WorkspaceModel::new();
        workspace_model.set_workspace_root(root.clone());
        let consumer_url = Url::from_file_path(&consumer_path).expect("consumer url");
        let dep_url = Url::from_file_path(&dep_path).expect("dep url");
        workspace_model.update_file(consumer_url.clone(), consumer_source.clone());
        workspace_model.update_file(
            dep_url,
            std::fs::read_to_string(&dep_path).expect("read dep"),
        );

        let mut ctx = Context::new();
        let insert_errors = workspace_model.insert_specs_into_context(&mut ctx);
        assert!(
            insert_errors.is_empty(),
            "workspace must insert both files cleanly, got: {insert_errors:?}",
        );
        assert!(
            ctx.find_repository("@lemma/std").is_some(),
            "ctx must contain @lemma/std after inserting lemma_deps/@lemma/std.lemma",
        );

        let parse_result = workspace_model
            .parse_success_for_url(&consumer_url)
            .expect("consumer must parse cleanly");

        let mut links: Vec<DocumentLink> = Vec::new();
        for specs in parse_result.repositories.values() {
            for consumer in specs {
                for data in &consumer.data {
                    let DataValue::Import(spec_ref) = &data.value else {
                        continue;
                    };
                    if let Some(link) = build_uses_document_link(
                        spec_ref,
                        &consumer.effective_from,
                        &consumer_source,
                        &root,
                        &ctx,
                    ) {
                        links.push(link);
                    }
                }
            }
        }

        assert_eq!(
            links.len(),
            1,
            "expected exactly one unified DocumentLink, got {}: {links:?}",
            links.len(),
        );
        let link = &links[0];

        let qualifier_start = consumer_source
            .find("@lemma/std")
            .expect("qualifier present");
        let target_end =
            consumer_source.find("2026-01-01").expect("date present") + "2026-01-01".len();
        let expected_range =
            diagnostics::span_to_range(&consumer_source, qualifier_start, target_end);
        assert_eq!(
            link.range, expected_range,
            "DocumentLink range must cover `@lemma/std finance 2026-01-01` as one region",
        );

        let target = link.target.as_ref().expect("link must have a target");
        assert_eq!(target.scheme(), "file", "target must be a file:// URL");
        let target_path = target.to_file_path().expect("target must be a file path");
        assert_eq!(
            target_path, dep_path,
            "DocumentLink must point at the on-disk lemma_deps/@lemma/std.lemma",
        );
        assert_eq!(
            target.fragment(),
            Some("L1"),
            "target must carry a #L<line> fragment for the resolved spec, got: {:?}",
            target.fragment(),
        );
    }
}