More unit tests.
[platal.git] / classes / plfilter.php
index 6cdeb70..05c1edb 100644 (file)
  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA                *
  ***************************************************************************/
 
+// {{{ class PlLimit
+class PlLimit
+{
+    private $count = null;
+    private $from  = null;
+
+    public function __construct($count = null, $from = null)
+    {
+        $this->count = $count;
+        $this->from  = $from;
+    }
+
+    public function getSql()
+    {
+        if (!is_null($this->count) && $this->count != 0) {
+            if (!is_null($this->from) && $this->from != 0) {
+                return XDB::format('LIMIT {?}, {?}', (int)$this->from, (int)$this->count);
+            } else {
+                return XDB::format('LIMIT {?}', (int)$this->count);
+            }
+        }
+        return '';
+    }
+}
+// }}}
+
+// {{{ class PlSqlJoin
 class PlSqlJoin
 {
     private $mode;
@@ -29,10 +56,12 @@ class PlSqlJoin
     const MODE_RIGHT = 'RIGHT';
     const MODE_INNER = 'INNER';
 
-    public function __construct($mode, $table, $condition)
+    private function __construct($mode, $params)
     {
+        $table = array_shift($params);
+        $condition = call_user_func_array(array('XDB', 'format'), $params);
         if ($mode != self::MODE_LEFT && $mode != self::MODE_RIGHT && $mode != self::MODE_INNER) {
-            Platal::page()->kill("Join mode error : unknown mode $mode");
+            Platal::page()->kill("Join mode error: unknown mode $mode");
             return;
         }
         $this->mode = $mode;
@@ -50,52 +79,333 @@ class PlSqlJoin
         return $this->table;
     }
 
-    public function condiftion()
+    public function condition()
     {
         return $this->condition;
     }
-}
-
-abstract class PlFilter
-{
-    public abstract function filter(array $objects, $count = null, $offset = null);
-
-    public abstract function setCondition(PlFilterCondition &$cond);
-
-    public abstract function addSort(PlFilterOrder &$sort);
 
-    private function replaceJoinMetas($cond, $key)
+    /** Replace all "metas" in the condition with their translation.
+     * $ME always becomes the alias of the table
+     * @param $key The name the joined table will have in the final query
+     * @param $joinMetas An array of meta => conversion to apply to the condition
+     */
+    public function replaceJoinMetas($key, $joinMetas = array())
     {
-        return str_replace(array('$ME'), array($key), $cond);
+        $joinMetas['$ME'] = $key;
+        return str_replace(array_keys($joinMetas), $joinMetas, $this->condition);
     }
 
-    private function formatJoin(array $joins)
+    /** Create a join command from an array of PlSqlJoin
+     * @param $joins The list of 'join' to convert into an SQL query
+     * @param $joinMetas An array of ('$META' => 'conversion') to apply to the joins.
+     */
+    public static function formatJoins(array $joins, array $joinMetas)
     {
         $str = '';
         foreach ($joins as $key => $join) {
-            if (! $join instanceof PlSqlJoin) {
-                Platal::page()->kill("Error : not a join : $join");
+            if (!($join instanceof PlSqlJoin)) {
+                Platal::page()->kill("Error: not a join: $join");
             }
             $mode  = $join->mode();
             $table = $join->table();
             $str .= ' ' . $mode . ' JOIN ' . $table . ' AS ' . $key;
             if ($join->condition() != null) {
-                $str .= ' ON (' . $this->replaceJoinMetas($join->condition(), $key) . ')';
+                $str .= ' ON (' . $join->replaceJoinMetas($key, $joinMetas) . ')';
             }
             $str .= "\n";
         }
         return $str;
     }
 
-    private function buildJoins()
+    /** Build a left join
+     * @param table The name of the table.
+     * @param condition The condition of the jointure
+     */
+    public static function left()
+    {
+        $params = func_get_args();
+        return new PlSqlJoin(self::MODE_LEFT, $params);
+    }
+
+    /** Build a right join
+     * @param table The name of the table.
+     * @param condition The condition of the jointure
+     */
+    public static function right()
+    {
+        $params = func_get_args();
+        return new PlSqlJoin(self::MODE_RIGHT, $params);
+    }
+
+    /** Build a inner join
+     * @param table The name of the table.
+     * @param condition The condition of the jointure
+     */
+    public static function inner()
+    {
+        $params = func_get_args();
+        return new PlSqlJoin(self::MODE_INNER, $params);
+    }
+}
+// }}}
+
+// {{{ class PlFilterOrder
+abstract class PlFilterOrder
+{
+    protected $desc = false;
+    public function __construct($desc = false)
+    {
+        $this->desc = $desc;
+    }
+
+    public function toggleDesc()
+    {
+        $this->desc = !$desc;
+    }
+
+    public function setDescending($desc = true)
+    {
+        $this->desc = $desc;
+    }
+
+    public function buildSort(PlFilter &$pf)
+    {
+        $sel = $this->getSortTokens($pf);
+        if (!is_array($sel)) {
+            $sel = array($sel);
+        }
+        if ($this->desc) {
+            foreach ($sel as $k => $s) {
+                $sel[$k] = $s . ' DESC';
+            }
+        }
+        return $sel;
+    }
+
+    abstract protected function getSortTokens(PlFilter &$pf);
+}
+// }}}
+
+// {{{ class PFO_Random
+class PFO_Random extends PlFilterOrder
+{
+    private $seed = null;
+
+    public function __construct($seed = null, $desc = false)
+    {
+        parent::__construct($desc);
+        $this->seed = $seed;
+    }
+
+    protected function getSortTokens(PlFilter &$pf)
+    {
+        if ($this->seed == null) {
+            return 'RAND()';
+        } else {
+            return XDB::format('RAND({?})', $this->seed);
+        }
+    }
+}
+// }}}
+
+// {{{ interface PlFilterCondition
+interface PlFilterCondition
+{
+    const COND_TRUE  = 'TRUE';
+    const COND_FALSE = 'FALSE';
+
+    public function buildCondition(PlFilter &$pf);
+}
+// }}}
+
+// {{{ class PFC_OneChild
+abstract class PFC_OneChild implements PlFilterCondition
+{
+    protected $child;
+
+    public function __construct(&$child = null)
+    {
+        if (!is_null($child) && ($child instanceof PlFilterCondition)) {
+            $this->setChild($child);
+        }
+    }
+
+    public function setChild(PlFilterCondition &$cond)
+    {
+        $this->child =& $cond;
+    }
+}
+// }}}
+
+// {{{ class PFC_NChildren
+abstract class PFC_NChildren implements PlFilterCondition
+{
+    protected $children = array();
+
+    public function __construct()
+    {
+        $children = func_get_args();
+        foreach ($children as &$child) {
+            if (!is_null($child) && ($child instanceof PlFilterCondition)) {
+                $this->addChild($child);
+            }
+        }
+    }
+
+    public function addChild(PlFilterCondition &$cond)
+    {
+        $this->children[] =& $cond;
+    }
+
+    protected function catConds(array $cond, $op, $fallback)
+    {
+        if (count($cond) == 0) {
+            return $fallback;
+        } else if (count($cond) == 1) {
+            return $cond[0];
+        } else {
+            return '(' . implode(') ' . $op . ' (', $cond) . ')';
+        }
+    }
+}
+// }}}
+
+// {{{ class PFC_True
+class PFC_True implements PlFilterCondition
+{
+    public function buildCondition(PlFilter &$uf)
+    {
+        return self::COND_TRUE;
+    }
+}
+// }}}
+
+// {{{ class PFC_False
+class PFC_False implements PlFilterCondition
+{
+    public function buildCondition(PlFilter &$uf)
+    {
+        return self::COND_FALSE;
+    }
+}
+// }}}
+
+// {{{ class PFC_Not
+class PFC_Not extends PFC_OneChild
+{
+    public function buildCondition(PlFilter &$uf)
+    {
+        $val = $this->child->buildCondition($uf);
+        if ($val == self::COND_TRUE) {
+            return self::COND_FALSE;
+        } else if ($val == self::COND_FALSE) {
+            return self::COND_TRUE;
+        } else {
+            return 'NOT (' . $val . ')';
+        }
+    }
+}
+// }}}
+
+// {{{ class PFC_And
+class PFC_And extends PFC_NChildren
+{
+    public function buildCondition(PlFilter &$uf)
+    {
+        if (empty($this->children)) {
+            return self::COND_FALSE;
+        } else {
+            $true = self::COND_FALSE;
+            $conds = array();
+            foreach ($this->children as &$child) {
+                $val = $child->buildCondition($uf);
+                if ($val == self::COND_TRUE) {
+                    $true = self::COND_TRUE;
+                } else if ($val == self::COND_FALSE) {
+                    return self::COND_FALSE;
+                } else {
+                    $conds[] = $val;
+                }
+            }
+            return $this->catConds($conds, 'AND', $true);
+        }
+    }
+}
+// }}}
+
+// {{{ class PFC_Or
+class PFC_Or extends PFC_NChildren
+{
+    public function buildCondition(PlFilter &$uf)
+    {
+        if (empty($this->children)) {
+            return self::COND_TRUE;
+        } else {
+            $true = self::COND_TRUE;
+            $conds = array();
+            foreach ($this->children as &$child) {
+                $val = $child->buildCondition($uf);
+                if ($val == self::COND_TRUE) {
+                    return self::COND_TRUE;
+                } else if ($val == self::COND_FALSE) {
+                    $true = self::COND_FALSE;
+                } else {
+                    $conds[] = $val;
+                }
+            }
+            return $this->catConds($conds, 'OR', $true);
+        }
+    }
+}
+// }}}
+
+// {{{ class PlFilter
+abstract class PlFilter
+{
+    /** Filters objects matching the PlFilter
+     * @param $objects The objects to filter
+     * @param $limit The portion of the matching objects to show
+     */
+    public abstract function filter(array $objects, $limit = null);
+
+    public abstract function setCondition(PlFilterCondition &$cond);
+
+    public abstract function addSort(PlFilterOrder &$sort);
+
+    public abstract function getTotalCount();
+
+    /** Get objects, selecting only those within a limit
+     * @param $limit The portion of the matching objects to select
+     */
+    public abstract function get($limit = null);
+
+    /** PRIVATE FUNCTIONS
+     */
+
+    /** List of metas to replace in joins:
+     * '$COIN' => 'pan.x' means 'replace $COIN with pan.x in the condition of the joins'
+     *
+     * "$ME" => "joined table alias" is always added to these.
+     */
+    protected $joinMetas = array();
+
+    protected $joinMethods = array();
+
+    /** Build the 'join' part of the query
+     * This function will call all methods declared in self::$joinMethods
+     * to get an array of PlSqlJoin objects to merge
+     */
+    protected function buildJoins()
     {
         $joins = array();
-        foreach (self::$joinMethods as $method) {
+        foreach ($this->joinMethods as $method) {
             $joins = array_merge($joins, $this->$method());
         }
-        return $this->formatJoin($joins);
+        return PlSqlJoin::formatJoins($joins, $this->joinMetas);
     }
 
 }
+// }}}
 
+// vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 ?>