Skip to main content

oak_jasm/lsp/
mod.rs

1#![doc = include_str!("readme.md")]
2
3/// Formatter module.
4#[cfg(feature = "oak-pretty-print")]
5pub mod formatter;
6#[cfg(feature = "oak-highlight")]
7pub mod highlighter;
8
9use crate::JasmLanguage;
10use core::range::Range;
11use oak_core::tree::RedNode;
12#[cfg(feature = "lsp")]
13use {
14    futures::Future,
15    oak_hover::{Hover, HoverProvider},
16    oak_lsp::service::LanguageService,
17    oak_vfs::Vfs,
18};
19#[cfg(feature = "lsp")]
20/// Hover provider for Jasm.
21pub struct JasmHoverProvider;
22#[cfg(feature = "lsp")]
23impl HoverProvider<JasmLanguage> for JasmHoverProvider {
24    fn hover(&self, node: &RedNode<JasmLanguage>, _range: Range<usize>) -> Option<Hover> {
25        let kind = node.green.kind;
26        let contents = format!("### JASM Node\nKind: {:?}", kind);
27        Some(Hover { contents, range: Some(node.span()) })
28    }
29}
30#[cfg(feature = "lsp")]
31/// Language service for Jasm.
32pub struct JasmLanguageService<V: Vfs> {
33    /// The virtual file system.
34    pub vfs: V,
35    /// The workspace manager.
36    pub workspace: oak_lsp::workspace::WorkspaceManager,
37    /// The hover provider.
38    pub hover_provider: JasmHoverProvider,
39}
40#[cfg(feature = "lsp")]
41impl<V: Vfs> JasmLanguageService<V> {
42    /// Creates a new instance of `JasmLanguageService`.
43    pub fn new(vfs: V) -> Self {
44        Self { vfs, workspace: oak_lsp::workspace::WorkspaceManager::default(), hover_provider: JasmHoverProvider }
45    }
46}
47#[cfg(feature = "lsp")]
48impl<V: Vfs + Send + Sync + 'static + oak_vfs::WritableVfs> LanguageService for JasmLanguageService<V> {
49    type Lang = JasmLanguage;
50    type Vfs = V;
51    fn vfs(&self) -> &Self::Vfs {
52        &self.vfs
53    }
54    fn workspace(&self) -> &oak_lsp::workspace::WorkspaceManager {
55        &self.workspace
56    }
57    fn get_root(&self, uri: &str) -> impl Future<Output = Option<RedNode<'_, JasmLanguage>>> + Send + '_ {
58        let source = self.vfs().get_source(uri);
59        async move {
60            let _source = source?;
61            None
62        }
63    }
64    fn hover(&self, uri: &str, range: Range<usize>) -> impl Future<Output = Option<oak_lsp::Hover>> + Send + '_ {
65        let uri = uri.to_string();
66        async move { self.with_root(&uri, |root| self.hover_provider.hover(&root, range).map(|h| oak_lsp::Hover { contents: h.contents, range: h.range })).await.flatten() }
67    }
68}