cairo-language-server 2.18.0

//! Module for collecting static and dynamic capabilities the server wants to register.
//! A capability can be registered statically ONLY
//! if the client does not support dynamic registration for this capability, as per LSP spec
//! <https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#client_registerCapability>:
//!
//! > Server must not register the same capability both statically through the initialize result and
//! > dynamically for the same document selector. If a server wants to support both static and
//! > dynamic
//! > registration it needs to check the client capability in the initialize request and only
//! > register
//! > the capability statically if the client doesn’t support dynamic registration for that
//! > capability.

use std::ops::Not;

use itertools::{Itertools, chain};
use lsp_types::notification::{
    DidChangeTextDocument, DidChangeWatchedFiles, DidCloseTextDocument, DidOpenTextDocument,
    DidSaveTextDocument, Notification,
};
use lsp_types::request::{
    CodeActionRequest, CodeLensRequest, Completion, DocumentHighlightRequest, ExecuteCommand,
    Formatting, GotoDefinition, HoverRequest, InlayHintRequest, References, Rename, Request,
};
use lsp_types::{
    ClientCapabilities, CodeActionProviderCapability, CodeLensOptions, CompletionOptions,
    CompletionRegistrationOptions, DefinitionOptions, DidChangeWatchedFilesRegistrationOptions,
    DocumentFilter, DocumentHighlightOptions, ExecuteCommandOptions,
    ExecuteCommandRegistrationOptions, FileOperationFilter, FileOperationPattern,
    FileOperationPatternKind, FileOperationRegistrationOptions, FileSystemWatcher, GlobPattern,
    HoverProviderCapability, HoverRegistrationOptions, InlayHintOptions,
    InlayHintRegistrationOptions, OneOf, ReferencesOptions, Registration, RenameOptions,
    SaveOptions, SemanticTokensFullOptions, SemanticTokensLegend, SemanticTokensOptions,
    SemanticTokensRegistrationOptions, ServerCapabilities, TextDocumentChangeRegistrationOptions,
    TextDocumentRegistrationOptions, TextDocumentSaveRegistrationOptions,
    TextDocumentSyncCapability, TextDocumentSyncKind, TextDocumentSyncOptions,
    TextDocumentSyncSaveOptions, WorkspaceFileOperationsServerCapabilities,
    WorkspaceServerCapabilities,
};
use missing_lsp_types::{
    CodeActionRegistrationOptions, CodeLensRegistrationOptions, DefinitionRegistrationOptions,
    DocumentFormattingRegistrationOptions, DocumentHighlightRegistrationOptions,
    ReferencesRegistrationOptions, RenameRegistrationOptions,
};
use serde::Serialize;

use crate::ide::semantic_highlighting::SemanticTokenKind;
use crate::lsp::capabilities::client::ClientCapabilitiesExt;
use crate::lsp::ext::ViewSyntaxTree;
use crate::server::commands::ServerCommand;

/// Returns capabilities the server wants to register statically.
pub fn collect_server_capabilities(client_capabilities: &ClientCapabilities) -> ServerCapabilities {
    ServerCapabilities {
        workspace: client_capabilities.workspace_will_rename_files_support().then(|| {
            WorkspaceServerCapabilities {
                file_operations: Some(WorkspaceFileOperationsServerCapabilities {
                    will_rename: Some(FileOperationRegistrationOptions {
                        filters: vec![
                            FileOperationFilter {
                                scheme: Some(String::from("file")),
                                pattern: FileOperationPattern {
                                    glob: String::from("**/*.cairo"),
                                    matches: Some(FileOperationPatternKind::File),
                                    options: None,
                                },
                            },
                            FileOperationFilter {
                                scheme: Some(String::from("file")),
                                pattern: FileOperationPattern {
                                    glob: String::from("**"),
                                    matches: Some(FileOperationPatternKind::Folder),
                                    options: None,
                                },
                            },
                        ],
                    }),
                    ..Default::default()
                }),
                ..Default::default()
            }
        }),
        text_document_sync: client_capabilities
            .text_document_synchronization_dynamic_registration()
            .not()
            .then_some(TextDocumentSyncCapability::Options(TextDocumentSyncOptions {
                open_close: Some(true),
                change: Some(TextDocumentSyncKind::FULL),
                will_save: Some(false),
                will_save_wait_until: Some(false),
                save: Some(TextDocumentSyncSaveOptions::SaveOptions(SaveOptions {
                    include_text: Some(false),
                })),
            })),
        completion_provider: client_capabilities.completion_dynamic_registration().not().then(
            || CompletionOptions {
                resolve_provider: Some(false),
                trigger_characters: Some(vec![".".to_string(), ":".to_string()]),
                all_commit_characters: None,
                work_done_progress_options: Default::default(),
                completion_item: None,
            },
        ),
        execute_command_provider: client_capabilities
            .execute_command_dynamic_registration()
            .not()
            .then(|| ExecuteCommandOptions {
                commands: vec![
                    ServerCommand::Reload.as_str().to_string(),
                    ServerCommand::ExecuteCodeLens.as_str().to_string(),
                ],
                work_done_progress_options: Default::default(),
            }),
        semantic_tokens_provider: client_capabilities
            .semantic_tokens_dynamic_registration()
            .not()
            .then(|| {
                SemanticTokensOptions {
                    legend: SemanticTokensLegend {
                        token_types: SemanticTokenKind::legend(),
                        token_modifiers: vec![],
                    },
                    full: Some(SemanticTokensFullOptions::Bool(true)),
                    ..SemanticTokensOptions::default()
                }
                .into()
            }),
        document_formatting_provider: client_capabilities
            .formatting_dynamic_registration()
            .not()
            .then_some(OneOf::Left(true)),
        hover_provider: client_capabilities
            .hover_dynamic_registration()
            .not()
            .then_some(HoverProviderCapability::Simple(true)),
        definition_provider: client_capabilities
            .definition_dynamic_registration()
            .not()
            .then_some(OneOf::Left(true)),
        code_action_provider: client_capabilities
            .code_action_dynamic_registration()
            .not()
            .then_some(CodeActionProviderCapability::Simple(true)),
        references_provider: client_capabilities
            .references_provider_dynamic_registration()
            .not()
            .then_some(OneOf::Left(true)),
        rename_provider: client_capabilities
            .rename_provider_dynamic_registration()
            .not()
            .then_some(OneOf::Left(true)),
        document_highlight_provider: client_capabilities
            .document_highlight_provider_dynamic_registration()
            .not()
            .then_some(OneOf::Left(true)),
        code_lens_provider: client_capabilities
            .code_lens_provider_dynamic_registration()
            .not()
            .then_some(CodeLensOptions { resolve_provider: Some(false) }),
        inlay_hint_provider: client_capabilities
            .text_document_inlay_hints_dynamic_registration()
            .not()
            .then_some(OneOf::Left(true)),
        ..ServerCapabilities::default()
    }
}

/// Returns registrations of capabilities the server wants to register dynamically.
pub fn collect_dynamic_registrations(
    client_capabilities: &ClientCapabilities,
) -> Vec<Registration> {
    let mut registrations = vec![];

    // Relevant files.
    let cairo_files_filters = vec![
        DocumentFilter {
            language: Some("cairo".to_string()),
            scheme: Some("file".to_string()),
            pattern: None,
        },
        DocumentFilter {
            language: Some("cairo".to_string()),
            scheme: Some("vfs".to_string()),
            pattern: None,
        },
    ];
    // .toml files are added here to properly trigger project updates on opening the files in the editor.
    let sync_document_selector = Some(
        chain!(
            cairo_files_filters.clone(),
            vec![DocumentFilter {
                language: Some("toml".to_string()),
                scheme: Some("file".to_string()),
                pattern: Some("**/Scarb.toml".to_string()),
            },]
        )
        .collect_vec(),
    );
    let text_document_registration_options =
        TextDocumentRegistrationOptions { document_selector: Some(cairo_files_filters.clone()) };
    let open_text_document_registration_options =
        TextDocumentRegistrationOptions { document_selector: sync_document_selector.clone() };

    if client_capabilities.did_change_watched_files_dynamic_registration() {
        // Register patterns for the client file watcher.
        // This is used to detect changes to config files and invalidate .cairo files.
        let registration_options = DidChangeWatchedFilesRegistrationOptions {
            watchers: ["/**/*.cairo", "/**/Scarb.toml", "/**/Scarb.lock", "/**/cairo_project.toml"]
                .map(|glob_pattern| FileSystemWatcher {
                    glob_pattern: GlobPattern::String(glob_pattern.to_string()),
                    kind: None,
                })
                .into(),
        };

        registrations
            .push(create_registration(DidChangeWatchedFiles::METHOD, registration_options));
    }

    if client_capabilities.text_document_synchronization_dynamic_registration() {
        registrations.push(create_registration(
            DidOpenTextDocument::METHOD,
            &open_text_document_registration_options,
        ));

        registrations.push(create_registration(
            DidChangeTextDocument::METHOD,
            TextDocumentChangeRegistrationOptions {
                document_selector: Some(cairo_files_filters.clone()),
                sync_kind: 1, // TextDocumentSyncKind::FULL
            },
        ));

        registrations.push(create_registration(
            DidSaveTextDocument::METHOD,
            TextDocumentSaveRegistrationOptions {
                include_text: Some(false),
                text_document_registration_options: text_document_registration_options.clone(),
            },
        ));

        registrations.push(create_registration(
            DidCloseTextDocument::METHOD,
            &text_document_registration_options,
        ));
    }

    if client_capabilities.completion_dynamic_registration() {
        let registration_options = CompletionRegistrationOptions {
            text_document_registration_options: text_document_registration_options.clone(),
            completion_options: CompletionOptions {
                resolve_provider: Some(false),
                trigger_characters: Some(vec![".".to_string(), ":".to_string()]),
                all_commit_characters: None,
                work_done_progress_options: Default::default(),
                completion_item: None,
            },
        };

        registrations.push(create_registration(Completion::METHOD, registration_options));
    }

    if client_capabilities.execute_command_dynamic_registration() {
        let registration_options = ExecuteCommandRegistrationOptions {
            commands: vec![
                ServerCommand::Reload.as_str().to_string(),
                ServerCommand::ExecuteCodeLens.as_str().to_string(),
            ],
            execute_command_options: ExecuteCommandOptions {
                commands: vec![
                    ServerCommand::Reload.as_str().to_string(),
                    ServerCommand::ExecuteCodeLens.as_str().to_string(),
                ],
                work_done_progress_options: Default::default(),
            },
        };

        registrations.push(create_registration(ExecuteCommand::METHOD, registration_options));
    }

    if client_capabilities.semantic_tokens_dynamic_registration() {
        let registration_options = SemanticTokensRegistrationOptions {
            text_document_registration_options: text_document_registration_options.clone(),
            semantic_tokens_options: SemanticTokensOptions {
                legend: SemanticTokensLegend {
                    token_types: SemanticTokenKind::legend(),
                    token_modifiers: vec![],
                },
                full: Some(SemanticTokensFullOptions::Bool(true)),
                ..SemanticTokensOptions::default()
            },
            static_registration_options: Default::default(),
        };

        registrations
            .push(create_registration("textDocument/semanticTokens", registration_options));
    }

    if client_capabilities.formatting_dynamic_registration() {
        let registration_options = DocumentFormattingRegistrationOptions {
            text_document_registration_options: text_document_registration_options.clone(),
            document_formatting_options: Default::default(),
        };

        registrations.push(create_registration(Formatting::METHOD, registration_options));
    }

    let hover_text_document_registration_options = TextDocumentRegistrationOptions {
        document_selector: Some(vec![
            DocumentFilter {
                language: Some("cairo".to_string()),
                scheme: Some("file".to_string()),
                pattern: None,
            },
            DocumentFilter {
                language: Some("cairo".to_string()),
                scheme: Some("vfs".to_string()),
                pattern: None,
            },
            DocumentFilter {
                language: Some("toml".to_string()),
                scheme: Some("file".into()),
                pattern: Some("**/Scarb.toml".into()),
            },
        ]),
    };

    if client_capabilities.hover_dynamic_registration() {
        let registration_options = HoverRegistrationOptions {
            text_document_registration_options: hover_text_document_registration_options,
            hover_options: Default::default(),
        };

        registrations.push(create_registration(HoverRequest::METHOD, registration_options));
    }

    if client_capabilities.definition_dynamic_registration() {
        let registration_options = DefinitionRegistrationOptions {
            text_document_registration_options: text_document_registration_options.clone(),
            definition_options: DefinitionOptions {
                work_done_progress_options: Default::default(),
            },
        };

        registrations.push(create_registration(GotoDefinition::METHOD, registration_options));
    }

    if client_capabilities.code_action_dynamic_registration() {
        let registration_options = CodeActionRegistrationOptions {
            text_document_registration_options: text_document_registration_options.clone(),
            code_action_options: Default::default(),
        };

        registrations.push(create_registration(CodeActionRequest::METHOD, registration_options));
    }

    if client_capabilities.references_provider_dynamic_registration() {
        registrations.push(create_registration(
            References::METHOD,
            ReferencesRegistrationOptions {
                text_document_registration_options: text_document_registration_options.clone(),
                references_options: ReferencesOptions {
                    work_done_progress_options: Default::default(),
                },
            },
        ));
    }

    if client_capabilities.rename_provider_dynamic_registration() {
        registrations.push(create_registration(
            Rename::METHOD,
            RenameRegistrationOptions {
                text_document_registration_options: text_document_registration_options.clone(),
                rename_options: RenameOptions {
                    prepare_provider: Some(false),
                    work_done_progress_options: Default::default(),
                },
            },
        ));
    }

    if client_capabilities.document_highlight_provider_dynamic_registration() {
        registrations.push(create_registration(
            DocumentHighlightRequest::METHOD,
            DocumentHighlightRegistrationOptions {
                text_document_registration_options: text_document_registration_options.clone(),
                document_highlight_options: DocumentHighlightOptions {
                    work_done_progress_options: Default::default(),
                },
            },
        ));
    }

    if client_capabilities.code_lens_provider_dynamic_registration() {
        registrations.push(create_registration(
            CodeLensRequest::METHOD,
            CodeLensRegistrationOptions {
                text_document_registration_options: text_document_registration_options.clone(),
                code_lens_options: CodeLensOptions { resolve_provider: Some(false) },
            },
        ));
    }

    if client_capabilities.text_document_inlay_hints_dynamic_registration() {
        registrations.push(create_registration(
            InlayHintRequest::METHOD,
            InlayHintRegistrationOptions {
                text_document_registration_options: text_document_registration_options.clone(),
                inlay_hint_options: InlayHintOptions {
                    resolve_provider: Some(false),
                    ..Default::default()
                },
                ..Default::default()
            },
        ));
    }

    registrations.push(create_registration(ViewSyntaxTree::METHOD, ()));

    registrations
}

fn create_registration(method: &str, registration_options: impl Serialize) -> Registration {
    Registration {
        id: method.to_string(),
        method: method.to_string(),
        register_options: Some(serde_json::to_value(registration_options).unwrap()),
    }
}

mod missing_lsp_types {
    use lsp_types::{
        CodeActionOptions, CodeLensOptions, DefinitionOptions, DocumentFormattingOptions,
        DocumentHighlightOptions, ReferencesOptions, RenameOptions,
        TextDocumentRegistrationOptions,
    };
    use serde::{Deserialize, Serialize};

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct DocumentFormattingRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub document_formatting_options: DocumentFormattingOptions,
    }

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct DefinitionRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub definition_options: DefinitionOptions,
    }

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct CodeActionRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub code_action_options: CodeActionOptions,
    }

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct ReferencesRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub references_options: ReferencesOptions,
    }

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct RenameRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub rename_options: RenameOptions,
    }

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct DocumentHighlightRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub document_highlight_options: DocumentHighlightOptions,
    }

    #[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
    #[serde(rename_all = "camelCase")]
    pub struct CodeLensRegistrationOptions {
        #[serde(flatten)]
        pub text_document_registration_options: TextDocumentRegistrationOptions,

        #[serde(flatten)]
        pub code_lens_options: CodeLensOptions,
    }
}