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
9pub 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
28pub 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 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}