ToolRunner.java

/*
 * Copyright (c) 2024 Christian Stein
 * Licensed under the Universal Permissive License v 1.0 -> https://opensource.org/license/upl
 */

package run.bach;

import java.lang.System.Logger.Level;
import java.util.function.UnaryOperator;

/**
 * A runner of tool calls.
 *
 * <p>Usage example:
 *
 * <pre>{@code
 * ToolRunner.ofSystem().run("java", "--version");
 * }</pre>
 */
@FunctionalInterface
public interface ToolRunner {
  /**
   * {@return an instance of the default tool runner using the given tool finder}
   *
   * @param finder the finder instance to be used for finding tools by name
   */
  static ToolRunner of(ToolFinder finder) {
    return new ToolSpace(finder);
  }

  /** {@return an instance of the default tool runner using the system tool finder} */
  static ToolRunner ofSilence() {
    return new ToolSpace(ToolFinder.ofSystem(), Level.OFF, ToolSpace.Flag.SILENT);
  }

  /** {@return an instance of the default tool runner using the system tool finder} */
  static ToolRunner ofSystem() {
    class SystemRunner {
      static final ToolRunner SINGLETON = ToolRunner.of(ToolFinder.ofSystem());
    }
    return SystemRunner.SINGLETON;
  }

  ToolRun run(ToolCall call);

  default void log(Level level, String message) {
    System.out.printf("[%s] %s".formatted(level.name().charAt(0), message));
  }

  default ToolRun run(Tool tool, String... args) {
    return run(ToolCall.of(tool).addAll(args));
  }

  default ToolRun run(Tool tool, UnaryOperator<ToolCall> args) {
    return run(args.apply(ToolCall.of(tool)));
  }

  default ToolRun run(String tool, String... args) {
    return run(ToolCall.of(tool).addAll(args));
  }

  default ToolRun run(String tool, UnaryOperator<ToolCall> args) {
    return run(args.apply(ToolCall.of(tool)));
  }
}