use crate::{
helpers,
uri::{InternUri, UrisCtx},
LanguageService,
};
use itertools::Itertools;
use line_index::LineIndex;
use lsp_types::{CodeAction, CodeActionKind, TextEdit, WorkspaceEdit};
use std::collections::HashMap;
use wat_syntax::{SyntaxKind, SyntaxNode};
pub fn act(
service: &LanguageService,
uri: InternUri,
line_index: &LineIndex,
node: &SyntaxNode,
kind: SyntaxKind,
) -> Option<CodeAction> {
let types = node.children().collect::<Vec<_>>();
if types.len() <= 1 {
return None;
}
let keyword = match kind {
SyntaxKind::PARAM => "param",
SyntaxKind::RESULT => "result",
SyntaxKind::LOCAL => "local",
_ => return None,
};
let new_text = types
.into_iter()
.map(|ty| format!("({keyword} {ty})"))
.join(" ");
#[expect(clippy::mutable_key_type)]
let mut changes = HashMap::with_capacity(1);
changes.insert(
service.lookup_uri(uri),
vec![TextEdit {
range: helpers::rowan_range_to_lsp_range(line_index, node.text_range()),
new_text,
}],
);
let title = match kind {
SyntaxKind::PARAM => "Split parameters".into(),
SyntaxKind::RESULT => "Split results".into(),
SyntaxKind::LOCAL => "Split locals".into(),
_ => return None,
};
Some(CodeAction {
title,
kind: Some(CodeActionKind::REFACTOR_REWRITE),
edit: Some(WorkspaceEdit {
changes: Some(changes),
..Default::default()
}),
..Default::default()
})
}