Fixes #3

jimbojsb · Jan 4, 2013 · ec92b64 · ec92b64
1 parent 6b7f5bb
commit ec92b64
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/lib/Bstools/App.php b/lib/Bstools/App.php
@@ -19,5 +19,6 @@ public function __construct()
         $this->add(new \Bstools\Command\Drain());
         $this->add(new \Bstools\Command\Peek());
         $this->add(new \Bstools\Command\Bury());
+        $this->add(new \Bstools\Command\StatsJob());
     }
 }
diff --git a/lib/Bstools/Command/Peek.php b/lib/Bstools/Command/Peek.php
@@ -14,21 +14,22 @@ public function configure()
         $this->setName('peek')
              ->setDescription('Peek at the job on top of the ready or buried queue');
         $this->addArgument('tube', InputArgument::REQUIRED, 'the tube to drain from');
-        $this->addOption('buried', null, InputOption::VALUE_NONE, 'drain from buried instead of ready');
+        $this->addArgument('state', InputArgument::OPTIONAL, 'peek from buried instead of ready', 'ready');
     }
 
     public function execute(InputInterface $input, OutputInterface $output)
     {
         $pheanstalk = new \Pheanstalk($input->getOption('host'));
 
         $tube = $input->getArgument('tube');
-        $buried = $input->getOption('buried');
+        $state = $input->getArgument('state');
+        $permissibleStates = array('ready', 'buried', 'delayed');
+        if (!in_array($state, $permissibleStates)) {
+            throw new \Exception("$state is not a valid job state");
+        }
         try {
-            if ($buried) {
-                $job = $pheanstalk->peekBuried($tube);
-            } else {
-                $job = $pheanstalk->peekReady($tube);
-            }
+            $cmd = "peek" . ucfirst(strtolower($state));
+            $job = $pheanstalk->$cmd($tube);
         } catch (\Exception $e) {
             return;
         }