silverstripe-blog/code/extensions/URLSegmentExtension.php

<?php

/**
 * Adds URLSegment functionality to Tags & Categories.
 *
 * @package silverstripe
 * @subpackage blog
 */
class URLSegmentExtension extends DataExtension
{
    /**
     * @var array
     */
    private static $db = array(
        'URLSegment' => 'Varchar(255)',
    );

    /**
     * {@inheritdoc}
     */
    public function onBeforeWrite()
    {
        if ($this->owner->BlogID) {
            $this->owner->generateURLSegment();
        }
    }

    /**
     * Generates a unique URLSegment from the title.
     *
     * @param int $increment
     *
     * @return string
     */
    public function generateURLSegment($increment = null)
    {
        $filter = new URLSegmentFilter();

        $this->owner->URLSegment = $filter->filter($this->owner->Title);

        if (is_int($increment)) {
            $this->owner->URLSegment .= '-' . $increment;
        }

        // Postgres use '' instead of 0 as an emtpy blog ID
        // Without this all the tests fail
        if (!$this->owner->BlogID) {
            $this->owner->BlogID = 0;
        }

        $duplicate = DataList::create($this->owner->ClassName)->filter(array(
            'URLSegment' => $this->owner->URLSegment,
            'BlogID' => $this->owner->BlogID,
        ));

        if ($this->owner->ID) {
            $duplicate = $duplicate->exclude('ID', $this->owner->ID);
        }

        if ($duplicate->count() > 0) {
            if (is_int($increment)) {
                $increment += 1;
            } else {
                $increment = 0;
            }

            $this->owner->generateURLSegment((int) $increment);
        }

        return $this->owner->URLSegment;
    }
}