com.github._1c_syntax.bsl.languageserver.diagnostics.CognitiveComplexityDiagnostic Maven / Gradle / Ivy

Go to download
/*
 * 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.diagnostics;

import com.github._1c_syntax.bsl.languageserver.context.computer.ComplexitySecondaryLocation;
import com.github._1c_syntax.bsl.languageserver.context.symbol.MethodSymbol;
import com.github._1c_syntax.bsl.languageserver.diagnostics.metadata.DiagnosticMetadata;
import com.github._1c_syntax.bsl.languageserver.diagnostics.metadata.DiagnosticParameter;
import com.github._1c_syntax.bsl.languageserver.diagnostics.metadata.DiagnosticSeverity;
import com.github._1c_syntax.bsl.languageserver.diagnostics.metadata.DiagnosticTag;
import com.github._1c_syntax.bsl.languageserver.diagnostics.metadata.DiagnosticType;
import com.github._1c_syntax.bsl.languageserver.utils.Ranges;
import com.github._1c_syntax.bsl.languageserver.utils.RelatedInformation;
import com.github._1c_syntax.bsl.parser.BSLParser;
import com.github._1c_syntax.bsl.parser.BSLParserRuleContext;
import org.antlr.v4.runtime.tree.ParseTree;
import org.eclipse.lsp4j.DiagnosticRelatedInformation;

import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

@DiagnosticMetadata(
  type = DiagnosticType.CODE_SMELL,
  severity = DiagnosticSeverity.CRITICAL,
  minutesToFix = 15,
  tags = {
    DiagnosticTag.BRAINOVERLOAD
  }
)
public class CognitiveComplexityDiagnostic extends AbstractVisitorDiagnostic {

  private static final int COMPLEXITY_THRESHOLD = 15;
  private static final boolean CHECK_MODULE_BODY = true;

  @DiagnosticParameter(
    type = Integer.class,
    defaultValue = "" + COMPLEXITY_THRESHOLD
  )
  private int complexityThreshold = COMPLEXITY_THRESHOLD;

  @DiagnosticParameter(
    type = Boolean.class,
    defaultValue = "" + CHECK_MODULE_BODY
  )
  private boolean checkModuleBody = CHECK_MODULE_BODY;

  private boolean fileCodeBlockChecked;

  private List makeRelations(MethodSymbol methodSymbol, Integer methodComplexity) {
    List relatedInformation = new ArrayList<>();

    relatedInformation.add(RelatedInformation.create(
      documentContext.getUri(),
      methodSymbol.getSubNameRange(),
      info.getMessage(methodSymbol.getName(), methodComplexity, complexityThreshold)
    ));

    List secondaryLocations =
      documentContext.getCognitiveComplexityData().getMethodsComplexitySecondaryLocations().get(methodSymbol);

    secondaryLocations.stream()
      .map((ComplexitySecondaryLocation secondaryLocation) ->
        RelatedInformation.create(
          documentContext.getUri(),
          secondaryLocation.getRange(),
          secondaryLocation.getMessage()
        )
      )
      .collect(Collectors.toCollection(() -> relatedInformation));

    return relatedInformation;
  }

  @Override
  public ParseTree visitSub(BSLParser.SubContext ctx) {
    Optional optionalMethodSymbol = documentContext.getSymbolTree().getMethodSymbol(ctx);
    optionalMethodSymbol.ifPresent((MethodSymbol methodSymbol) -> {
      Integer methodComplexity = documentContext.getCognitiveComplexityData().getMethodsComplexity().get(methodSymbol);

      if (methodComplexity > complexityThreshold) {

        List relatedInformation = new ArrayList<>();

        relatedInformation.add(RelatedInformation.create(
          documentContext.getUri(),
          methodSymbol.getSubNameRange(),
          info.getMessage(methodSymbol.getName(), methodComplexity, complexityThreshold)
        ));

        List secondaryLocations =
          documentContext.getCognitiveComplexityData().getMethodsComplexitySecondaryLocations().get(methodSymbol);

        secondaryLocations.stream()
          .map((ComplexitySecondaryLocation secondaryLocation) ->
            RelatedInformation.create(
              documentContext.getUri(),
              secondaryLocation.getRange(),
              secondaryLocation.getMessage()
            )
          )
          .collect(Collectors.toCollection(() -> relatedInformation));

        diagnosticStorage.addDiagnostic(
          methodSymbol.getSubNameRange(),
          info.getMessage(methodSymbol.getName(), methodComplexity, complexityThreshold),
          makeRelations(methodSymbol, methodComplexity)
        );
      }
    });
    return ctx;
  }

  @Override
  public ParseTree visitFileCodeBlockBeforeSub(BSLParser.FileCodeBlockBeforeSubContext ctx) {
    checkFileCodeBlock(ctx);
    fileCodeBlockChecked = ctx.getChildCount() > 0;
    return ctx;
  }

  @Override
  public ParseTree visitFileCodeBlock(BSLParser.FileCodeBlockContext ctx) {
    if (!fileCodeBlockChecked) {
      checkFileCodeBlock(ctx);
    }
    return ctx;
  }

  private void checkFileCodeBlock(BSLParserRuleContext ctx) {
    if (!checkModuleBody) {
      return;
    }

    Integer fileCodeBlockComplexity = documentContext.getCognitiveComplexityData().getFileCodeBlockComplexity();

    if (fileCodeBlockComplexity > complexityThreshold) {

      List relatedInformation = new ArrayList<>();

      relatedInformation.add(RelatedInformation.create(
        documentContext.getUri(),
        Ranges.create(ctx.getStart()),
        info.getMessage("body", fileCodeBlockComplexity, complexityThreshold)
      ));

      List secondaryLocations =
        documentContext.getCognitiveComplexityData().getFileBlockComplexitySecondaryLocations();

      secondaryLocations.stream()
        .map((ComplexitySecondaryLocation secondaryLocation) ->
          RelatedInformation.create(
            documentContext.getUri(),
            secondaryLocation.getRange(),
            secondaryLocation.getMessage()
          )
        )
        .collect(Collectors.toCollection(() -> relatedInformation));

      diagnosticStorage.addDiagnostic(
        ctx.getStart(),
        info.getMessage("body", fileCodeBlockComplexity, complexityThreshold),
        relatedInformation
      );
    }
  }

}