Skip to main content

oak_sql/lsp/
mod.rs

1#![doc = include_str!("readme.md")]
2#[cfg(feature = "oak-highlight")]
3pub mod highlighter;
4
5#[cfg(feature = "lsp")]
6use {
7    futures::Future,
8    oak_hover::{Hover, HoverProvider},
9    oak_lsp::service::LanguageService,
10    oak_vfs::Vfs,
11};
12#[cfg(feature = "oak-pretty-print")]
13pub mod formatter;
14use crate::language::SqlLanguage;
15use core::range::Range;
16use oak_core::tree::RedNode;
17#[cfg(feature = "lsp")]
18pub struct SqlHoverProvider;
19#[cfg(feature = "lsp")]
20impl HoverProvider<SqlLanguage> for SqlHoverProvider {
21    fn hover(&self, node: &RedNode<SqlLanguage>, _range: Range<usize>) -> Option<Hover> {
22        let _kind = node.green.kind;
23        None
24    }
25}
26#[cfg(feature = "lsp")]
27pub struct SqlLanguageService<V: Vfs> {
28    vfs: V,
29    workspace: oak_lsp::workspace::WorkspaceManager,
30    hover_provider: SqlHoverProvider,
31}
32impl<V: Vfs> SqlLanguageService<V> {
33    pub fn new(vfs: V) -> Self {
34        Self { vfs, workspace: oak_lsp::workspace::WorkspaceManager::default(), hover_provider: SqlHoverProvider }
35    }
36}
37impl<V: Vfs + Send + Sync + 'static + oak_vfs::WritableVfs> LanguageService for SqlLanguageService<V> {
38    type Lang = SqlLanguage;
39    type Vfs = V;
40    fn vfs(&self) -> &Self::Vfs {
41        &self.vfs
42    }
43    fn workspace(&self) -> &oak_lsp::workspace::WorkspaceManager {
44        &self.workspace
45    }
46    fn get_root(&self, uri: &str) -> impl Future<Output = Option<RedNode<'_, SqlLanguage>>> + Send + '_ {
47        let source = self.vfs().get_source(uri);
48        async move {
49            let source = source?;
50            let language = SqlLanguage::default();
51            let parser = crate::parser::SqlParser::new(&language);
52            let lexer = crate::lexer::SqlLexer::new(&language);
53            let mut cache = oak_core::parser::session::ParseSession::<SqlLanguage>::default();
54            let _output = oak_core::parser::parse(&parser, &lexer, &source, &[], &mut cache);
55            None
56        }
57    }
58    fn hover(&self, uri: &str, range: Range<usize>) -> impl Future<Output = Option<oak_lsp::Hover>> + Send + '_ {
59        let uri = uri.to_string();
60        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() }
61    }
62}