135 lines
4.4 KiB
PHP
135 lines
4.4 KiB
PHP
<?php
|
|
class GridFieldSortableObject extends DataExtension {
|
|
public static $db=array(
|
|
'SortOrder'=>'Int'
|
|
);
|
|
|
|
protected static $sortable_classes = array();
|
|
protected static $many_many_sortable_relations = array();
|
|
protected static $sort_dir = "ASC";
|
|
|
|
|
|
public static function set_sort_dir($dir) {
|
|
self::$sort_dir=$dir;
|
|
}
|
|
|
|
|
|
public static function add_sortable_class($className) {
|
|
if(!self::is_sortable_class($className)) {
|
|
Object::add_extension($className, 'GridFieldSortableObject');
|
|
|
|
self::$sortable_classes[]=$className;
|
|
}
|
|
}
|
|
|
|
public static function add_sortable_classes(array $classes) {
|
|
foreach($classes as $class) {
|
|
self::add_sortable_class($class);
|
|
}
|
|
}
|
|
|
|
public static function add_sortable_many_many_relation($ownerClass, $componentName) {
|
|
list($parentClass, $componentClass, $parentField, $componentField, $table)=singleton($ownerClass)->many_many($componentName);
|
|
|
|
Object::add_static_var($ownerClass, 'many_many_extraFields', array(
|
|
$componentName=>array(
|
|
'SortOrder'=>'Int'
|
|
)));
|
|
|
|
|
|
if(!isset(self::$many_many_sortable_relations[$componentClass])) {
|
|
self::$many_many_sortable_relations[$componentClass] = array();
|
|
}
|
|
|
|
|
|
self::$many_many_sortable_relations[$componentClass][$parentClass]=$table;
|
|
self::add_sortable_class($componentClass);
|
|
}
|
|
|
|
public static function remove_sortable_class($class) {
|
|
Object::remove_extension($class, 'GridFieldSortableObject');
|
|
}
|
|
|
|
public static function is_sortable_class($classname) {
|
|
if(in_array($classname, self::$sortable_classes)) {
|
|
return true;
|
|
}
|
|
|
|
foreach(self::$sortable_classes as $class) {
|
|
if(is_subclass_of($classname, $class)) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
|
|
return Object::has_extension($classname, 'GridFieldSortableObject');
|
|
}
|
|
|
|
public static function is_sortable_many_many($componentClass, $parentClass=null) {
|
|
$map=self::$many_many_sortable_relations;
|
|
if($parentClass===null) {
|
|
return isset($map[$componentClass]);
|
|
}else {
|
|
if(isset($map[$componentClass])) {
|
|
return isset($map[$componentClass][$parentClass]);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
public static function get_join_tables($classname) {
|
|
if(isset(self::$many_many_sortable_relations[$classname])) {
|
|
return self::$many_many_sortable_relations[$classname];
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public function augmentSQL(SQLQuery &$query) {
|
|
if(empty($query->select) || $query->delete || in_array("COUNT(*)", $query->select) || in_array("count(*)", $query->select)) {
|
|
return;
|
|
}
|
|
|
|
|
|
$sort_field=false;
|
|
if($join_tables=self::get_join_tables($this->owner->class)) {
|
|
foreach($query->from as $from) {
|
|
if($sort_field) {
|
|
break;
|
|
}
|
|
|
|
foreach($join_tables as $join_table) {
|
|
if(stristr($from,$join_table)) {
|
|
$sort_field="\"$join_table\".\"SortOrder\"";
|
|
|
|
if(isset($query->select['SortOrder'])) {
|
|
$query->select['SortOrder']="\"{$this->owner->class}\".SortOrder AS LocalSort";
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if(!$sort_field) {
|
|
$sort_field="\"SortOrder\"";
|
|
}
|
|
|
|
if(!$query->orderby || ($query->orderby==$this->owner->stat('default_sort'))) {
|
|
$query->orderby="$sort_field ".self::$sort_dir;
|
|
}
|
|
}
|
|
|
|
public function onBeforeWrite() {
|
|
if(!$this->owner->ID) {
|
|
if($peers=DataList::create($this->owner->class)) {
|
|
$this->owner->SortOrder=$peers->Count()+1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
?>
|