diff --git a/framework/console/Controller.php b/framework/console/Controller.php index 042a084..1a6b047 100644 --- a/framework/console/Controller.php +++ b/framework/console/Controller.php @@ -24,10 +24,42 @@ use yii\base\Exception; * ~~~ * * @author Qiang Xue + * @author Carsten Brandt * @since 2.0 */ class Controller extends \yii\base\Controller { + const FG_COLOR_BLACK = 30; + const FG_COLOR_RED = 31; + const FG_COLOR_GREEN = 32; + const FG_COLOR_YELLOW = 33; + const FG_COLOR_BLUE = 34; + const FG_COLOR_PURPLE = 35; + const FG_COLOR_CYAN = 36; + const FG_COLOR_GREY = 37; + + const BG_COLOR_BLACK = 40; + const BG_COLOR_RED = 41; + const BG_COLOR_GREEN = 42; + const BG_COLOR_YELLOW = 43; + const BG_COLOR_BLUE = 44; + const BG_COLOR_PURPLE = 45; + const BG_COLOR_CYAN = 46; + const BG_COLOR_GREY = 47; + + const TEXT_BOLD = 1; + const TEXT_ITALIC = 3; + const TEXT_UNDERLINE = 4; + const TEXT_BLINK = 5; + const TEXT_NEGATIVE = 7; + const TEXT_CONCEALED = 8; + const TEXT_CROSSED_OUT = 9; + const TEXT_FRAMED = 51; + const TEXT_ENCIRCLED = 52; + const TEXT_OVERLINED = 53; + + public $color = null; + /** * This method is invoked when the request parameters do not satisfy the requirement of the specified action. * The default implementation will throw an exception. @@ -112,4 +144,426 @@ class Controller extends \yii\base\Controller $input = trim(fgets(STDIN)); return empty($input) ? $default : !strncasecmp($input, 'y', 1); } + + /** + * Moves the terminal cursor up by sending ANSI code CUU to the terminal. + * If the cursor is already at the edge of the screen, this has no effect. + * @param integer $rows number of rows the cursor should be moved up + */ + public function moveCursorUp($rows=1) + { + echo "\033[" . (int) $rows . 'A'; + } + + /** + * Moves the terminal cursor down by sending ANSI code CUD to the terminal. + * If the cursor is already at the edge of the screen, this has no effect. + * @param integer $rows number of rows the cursor should be moved down + */ + public function moveCursorDown($rows=1) + { + echo "\033[" . (int) $rows . 'B'; + } + + /** + * Moves the terminal cursor forward by sending ANSI code CUF to the terminal. + * If the cursor is already at the edge of the screen, this has no effect. + * @param integer $steps number of steps the cursor should be moved forward + */ + public function moveCursorForward($steps=1) + { + echo "\033[" . (int) $steps . 'C'; + } + + /** + * Moves the terminal cursor backward by sending ANSI code CUB to the terminal. + * If the cursor is already at the edge of the screen, this has no effect. + * @param integer $steps number of steps the cursor should be moved backward + */ + public function moveCursorBackward($steps=1) + { + echo "\033[" . (int) $steps . 'D'; + } + + /** + * Moves the terminal cursor to the beginning of the next line by sending ANSI code CNL to the terminal. + * @param integer $lines number of lines the cursor should be moved down + */ + public function moveCursorNextLine($lines=1) + { + echo "\033[" . (int) $lines . 'E'; + } + + /** + * Moves the terminal cursor to the beginning of the previous line by sending ANSI code CPL to the terminal. + * @param integer $lines number of lines the cursor should be moved up + */ + public function moveCursorPrevLine($lines=1) + { + echo "\033[" . (int) $lines . 'F'; + } + + /** + * Moves the cursor to an absolute position given as column and row by sending ANSI code CUP or CHA to the terminal. + * @param integer $column 1-based column number, 1 is the left edge of the screen. + * @param integer|null $row 1-based row number, 1 is the top edge of the screen. if not set, will move cursor only in current line. + */ + public function moveCursorTo($column, $row=null) + { + if ($row === null) { + echo "\033[" . (int) $column . 'G'; + } else { + echo "\033[" . (int) $row . ';' . (int) $column . 'H'; + } + } + + /** + * Scrolls whole page up by sending ANSI code SU to the terminal. + * New lines are added at the bottom. This is not supported by ANSI.SYS used in windows. + * @param int $lines number of lines to scroll up + */ + public function scrollUp($lines=1) + { + echo "\033[".(int)$lines."S"; + } + + /** + * Scrolls whole page down by sending ANSI code SD to the terminal. + * New lines are added at the top. This is not supported by ANSI.SYS used in windows. + * @param int $lines number of lines to scroll down + */ + public function scrollDown($lines=1) + { + echo "\033[".(int)$lines."T"; + } + + /** + * Saves the current cursor position by sending ANSI code SCP to the terminal. + * Position can then be restored with {@link restoreCursorPosition}. + */ + public function saveCursorPosition() + { + echo "\033[s"; + } + + /** + * Restores the cursor position saved with {@link saveCursorPosition} by sending ANSI code RCP to the terminal. + */ + public function restoreCursorPosition() + { + echo "\033[u"; + } + + /** + * Hides the cursor by sending ANSI DECTCEM code ?25l to the terminal. + * Use {@link showCursor} to bring it back. + * Do not forget to show cursor when your application exits. Cursor might stay hidden in terminal after exit. + */ + public function hideCursor() + { + echo "\033[?25l"; + } + + /** + * Will show a cursor again when it has been hidden by {@link hideCursor} by sending ANSI DECTCEM code ?25h to the terminal. + */ + public function showCursor() + { + echo "\033[?25h"; + } + + /** + * Clears entire screen content by sending ANSI code ED with argument 2 to the terminal. + * Cursor position will not be changed. + * **Note:** ANSI.SYS implementation used in windows will reset cursor position to upper left corner of the screen. + */ + public function clearScreen() + { + echo "\033[2J"; + } + + /** + * Clears text from cursor to the beginning of the screen by sending ANSI code ED with argument 1 to the terminal. + * Cursor position will not be changed. + */ + public function clearScreenBeforeCursor() + { + echo "\033[1J"; + } + + /** + * Clears text from cursor to the end of the screen by sending ANSI code ED with argument 0 to the terminal. + * Cursor position will not be changed. + */ + public function clearScreenAfterCursor() + { + echo "\033[0J"; + } + + + /** + * Clears the line, the cursor is currently on by sending ANSI code EL with argument 2 to the terminal. + * Cursor position will not be changed. + */ + public function clearLine() + { + echo "\033[2K"; + } + + /** + * Clears text from cursor position to the beginning of the line by sending ANSI code EL with argument 1 to the terminal. + * Cursor position will not be changed. + */ + public function clearLineBeforeCursor() + { + echo "\033[1K"; + } + + /** + * Clears text from cursor position to the end of the line by sending ANSI code EL with argument 0 to the terminal. + * Cursor position will not be changed. + */ + public function clearLineAfterCursor() + { + echo "\033[0K"; + } + + /** + * Will send ANSI format for following output + * + * You can pass any of the FG_*, BG_* and TEXT_* constants and also xterm256ColorBg + * TODO: documentation + */ + public function ansiStyle() + { + echo "\033[" . implode(';', func_get_args()) . 'm'; + } + + /** + * Will return a string formatted with the given ANSI style + * + * See {@link ansiStyle} for possible arguments. + * @param string $string the string to be formatted + * @return string + */ + public function ansiStyleString($string) + { + $args = func_get_args(); + array_shift($args); + $code = implode(';', $args); + return "\033[0m" . ($code !== '' ? "\033[" . $code . "m" : '') . $string."\033[0m"; + } + + //const COLOR_XTERM256 = 38;// http://en.wikipedia.org/wiki/Talk:ANSI_escape_code#xterm-256colors + public function xterm256ColorFg($i) // TODO naming! + { + return '38;5;'.$i; + } + + public function xterm256ColorBg($i) // TODO naming! + { + return '48;5;'.$i; + } + + /** + * Usage: list($w, $h) = $this->getScreenSize(); + * + * @return array + */ + public function getScreenSize() + { + // TODO implement + return array(150,50); + } + + /** + * resets any ansi style set by previous method {@link ansiStyle} + * Any output after this is will have default text style. + */ + public function reset() + { + echo "\033[0m"; + } + + /** + * Strips ANSI control codes from a string + * + * @param string $string String to strip + * @return string + */ + function strip($string) + { + return preg_replace('/\033\[[\d;]+m/', '', $string); // TODO currently only strips color + } + + // TODO refactor and review + public function ansiToHtml($string) + { + $tags = 0; + return preg_replace_callback('/\033\[[\d;]+m/', function($ansi) use (&$tags) { + $styleA = array(); + foreach(explode(';', $ansi) as $controlCode) + { + switch($controlCode) + { + case static::FG_COLOR_BLACK: $style = array('color' => '#000000'); break; + case static::FG_COLOR_BLUE: $style = array('color' => '#000078'); break; + case static::FG_COLOR_CYAN: $style = array('color' => '#007878'); break; + case static::FG_COLOR_GREEN: $style = array('color' => '#007800'); break; + case static::FG_COLOR_GREY: $style = array('color' => '#787878'); break; + case static::FG_COLOR_PURPLE: $style = array('color' => '#780078'); break; + case static::FG_COLOR_RED: $style = array('color' => '#780000'); break; + case static::FG_COLOR_YELLOW: $style = array('color' => '#787800'); break; + case static::BG_COLOR_BLACK: $style = array('background-color' => '#000000'); break; + case static::BG_COLOR_BLUE: $style = array('background-color' => '#000078'); break; + case static::BG_COLOR_CYAN: $style = array('background-color' => '#007878'); break; + case static::BG_COLOR_GREEN: $style = array('background-color' => '#007800'); break; + case static::BG_COLOR_GREY: $style = array('background-color' => '#787878'); break; + case static::BG_COLOR_PURPLE: $style = array('background-color' => '#780078'); break; + case static::BG_COLOR_RED: $style = array('background-color' => '#780000'); break; + case static::BG_COLOR_YELLOW: $style = array('background-color' => '#787800'); break; + case static::TEXT_BOLD: $style = array('font-weight' => 'bold'); break; + case static::TEXT_ITALIC: $style = array('font-style' => 'italic'); break; + case static::TEXT_UNDERLINE: $style = array('text-decoration' => array('underline')); break; + case static::TEXT_OVERLINED: $style = array('text-decoration' => array('overline')); break; + case static::TEXT_CROSSED_OUT:$style = array('text-decoration' => array('line-through')); break; + case static::TEXT_BLINK: $style = array('text-decoration' => array('blink')); break; + case static::TEXT_NEGATIVE: // ??? + case static::TEXT_CONCEALED: + case static::TEXT_ENCIRCLED: + case static::TEXT_FRAMED: + // TODO allow resetting codes + break; + case 0: // ansi reset + $return = ''; + for($n=$tags; $tags>0; $tags--) { + $return .= ''; + } + return $return; + } + $styleA = \yii\util\ArrayHelper::merge($styleA, $style); + } + $styleString[] = array(); + foreach($styleA as $name => $content) { + if ($name = 'text-decoration') { + $content = implode(' ', $content); + } + $styleString[] = $name.':'.$content; + } + $tags++; + return ' 1) { + return $this->buildCompositeInCondition($operator, $column, $values); + } else { + $column = reset($column); + foreach ($values as $i => $value) { + if (is_array($value)) { + $values[$i] = isset($value[$column]) ? $value[$column] : null; + } else { + $values[$i] = null; + } + } + } + } + foreach ($values as $i => $value) { - $values[$i] = is_string($value) ? $this->connection->quoteValue($value) : (string)$value; + if ($value === null) { + $values[$i] = 'NULL'; + } else { + $values[$i] = is_string($value) ? $this->connection->quoteValue($value) : (string)$value; + } } if (strpos($column, '(') === false) { $column = $this->quoteColumnName($column); } - $operator = strtoupper($operator); return "$column $operator (" . implode(', ', $values) . ')'; } + protected function buildCompositeInCondition($operator, $columns, $values) + { + foreach ($columns as $i => $column) { + if (strpos($column, '(') === false) { + $columns[$i] = $this->quoteColumnName($column); + } + } + $vss = array(); + foreach ($values as $value) { + $vs = array(); + foreach ($columns as $column) { + if (isset($value[$column])) { + $vs[] = is_string($value[$column]) ? $this->connection->quoteValue($value[$column]) : (string)$value[$column]; + } else { + $vs[] = 'NULL'; + } + } + $vss[] = '(' . implode(', ', $vs) . ')'; + } + return '(' . implode(', ', $columns) . ") $operator (" . implode(', ', $vss) . ')'; + } + private function buildLikeCondition($operator, $operands) { if (!isset($operands[0], $operands[1])) { @@ -595,21 +634,20 @@ class QueryBuilder extends \yii\base\Object $values = (array)$values; if ($values === array()) { - return $operator === 'like' || $operator === 'or like' ? '0=1' : ''; + return $operator === 'LIKE' || $operator === 'OR LIKE' ? '0=1' : ''; } - if ($operator === 'like' || $operator === 'not like') { + if ($operator === 'LIKE' || $operator === 'NOT LIKE') { $andor = ' AND '; } else { $andor = ' OR '; - $operator = $operator === 'or like' ? 'like' : 'not like'; + $operator = $operator === 'OR LIKE' ? 'LIKE' : 'NOT LIKE'; } if (strpos($column, '(') === false) { $column = $this->quoteColumnName($column); } - $operator = strtoupper($operator); $parts = array(); foreach ($values as $value) { $parts[] = "$column $operator " . $this->connection->quoteValue($value); @@ -726,7 +764,7 @@ class QueryBuilder extends \yii\base\Object $table = $driver->quoteTableName($table); } } - $joins[$i] = strtoupper($join[0]) . ' ' . $table; + $joins[$i] = $join[0] . ' ' . $table; if (isset($join[2])) { $condition = $this->buildCondition($join[2]); if ($condition !== '') { diff --git a/tests/unit/framework/db/ar/ActiveRecordTest.php b/tests/unit/framework/db/ar/ActiveRecordTest.php index 9b169eb..923345b 100644 --- a/tests/unit/framework/db/ar/ActiveRecordTest.php +++ b/tests/unit/framework/db/ar/ActiveRecordTest.php @@ -23,42 +23,14 @@ class ActiveRecordTest extends \yiiunit\MysqlTestCase $this->assertTrue($result instanceof ActiveQuery); $customer = $result->one(); $this->assertTrue($customer instanceof Customer); - $this->assertEquals(1, $result->count); - $this->assertEquals(1, count($result)); // find all $result = Customer::find(); $customers = $result->all(); - $this->assertTrue(is_array($customers)); $this->assertEquals(3, count($customers)); $this->assertTrue($customers[0] instanceof Customer); $this->assertTrue($customers[1] instanceof Customer); $this->assertTrue($customers[2] instanceof Customer); - $this->assertEquals(3, $result->count); - $this->assertEquals(3, count($result)); - - // check count first - $result = Customer::find(); - $this->assertEquals(3, $result->count); - $customer = $result->one(); - $this->assertTrue($customer instanceof Customer); - $this->assertEquals(3, $result->count); - - // iterator - $result = Customer::find(); - $count = 0; - foreach ($result as $customer) { - $this->assertTrue($customer instanceof Customer); - $count++; - } - $this->assertEquals($count, $result->count); - - // array access - $result = Customer::find(); - $this->assertTrue($result[0] instanceof Customer); - $this->assertTrue($result[1] instanceof Customer); - $this->assertTrue($result[2] instanceof Customer); - $this->assertEquals(3, count($result)); // find by a single primary key $customer = Customer::find(2); @@ -70,7 +42,7 @@ class ActiveRecordTest extends \yiiunit\MysqlTestCase $this->assertTrue($customer instanceof Customer); $this->assertEquals(2, $customer->id); - // find by Query + // find by Query array $query = array( 'where' => 'id=:id', 'params' => array(':id' => 2), @@ -80,13 +52,59 @@ class ActiveRecordTest extends \yiiunit\MysqlTestCase $this->assertEquals('user2', $customer->name); // find count - $this->assertEquals(3, Customer::find()->count()); - $this->assertEquals(3, Customer::count()); - $this->assertEquals(2, Customer::count(array( - 'where' => 'id=1 OR id=2', - ))); - $this->assertEquals(2, Customer::count()->where('id=1 OR id=2')); +// $this->assertEquals(3, Customer::count()); +// $this->assertEquals(2, Customer::count(array( +// 'where' => 'id=1 OR id=2', +// ))); +// $this->assertEquals(2, Customer::find()->select('COUNT(*)')->where('id=1 OR id=2')->value()); + } + + public function testFindBySql() + { + // find one + $customer = Customer::findBySql('SELECT * FROM tbl_customer ORDER BY id DESC')->one(); + $this->assertTrue($customer instanceof Customer); + $this->assertEquals('user3', $customer->name); + + // find all + $customers = Customer::findBySql('SELECT * FROM tbl_customer')->all(); + $this->assertEquals(3, count($customers)); + + // find with parameter binding + $customer = Customer::findBySql('SELECT * FROM tbl_customer WHERE id=:id', array(':id' => 2))->one(); + $this->assertTrue($customer instanceof Customer); + $this->assertEquals('user2', $customer->name); + } + + public function testScope() + { + $customers = Customer::find(array( + 'scopes' => array('active'), + ))->all(); + $this->assertEquals(2, count($customers)); + + $customers = Customer::find()->active()->all(); + $this->assertEquals(2, count($customers)); } +// +// public function testFindLazy() +// { +// $customer = Customer::find(2); +// $orders = $customer->orders; +// $this->assertEquals(2, count($orders)); +// +// $orders = $customer->orders()->where('id=3')->all(); +// $this->assertEquals(1, count($orders)); +// $this->assertEquals(3, $orders[0]->id); +// } +// +// public function testFindEager() +// { +// $customers = Customer::find()->with('orders')->all(); +// $this->assertEquals(3, count($customers)); +// $this->assertEquals(1, count($customers[0]->orders)); +// $this->assertEquals(2, count($customers[1]->orders)); +// } // public function testInsert() // {