Database\Selector\TableRow.php

  1: <?php
  2: 
  3: /**
  4:  * This file is part of the Nette Framework (http://nette.org)
  5:  *
  6:  * Copyright (c) 2004, 2011 David Grudl (http://davidgrudl.com)
  7:  *
  8:  * For the full copyright and license information, please view
  9:  * the file license.txt that was distributed with this source code.
 10:  */
 11: 
 12: namespace Nette\Database\Selector;
 13: 
 14: use Nette;
 15: 
 16: 
 17: 
 18: /**
 19:  * Single row representation.
 20:  * Selector is based on the great library NotORM http://www.notorm.com written by Jakub Vrana.
 21:  *
 22:  * @author     Jakub Vrana
 23:  */
 24: class TableRow extends Nette\Object implements \IteratorAggregate, \ArrayAccess
 25: {
 26:     /** @var TableSelection */
 27:     protected $table;
 28: 
 29:     /** @var array of row data */
 30:     protected $data;
 31: 
 32:     /** @var array of new values {@see TableRow::update()} */
 33:     private $modified = array();
 34: 
 35: 
 36: 
 37:     public function __construct(array $data, TableSelection $table)
 38:     {
 39:         $this->data = $data;
 40:         $this->table = $table;
 41:     }
 42: 
 43: 
 44: 
 45:     /**
 46:      * Returns primary key value.
 47:      * @return string
 48:      */
 49:     public function __toString()
 50:     {
 51:         return (string) $this[$this->table->primary]; // (string) - PostgreSQL returns int
 52:     }
 53: 
 54: 
 55: 
 56:     /**
 57:      * @return array
 58:      */
 59:     public function toArray()
 60:     {
 61:         $this->access(NULL);
 62:         return $this->data;
 63:     }
 64: 
 65: 
 66: 
 67:     /**
 68:      * Returns referenced row.
 69:      * @param  string
 70:      * @return TableRow or NULL if the row does not exist
 71:      */
 72:     public function ref($name)
 73:     {
 74:         $referenced = $this->table->getReferencedTable($name, $column);
 75:         if (isset($referenced[$this[$column]])) { // referenced row may not exist
 76:             $res = $referenced[$this[$column]];
 77:             return $res;
 78:         }
 79:     }
 80: 
 81: 
 82: 
 83:     /**
 84:      * Returns referencing rows.
 85:      * @param  string table name
 86:      * @return GroupedTableSelection
 87:      */
 88:     public function related($table)
 89:     {
 90:         $referencing = $this->table->getReferencingTable($table);
 91:         $referencing->active = $this[$this->table->primary];
 92:         return $referencing;
 93:     }
 94: 
 95: 
 96: 
 97:     /**
 98:      * Updates row.
 99:      * @param  array or NULL for all modified values
100:      * @return int number of affected rows or FALSE in case of an error
101:      */
102:     public function update($data = NULL)
103:     {
104:         if ($data === NULL) {
105:             $data = $this->modified;
106:         }
107:         return $this->table->connection->table($this->table->name)->where($this->table->primary, $this[$this->table->primary])->update($data);
108:     }
109: 
110: 
111: 
112:     /**
113:      * Deletes row.
114:      * @return int number of affected rows or FALSE in case of an error
115:      */
116:     public function delete()
117:     {
118:         return $this->table->connection->table($this->table->name)->where($this->table->primary, $this[$this->table->primary])->delete();
119:     }
120: 
121: 
122: 
123:     /********************* interface IteratorAggregate ****************d*g**/
124: 
125: 
126: 
127:     public function getIterator()
128:     {
129:         $this->access(NULL);
130:         return new \ArrayIterator($this->data);
131:     }
132: 
133: 
134: 
135:     /********************* interface ArrayAccess & magic accessors ****************d*g**/
136: 
137: 
138: 
139:     /**
140:      * Stores value in column.
141:      * @param  string column name
142:      * @return NULL
143:      */
144:     public function offsetSet($key, $value)
145:     {
146:         $this->__set($key, $value);
147:     }
148: 
149: 
150: 
151:     /**
152:      * Returns value of column.
153:      * @param  string column name
154:      * @return string
155:      */
156:     public function offsetGet($key)
157:     {
158:         return $this->__get($key);
159:     }
160: 
161: 
162: 
163:     /**
164:      * Tests if column exists.
165:      * @param  string column name
166:      * @return bool
167:      */
168:     public function offsetExists($key)
169:     {
170:         return $this->__isset($key);
171:     }
172: 
173: 
174: 
175:     /**
176:      * Removes column from data.
177:      * @param  string column name
178:      * @return NULL
179:      */
180:     public function offsetUnset($key)
181:     {
182:         $this->__unset($key);
183:     }
184: 
185: 
186: 
187:     public function __set($key, $value)
188:     {
189:         $this->data[$key] = $value;
190:         $this->modified[$key] = $value;
191:     }
192: 
193: 
194: 
195:     public function &__get($key)
196:     {
197:         if (array_key_exists($key, $this->data)) {
198:             $this->access($key);
199:             return $this->data[$key];
200:         }
201: 
202:         $column = $this->table->connection->databaseReflection->getReferencedColumn($key, $this->table->name);
203:         if (array_key_exists($column, $this->data)) {
204:             $value = $this->data[$column];
205:             $referenced = $this->table->getReferencedTable($key);
206:             $ret = isset($referenced[$value]) ? $referenced[$value] : NULL; // referenced row may not exist
207:             return $ret;
208:         }
209: 
210:         $this->access($key);
211:         if (array_key_exists($key, $this->data)) {
212:             return $this->data[$key];
213: 
214:         } else {
215:             $this->access($key, TRUE);
216: 
217:             $this->access($column);
218:             if (array_key_exists($column, $this->data)) {
219:                 $value = $this->data[$column];
220:                 $referenced = $this->table->getReferencedTable($key);
221:                 $ret = isset($referenced[$value]) ? $referenced[$value] : NULL; // referenced row may not exist
222: 
223:             } else {
224:                 $this->access($column, TRUE);
225:                 trigger_error("Unknown column $key", E_USER_WARNING);
226:                 $ret = NULL;
227:             }
228:             return $ret;
229:         }
230:     }
231: 
232: 
233: 
234:     public function __isset($key)
235:     {
236:         $this->access($key);
237:         $return = array_key_exists($key, $this->data);
238:         if (!$return) {
239:             $this->access($key, TRUE);
240:         }
241:         return $return;
242:     }
243: 
244: 
245: 
246:     public function __unset($key)
247:     {
248:         unset($this->data[$key]);
249:         unset($this->modified[$key]);
250:     }
251: 
252: 
253: 
254:     public function access($key, $delete = FALSE)
255:     {
256:         if ($this->table->connection->cache && $this->table->access($key, $delete)) {
257:             $this->data = $this->table[$this->data[$this->table->primary]]->data;
258:         }
259:     }
260: 
261: }
262:
Namespaces

Classes

Interfaces
