123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617 |
- <?php
- /*
- * This file is part of Psy Shell.
- *
- * (c) 2012-2023 Justin Hileman
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Psy;
- use Psy\CodeCleaner\NoReturnValue;
- use Psy\Exception\BreakException;
- use Psy\Exception\ErrorException;
- use Psy\Exception\Exception as PsyException;
- use Psy\Exception\RuntimeException;
- use Psy\Exception\ThrowUpException;
- use Psy\ExecutionLoop\ProcessForker;
- use Psy\ExecutionLoop\RunkitReloader;
- use Psy\Formatter\TraceFormatter;
- use Psy\Input\ShellInput;
- use Psy\Input\SilentInput;
- use Psy\Output\ShellOutput;
- use Psy\TabCompletion\Matcher;
- use Psy\VarDumper\PresenterAware;
- use Symfony\Component\Console\Application;
- use Symfony\Component\Console\Command\Command as BaseCommand;
- use Symfony\Component\Console\Exception\ExceptionInterface as SymfonyConsoleException;
- use Symfony\Component\Console\Formatter\OutputFormatter;
- use Symfony\Component\Console\Input\ArrayInput;
- use Symfony\Component\Console\Input\InputArgument;
- use Symfony\Component\Console\Input\InputDefinition;
- use Symfony\Component\Console\Input\InputInterface;
- use Symfony\Component\Console\Input\InputOption;
- use Symfony\Component\Console\Input\StringInput;
- use Symfony\Component\Console\Output\ConsoleOutput;
- use Symfony\Component\Console\Output\OutputInterface;
- /**
- * The Psy Shell application.
- *
- * Usage:
- *
- * $shell = new Shell;
- * $shell->run();
- *
- * @author Justin Hileman <justin@justinhileman.info>
- */
- class Shell extends Application
- {
- const VERSION = 'v0.12.0';
- private $config;
- private $cleaner;
- private $output;
- private $originalVerbosity;
- private $readline;
- private $inputBuffer;
- private $code;
- private $codeBuffer;
- private $codeBufferOpen;
- private $codeStack;
- private $stdoutBuffer;
- private $context;
- private $includes;
- private $outputWantsNewline = false;
- private $loopListeners;
- private $autoCompleter;
- private $matchers = [];
- private $commandsMatcher;
- private $lastExecSuccess = true;
- private $nonInteractive = false;
- private $errorReporting;
- /**
- * Create a new Psy Shell.
- *
- * @param Configuration|null $config (default: null)
- */
- public function __construct(Configuration $config = null)
- {
- $this->config = $config ?: new Configuration();
- $this->cleaner = $this->config->getCodeCleaner();
- $this->context = new Context();
- $this->includes = [];
- $this->readline = $this->config->getReadline();
- $this->inputBuffer = [];
- $this->codeStack = [];
- $this->stdoutBuffer = '';
- $this->loopListeners = $this->getDefaultLoopListeners();
- parent::__construct('Psy Shell', self::VERSION);
- $this->config->setShell($this);
- // Register the current shell session's config with \Psy\info
- \Psy\info($this->config);
- }
- /**
- * Check whether the first thing in a backtrace is an include call.
- *
- * This is used by the psysh bin to decide whether to start a shell on boot,
- * or to simply autoload the library.
- */
- public static function isIncluded(array $trace): bool
- {
- $isIncluded = isset($trace[0]['function']) &&
- \in_array($trace[0]['function'], ['require', 'include', 'require_once', 'include_once']);
- // Detect Composer PHP bin proxies.
- if ($isIncluded && \array_key_exists('_composer_autoload_path', $GLOBALS) && \preg_match('{[\\\\/]psysh$}', $trace[0]['file'])) {
- // If we're in a bin proxy, we'll *always* see one include, but we
- // care if we see a second immediately after that.
- return isset($trace[1]['function']) &&
- \in_array($trace[1]['function'], ['require', 'include', 'require_once', 'include_once']);
- }
- return $isIncluded;
- }
- /**
- * Check if the currently running PsySH bin is a phar archive.
- */
- public static function isPhar(): bool
- {
- return \class_exists("\Phar") && \Phar::running() !== '' && \strpos(__FILE__, \Phar::running(true)) === 0;
- }
- /**
- * Invoke a Psy Shell from the current context.
- *
- * @see Psy\debug
- * @deprecated will be removed in 1.0. Use \Psy\debug instead
- *
- * @param array $vars Scope variables from the calling context (default: [])
- * @param object|string $bindTo Bound object ($this) or class (self) value for the shell
- *
- * @return array Scope variables from the debugger session
- */
- public static function debug(array $vars = [], $bindTo = null): array
- {
- @\trigger_error('`Psy\\Shell::debug` is deprecated; call `Psy\\debug` instead.', \E_USER_DEPRECATED);
- return \Psy\debug($vars, $bindTo);
- }
- /**
- * Adds a command object.
- *
- * {@inheritdoc}
- *
- * @param BaseCommand $command A Symfony Console Command object
- *
- * @return BaseCommand The registered command
- */
- public function add(BaseCommand $command): BaseCommand
- {
- if ($ret = parent::add($command)) {
- if ($ret instanceof ContextAware) {
- $ret->setContext($this->context);
- }
- if ($ret instanceof PresenterAware) {
- $ret->setPresenter($this->config->getPresenter());
- }
- if (isset($this->commandsMatcher)) {
- $this->commandsMatcher->setCommands($this->all());
- }
- }
- return $ret;
- }
- /**
- * Gets the default input definition.
- *
- * @return InputDefinition An InputDefinition instance
- */
- protected function getDefaultInputDefinition(): InputDefinition
- {
- return new InputDefinition([
- new InputArgument('command', InputArgument::REQUIRED, 'The command to execute'),
- new InputOption('--help', '-h', InputOption::VALUE_NONE, 'Display this help message.'),
- ]);
- }
- /**
- * Gets the default commands that should always be available.
- *
- * @return array An array of default Command instances
- */
- protected function getDefaultCommands(): array
- {
- $sudo = new Command\SudoCommand();
- $sudo->setReadline($this->readline);
- $hist = new Command\HistoryCommand();
- $hist->setReadline($this->readline);
- return [
- new Command\HelpCommand(),
- new Command\ListCommand(),
- new Command\DumpCommand(),
- new Command\DocCommand(),
- new Command\ShowCommand(),
- new Command\WtfCommand(),
- new Command\WhereamiCommand(),
- new Command\ThrowUpCommand(),
- new Command\TimeitCommand(),
- new Command\TraceCommand(),
- new Command\BufferCommand(),
- new Command\ClearCommand(),
- new Command\EditCommand($this->config->getRuntimeDir()),
- // new Command\PsyVersionCommand(),
- $sudo,
- $hist,
- new Command\ExitCommand(),
- ];
- }
- /**
- * @return Matcher\AbstractMatcher[]
- */
- protected function getDefaultMatchers(): array
- {
- // Store the Commands Matcher for later. If more commands are added,
- // we'll update the Commands Matcher too.
- $this->commandsMatcher = new Matcher\CommandsMatcher($this->all());
- return [
- $this->commandsMatcher,
- new Matcher\KeywordsMatcher(),
- new Matcher\VariablesMatcher(),
- new Matcher\ConstantsMatcher(),
- new Matcher\FunctionsMatcher(),
- new Matcher\ClassNamesMatcher(),
- new Matcher\ClassMethodsMatcher(),
- new Matcher\ClassAttributesMatcher(),
- new Matcher\ObjectMethodsMatcher(),
- new Matcher\ObjectAttributesMatcher(),
- new Matcher\ClassMethodDefaultParametersMatcher(),
- new Matcher\ObjectMethodDefaultParametersMatcher(),
- new Matcher\FunctionDefaultParametersMatcher(),
- ];
- }
- /**
- * Gets the default command loop listeners.
- *
- * @return array An array of Execution Loop Listener instances
- */
- protected function getDefaultLoopListeners(): array
- {
- $listeners = [];
- if (ProcessForker::isSupported() && $this->config->usePcntl()) {
- $listeners[] = new ProcessForker();
- }
- if (RunkitReloader::isSupported()) {
- $listeners[] = new RunkitReloader();
- }
- return $listeners;
- }
- /**
- * Add tab completion matchers.
- *
- * @param array $matchers
- */
- public function addMatchers(array $matchers)
- {
- $this->matchers = \array_merge($this->matchers, $matchers);
- if (isset($this->autoCompleter)) {
- $this->addMatchersToAutoCompleter($matchers);
- }
- }
- /**
- * @deprecated Call `addMatchers` instead
- *
- * @param array $matchers
- */
- public function addTabCompletionMatchers(array $matchers)
- {
- @\trigger_error('`addTabCompletionMatchers` is deprecated; call `addMatchers` instead.', \E_USER_DEPRECATED);
- $this->addMatchers($matchers);
- }
- /**
- * Set the Shell output.
- *
- * @param OutputInterface $output
- */
- public function setOutput(OutputInterface $output)
- {
- $this->output = $output;
- $this->originalVerbosity = $output->getVerbosity();
- }
- /**
- * Runs PsySH.
- *
- * @param InputInterface|null $input An Input instance
- * @param OutputInterface|null $output An Output instance
- *
- * @return int 0 if everything went fine, or an error code
- */
- public function run(InputInterface $input = null, OutputInterface $output = null): int
- {
- // We'll just ignore the input passed in, and set up our own!
- $input = new ArrayInput([]);
- if ($output === null) {
- $output = $this->config->getOutput();
- }
- $this->setAutoExit(false);
- $this->setCatchExceptions(false);
- try {
- return parent::run($input, $output);
- } catch (\Throwable $e) {
- $this->writeException($e);
- }
- return 1;
- }
- /**
- * Runs PsySH.
- *
- * @throws \Throwable if thrown via the `throw-up` command
- *
- * @param InputInterface $input An Input instance
- * @param OutputInterface $output An Output instance
- *
- * @return int 0 if everything went fine, or an error code
- */
- public function doRun(InputInterface $input, OutputInterface $output): int
- {
- $this->setOutput($output);
- $this->resetCodeBuffer();
- if ($input->isInteractive()) {
- // @todo should it be possible to have raw output in an interactive run?
- return $this->doInteractiveRun();
- } else {
- return $this->doNonInteractiveRun($this->config->rawOutput());
- }
- }
- /**
- * Run PsySH in interactive mode.
- *
- * Initializes tab completion and readline history, then spins up the
- * execution loop.
- *
- * @throws \Throwable if thrown via the `throw-up` command
- *
- * @return int 0 if everything went fine, or an error code
- */
- private function doInteractiveRun(): int
- {
- $this->initializeTabCompletion();
- $this->readline->readHistory();
- $this->output->writeln($this->getHeader());
- $this->writeVersionInfo();
- $this->writeStartupMessage();
- try {
- $this->beforeRun();
- $this->loadIncludes();
- $loop = new ExecutionLoopClosure($this);
- $loop->execute();
- $this->afterRun();
- } catch (ThrowUpException $e) {
- throw $e->getPrevious();
- } catch (BreakException $e) {
- // The ProcessForker throws a BreakException to finish the main thread.
- }
- return 0;
- }
- /**
- * Run PsySH in non-interactive mode.
- *
- * Note that this isn't very useful unless you supply "include" arguments at
- * the command line, or code via stdin.
- *
- * @param bool $rawOutput
- *
- * @return int 0 if everything went fine, or an error code
- */
- private function doNonInteractiveRun(bool $rawOutput): int
- {
- $this->nonInteractive = true;
- // If raw output is enabled (or output is piped) we don't want startup messages.
- if (!$rawOutput && !$this->config->outputIsPiped()) {
- $this->output->writeln($this->getHeader());
- $this->writeVersionInfo();
- $this->writeStartupMessage();
- }
- $this->beforeRun();
- $this->loadIncludes();
- // For non-interactive execution, read only from the input buffer or from piped input.
- // Otherwise it'll try to readline and hang, waiting for user input with no indication of
- // what's holding things up.
- if (!empty($this->inputBuffer) || $this->config->inputIsPiped()) {
- $this->getInput(false);
- }
- if ($this->hasCode()) {
- $ret = $this->execute($this->flushCode());
- $this->writeReturnValue($ret, $rawOutput);
- }
- $this->afterRun();
- $this->nonInteractive = false;
- return 0;
- }
- /**
- * Configures the input and output instances based on the user arguments and options.
- */
- protected function configureIO(InputInterface $input, OutputInterface $output): void
- {
- // @todo overrides via environment variables (or should these happen in config? ... probably config)
- $input->setInteractive($this->config->getInputInteractive());
- if ($this->config->getOutputDecorated() !== null) {
- $output->setDecorated($this->config->getOutputDecorated());
- }
- $output->setVerbosity($this->config->getOutputVerbosity());
- }
- /**
- * Load user-defined includes.
- */
- private function loadIncludes()
- {
- // Load user-defined includes
- $load = function (self $__psysh__) {
- \set_error_handler([$__psysh__, 'handleError']);
- foreach ($__psysh__->getIncludes() as $__psysh_include__) {
- try {
- include_once $__psysh_include__;
- } catch (\Exception $_e) {
- $__psysh__->writeException($_e);
- }
- }
- \restore_error_handler();
- unset($__psysh_include__);
- // Override any new local variables with pre-defined scope variables
- \extract($__psysh__->getScopeVariables(false));
- // ... then add the whole mess of variables back.
- $__psysh__->setScopeVariables(\get_defined_vars());
- };
- $load($this);
- }
- /**
- * Read user input.
- *
- * This will continue fetching user input until the code buffer contains
- * valid code.
- *
- * @throws BreakException if user hits Ctrl+D
- *
- * @param bool $interactive
- */
- public function getInput(bool $interactive = true)
- {
- $this->codeBufferOpen = false;
- do {
- // reset output verbosity (in case it was altered by a subcommand)
- $this->output->setVerbosity($this->originalVerbosity);
- $input = $this->readline();
- /*
- * Handle Ctrl+D. It behaves differently in different cases:
- *
- * 1) In an expression, like a function or "if" block, clear the input buffer
- * 2) At top-level session, behave like the exit command
- * 3) When non-interactive, return, because that's the end of stdin
- */
- if ($input === false) {
- if (!$interactive) {
- return;
- }
- $this->output->writeln('');
- if ($this->hasCode()) {
- $this->resetCodeBuffer();
- } else {
- throw new BreakException('Ctrl+D');
- }
- }
- // handle empty input
- if (\trim($input) === '' && !$this->codeBufferOpen) {
- continue;
- }
- $input = $this->onInput($input);
- // If the input isn't in an open string or comment, check for commands to run.
- if ($this->hasCommand($input) && !$this->inputInOpenStringOrComment($input)) {
- $this->addHistory($input);
- $this->runCommand($input);
- continue;
- }
- $this->addCode($input);
- } while (!$interactive || !$this->hasValidCode());
- }
- /**
- * Check whether the code buffer (plus current input) is in an open string or comment.
- *
- * @param string $input current line of input
- *
- * @return bool true if the input is in an open string or comment
- */
- private function inputInOpenStringOrComment(string $input): bool
- {
- if (!$this->hasCode()) {
- return false;
- }
- $code = $this->codeBuffer;
- $code[] = $input;
- $tokens = @\token_get_all('<?php '.\implode("\n", $code));
- $last = \array_pop($tokens);
- return $last === '"' || $last === '`' ||
- (\is_array($last) && \in_array($last[0], [\T_ENCAPSED_AND_WHITESPACE, \T_START_HEREDOC, \T_COMMENT]));
- }
- /**
- * Run execution loop listeners before the shell session.
- */
- protected function beforeRun()
- {
- foreach ($this->loopListeners as $listener) {
- $listener->beforeRun($this);
- }
- }
- /**
- * Run execution loop listeners at the start of each loop.
- */
- public function beforeLoop()
- {
- foreach ($this->loopListeners as $listener) {
- $listener->beforeLoop($this);
- }
- }
- /**
- * Run execution loop listeners on user input.
- *
- * @param string $input
- */
- public function onInput(string $input): string
- {
- foreach ($this->loopListeners as $listeners) {
- if (($return = $listeners->onInput($this, $input)) !== null) {
- $input = $return;
- }
- }
- return $input;
- }
- /**
- * Run execution loop listeners on code to be executed.
- *
- * @param string $code
- */
- public function onExecute(string $code): string
- {
- $this->errorReporting = \error_reporting();
- foreach ($this->loopListeners as $listener) {
- if (($return = $listener->onExecute($this, $code)) !== null) {
- $code = $return;
- }
- }
- $output = $this->output;
- if ($output instanceof ConsoleOutput) {
- $output = $output->getErrorOutput();
- }
- $output->writeln(\sprintf('<whisper>%s</whisper>', OutputFormatter::escape($code)), ConsoleOutput::VERBOSITY_DEBUG);
- return $code;
- }
- /**
- * Run execution loop listeners after each loop.
- */
- public function afterLoop()
- {
- foreach ($this->loopListeners as $listener) {
- $listener->afterLoop($this);
- }
- }
- /**
- * Run execution loop listers after the shell session.
- */
- protected function afterRun()
- {
- foreach ($this->loopListeners as $listener) {
- $listener->afterRun($this);
- }
- }
- /**
- * Set the variables currently in scope.
- *
- * @param array $vars
- */
- public function setScopeVariables(array $vars)
- {
- $this->context->setAll($vars);
- }
- /**
- * Return the set of variables currently in scope.
- *
- * @param bool $includeBoundObject Pass false to exclude 'this'. If you're
- * passing the scope variables to `extract`
- * you _must_ exclude 'this'
- *
- * @return array Associative array of scope variables
- */
- public function getScopeVariables(bool $includeBoundObject = true): array
- {
- $vars = $this->context->getAll();
- if (!$includeBoundObject) {
- unset($vars['this']);
- }
- return $vars;
- }
- /**
- * Return the set of magic variables currently in scope.
- *
- * @param bool $includeBoundObject Pass false to exclude 'this'. If you're
- * passing the scope variables to `extract`
- * you _must_ exclude 'this'
- *
- * @return array Associative array of magic scope variables
- */
- public function getSpecialScopeVariables(bool $includeBoundObject = true): array
- {
- $vars = $this->context->getSpecialVariables();
- if (!$includeBoundObject) {
- unset($vars['this']);
- }
- return $vars;
- }
- /**
- * Return the set of variables currently in scope which differ from the
- * values passed as $currentVars.
- *
- * This is used inside the Execution Loop Closure to pick up scope variable
- * changes made by commands while the loop is running.
- *
- * @param array $currentVars
- *
- * @return array Associative array of scope variables which differ from $currentVars
- */
- public function getScopeVariablesDiff(array $currentVars): array
- {
- $newVars = [];
- foreach ($this->getScopeVariables(false) as $key => $value) {
- if (!\array_key_exists($key, $currentVars) || $currentVars[$key] !== $value) {
- $newVars[$key] = $value;
- }
- }
- return $newVars;
- }
- /**
- * Get the set of unused command-scope variable names.
- *
- * @return array Array of unused variable names
- */
- public function getUnusedCommandScopeVariableNames(): array
- {
- return $this->context->getUnusedCommandScopeVariableNames();
- }
- /**
- * Get the set of variable names currently in scope.
- *
- * @return array Array of variable names
- */
- public function getScopeVariableNames(): array
- {
- return \array_keys($this->context->getAll());
- }
- /**
- * Get a scope variable value by name.
- *
- * @param string $name
- *
- * @return mixed
- */
- public function getScopeVariable(string $name)
- {
- return $this->context->get($name);
- }
- /**
- * Set the bound object ($this variable) for the interactive shell.
- *
- * @param object|null $boundObject
- */
- public function setBoundObject($boundObject)
- {
- $this->context->setBoundObject($boundObject);
- }
- /**
- * Get the bound object ($this variable) for the interactive shell.
- *
- * @return object|null
- */
- public function getBoundObject()
- {
- return $this->context->getBoundObject();
- }
- /**
- * Set the bound class (self) for the interactive shell.
- *
- * @param string|null $boundClass
- */
- public function setBoundClass($boundClass)
- {
- $this->context->setBoundClass($boundClass);
- }
- /**
- * Get the bound class (self) for the interactive shell.
- *
- * @return string|null
- */
- public function getBoundClass()
- {
- return $this->context->getBoundClass();
- }
- /**
- * Add includes, to be parsed and executed before running the interactive shell.
- *
- * @param array $includes
- */
- public function setIncludes(array $includes = [])
- {
- $this->includes = $includes;
- }
- /**
- * Get PHP files to be parsed and executed before running the interactive shell.
- *
- * @return string[]
- */
- public function getIncludes(): array
- {
- return \array_merge($this->config->getDefaultIncludes(), $this->includes);
- }
- /**
- * Check whether this shell's code buffer contains code.
- *
- * @return bool True if the code buffer contains code
- */
- public function hasCode(): bool
- {
- return !empty($this->codeBuffer);
- }
- /**
- * Check whether the code in this shell's code buffer is valid.
- *
- * If the code is valid, the code buffer should be flushed and evaluated.
- *
- * @return bool True if the code buffer content is valid
- */
- protected function hasValidCode(): bool
- {
- return !$this->codeBufferOpen && $this->code !== false;
- }
- /**
- * Add code to the code buffer.
- *
- * @param string $code
- * @param bool $silent
- */
- public function addCode(string $code, bool $silent = false)
- {
- try {
- // Code lines ending in \ keep the buffer open
- if (\substr(\rtrim($code), -1) === '\\') {
- $this->codeBufferOpen = true;
- $code = \substr(\rtrim($code), 0, -1);
- } else {
- $this->codeBufferOpen = false;
- }
- $this->codeBuffer[] = $silent ? new SilentInput($code) : $code;
- $this->code = $this->cleaner->clean($this->codeBuffer, $this->config->requireSemicolons());
- } catch (\Throwable $e) {
- // Add failed code blocks to the readline history.
- $this->addCodeBufferToHistory();
- throw $e;
- }
- }
- /**
- * Set the code buffer.
- *
- * This is mostly used by `Shell::execute`. Any existing code in the input
- * buffer is pushed onto a stack and will come back after this new code is
- * executed.
- *
- * @throws \InvalidArgumentException if $code isn't a complete statement
- *
- * @param string $code
- * @param bool $silent
- */
- private function setCode(string $code, bool $silent = false)
- {
- if ($this->hasCode()) {
- $this->codeStack[] = [$this->codeBuffer, $this->codeBufferOpen, $this->code];
- }
- $this->resetCodeBuffer();
- try {
- $this->addCode($code, $silent);
- } catch (\Throwable $e) {
- $this->popCodeStack();
- throw $e;
- }
- if (!$this->hasValidCode()) {
- $this->popCodeStack();
- throw new \InvalidArgumentException('Unexpected end of input');
- }
- }
- /**
- * Get the current code buffer.
- *
- * This is useful for commands which manipulate the buffer.
- *
- * @return string[]
- */
- public function getCodeBuffer(): array
- {
- return $this->codeBuffer;
- }
- /**
- * Run a Psy Shell command given the user input.
- *
- * @throws \InvalidArgumentException if the input is not a valid command
- *
- * @param string $input User input string
- *
- * @return mixed Who knows?
- */
- protected function runCommand(string $input)
- {
- $command = $this->getCommand($input);
- if (empty($command)) {
- throw new \InvalidArgumentException('Command not found: '.$input);
- }
- $input = new ShellInput(\str_replace('\\', '\\\\', \rtrim($input, " \t\n\r\0\x0B;")));
- if ($input->hasParameterOption(['--help', '-h'])) {
- $helpCommand = $this->get('help');
- if (!$helpCommand instanceof Command\HelpCommand) {
- throw new RuntimeException('Invalid help command instance');
- }
- $helpCommand->setCommand($command);
- return $helpCommand->run(new StringInput(''), $this->output);
- }
- return $command->run($input, $this->output);
- }
- /**
- * Reset the current code buffer.
- *
- * This should be run after evaluating user input, catching exceptions, or
- * on demand by commands such as BufferCommand.
- */
- public function resetCodeBuffer()
- {
- $this->codeBuffer = [];
- $this->code = false;
- }
- /**
- * Inject input into the input buffer.
- *
- * This is useful for commands which want to replay history.
- *
- * @param string|array $input
- * @param bool $silent
- */
- public function addInput($input, bool $silent = false)
- {
- foreach ((array) $input as $line) {
- $this->inputBuffer[] = $silent ? new SilentInput($line) : $line;
- }
- }
- /**
- * Flush the current (valid) code buffer.
- *
- * If the code buffer is valid, resets the code buffer and returns the
- * current code.
- *
- * @return string|null PHP code buffer contents
- */
- public function flushCode()
- {
- if ($this->hasValidCode()) {
- $this->addCodeBufferToHistory();
- $code = $this->code;
- $this->popCodeStack();
- return $code;
- }
- }
- /**
- * Reset the code buffer and restore any code pushed during `execute` calls.
- */
- private function popCodeStack()
- {
- $this->resetCodeBuffer();
- if (empty($this->codeStack)) {
- return;
- }
- list($codeBuffer, $codeBufferOpen, $code) = \array_pop($this->codeStack);
- $this->codeBuffer = $codeBuffer;
- $this->codeBufferOpen = $codeBufferOpen;
- $this->code = $code;
- }
- /**
- * (Possibly) add a line to the readline history.
- *
- * Like Bash, if the line starts with a space character, it will be omitted
- * from history. Note that an entire block multi-line code input will be
- * omitted iff the first line begins with a space.
- *
- * Additionally, if a line is "silent", i.e. it was initially added with the
- * silent flag, it will also be omitted.
- *
- * @param string|SilentInput $line
- */
- private function addHistory($line)
- {
- if ($line instanceof SilentInput) {
- return;
- }
- // Skip empty lines and lines starting with a space
- if (\trim($line) !== '' && \substr($line, 0, 1) !== ' ') {
- $this->readline->addHistory($line);
- }
- }
- /**
- * Filter silent input from code buffer, write the rest to readline history.
- */
- private function addCodeBufferToHistory()
- {
- $codeBuffer = \array_filter($this->codeBuffer, function ($line) {
- return !$line instanceof SilentInput;
- });
- $this->addHistory(\implode("\n", $codeBuffer));
- }
- /**
- * Get the current evaluation scope namespace.
- *
- * @see CodeCleaner::getNamespace
- *
- * @return string|null Current code namespace
- */
- public function getNamespace()
- {
- if ($namespace = $this->cleaner->getNamespace()) {
- return \implode('\\', $namespace);
- }
- }
- /**
- * Write a string to stdout.
- *
- * This is used by the shell loop for rendering output from evaluated code.
- *
- * @param string $out
- * @param int $phase Output buffering phase
- */
- public function writeStdout(string $out, int $phase = \PHP_OUTPUT_HANDLER_END)
- {
- if ($phase & \PHP_OUTPUT_HANDLER_START) {
- if ($this->output instanceof ShellOutput) {
- $this->output->startPaging();
- }
- }
- $isCleaning = $phase & \PHP_OUTPUT_HANDLER_CLEAN;
- // Incremental flush
- if ($out !== '' && !$isCleaning) {
- $this->output->write($out, false, OutputInterface::OUTPUT_RAW);
- $this->outputWantsNewline = (\substr($out, -1) !== "\n");
- $this->stdoutBuffer .= $out;
- }
- // Output buffering is done!
- if ($phase & \PHP_OUTPUT_HANDLER_END) {
- // Write an extra newline if stdout didn't end with one
- if ($this->outputWantsNewline) {
- if (!$this->config->rawOutput() && !$this->config->outputIsPiped()) {
- $this->output->writeln(\sprintf('<whisper>%s</whisper>', $this->config->useUnicode() ? '⏎' : '\\n'));
- } else {
- $this->output->writeln('');
- }
- $this->outputWantsNewline = false;
- }
- // Save the stdout buffer as $__out
- if ($this->stdoutBuffer !== '') {
- $this->context->setLastStdout($this->stdoutBuffer);
- $this->stdoutBuffer = '';
- }
- if ($this->output instanceof ShellOutput) {
- $this->output->stopPaging();
- }
- }
- }
- /**
- * Write a return value to stdout.
- *
- * The return value is formatted or pretty-printed, and rendered in a
- * visibly distinct manner (in this case, as cyan).
- *
- * @see self::presentValue
- *
- * @param mixed $ret
- * @param bool $rawOutput Write raw var_export-style values
- */
- public function writeReturnValue($ret, bool $rawOutput = false)
- {
- $this->lastExecSuccess = true;
- if ($ret instanceof NoReturnValue) {
- return;
- }
- $this->context->setReturnValue($ret);
- if ($rawOutput) {
- $formatted = \var_export($ret, true);
- } else {
- $prompt = $this->config->theme()->returnValue();
- $indent = \str_repeat(' ', \strlen($prompt));
- $formatted = $this->presentValue($ret);
- $formattedRetValue = \sprintf('<whisper>%s</whisper>', $prompt);
- $formatted = $formattedRetValue.\str_replace(\PHP_EOL, \PHP_EOL.$indent, $formatted);
- }
- if ($this->output instanceof ShellOutput) {
- $this->output->page($formatted.\PHP_EOL);
- } else {
- $this->output->writeln($formatted);
- }
- }
- /**
- * Renders a caught Exception or Error.
- *
- * Exceptions are formatted according to severity. ErrorExceptions which were
- * warnings or Strict errors aren't rendered as harshly as real errors.
- *
- * Stores $e as the last Exception in the Shell Context.
- *
- * @param \Throwable $e An exception or error instance
- */
- public function writeException(\Throwable $e)
- {
- // No need to write the break exception during a non-interactive run.
- if ($e instanceof BreakException && $this->nonInteractive) {
- $this->resetCodeBuffer();
- return;
- }
- // Break exceptions don't count :)
- if (!$e instanceof BreakException) {
- $this->lastExecSuccess = false;
- $this->context->setLastException($e);
- }
- $output = $this->output;
- if ($output instanceof ConsoleOutput) {
- $output = $output->getErrorOutput();
- }
- if (!$this->config->theme()->compact()) {
- $output->writeln('');
- }
- $output->writeln($this->formatException($e));
- if (!$this->config->theme()->compact()) {
- $output->writeln('');
- }
- // Include an exception trace (as long as this isn't a BreakException).
- if (!$e instanceof BreakException && $output->getVerbosity() >= OutputInterface::VERBOSITY_VERBOSE) {
- $trace = TraceFormatter::formatTrace($e);
- if (\count($trace) !== 0) {
- $output->writeln('--');
- $output->write($trace, true);
- $output->writeln('');
- }
- }
- $this->resetCodeBuffer();
- }
- /**
- * Check whether the last exec was successful.
- *
- * Returns true if a return value was logged rather than an exception.
- */
- public function getLastExecSuccess(): bool
- {
- return $this->lastExecSuccess;
- }
- /**
- * Helper for formatting an exception or error for writeException().
- *
- * @todo extract this to somewhere it makes more sense
- *
- * @param \Throwable $e
- */
- public function formatException(\Throwable $e): string
- {
- $indent = $this->config->theme()->compact() ? '' : ' ';
- if ($e instanceof BreakException) {
- return \sprintf('%s<info> INFO </info> %s.', $indent, \rtrim($e->getRawMessage(), '.'));
- } elseif ($e instanceof PsyException) {
- $message = $e->getLine() > 1
- ? \sprintf('%s in %s on line %d', $e->getRawMessage(), $e->getFile(), $e->getLine())
- : \sprintf('%s in %s', $e->getRawMessage(), $e->getFile());
- $messageLabel = \strtoupper($this->getMessageLabel($e));
- } else {
- $message = $e->getMessage();
- $messageLabel = $this->getMessageLabel($e);
- }
- $message = \preg_replace(
- "#(\\w:)?([\\\\/]\\w+)*[\\\\/]src[\\\\/]Execution(?:Loop)?Closure.php\(\d+\) : eval\(\)'d code#",
- "eval()'d code",
- $message
- );
- $message = \str_replace(" in eval()'d code", '', $message);
- $message = \trim($message);
- // Ensures the given string ends with punctuation...
- if (!empty($message) && !\in_array(\substr($message, -1), ['.', '?', '!', ':'])) {
- $message = "$message.";
- }
- // Ensures the given message only contains relative paths...
- $message = \str_replace(\getcwd().\DIRECTORY_SEPARATOR, '', $message);
- $severity = ($e instanceof \ErrorException) ? $this->getSeverity($e) : 'error';
- return \sprintf('%s<%s> %s </%s> %s', $indent, $severity, $messageLabel, $severity, OutputFormatter::escape($message));
- }
- /**
- * Helper for getting an output style for the given ErrorException's level.
- *
- * @param \ErrorException $e
- */
- protected function getSeverity(\ErrorException $e): string
- {
- $severity = $e->getSeverity();
- if ($severity & \error_reporting()) {
- switch ($severity) {
- case \E_WARNING:
- case \E_NOTICE:
- case \E_CORE_WARNING:
- case \E_COMPILE_WARNING:
- case \E_USER_WARNING:
- case \E_USER_NOTICE:
- case \E_USER_DEPRECATED:
- case \E_DEPRECATED:
- case \E_STRICT:
- return 'warning';
- default:
- return 'error';
- }
- } else {
- // Since this is below the user's reporting threshold, it's always going to be a warning.
- return 'warning';
- }
- }
- /**
- * Helper for getting an output style for the given ErrorException's level.
- *
- * @param \Throwable $e
- */
- protected function getMessageLabel(\Throwable $e): string
- {
- if ($e instanceof \ErrorException) {
- $severity = $e->getSeverity();
- if ($severity & \error_reporting()) {
- switch ($severity) {
- case \E_WARNING:
- return 'Warning';
- case \E_NOTICE:
- return 'Notice';
- case \E_CORE_WARNING:
- return 'Core Warning';
- case \E_COMPILE_WARNING:
- return 'Compile Warning';
- case \E_USER_WARNING:
- return 'User Warning';
- case \E_USER_NOTICE:
- return 'User Notice';
- case \E_USER_DEPRECATED:
- return 'User Deprecated';
- case \E_DEPRECATED:
- return 'Deprecated';
- case \E_STRICT:
- return 'Strict';
- }
- }
- }
- if ($e instanceof PsyException || $e instanceof SymfonyConsoleException) {
- $exceptionShortName = (new \ReflectionClass($e))->getShortName();
- $typeParts = \preg_split('/(?=[A-Z])/', $exceptionShortName);
- switch ($exceptionShortName) {
- case 'RuntimeException':
- case 'LogicException':
- // These ones look weird without 'Exception'
- break;
- default:
- if (\end($typeParts) === 'Exception') {
- \array_pop($typeParts);
- }
- break;
- }
- return \trim(\strtoupper(\implode(' ', $typeParts)));
- }
- return \get_class($e);
- }
- /**
- * Execute code in the shell execution context.
- *
- * @param string $code
- * @param bool $throwExceptions
- *
- * @return mixed
- */
- public function execute(string $code, bool $throwExceptions = false)
- {
- $this->setCode($code, true);
- $closure = new ExecutionClosure($this);
- if ($throwExceptions) {
- return $closure->execute();
- }
- try {
- return $closure->execute();
- } catch (\Throwable $_e) {
- $this->writeException($_e);
- }
- }
- /**
- * Helper for throwing an ErrorException.
- *
- * This allows us to:
- *
- * set_error_handler([$psysh, 'handleError']);
- *
- * Unlike ErrorException::throwException, this error handler respects error
- * levels; i.e. it logs warnings and notices, but doesn't throw exceptions.
- * This should probably only be used in the inner execution loop of the
- * shell, as most of the time a thrown exception is much more useful.
- *
- * If the error type matches the `errorLoggingLevel` config, it will be
- * logged as well, regardless of the `error_reporting` level.
- *
- * @see \Psy\Exception\ErrorException::throwException
- * @see \Psy\Shell::writeException
- *
- * @throws \Psy\Exception\ErrorException depending on the error level
- *
- * @param int $errno Error type
- * @param string $errstr Message
- * @param string $errfile Filename
- * @param int $errline Line number
- */
- public function handleError($errno, $errstr, $errfile, $errline)
- {
- // This is an error worth throwing.
- //
- // n.b. Technically we can't handle all of these in userland code, but
- // we'll list 'em all for good measure
- if ($errno & (\E_ERROR | \E_PARSE | \E_CORE_ERROR | \E_COMPILE_ERROR | \E_USER_ERROR | \E_RECOVERABLE_ERROR)) {
- ErrorException::throwException($errno, $errstr, $errfile, $errline);
- }
- // When errors are suppressed, the error_reporting value will differ
- // from when we started executing. In that case, we won't log errors.
- $errorsSuppressed = $this->errorReporting !== null && $this->errorReporting !== \error_reporting();
- // Otherwise log it and continue.
- if ($errno & \error_reporting() || (!$errorsSuppressed && ($errno & $this->config->errorLoggingLevel()))) {
- $this->writeException(new ErrorException($errstr, 0, $errno, $errfile, $errline));
- }
- }
- /**
- * Format a value for display.
- *
- * @see Presenter::present
- *
- * @param mixed $val
- *
- * @return string Formatted value
- */
- protected function presentValue($val): string
- {
- return $this->config->getPresenter()->present($val);
- }
- /**
- * Get a command (if one exists) for the current input string.
- *
- * @param string $input
- *
- * @return BaseCommand|null
- */
- protected function getCommand(string $input)
- {
- $input = new StringInput($input);
- if ($name = $input->getFirstArgument()) {
- return $this->get($name);
- }
- }
- /**
- * Check whether a command is set for the current input string.
- *
- * @param string $input
- *
- * @return bool True if the shell has a command for the given input
- */
- protected function hasCommand(string $input): bool
- {
- if (\preg_match('/([^\s]+?)(?:\s|$)/A', \ltrim($input), $match)) {
- return $this->has($match[1]);
- }
- return false;
- }
- /**
- * Get the current input prompt.
- *
- * @return string|null
- */
- protected function getPrompt()
- {
- if ($this->output->isQuiet()) {
- return null;
- }
- $theme = $this->config->theme();
- if ($this->hasCode()) {
- return $theme->bufferPrompt();
- }
- return $theme->prompt();
- }
- /**
- * Read a line of user input.
- *
- * This will return a line from the input buffer (if any exist). Otherwise,
- * it will ask the user for input.
- *
- * If readline is enabled, this delegates to readline. Otherwise, it's an
- * ugly `fgets` call.
- *
- * @param bool $interactive
- *
- * @return string|false One line of user input
- */
- protected function readline(bool $interactive = true)
- {
- $prompt = $this->config->theme()->replayPrompt();
- if (!empty($this->inputBuffer)) {
- $line = \array_shift($this->inputBuffer);
- if (!$line instanceof SilentInput) {
- $this->output->writeln(\sprintf('<whisper>%s</whisper><aside>%s</aside>', $prompt, OutputFormatter::escape($line)));
- }
- return $line;
- }
- $bracketedPaste = $interactive && $this->config->useBracketedPaste();
- if ($bracketedPaste) {
- \printf("\e[?2004h"); // Enable bracketed paste
- }
- $line = $this->readline->readline($this->getPrompt());
- if ($bracketedPaste) {
- \printf("\e[?2004l"); // ... and disable it again
- }
- return $line;
- }
- /**
- * Get the shell output header.
- */
- protected function getHeader(): string
- {
- return \sprintf('<whisper>%s by Justin Hileman</whisper>', $this->getVersion());
- }
- /**
- * Get the current version of Psy Shell.
- *
- * @deprecated call self::getVersionHeader instead
- */
- public function getVersion(): string
- {
- @\trigger_error('`getVersion` is deprecated; call `self::getVersionHeader` instead.', \E_USER_DEPRECATED);
- return self::getVersionHeader($this->config->useUnicode());
- }
- /**
- * Get a pretty header including the current version of Psy Shell.
- *
- * @param bool $useUnicode
- */
- public static function getVersionHeader(bool $useUnicode = false): string
- {
- $separator = $useUnicode ? '—' : '-';
- return \sprintf('Psy Shell %s (PHP %s %s %s)', self::VERSION, \PHP_VERSION, $separator, \PHP_SAPI);
- }
- /**
- * Get a PHP manual database instance.
- *
- * @return \PDO|null
- */
- public function getManualDb()
- {
- return $this->config->getManualDb();
- }
- /**
- * Initialize tab completion matchers.
- *
- * If tab completion is enabled this adds tab completion matchers to the
- * auto completer and sets context if needed.
- */
- protected function initializeTabCompletion()
- {
- if (!$this->config->useTabCompletion()) {
- return;
- }
- $this->autoCompleter = $this->config->getAutoCompleter();
- // auto completer needs shell to be linked to configuration because of
- // the context aware matchers
- $this->addMatchersToAutoCompleter($this->getDefaultMatchers());
- $this->addMatchersToAutoCompleter($this->matchers);
- $this->autoCompleter->activate();
- }
- /**
- * Add matchers to the auto completer, setting context if needed.
- *
- * @param array $matchers
- */
- private function addMatchersToAutoCompleter(array $matchers)
- {
- foreach ($matchers as $matcher) {
- if ($matcher instanceof ContextAware) {
- $matcher->setContext($this->context);
- }
- $this->autoCompleter->addMatcher($matcher);
- }
- }
- /**
- * @todo Implement prompt to start update
- *
- * @return void|string
- */
- protected function writeVersionInfo()
- {
- if (\PHP_SAPI !== 'cli') {
- return;
- }
- try {
- $client = $this->config->getChecker();
- if (!$client->isLatest()) {
- $this->output->writeln(\sprintf('<whisper>New version is available at psysh.org/psysh (current: %s, latest: %s)</whisper>', self::VERSION, $client->getLatest()));
- }
- } catch (\InvalidArgumentException $e) {
- $this->output->writeln($e->getMessage());
- }
- }
- /**
- * Write a startup message if set.
- */
- protected function writeStartupMessage()
- {
- $message = $this->config->getStartupMessage();
- if ($message !== null && $message !== '') {
- $this->output->writeln($message);
- }
- }
- }
|