openide/platform/lang-impl/testSources/com/intellij/codeInsight/highlighting/HeavyBraceHighlighterTest.java

// Copyright 2000-2022 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
package com.intellij.codeInsight.highlighting;

import com.intellij.injected.editor.EditorWindow;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.markup.RangeHighlighter;
import com.intellij.openapi.extensions.ExtensionPoint;
import com.intellij.openapi.fileTypes.FileType;
import com.intellij.openapi.util.Disposer;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiFile;
import com.intellij.psi.util.PsiUtilBase;
import com.intellij.testFramework.FileBasedTestCaseHelper;
import com.intellij.testFramework.LightPlatformCodeInsightTestCase;
import com.intellij.testFramework.TestDataPath;
import com.intellij.testFramework.UsefulTestCase;
import com.intellij.util.Alarm;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;

@RunWith(com.intellij.testFramework.Parameterized.class)
@TestDataPath("/testData/../../../platform/lang-impl/testData/editor/heavyHighlighter/")
public class HeavyBraceHighlighterTest extends LightPlatformCodeInsightTestCase implements FileBasedTestCaseHelper {
  @Override
  protected void setUp() throws Exception {
    super.setUp();

    ExtensionPoint<HeavyBraceHighlighter> point = HeavyBraceHighlighter.EP_NAME.getPoint();
    point.registerExtension(new HeavyHighlighterSample(), getTestRootDisposable());
  }

  @Override
  public @Nullable String getFileSuffix(@NotNull String fileName) {
    return StringUtil.endsWith(fileName, ".txt") ? null : fileName;
  }

  @Override
  public @Nullable String getBaseName(@NotNull String fileAfterSuffix) {
    return StringUtil.endsWith(fileAfterSuffix, ".txt") ? fileAfterSuffix.substring(0, fileAfterSuffix.length() - 4) : null;
  }

  @Test
  public void testAction() {
    configureByFile(myFileSuffix);
    String result = getEditorTextWithHighlightedBraces(getEditor(), getFile());
    UsefulTestCase.assertSameLinesWithFile(getAnswerFilePath(), result);
  }

  /**
   * @see BraceHighlightingHandlerTest
   */
  public static String getEditorTextWithHighlightedBraces(@NotNull Editor editor, @NotNull PsiFile psiFile) {
    Editor hostEditor = editor instanceof EditorWindow ? ((EditorWindow)editor).getDelegate() : editor;
    List<Pair<Integer, String>> markers = new ArrayList<>();
    Alarm alarm = new Alarm();
    try {
      Pair<TextRange, TextRange> match = HeavyBraceHighlighter.match(psiFile, editor.getCaretModel().getOffset());
      if (match != null) {
        FileType fileType = PsiUtilBase.getPsiFileAtOffset(psiFile, match.first.getStartOffset()).getFileType();

        new BraceHighlightingHandler(psiFile.getProject(), editor, alarm, psiFile)
          .highlightBraces(match.first, match.second, true, false, fileType);
      }
      RangeHighlighter[] highlighters = editor.getMarkupModel().getAllHighlighters();
      for (RangeHighlighter highlighter : highlighters) {
        if (highlighter.getLayer() == BraceHighlightingHandler.LAYER) {
          markers.add(Pair.create(highlighter.getStartOffset(), "<brace>"));
          markers.add(Pair.create(highlighter.getEndOffset(), "</brace>"));
        }
      }
    }
    finally {
      Disposer.dispose(alarm);
    }

    hostEditor.getCaretModel().getAllCarets().forEach(it -> markers.add(Pair.create(it.getOffset(), "<caret>")));

    StringBuilder result = new StringBuilder(hostEditor.getDocument().getCharsSequence());
    markers.stream()
      .sorted(Comparator.comparingInt(it -> -it.first))
      .forEach(it -> result.insert(it.first, it.second));
    return result.toString();
  }
}