com.github._1c_syntax.bsl.languageserver.utils.MdoRefBuilder 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.utils;

import com.github._1c_syntax.bsl.languageserver.context.DocumentContext;
import com.github._1c_syntax.bsl.parser.BSLParser;
import com.github._1c_syntax.bsl.types.MDOType;
import com.github._1c_syntax.bsl.types.MdoReference;
import com.github._1c_syntax.bsl.types.ModuleType;
import com.github._1c_syntax.mdclasses.mdo.AbstractMDObjectBase;
import com.github._1c_syntax.mdclasses.mdo.MDCommonModule;
import com.github._1c_syntax.mdclasses.utils.MDOUtils;
import com.github._1c_syntax.utils.StringInterner;
import lombok.experimental.UtilityClass;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.TerminalNode;

import javax.annotation.Nullable;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.atomic.AtomicReference;

@UtilityClass
public class MdoRefBuilder {

  private final StringInterner stringInterner = new StringInterner();

  public String getMdoRef(DocumentContext documentContext, BSLParser.CallStatementContext callStatement) {
    if (callStatement.globalMethodCall() != null) {
      return getMdoRef(documentContext);
    } else {
      return getMdoRef(documentContext, callStatement.IDENTIFIER(), callStatement.modifier());
    }
  }

  public static String getMdoRef(DocumentContext documentContext) {
    var mdoRef = documentContext.getMdObject()
      .map(AbstractMDObjectBase::getMdoReference)
      .map(MdoReference::getMdoRef)
      .orElseGet(() -> documentContext.getUri().toString());
    return stringInterner.intern(mdoRef);
  }

  public String getMdoRef(DocumentContext documentContext, BSLParser.ComplexIdentifierContext complexIdentifier) {
    return getMdoRef(documentContext, complexIdentifier.IDENTIFIER(), complexIdentifier.modifier());
  }

  public String getMdoRef(
    DocumentContext documentContext,
    @Nullable
      TerminalNode identifier,
    List modifiers
  ) {

    AtomicReference mdoRef = new AtomicReference<>("");

    Optional.ofNullable(identifier)
      .map(ParseTree::getText)
      .flatMap(commonModuleName -> getCommonModuleMdoRef(documentContext, commonModuleName))
      .or(() ->
        Optional.ofNullable(identifier)
          .map(ParseTree::getText)
          .flatMap(MDOType::fromValue)
          .filter(mdoType -> MDOUtils.getModuleTypesForMdoTypes()
            .getOrDefault(mdoType, Collections.emptySet())
            .contains(ModuleType.ManagerModule))
          .map(mdoType -> getMdoRef(mdoType, getMdoName(modifiers)))
      )
      .ifPresent(mdoRef::set);

    return stringInterner.intern(mdoRef.get());
  }

  private Optional getCommonModuleMdoRef(DocumentContext documentContext, String commonModuleName) {
    return documentContext.getServerContext()
      .getConfiguration()
      .getCommonModule(commonModuleName)
      .map(MDCommonModule::getMdoReference)
      .map(MdoReference::getMdoRef);
  }

  private String getMdoRef(MDOType mdoType, String identifier) {
    if (identifier.isEmpty()) {
      return "";
    }

    return mdoType.getName() + "." + identifier;
  }

  private String getMdoName(List modifiers) {
    return modifiers.stream()
      .limit(1)
      .findAny()
      .map(BSLParser.ModifierContext::accessProperty)
      .map(BSLParser.AccessPropertyContext::IDENTIFIER)
      .map(ParseTree::getText)
      .orElse("");
  }
}