sway_lsp/capabilities/code_actions/enum_decl/
mod.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
pub(crate) mod enum_impl;

use self::enum_impl::EnumImplCodeAction;
use crate::capabilities::code_actions::{CodeAction, CodeActionContext};
use lsp_types::CodeActionOrCommand;
use sway_core::{decl_engine::id::DeclId, language::ty};

use super::common::basic_doc_comment::BasicDocCommentCodeAction;

pub(crate) fn code_actions(
    decl_id: &DeclId<ty::TyEnumDecl>,
    ctx: &CodeActionContext,
) -> Vec<CodeActionOrCommand> {
    let decl = (*ctx.engines.de().get_enum(decl_id)).clone();
    vec![
        EnumImplCodeAction::new(ctx, &decl).code_action(),
        BasicDocCommentCodeAction::new(ctx, &decl).code_action(),
    ]
}