1. <?php
  4. class Sluggable extends DataExtension {
  6. private static $db = array (
  7. 'Slug' => 'Varchar(255)'
  8. );
  10. private static $indexes = array (
  11. 'Slug' => true
  12. );
  14. public static function create_slug($obj, $field) {
  15. if(!isset($_REQUEST[$field])) {
  16. $title = $obj->$field;
  17. }
  18. else {
  19. $title = $_REQUEST[$field];
  20. }
  21. $slug = singleton('SiteTree')->generateURLSegment($title);
  22. $original_slug = $slug;
  23. $i = 0;
  24. $class = $obj->class;
  25. if(!Object::has_extension($class, "Sluggable")) {
  26. while($parent = get_parent_class($obj)) {
  27. if(Object::has_extension($parent,"Sluggable")) {
  28. $class = $parent;
  29. break;
  30. }
  31. else {
  32. $obj = singleton($parent);
  33. }
  34. }
  35. }
  36. while($t = DataList::create($class)
  37. ->filter(array("Slug" => "$slug"))
  38. ->exclude(array("{$class}.ID" => $obj->ID))
  39. ->first()) {
  40. $i++;
  41. $slug = $original_slug."-{$i}";
  42. }
  43. $obj->Slug = $slug;
  44. }
  47. public static function get_by_slug($class, $slug) {
  48. return DataList::create($class)->filter("Slug",$slug)->first();
  49. }
  52. public function onBeforeWrite() {
  53. $f = $this->owner->config()->slug_field;
  54. $field = $f ? $f : "Title";
  55. self::create_slug($this->owner, $field);
  56. }
  57. }