silverstripe-blog/code/extensions/BlogMemberExtension.php

<?php

/**
 * This class is responsible for add Blog specific behaviour to Members.
 *
 * @package silverstripe
 * @subpackage blog
 */
class BlogMemberExtension extends DataExtension {
	/**
	 * @var array
	 */
	private static $db = array(
		'URLSegment' => 'Varchar',
		'BlogProfileSummary' => 'Text',
	);

	/**
	 * @var array
	 */
	private static $has_one = array(
		'BlogProfileImage' => 'Image',
	);

	/**
	 * @var array
	 */
	private static $belongs_many_many = array(
		'BlogPosts' => 'BlogPost',
	);

	/**
	 * {@inheritdoc}
	 */
	public function onBeforeWrite() {
		$count = 1;

		$this->owner->URLSegment = $this->generateURLSegment();

		while(!$this->validURLSegment()) {
			$this->owner->URLSegment = preg_replace('/-[0-9]+$/', null, $this->owner->URLSegment) . '-' . $count;
			$count++;
		}
	}

	/**
	 * Generate a unique URL segment based on the Member's name.
	 *
	 * @return string
	 */
	public function generateURLSegment() {
		$filter = URLSegmentFilter::create();
		$name = $this->owner->FirstName . ' ' . $this->owner->Surname;
		$urlSegment = $filter->filter($name);

		if(!$urlSegment || $urlSegment == '-' || $urlSegment == '-1') {
			$urlSegment = 'profile-' . $this->owner->ID;
		}

		return $urlSegment;
	}

	/**
	 * Returns TRUE if this object has a URL segment value that does not conflict with any other
	 * objects.
	 *
	 * @return bool
	 */
	public function validURLSegment() {
		$conflict = Member::get()->filter('URLSegment', $this->owner->URLSegment);

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

		return $conflict->count() == 0;
	}


	/**
	 * {@inheritdoc}
	 */
	public function updateCMSFields(FieldList $fields) {
		$fields->removeByName('URLSegment');

		// Remove the automatically-generated posts tab.

		$fields->removeFieldFromTab('Root', 'BlogPosts');

		// Construct a better posts tab.

		Requirements::css(BLOGGER_DIR . '/css/cms.css');
		Requirements::javascript(BLOGGER_DIR . '/js/cms.js');

		$tab = new Tab('BlogPosts', 'Blog Posts');

		$gridField = new GridField(
			'BlogPosts',
			'Blog Posts',
			$this->owner->BlogPosts(),
			new GridFieldConfig_BlogPost()
		);

		$tab->Fields()->add($gridField);

		$fields->addFieldToTab('Root', $tab);

		return $fields;
	}
}