org.aya.lsp.models.HighlightResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ide-lsp Show documentation
The Aya proof assistant
The newest version!
// Copyright (c) 2020-2024 Tesla (Yinsen) Zhang.
// Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
package org.aya.lsp.models;

import kala.collection.Seq;
import kala.control.Option;
import kala.value.TransientVar;
import org.aya.cli.literate.HighlightInfo;
import org.aya.lsp.utils.LspRange;
import org.aya.util.error.SourcePos;
import org.javacs.lsp.Range;
import org.jetbrains.annotations.NotNull;

import java.net.URI;
import java.util.List;

public record HighlightResult(@NotNull URI uri, @NotNull List symbols) {
  public HighlightResult(@NotNull URI uri, @NotNull Seq symbols) {
    this(uri, symbols.asJava());
  }

  public enum Kind {
    // definitions
    ModuleDef, FnDef, DataDef, StructDef, ConDef, FieldDef, PrimDef, GeneralizeDef,
    // expressions
    FnRef, DataRef, StructRef, ConRef, FieldRef, PrimRef, ModuleRef, GeneralizeRef,
  }

  public record Symbol(
    @NotNull Range range,
    @NotNull HighlightResult.Kind kind,
    @NotNull TransientVar sourcePos
  ) {
    public static @NotNull Option from(@NotNull HighlightInfo info) {
      return kindOf(info).map(k -> {
        var pos = info.sourcePos();
        return new Symbol(LspRange.toRange(pos), k, new TransientVar<>(pos));
      });
    }

    private static @NotNull Option kindOf(@NotNull HighlightInfo symbol) {
      return switch (symbol) {
        case HighlightInfo.Def def -> switch (def.kind()) {
          case LocalVar -> Option.none(); // maybe rainbow local variables
          case Generalized -> Option.some(Kind.GeneralizeDef);
          case Module -> Option.some(Kind.ModuleDef);
          case Fn -> Option.some(Kind.FnDef);
          case Data -> Option.some(Kind.DataDef);
          case Clazz -> Option.some(Kind.StructDef);
          case Con -> Option.some(Kind.ConDef);
          case Member -> Option.some(Kind.FieldDef);
          case Prim -> Option.some(Kind.PrimDef);
          case Unknown -> Option.none();
        };
        case HighlightInfo.Ref ref -> switch (ref.kind()) {
          case LocalVar -> Option.none(); // maybe rainbow local variables
          case Generalized -> Option.some(Kind.GeneralizeRef);
          case Module -> Option.some(Kind.ModuleRef);
          case Fn -> Option.some(Kind.FnRef);
          case Data -> Option.some(Kind.DataRef);
          case Clazz -> Option.some(Kind.StructRef);
          case Con -> Option.some(Kind.ConRef);
          case Member -> Option.some(Kind.FieldRef);
          case Prim -> Option.some(Kind.PrimRef);
          case Unknown -> Option.none();
        };
        case HighlightInfo.Lit _, HighlightInfo.Err _ -> Option.none(); // handled by client
        case HighlightInfo.UserMeta _ -> Option.none(); // handled by something else
      };
    }
  }
}