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

import com.intellij.commandInterface.command.*;
import com.intellij.commandInterface.commandLine.psi.CommandLineFile;
import com.intellij.openapi.Disposable;
import com.intellij.openapi.util.Pair;
import com.intellij.testFramework.fixtures.CodeInsightTestFixture;
import com.intellij.testFramework.fixtures.impl.CodeInsightTestFixtureImpl;
import com.jetbrains.python.PythonTestUtil;
import org.easymock.EasyMock;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.TestOnly;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * Shared tools to testing command line
 *
 * @author Ilya.Kazakevich
 */
final class CommandTestTools {
  @NotNull
  static final String TEST_PATH = PythonTestUtil.getTestDataPath() + "/commandLine/";

  private CommandTestTools() {
  }

  /**
   * <pre>command --available-option=available_argument --option-no-argument positional_argument</pre>
   *
   * @return list cosists of fake command with opts and arguments
   */
  @TestOnly
  @NotNull
  static List<Command> createCommands() {
    //command positional_argument --available-option=available_argument
    final Command command = EasyMock.createMock(Command.class);
    EasyMock.expect(command.getName()).andReturn("command").anyTimes();
    EasyMock.expect(command.getHelp(true)).andReturn(new Help("some_text")).anyTimes();
    EasyMock.expect(command.getHelp(false)).andReturn(new Help("some_text")).anyTimes();
    final List<Option> options = new ArrayList<>();


    final Pair<List<String>, Boolean> argument = Pair.create(Collections.singletonList("available_argument"), true);
    options.add(new Option(Pair.create(1, new Argument(new Help("option argument"), argument)), new Help(""),
                           Collections.emptyList(),
                           Collections.singletonList("--available-option")));


    options.add(new Option(null, new Help(""),
                           Collections.emptyList(),
                           Collections.singletonList("--option-no-argument")));

    EasyMock.expect(command.getOptions()).andReturn(options).anyTimes();


    final ArgumentsInfo argumentInfo = new KnownArgumentsInfo(Collections.singletonList(
      new Argument(new Help("positional_argument"),
                   Pair.create(Collections.singletonList("positional_argument"), true))), 1, 1);

    EasyMock.expect(command.getArgumentsInfo()).andReturn(argumentInfo)
      .anyTimes();

    EasyMock.replay(command);
    return Collections.singletonList(command);
  }

  /**
   * Hack to register file type (not registered for some reason?)
   */
  static void initFileType(@NotNull Disposable parentDisposable) {
    CodeInsightTestFixtureImpl.associateExtensionTemporarily(CommandLineFileType.INSTANCE, CommandLineFileType.EXTENSION, parentDisposable);
  }

  /**
   * Creates command file by text and  {@link #createCommands() fills it with commands}
   * @param testFixture fixture
   * @param text command text
   * @return command file
   * @see #createCommands()
   */
  @NotNull
  static CommandLineFile createFileByText(@NotNull final CodeInsightTestFixture testFixture, @NotNull final String text) {
    final CommandLineFile file =
      (CommandLineFile)testFixture.configureByText(CommandLineFileType.INSTANCE, text);
    file.setCommands(createCommands());
    return file;
  }
}