pronto/core/record_selector.php

Go to the documentation of this file.

<?php
class RecordSelector
{
        var $model    = null;

        var $sql_where = null;
        var $sql_order = null;
        var $sql_limit = null;

        var $it_ptr;
        var $id_cache;

        function RecordSelector($where, $args, &$model)
        {
                $this->_reset_state();
                $this->model = $model;
                if($where) {
                        $this->sql_where = "WHERE (".$this->model->db->query($where, $args).")";
                }
        }

        function where()
        {
                $this->_reset_state();
                $args = func_get_args();
                $q    = array_shift($args);
                if(empty($q)) return $this;

                if($this->sql_where) {
                        $this->sql_where .= " AND (".$this->model->db->query($q, $args).")";
                } else {
                        $this->sql_where = "WHERE (".$this->model->db->query($q, $args).")";
                }
                return $this;
        }

        function eq($col, $val)
        {
                return $this->where("$col='%s'", $val);
        }

        function match($record)
        {
                $t = $this;
                foreach($record as $k=>$v) {
                        $t = $this->eq($k, $v);
                }
                return $t;
        }

        function order($orderby)
        {
                $this->_reset_state();
                if($this->sql_order) {
                        $this->sql_order .= ",$orderby";
                } else {
                        $this->sql_order = "ORDER BY $orderby";
                }
                return $this;
        }

        function limit($num, $offset=0)
        {
                $this->_reset_state();
                $this->sql_limit = "LIMIT $num OFFSET $offset";
                return $this;
        }

        function fetch()
        {
                $q = "SELECT * FROM {$this->model->table}".$this->_build_clause();
                $ret = $this->model->db->get_all($q);
                switch(count($ret)) {
                        case 0:  return false;
                        case 1:  return $ret[0];
                        default: return $ret;
                }
        }

        function count()
        {
                $q = "SELECT COUNT (*) FROM {$this->model->table}".$this->_build_clause();
                return $this->model->db->get_value($q);
        }

        function load($ret2d=false)
        {
                // let the model do the load
                $ids = $this->_get_ids();
                $ret = array();
                foreach($ids as $id) $ret[] = $this->model->load($id);
                switch(count($ret)) {
                        case 0:  return $ret2d ? array() : false;
                        case 1:  return $ret2d ? $ret : $ret[0];
                        default: return $ret;
                }
        }

        function load_all()
        {
                return $this->load(true);
        }

        function load_one()
        {
                $ids = $this->_get_ids();
                if(!isset($ids[$this->it_ptr])) return false;
                return $this->model->load($ids[$this->it_ptr++]);
        }

        function one()
        {
                return $this->load_one();
        }

        function delete()
        {
                // let the model remove the rows
                $ids = $this->_get_ids();
                foreach($ids as $id) $this->model->delete($id);
        }

        function set($key, $val=null)
        {
                if(is_array($key)) {
                        $kp = $args = array();
                        foreach($key as $k=>$v) {
                                $kp[]   = "\"$k\"='%s'";
                                $args[] = $v;
                        }
                        $q = $this->model->db->query("UPDATE {$this->model->table} SET ".implode(',',$kp).$this->_build_clause(), $args);
                } else {
                        $q = $this->model->db->query("UPDATE {$this->model->table} SET \"$key\"='%s'".$this->_build_clause(), array($val));
                }
                // invalidate cache entries
                if($this->model->enable_cache && $this->model->cache) {
                        $ids = $this->_get_ids();
                        foreach($ids as $id) $this->model->invalidate($id);
                }
                return $this->model->db->execute($q);
        }

        function get()
        {
                $cols = array();
                foreach(func_get_args() as $a) $cols[] = "\"$a\"";
                $q = "SELECT ".implode(',',$cols)." FROM {$this->model->table}".$this->_build_clause();
                $rows = $this->model->db->get_all($q);

                if(empty($rows)) return false;

                $cols = func_get_args();
                $ret  = array();
                if(count($cols) > 1) {
                        foreach($rows as $row) {
                                $rec = array();
                                foreach($cols as $col) $rec[$col] = $row[$col];
                                $ret[] = $rec;
                        }
                } else {
                        foreach($rows as $row) $ret[] = $row[$cols[0]];
                }

                if(count($rows) == 1) {
                        return $ret[0];
                }
                return $ret;
        }

        function pair()
        {
                if(func_num_args() < 2) return false;
                $args = func_get_args();
                $rows = call_user_func_array(array($this,'get'), $args);
                if(!isset($rows[0])) $rows = array($rows);

                $ret = array();
                foreach($rows as $row) {
                        if(count($row) > 2) {
                                $ret[$row[$args[0]]] = array();
                                for($i = 1; $i < count($args); $i++) $ret[$row[$args[0]]][] = $row[$args[$i]];
                        } else {
                                $ret[$row[$args[0]]] = $row[$args[1]];
                        }
                }
                return $ret;
        }

        function _get_ids()
        {
                if(!empty($this->id_cache)) {
                        return $this->id_cache;
                }
                $ids = $this->get('id');
                assert_type($ids, 'array');
                $this->id_cache = $ids;
                return $ids;
        }

        function _reset_state()
        {
                $this->it_ptr   = 0;
                $this->id_cache = array();
        }

        function _build_clause()
        {
                $sql = '';
                if($this->sql_where) $sql .= ' '.$this->sql_where;
                if($this->sql_order) $sql .= ' '.$this->sql_order;
                if($this->sql_limit) $sql .= ' '.$this->sql_limit;
                return $sql;
        }
}

?>

---