Skip to main content

oak_nim/
lsp.rs

1use crate::{kind::NimSyntaxKind, language::NimLanguage};
2use dashmap::DashMap;
3use oak_core::{Range, parser::session::ParseSession, tree::RedNode};
4use oak_hover::HoverProvider;
5use oak_lsp::service::LanguageService;
6use oak_vfs::Vfs;
7use std::{future::Future, sync::Arc};
8
9/// Hover provider implementation for Nim.
10pub struct NimHoverProvider;
11
12impl HoverProvider<NimLanguage> for NimHoverProvider {
13    fn hover(&self, node: &RedNode<'_, NimLanguage>, _range: Range<usize>) -> Option<oak_hover::Hover> {
14        let kind = node.green.kind;
15
16        let contents = match kind {
17            NimSyntaxKind::ProcKeyword => "### Nim Procedure\nDefines a callable procedure in Nim.",
18            NimSyntaxKind::TypeKeyword => "### Nim Type Definition\nDefines a new type or type alias.",
19            NimSyntaxKind::ConstKeyword => "### Nim Constant\nDefines a compile-time constant.",
20            NimSyntaxKind::Root => "### Nim Module\nThe root of a Nim module.",
21            _ => return None,
22        };
23
24        Some(oak_hover::Hover { contents: contents.to_string(), range: Some(node.span()) })
25    }
26}
27
28/// Language service implementation for Nim.
29pub struct NimLanguageService<V: Vfs> {
30    vfs: V,
31    workspace: oak_lsp::workspace::WorkspaceManager,
32    hover_provider: NimHoverProvider,
33    _cache: Arc<DashMap<String, ParseSession<NimLanguage>>>,
34}
35
36impl<V: Vfs> NimLanguageService<V> {
37    pub fn new(vfs: V) -> Self {
38        Self { vfs, workspace: oak_lsp::workspace::WorkspaceManager::default(), hover_provider: NimHoverProvider, _cache: Arc::new(DashMap::new()) }
39    }
40}
41
42impl<V: Vfs + Send + Sync + 'static + oak_vfs::WritableVfs> LanguageService for NimLanguageService<V> {
43    type Lang = NimLanguage;
44    type Vfs = V;
45
46    fn vfs(&self) -> &Self::Vfs {
47        &self.vfs
48    }
49
50    fn workspace(&self) -> &oak_lsp::workspace::WorkspaceManager {
51        &self.workspace
52    }
53
54    fn get_root(&self, uri: &str) -> impl Future<Output = Option<RedNode<'_, NimLanguage>>> + Send + '_ {
55        let source = self.get_source(uri);
56        async move {
57            let _source = source?;
58            // TODO: Implement proper caching and conversion to RedNode
59            None
60        }
61    }
62
63    fn hover(&self, uri: &str, range: Range<usize>) -> impl Future<Output = Option<oak_lsp::types::Hover>> + Send + '_ {
64        let uri = uri.to_string();
65        async move {
66            let hover = self.with_root(&uri, |root| self.hover_provider.hover(&root, range)).await.flatten()?;
67            Some(oak_lsp::types::Hover { contents: hover.contents, range: hover.range })
68        }
69    }
70}