use crate::{ContainerBody, Import, Language, LanguageSymbols, Visibility};
use tree_sitter::Node;
pub struct VB;
impl Language for VB {
fn name(&self) -> &'static str {
"Visual Basic"
}
fn extensions(&self) -> &'static [&'static str] {
&["vb", "vbs"]
}
fn grammar_name(&self) -> &'static str {
"vb"
}
fn as_symbols(&self) -> Option<&dyn LanguageSymbols> {
Some(self)
}
fn extract_imports(&self, node: &Node, content: &str) -> Vec<Import> {
if node.kind() != "imports_statement" {
return Vec::new();
}
let text = &content[node.byte_range()];
vec![Import {
module: text.trim().to_string(),
names: Vec::new(),
alias: None,
is_wildcard: false,
is_relative: false,
line: node.start_position().row + 1,
}]
}
fn format_import(&self, import: &Import, _names: Option<&[&str]>) -> String {
format!("Imports {}", import.module)
}
fn get_visibility(&self, node: &Node, content: &str) -> Visibility {
let text = &content[node.byte_range()];
let lower = text.to_lowercase();
if lower.contains("private") {
Visibility::Private
} else if lower.contains("protected") {
Visibility::Protected
} else {
Visibility::Public
}
}
fn is_test_symbol(&self, symbol: &crate::Symbol) -> bool {
let name = symbol.name.as_str();
match symbol.kind {
crate::SymbolKind::Function | crate::SymbolKind::Method => name.starts_with("test_"),
crate::SymbolKind::Module => name == "tests" || name == "test",
_ => false,
}
}
fn test_file_globs(&self) -> &'static [&'static str] {
&["**/*Test.vb", "**/*Tests.vb"]
}
fn container_body<'a>(&self, node: &'a Node<'a>) -> Option<Node<'a>> {
node.child_by_field_name("body")
}
fn analyze_container_body(
&self,
body_node: &Node,
content: &str,
inner_indent: &str,
) -> Option<ContainerBody> {
crate::body::analyze_end_body(body_node, content, inner_indent)
}
}
impl LanguageSymbols for VB {}
#[cfg(test)]
mod tests {
use super::*;
use crate::validate_unused_kinds_audit;
#[test]
fn unused_node_kinds_audit() {
#[rustfmt::skip]
let documented_unused: &[&str] = &[
"namespace_block",
"field_declaration", "constructor_declaration", "event_declaration",
"type_declaration", "const_declaration", "enum_member",
"statement", "assignment_statement", "compound_assignment_statement",
"call_statement", "dim_statement", "redim_statement", "re_dim_clause",
"exit_statement", "continue_statement", "return_statement", "goto_statement",
"label_statement", "throw_statement", "empty_statement",
"try_statement", "catch_block", "finally_block",
"case_block", "case_else_block", "else_clause", "elseif_clause",
"with_statement", "with_initializer",
"using_statement", "sync_lock_statement",
"expression", "binary_expression", "unary_expression", "ternary_expression",
"parenthesized_expression", "lambda_expression", "new_expression",
"type", "generic_type", "array_type", "primitive_type",
"type_parameters", "type_parameter", "type_constraint",
"type_argument_list", "array_rank_specifier",
"as_clause", "inherits_clause", "implements_clause",
"modifier", "modifiers",
"add_handler_block", "remove_handler_block", "raise_event_block",
"identifier", "attribute_block", "option_statements",
"relational_operator", "lambda_parameter",
"case_clause",
"while_statement",
"for_statement",
"for_each_statement",
"imports_statement",
"do_statement",
"if_statement",
"select_case_statement",
];
validate_unused_kinds_audit(&VB, documented_unused)
.expect("Visual Basic unused node kinds audit failed");
}
}