com.github._1c_syntax.bsl.languageserver.providers.CodeActionProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bsl-language-server Show documentation

Language Server Protocol implementation for 1C (BSL) - 1C:Enterprise 8 and OneScript languages.

There is a newer version: 0.24.0-rc.1

/*
 * This file is a part of BSL Language Server.
 *
 * Copyright (c) 2018-2022
 * Alexey Sosnoviy , Nikita Fedkin  and contributors
 *
 * SPDX-License-Identifier: LGPL-3.0-or-later
 *
 * BSL Language Server is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 *
 * BSL Language Server is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with BSL Language Server.
 */
package com.github._1c_syntax.bsl.languageserver.providers;

import com.github._1c_syntax.bsl.languageserver.codeactions.CodeActionSupplier;
import com.github._1c_syntax.bsl.languageserver.context.DocumentContext;
import lombok.RequiredArgsConstructor;
import org.eclipse.lsp4j.CodeAction;
import org.eclipse.lsp4j.CodeActionKind;
import org.eclipse.lsp4j.CodeActionParams;
import org.eclipse.lsp4j.Command;
import org.eclipse.lsp4j.Diagnostic;
import org.eclipse.lsp4j.TextEdit;
import org.eclipse.lsp4j.WorkspaceEdit;
import org.eclipse.lsp4j.jsonrpc.messages.Either;
import org.springframework.stereotype.Component;

import java.net.URI;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

@Component
@RequiredArgsConstructor
public final class CodeActionProvider {

  private final List codeActionSuppliers;

  public static List createCodeActions(
    List textEdits,
    String title,
    URI uri,
    List diagnostics
  ) {

    if (diagnostics.isEmpty()) {
      return Collections.emptyList();
    }

    WorkspaceEdit edit = new WorkspaceEdit();

    Map> changes = new HashMap<>();
    changes.put(uri.toString(), textEdits);
    edit.setChanges(changes);

    if (diagnostics.size() > 1) {
      title = "Fix all: " + title;
    }

    CodeAction codeAction = new CodeAction(title);
    codeAction.setDiagnostics(diagnostics);
    codeAction.setEdit(edit);
    codeAction.setKind(CodeActionKind.QuickFix);
    if (diagnostics.size() == 1) {
      codeAction.setIsPreferred(Boolean.TRUE);
    }

    return Collections.singletonList(codeAction);

  }

  public List> getCodeActions(
    CodeActionParams params,
    DocumentContext documentContext
  ) {

    List only = Optional.ofNullable(params.getContext().getOnly())
      .orElse(Collections.emptyList());

    return codeActionSuppliers.stream()
      .flatMap(codeActionSupplier -> codeActionSupplier.getCodeActions(params, documentContext).stream())
      .filter(codeAction -> only.isEmpty() || only.contains(codeAction.getKind()))
      .map(Either::forRight)
      .collect(Collectors.toList());
  }

}