Prompt can now be set before linereader is created

examples/simple.php

@@ -4,6 +4,7 @@ require_once __DIR__."/../vendor/autoload.php";
 
 use NoccyLabs\Shell\Shell;
 use NoccyLabs\Shell\Command;
+use NoccyLabs\Shell\Context;
 
 class MyCommand extends Command
 {
@@ -13,6 +14,42 @@ class MyCommand extends Command
     }
 }
 
+class MyContext extends Context
+{
+    /**
+     * @command testme
+     * @args
+     * @help Useful test!
+     * @global
+     */
+    public function test()
+    {
+        echo "Test\n";
+    }
+
+    /**
+     * @command context
+     * @help Create a new context
+     */
+    public function context()
+    {
+        return new OtherContext("newcontext");
+    }
+}
+
+class OtherContext extends Context
+{
+    /**
+     * @command other
+     * @args
+     * @help Other test
+     */
+    public function test()
+    {
+
+    }
+}
+
 class MyShell extends Shell
 {
 
@@ -20,7 +57,8 @@ class MyShell extends Shell
 
     protected function configure()
     {
-        $context = $this->createContext();
+        $context = new MyContext();
+        $this->pushContext($context);
         $context->addCommand("hello", function () {
             echo "world\nthis\nis\na\ntest\n";
         });

examples/timers.php

@@ -0,0 +1,54 @@
+<?php
+
+require_once __DIR__."/../vendor/autoload.php";
+
+use NoccyLabs\Shell\Shell;
+use NoccyLabs\Shell\Command;
+use NoccyLabs\Shell\Context;
+
+class MyContext extends Context
+{
+    /**
+     * @command testme
+     * @args
+     * @help Useful test!
+     */
+    public function test()
+    {
+
+    }
+}
+
+class MyShell extends Shell
+{
+
+    protected $seq = 0;
+
+    protected function configure()
+    {
+        $context = new MyContext();
+        $this->pushContext($context);
+        $this->updatePrompt();
+
+        $this->addTimer(5000, function () {
+            echo "5 seconds\n";
+        });
+    }
+    
+    protected function updatePrompt()
+    {
+        $this->setPrompt("test[{$this->seq}]: ");
+    }
+    
+    protected function onCommand($buffer)
+    {
+        $this->seq++;
+        $this->updatePrompt();
+        parent::onCommand($buffer);
+    }
+
+}
+
+$myShell = new MyShell();
+$myShell->run();
+echo "Exiting\n";

lib/AutoComplete/Hinter.php

@@ -1,14 +0,0 @@
-<?php
-
-namespace NoccyLabs\Shell\AutoComplete;
-
-class Hinter implement HinterInterface
-{
-    public function getHints()
-    {
-    }
-
-    public function addHinter(HinterInterface $hinter)
-    {
-    }
-}

lib/AutoComplete/HinterInterface.php

@@ -1,7 +0,0 @@
-<?php
-
-namespace NoccyLabs\Shell\AutoComplete;
-
-interface HinterInterface
-{
-}

lib/Context.php

@@ -8,8 +8,13 @@ class Context
 
     protected $commands = [];
 
+    protected $commandInfo = [];
+
     protected $data = [];
 
+    protected $parent;
+
+    protected $shell;
 
     public function __construct($name=null, array $data=[])
     {
@@ -18,14 +23,79 @@ class Context
         $this->configure();
     }
 
+    public function setShell(Shell $shell)
+    {
+        $this->shell = $shell;
+    }
+
+    public function getShell()
+    {
+        return $this->shell;
+    }
+
+    public function setParent(Context $parent=null)
+    {
+        $this->parent = $parent;
+    }
+
+    public function getParent()
+    {
+        return $this->parent;
+    }
+
+    public function getRoot()
+    {
+        if (!$this->parent) {
+            return $this;
+        }
+        $node = $this;
+        while ($parent = $node->getParent()) {
+            $node = $parent;
+        }
+        return $parent;
+    }
+
     protected function configure()
     {
         // Override this to do setup stuff
+        $this->findCommands();
     }
 
-    public function addCommand($command, callable $handler)
+    protected function findCommands()
+    {
+        $refl = new \ReflectionClass(get_called_class());
+        foreach ($refl->getMethods() as $method) {
+            $docblock = $method->getDocComment();
+            $lines = array_map(function ($line) {
+                return trim($line, "*/ \t");
+            }, explode("\n", $docblock));
+            $info = [];
+            foreach ($lines as $line) {
+                if (preg_match("/^@(command|help|args|global)\\s*(.*)$/", $line, $match)) {
+                    list($void,$key,$value) = $match;
+                    $info[$key] = $value; 
+                }
+                if (count($info)>0) {
+                    $cmdName = array_key_exists("command",$info)?$info["command"]:$method->getName();
+                    $this->addCommand($cmdName, [$this, $method->getName()], $info);
+                }
+            }
+        }
+    }
+
+    public function addCommand($command, callable $handler, array $info=[])
     {
         $this->commands[$command] = $handler;
+        $this->commandInfo[$command] = $info;
+        ksort($this->commands);
+    }
+
+    public function setCommandHelp($command, $help)
+    {
+        if (!array_key_exists($command, $this->commandInfo)) {
+            return;
+        }
+        $this->commandInfo[$command]['help'] = $help;
     }
 
     public function addCommands(array $commands)
@@ -50,6 +120,27 @@ class Context
         return $this->commands[$command];
     }
 
+    public function getCommandHelp()
+    {
+        $ret = [];
+        foreach ($this->commands as $command=>$handler) {
+            $info = $this->commandInfo[$command];
+            $args = array_key_exists("args",$info)?$info['args']:"";
+            $help = array_key_exists("help",$info)?$info['help']:"";
+            $ret[trim("{$command} {$args}")] = $help;
+        }
+        return $ret;
+    }
+
+    public function isCommandGlobal($command)
+    {
+        if (strpos($command," ")!==false) {
+            list($command, $void) = explode(" ",$command,2);
+        }
+        $info = $this->commandInfo[$command];
+        return array_key_exists('global', $info);
+    }
+
     public function getName()
     {
         return $this->name;

lib/LineRead.php

@@ -16,6 +16,8 @@ class LineRead
 
     protected $history = [];
 
+    protected $stashedBuffer = null;
+    
     protected $posHistory = 0;
     
     protected $posCursor = 0;
@@ -61,11 +63,16 @@ class LineRead
     {
         $prompt = $this->prompt;
         $buffer = $this->buffer;
-        $cursor = strlen($this->prompt) + 1 + $this->posCursor - $this->posScroll;
+
+        if ($this->posCursor > strlen($this->buffer)) {
+            $this->posCursor = strlen($this->buffer);
+        }
+
+        $cursor = strlen($this->prompt) + 2 + $this->posCursor - $this->posScroll;
         
         $endStyle = "\e[0m";
         
-        fprintf(STDOUT, "\r\e[2K%s%s\e[%dG{$endStyle}", ($this->promptStyle)($prompt), ($this->commandStyle)($buffer), $cursor);
+        fprintf(STDOUT, "\r\e[2K%s %s\e[%dG{$endStyle}", ($this->promptStyle)($prompt), ($this->commandStyle)($buffer), $cursor);
     }
     
     protected function styleToAnsi($style)
@@ -85,6 +92,10 @@ class LineRead
         $returnBuffer = null;
         while (strlen($keyBuffer)>0) {
             if ($keyBuffer[0] == "\e") {
+                if (strlen($keyBuffer)==1) {
+                    $keyBuffer = "";
+                    return "\e";
+                }
                 if ($keyBuffer[1] == "[") {
                     $ctrlChar = substr($keyBuffer, 0,3);
                     $keyBuffer = substr($keyBuffer, 3);
@@ -113,6 +124,7 @@ class LineRead
                     }
                 } elseif ($keyCode == 13) {
                     $returnBuffer = $this->buffer;
+                    array_unshift($this->history, $this->buffer);
                     $this->buffer = null;
                     $this->posCursor = 0;
                     printf("\n\r");
@@ -146,7 +158,25 @@ class LineRead
                 $this->redraw();
                 break;
             case "\e[A": // up
+                if ($this->posHistory == 0) {
+                    $this->stashedBuffer = $this->buffer;
+                }
+                if ($this->posHistory < count($this->history)) {
+                    $this->posHistory++;
+                    $this->buffer = $this->history[$this->posHistory-1];
+                    $this->redraw();
+                }
+                break;
             case "\e[B": // down
+                if ($this->posHistory > 1) {
+                    $this->posHistory--;
+                    $this->buffer = $this->history[$this->posHistory-1];
+                    $this->redraw();
+                } elseif ($this->posHistory > 0) {
+                    $this->posHistory--;
+                    $this->buffer = $this->stashedBuffer;
+                    $this->redraw();
+                }
                 break;
             default:
                 fprintf(STDERR, "\n%s\n", substr($code,1));

lib/Shell.php

@@ -17,6 +17,10 @@ class Shell
 
     protected $listeners = [];
     
+    protected $timers = [];
+
+    protected $prompt = ">";
+
     public function __construct()
     {
         $this->configure();
@@ -35,8 +39,10 @@ class Shell
     public function pushContext(Context $context)
     {
         if ($this->context) {
+            $context->setParent($this->context);
             array_unshift($this->contextStack, $this->context);
         }
+        $context->setShell($this);
         $this->context = $context;
         $this->dispatchEvent("context.update", [ "context"=>$this->context ]);
     }
@@ -124,7 +130,26 @@ class Shell
      */
     public function addTimer($interval, callable $handler, array $userdata=[])
     {
-
+        $timer = new class($interval, $handler, $userdata) {
+            private $next;
+            private $interval;
+            private $handler;
+            private $userdata;
+            public function __construct($interval, callable $handler, array $userdata) {
+                $this->interval = $interval / 1000;
+                $this->handler = $handler;
+                $this->userdata = $userdata;
+            }
+            public function update() {
+                $now = microtime(true);
+                if ($now > $this->next) {
+                    $this->next = $now + $this->interval;
+                    call_user_func($this->handler, $this->userdata);
+                }
+            }
+        };
+        $this->timers[] = $timer;
+        return $timer;
     }
 
     /**
@@ -139,11 +164,12 @@ class Shell
 
     public function setPrompt($text)
     {
-        if (!$this->lineReader) {
+        $this->prompt = $text;
-            return;
+    
-        }
+        if ($this->lineReader) {
             $this->lineReader->setPromptText($text);
         }
+    }
 
     /**
      * Find a command and return a closure.
@@ -153,12 +179,18 @@ class Shell
     private function findCommand($command)
     {
         // Go over current context and walk through stack until finding command
-        foreach(array_merge([ $this->context ] , $this->contextStack) as $context) {
+        if ($this->context->hasCommand($command)) {
-            if ($context->hasCommand($command)) {
+            $handler = $this->context->getCommand($command);
+        } else {
+            foreach($this->contextStack as $context) {
+                if ($context->hasCommand($command) && $context->isCommandGlobal($command)) {
                     $handler = $context->getCommand($command);
                     break;
                 }
             }
+        }
+
+        // No handler...
         if (empty($handler)) {
             return false;
         }
@@ -173,7 +205,7 @@ class Shell
      * Execute a command with arguments.
      *
      * @param string $command The command name to execute
-     * @param string.. $args Arguments
+     * @param string $args Arguments
      * @return mixed 
      * @throws Exception\BadCommandExcception
      */
@@ -199,19 +231,48 @@ class Shell
     {
         switch ($command) {
             case '.':
-                printf("context<%s>:\n", $this->context->getName());
+                $type = basename(strtr(get_class($this->context), "\\", "/"));
-                echo "  ".join("\n  ",explode("\n",json_encode($this->context->getData(),JSON_PRETTY_PRINT)))."\n";
+                printf("%s<%s>: %s\n", $type, $this->context->getName(), json_encode($this->context->getData()));
+                $level = 0;
+                foreach ($this->contextStack as $context) {
+                    $type = basename(strtr(get_class($context), "\\", "/"));
+                    printf(" %s- %s<%s>\n", str_repeat("  ",$level++), $type, $context->getName());
+                }
                 break;
             case '..':
                 if (count($this->contextStack)>0)
                     $this->popContext();
                 break;
             case 'help':
-                echo
+                $help = $this->context->getCommandHelp();
-                    "Built in commands:\n".
+                $ghelp = [];
-                    "  .        Show current context\n".
+                foreach ($this->contextStack as $context) {
-                    "  ..       Go to parent context\n".
+                    $commands = $context->getCommandHelp();
-                    "  exit     Exit the shell\n";
+                    foreach ($commands as $command=>$info) {
+                        if (strpos(" ",$command)!==false) {
+                            list ($cmd,$arg)=explode(" ",$command,2);
+                        } else {
+                            $cmd = $command;
+                        }
+                        if ($context->isCommandGlobal($cmd)) {
+                            $ghelp[$command] = $info;
+                        }
+                    }
+                }
+                ksort($ghelp);
+                printf("Commands in current context:\n");
+                foreach ($help as $command=>$info) {
+                    printf("  %-20s  %s\n", $command, $info);
+                }
+                if (count($ghelp)) {
+                    printf("\nImported from parent contexts:\n");
+                    foreach ($ghelp as $command=>$info) {
+                        printf("  %-20s  %s\n", $command, $info);
+                    }
+                }
+                printf("\nGlobal commands:\n");
+                printf("  %-20s  %s\n", "exit", "Leave the shell");
+                printf("  %-20s  %s\n", "..", "Discard the current context and go to parent");
                 break;
             case 'exit':
                 $this->stop();
@@ -256,7 +317,7 @@ class Shell
     {
         $this->lineReader = new LineRead();
 
-        $this->lineReader->setPromptText("shell>");
+        $this->lineReader->setPromptText($this->prompt);
         $this->lineReader->setPromptStyle(new Style(Style::BR_GREEN));
         $this->lineReader->setCommandStyle(new Style(Style::GREEN));
 
@@ -269,14 +330,23 @@ class Shell
             if (!($buffer = $this->lineReader->update())) {
                 usleep(10000);
             }
+            // Escape is handy too...
+            if ($buffer == "\e") {
+                $this->dispatchEvent("shell.abort");
+                continue;
+            }
             // we get a ^C on ^C, so deal with the ^C.
             if ($buffer == "\x03") {
+                $this->dispatchEvent("shell.stop");
                 $this->stop();
                 continue;
             }
             // Execute the buffer
             ob_start();
             $this->dispatchEvent("update");
+            foreach ($this->timers as $timer) {
+                $timer->update();
+            }
             if ($buffer) {
                 $this->executeBuffer($buffer);
             }