org.aya.repl.ReplCompleters Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tools-repl Show documentation

The Aya proof assistant

There is a newer version: 0.34.0

// 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.repl;

import kala.collection.Seq;
import org.jetbrains.annotations.NotNull;
import org.jline.reader.Candidate;
import org.jline.reader.Completer;
import org.jline.reader.LineReader;
import org.jline.reader.ParsedLine;

import java.util.List;
import java.util.function.Supplier;

/**
 * Basic completer for {@link Command}
 * @see Command
 * @see CommandManager
 */
public interface ReplCompleters {
  @NotNull Completer BOOL = (_, _, candidates) -> {
    candidates.add(new Candidate("true"));
    candidates.add(new Candidate("false"));
  };

  record EnumCompleter>(@NotNull Class enumClass) implements Completer {
    @Override public void complete(LineReader reader, ParsedLine line, List candidates) {
      Seq.of(enumClass.getEnumConstants()).view()
        .map(Enum::name)
        .map(Candidate::new)
        .forEach(candidates::add);
    }
  }

  record Help(@NotNull Supplier commandManager) implements Completer {
    @Override public void complete(LineReader reader, ParsedLine line, List candidates) {
      commandManager.get().cmd.view().map(CommandManager.CommandGen::owner)
        .flatMap(Command::names).map(Candidate::new).forEach(candidates::add);
    }
  }
}