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

oak_sql/lsp/
mod.rs
