| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448 |
- <?php
- namespace Laravel\Prompts;
- use Closure;
- use Laravel\Prompts\Exceptions\FormRevertedException;
- use Laravel\Prompts\Output\ConsoleOutput;
- use Laravel\Prompts\Support\Result;
- use RuntimeException;
- use Symfony\Component\Console\Output\OutputInterface;
- use Throwable;
- abstract class Prompt
- {
- use Concerns\Colors;
- use Concerns\Cursor;
- use Concerns\Erase;
- use Concerns\Events;
- use Concerns\FakesInputOutput;
- use Concerns\Fallback;
- use Concerns\Interactivity;
- use Concerns\Themes;
- /**
- * The current state of the prompt.
- */
- public string $state = 'initial';
- /**
- * The error message from the validator.
- */
- public string $error = '';
- /**
- * The cancel message displayed when this prompt is cancelled.
- */
- public string $cancelMessage = 'Cancelled.';
- /**
- * The previously rendered frame.
- */
- protected string $prevFrame = '';
- /**
- * How many new lines were written by the last output.
- */
- protected int $newLinesWritten = 1;
- /**
- * Whether user input is required.
- */
- public bool|string $required;
- /**
- * The transformation callback.
- */
- public ?Closure $transform = null;
- /**
- * The validator callback or rules.
- */
- public mixed $validate;
- /**
- * The cancellation callback.
- */
- protected static ?Closure $cancelUsing;
- /**
- * Indicates if the prompt has been validated.
- */
- protected bool $validated = false;
- /**
- * The custom validation callback.
- */
- protected static ?Closure $validateUsing;
- /**
- * The revert handler from the StepBuilder.
- */
- protected static ?Closure $revertUsing = null;
- /**
- * The output instance.
- */
- protected static OutputInterface $output;
- /**
- * The terminal instance.
- */
- protected static Terminal $terminal;
- /**
- * Get the value of the prompt.
- */
- abstract public function value(): mixed;
- /**
- * Render the prompt and listen for input.
- */
- public function prompt(): mixed
- {
- try {
- $this->capturePreviousNewLines();
- if (static::shouldFallback()) {
- return $this->fallback();
- }
- static::$interactive ??= stream_isatty(STDIN);
- if (! static::$interactive) {
- return $this->default();
- }
- $this->checkEnvironment();
- try {
- static::terminal()->setTty('-icanon -isig -echo');
- } catch (Throwable $e) {
- static::output()->writeln("<comment>{$e->getMessage()}</comment>");
- static::fallbackWhen(true);
- return $this->fallback();
- }
- $this->hideCursor();
- $this->render();
- $result = $this->runLoop(function (string $key): ?Result {
- $continue = $this->handleKeyPress($key);
- $this->render();
- if ($continue === false || $key === Key::CTRL_C) {
- if ($key === Key::CTRL_C) {
- if (isset(static::$cancelUsing)) {
- return Result::from((static::$cancelUsing)());
- } else {
- static::terminal()->exit();
- }
- }
- if ($key === Key::CTRL_U && self::$revertUsing) {
- throw new FormRevertedException;
- }
- return Result::from($this->transformedValue());
- }
- // Continue looping.
- return null;
- });
- return $result;
- } finally {
- $this->clearListeners();
- }
- }
- /**
- * Implementation of the prompt looping mechanism.
- *
- * @param callable(string $key): ?Result $callable
- */
- public function runLoop(callable $callable): mixed
- {
- while (($key = static::terminal()->read()) !== null) {
- /**
- * If $key is an empty string, Terminal::read
- * has failed. We can continue to the next
- * iteration of the loop, and try again.
- */
- if ($key === '') {
- continue;
- }
- $result = $callable($key);
- if ($result instanceof Result) {
- return $result->value;
- }
- }
- }
- /**
- * Register a callback to be invoked when a user cancels a prompt.
- */
- public static function cancelUsing(?Closure $callback): void
- {
- static::$cancelUsing = $callback;
- }
- /**
- * How many new lines were written by the last output.
- */
- public function newLinesWritten(): int
- {
- return $this->newLinesWritten;
- }
- /**
- * Capture the number of new lines written by the last output.
- */
- protected function capturePreviousNewLines(): void
- {
- $this->newLinesWritten = method_exists(static::output(), 'newLinesWritten')
- ? static::output()->newLinesWritten()
- : 1;
- }
- /**
- * Set the output instance.
- */
- public static function setOutput(OutputInterface $output): void
- {
- self::$output = $output;
- }
- /**
- * Get the current output instance.
- */
- protected static function output(): OutputInterface
- {
- return self::$output ??= new ConsoleOutput;
- }
- /**
- * Write output directly, bypassing newline capture.
- */
- protected static function writeDirectly(string $message): void
- {
- match (true) {
- method_exists(static::output(), 'writeDirectly') => static::output()->writeDirectly($message),
- method_exists(static::output(), 'getOutput') => static::output()->getOutput()->write($message),
- default => static::output()->write($message),
- };
- }
- /**
- * Get the terminal instance.
- */
- public static function terminal(): Terminal
- {
- return static::$terminal ??= new Terminal;
- }
- /**
- * Set the custom validation callback.
- */
- public static function validateUsing(Closure $callback): void
- {
- static::$validateUsing = $callback;
- }
- /**
- * Revert the prompt using the given callback.
- *
- * @internal
- */
- public static function revertUsing(Closure $callback): void
- {
- static::$revertUsing = $callback;
- }
- /**
- * Clear any previous revert callback.
- *
- * @internal
- */
- public static function preventReverting(): void
- {
- static::$revertUsing = null;
- }
- /**
- * Render the prompt.
- */
- protected function render(): void
- {
- $this->terminal()->initDimensions();
- $frame = $this->renderTheme();
- if ($frame === $this->prevFrame) {
- return;
- }
- if ($this->state === 'initial') {
- static::output()->write($frame);
- $this->state = 'active';
- $this->prevFrame = $frame;
- return;
- }
- $terminalHeight = $this->terminal()->lines();
- $previousFrameHeight = count(explode(PHP_EOL, $this->prevFrame));
- $renderableLines = array_slice(explode(PHP_EOL, $frame), abs(min(0, $terminalHeight - $previousFrameHeight)));
- $this->moveCursorToColumn(1);
- $this->moveCursorUp(min($terminalHeight, $previousFrameHeight) - 1);
- $this->eraseDown();
- $this->output()->write(implode(PHP_EOL, $renderableLines));
- $this->prevFrame = $frame;
- }
- /**
- * Submit the prompt.
- */
- protected function submit(): void
- {
- $this->validate($this->transformedValue());
- if ($this->state !== 'error') {
- $this->state = 'submit';
- }
- }
- /**
- * Handle a key press and determine whether to continue.
- */
- private function handleKeyPress(string $key): bool
- {
- if ($this->state === 'error') {
- $this->state = 'active';
- }
- $this->emit('key', $key);
- if ($this->state === 'submit') {
- return false;
- }
- if ($key === Key::CTRL_U) {
- if (! self::$revertUsing) {
- $this->state = 'error';
- $this->error = 'This cannot be reverted.';
- return true;
- }
- $this->state = 'cancel';
- $this->cancelMessage = 'Reverted.';
- call_user_func(self::$revertUsing);
- return false;
- }
- if ($key === Key::CTRL_C) {
- $this->state = 'cancel';
- return false;
- }
- if ($this->validated) {
- $this->validate($this->transformedValue());
- }
- return true;
- }
- /**
- * Transform the input.
- */
- private function transform(mixed $value): mixed
- {
- if (is_null($this->transform)) {
- return $value;
- }
- return call_user_func($this->transform, $value);
- }
- /**
- * Get the transformed value of the prompt.
- */
- protected function transformedValue(): mixed
- {
- return $this->transform($this->value());
- }
- /**
- * Validate the input.
- */
- private function validate(mixed $value): void
- {
- $this->validated = true;
- if ($this->required !== false && $this->isInvalidWhenRequired($value)) {
- $this->state = 'error';
- $this->error = is_string($this->required) && strlen($this->required) > 0 ? $this->required : 'Required.';
- return;
- }
- if (! isset($this->validate) && ! isset(static::$validateUsing)) {
- return;
- }
- $error = match (true) {
- is_callable($this->validate) => ($this->validate)($value),
- isset(static::$validateUsing) => (static::$validateUsing)($this),
- default => throw new RuntimeException('The validation logic is missing.'),
- };
- if (! is_string($error) && ! is_null($error)) {
- throw new RuntimeException('The validator must return a string or null.');
- }
- if (is_string($error) && strlen($error) > 0) {
- $this->state = 'error';
- $this->error = $error;
- }
- }
- /**
- * Determine whether the given value is invalid when the prompt is required.
- */
- protected function isInvalidWhenRequired(mixed $value): bool
- {
- return $value === '' || $value === [] || $value === false || $value === null;
- }
- /**
- * Check whether the environment can support the prompt.
- */
- private function checkEnvironment(): void
- {
- if (PHP_OS_FAMILY === 'Windows') {
- throw new RuntimeException('Prompts is not currently supported on Windows. Please use WSL or configure a fallback.');
- }
- }
- /**
- * Restore the cursor and terminal state.
- */
- public function __destruct()
- {
- $this->restoreCursor();
- static::terminal()->restoreTty();
- }
- }
|