Startwind\Forrest\CliCommand\Ai\AskCommand

Complexity

Total Complexity

15

Size/Duplication

Total Lines	91
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	configure()	0	7	1
A	formatCliText()	0	21	4
B	doExecute()	0	52	10

<?php

namespace Startwind\Forrest\CliCommand\Ai;

use Startwind\Forrest\CliCommand\Command\CommandCommand;
use Startwind\Forrest\Util\OutputHelper;
use Symfony\Component\Console\Command\Command as SymfonyCommand;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Startwind\Forrest\Command\Answer\Answer;
use Symfony\Component\Console\Question\Question;

class AskCommand extends CommandCommand
{
    public const COMMAND_NAME = 'ai:ask';

    protected static $defaultName = self::COMMAND_NAME;
    protected static $defaultDescription = 'Suggest an answer to a question.';

    protected function configure(): void
    {
        $this->addArgument('question', InputArgument::IS_ARRAY, 'The question you want to have answered.', []);
        $this->addOption('force', null, InputOption::VALUE_NONE, 'Run the command without asking for permission.');
        $this->addOption('silent', null, InputOption::VALUE_NONE, 'Run the command without explanation.');

        parent::configure();
    }

    protected function doExecute(InputInterface $input, OutputInterface $output): int
    {
        $this->enrichRepositories();

        $silent = $input->getOption('silent');

        if (!$silent) {
            \Startwind\Forrest\Output\OutputHelper::renderHeader($output);
        }

        $aiQuestion = trim(implode(' ', $input->getArgument('question')));

        if ($aiQuestion == 'how') {
            OutputHelper::writeErrorBox($output, ["Please provide a question."]);
            return SymfonyCommand::FAILURE;
        }

        if ($aiQuestion && !str_ends_with($aiQuestion, '?')) {
            $aiQuestion = $aiQuestion . '?';
        }

        /** @var \Symfony\Component\Console\Helper\QuestionHelper $questionHelper */
        $questionHelper = $this->getHelper('question');

        if (!$aiQuestion) {
            OutputHelper::writeInfoBox($output, ["Hi, I am Forrest, your AI command line helper. How can I help you?"]);
            $output->writeln('');
            $aiQuestion = $questionHelper->ask($input, $output, new Question('  Your question: '));
            $output->writeln(['', '']);
        } else {
            if (!$silent) {
                OutputHelper::writeInfoBox($output, [
                    "Question: " . ucfirst($aiQuestion)
                ]);
            }
        }

        $answers = $this->getRepositoryCollection()->ask($aiQuestion);

        foreach ($answers as $repoAnswers) {
            foreach ($repoAnswers as $answer) {
                /** @var Answer $answer */
                $output->writeln(OutputHelper::indentText($this->formatCliText($answer->getAnswer())));
                $command = $answer->getCommand();

                if ($command->getPrompt()) {
                    return $this->runCommand($answer->getCommand(), []);
                }
            }
        }

        return SymfonyCommand::SUCCESS;
    }

    private function formatCliText(string $text): string
    {
        preg_match_all('#```shell((.|\n)*?)```#', $text, $matches);

        if (count($matches[1]) == 1) {
            $shell = $matches[1][0];

            $shellNew = implode("\n", OutputHelper::indentText(trim($shell), 2, 100, ' | '));

            $text = str_replace($matches[0][0], $shellNew, $text);
        }

        preg_match_all('#`(.*)`#', $text, $matches);

        if (count($matches[1]) > 0) {
            foreach ($matches[0] as $key => $match) {
                $text = str_replace($match, '<options=bold>' . $matches[1][$key] . '</>', $text);
            }
        }

        return $text;
    }
}