use super::find_meaningful_token;
use crate::{
binder::{Symbol, SymbolKey, SymbolKind, SymbolTablesCtx},
data_set, helpers,
idx::{IdentsCtx, Idx},
syntax_tree::SyntaxTreeCtx,
types_analyzer::{self, TypesAnalyzerCtx},
uri::{InternUri, UrisCtx},
LanguageService,
};
use itertools::Itertools;
use lsp_types::{
Hover, HoverContents, HoverParams, LanguageString, MarkedString, MarkupContent, MarkupKind,
};
use rowan::ast::{support::child, AstNode};
use wat_syntax::{
ast::{GlobalType, PlainInstr},
SyntaxKind, SyntaxNode,
};
impl LanguageService {
pub fn hover(&self, params: HoverParams) -> Option<Hover> {
let uri = self.uri(
params
.text_document_position_params
.text_document
.uri
.clone(),
);
let root = SyntaxNode::new_root(self.root(uri));
let token = find_meaningful_token(
self,
uri,
&root,
params.text_document_position_params.position,
)?;
let line_index = self.line_index(uri);
match token.kind() {
SyntaxKind::IDENT | SyntaxKind::INT | SyntaxKind::UNSIGNED_INT => {
let symbol_table = self.symbol_table(uri);
let parent = token.parent()?;
let key = SymbolKey::new(&parent);
symbol_table
.find_param_or_local_def(key)
.map(|symbol| Hover {
contents: HoverContents::Scalar(create_param_or_local_hover(self, symbol)),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
.or_else(|| {
symbol_table
.symbols
.iter()
.find(|symbol| symbol.key == key)
.and_then(|symbol| match symbol.kind {
SymbolKind::Call => symbol_table.find_defs(key).map(|symbols| {
let contents = symbols
.map(|symbol| {
create_func_hover(self, uri, symbol.clone(), &root)
})
.join("\n---\n");
Hover {
contents: HoverContents::Markup(MarkupContent {
kind: MarkupKind::Markdown,
value: contents,
}),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
}
}),
SymbolKind::TypeUse => {
symbol_table.find_defs(key).map(|symbols| Hover {
contents: HoverContents::Array(
symbols
.map(|symbol| create_type_def_hover(self, symbol))
.collect(),
),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
}
SymbolKind::GlobalRef => {
symbol_table.find_defs(key).map(|symbols| Hover {
contents: HoverContents::Array(
symbols
.map(|symbol| {
create_global_def_hover(self, symbol, &root)
})
.collect(),
),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
}
SymbolKind::BlockRef => symbol_table
.find_block_def(key)
.and_then(|def_key| {
symbol_table
.symbols
.iter()
.find(|symbol| symbol.key == def_key)
})
.map(|block| Hover {
contents: HoverContents::Scalar(create_block_hover(
self, block, uri, &root,
)),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
}),
_ => None,
})
})
.or_else(|| {
symbol_table
.symbols
.iter()
.find(|symbol| symbol.key == key)
.and_then(|symbol| create_def_hover(self, uri, &root, symbol))
.map(|contents| Hover {
contents,
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
})
}
SyntaxKind::NUM_TYPE | SyntaxKind::VEC_TYPE | SyntaxKind::REF_TYPE => {
let ty = token.text();
data_set::get_value_type_description(token.text()).map(|doc| Hover {
contents: HoverContents::Markup(MarkupContent {
kind: MarkupKind::Markdown,
value: format!("```wat\n{ty}\n```\n\n{doc}"),
}),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
}
SyntaxKind::KEYWORD => {
let parent = token.parent()?;
let key = SymbolKey::new(&parent);
let symbol_table = self.symbol_table(uri);
symbol_table
.symbols
.iter()
.find(|symbol| symbol.key == key)
.and_then(|symbol| create_def_hover(self, uri, &root, symbol))
.map(|contents| Hover {
contents,
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
}
SyntaxKind::INSTR_NAME => {
let name = token.text();
let key = if name == "select" {
let parent = token.parent().and_then(PlainInstr::cast)?;
if parent.immediates().count() > 0 {
"select."
} else {
"select"
}
} else {
name
};
data_set::INSTR_OP_CODES.get(key).map(|code| Hover {
contents: HoverContents::Markup(MarkupContent {
kind: MarkupKind::Markdown,
value: format!(
"```wat\n{name}\n```\nBinary Opcode: {}",
format_op_code(*code)
),
}),
range: Some(helpers::rowan_range_to_lsp_range(
&line_index,
token.text_range(),
)),
})
}
_ => None,
}
}
}
fn create_def_hover(
service: &LanguageService,
uri: InternUri,
root: &SyntaxNode,
symbol: &Symbol,
) -> Option<HoverContents> {
match symbol.kind {
SymbolKind::Param | SymbolKind::Local => Some(HoverContents::Scalar(
create_param_or_local_hover(service, symbol),
)),
SymbolKind::Func => Some(HoverContents::Markup(MarkupContent {
kind: MarkupKind::Markdown,
value: create_func_hover(service, uri, symbol.clone(), root),
})),
SymbolKind::Type => Some(HoverContents::Scalar(create_type_def_hover(
service, symbol,
))),
SymbolKind::GlobalDef => Some(HoverContents::Scalar(create_global_def_hover(
service, symbol, root,
))),
SymbolKind::BlockDef => Some(HoverContents::Scalar(create_block_hover(
service, symbol, uri, root,
))),
_ => None,
}
}
fn create_func_hover(
service: &LanguageService,
uri: InternUri,
symbol: Symbol,
root: &SyntaxNode,
) -> String {
let node = symbol.key.to_node(root);
let doc = helpers::ast::get_doc_comment(&node);
let mut content = format!(
"```wat\n{}\n```",
service.render_func_header(
symbol.idx.name,
service.get_func_sig(uri, symbol.key, symbol.green)
)
);
if !doc.is_empty() {
content.push_str("\n---\n");
content.push_str(&doc);
}
content
}
fn create_param_or_local_hover(service: &LanguageService, symbol: &Symbol) -> MarkedString {
let mut content_value = '('.to_string();
match symbol.kind {
SymbolKind::Param => {
content_value.push_str("param");
if let Some(name) = symbol.idx.name {
content_value.push(' ');
content_value.push_str(&service.lookup_ident(name));
}
}
SymbolKind::Local => {
content_value.push_str("local");
if let Some(name) = symbol.idx.name {
content_value.push(' ');
content_value.push_str(&service.lookup_ident(name));
}
}
_ => {}
}
if let Some(ty) = service.extract_type(symbol.green.clone()) {
content_value.push(' ');
content_value.push_str(&ty.to_string());
}
content_value.push(')');
create_marked_string(content_value)
}
fn create_global_def_hover(
service: &LanguageService,
symbol: &Symbol,
root: &SyntaxNode,
) -> MarkedString {
let mut content_value = '('.to_string();
if symbol.kind == SymbolKind::GlobalDef {
content_value.push_str("global");
if let Some(name) = symbol.idx.name {
content_value.push(' ');
content_value.push_str(&service.lookup_ident(name));
}
}
let node = symbol.key.to_node(root);
if let Some(global_type) = child::<GlobalType>(&node) {
let mutable = global_type.mut_keyword().is_some();
if mutable {
content_value.push_str(" (mut");
}
if let Some(val_type) = global_type.val_type() {
content_value.push(' ');
content_value.push_str(&val_type.syntax().to_string());
}
if mutable {
content_value.push(')');
}
}
content_value.push(')');
create_marked_string(content_value)
}
fn create_type_def_hover(service: &LanguageService, symbol: &Symbol) -> MarkedString {
let mut content_value = "(type".to_string();
if let Symbol {
kind: SymbolKind::Type,
idx: Idx {
name: Some(name), ..
},
..
} = symbol
{
content_value.push(' ');
content_value.push_str(&service.lookup_ident(*name));
}
if let Some(func_type) = helpers::ast::find_func_type_of_type_def(&symbol.green) {
let sig = service.extract_sig(func_type.to_owned());
content_value.push_str(" (func");
if !sig.params.is_empty() || !sig.results.is_empty() {
content_value.push(' ');
content_value.push_str(&service.render_sig(sig));
}
content_value.push(')');
}
content_value.push(')');
create_marked_string(content_value)
}
fn create_block_hover(
service: &LanguageService,
symbol: &Symbol,
uri: InternUri,
root: &SyntaxNode,
) -> MarkedString {
create_marked_string(service.render_block_header(
symbol.key.kind(),
symbol.idx.name,
types_analyzer::get_block_sig(service, uri, &symbol.key.to_node(root)),
))
}
fn create_marked_string(value: String) -> MarkedString {
MarkedString::LanguageString(LanguageString {
language: "wat".into(),
value,
})
}
fn format_op_code(code: u32) -> String {
if code >> 16 > 0 {
format!(
"0x{:02X} 0x{:02X} 0x{:02X}",
code >> 16,
(code >> 8) & 0xFF,
code & 0xFF
)
} else if code >> 8 > 0 {
format!("0x{:02X} 0x{:02X}", code >> 8, code & 0xFF)
} else {
format!("0x{:02X}", code)
}
}