sintonia/library/propel/generator/lib/behavior/TimestampableBehavior.php

<?php

/**
 * This file is part of the Propel package.
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 *
 * @license    MIT License
 */

/**
 * Gives a model class the ability to track creation and last modification dates
 * Uses two additional columns storing the creation and update date
 *
 * @author     François Zaninotto
 * @version    $Revision: 1683 $
 * @package    propel.generator.behavior
 */
class TimestampableBehavior extends Behavior
{
	// default parameters value
	protected $parameters = array(
		'create_column' => 'created_at',
		'update_column' => 'updated_at'
	);
	
	/**
	 * Add the create_column and update_columns to the current table
	 */
	public function modifyTable()
	{
		if(!$this->getTable()->containsColumn($this->getParameter('create_column'))) {
			$this->getTable()->addColumn(array(
				'name' => $this->getParameter('create_column'),
				'type' => 'TIMESTAMP'
			));
		}
		if(!$this->getTable()->containsColumn($this->getParameter('update_column'))) {
			$this->getTable()->addColumn(array(
				'name' => $this->getParameter('update_column'),
				'type' => 'TIMESTAMP'
			));
		}
	}
	
	/**
	 * Get the setter of one of the columns of the behavior
	 * 
	 * @param     string $column One of the behavior colums, 'create_column' or 'update_column'
	 * @return    string The related setter, 'setCreatedOn' or 'setUpdatedOn'
	 */
	protected function getColumnSetter($column)
	{
		return 'set' . $this->getColumnForParameter($column)->getPhpName();
	}
	
	/**
	 * Add code in ObjectBuilder::preUpdate
	 *
	 * @return    string The code to put at the hook
	 */
	public function preUpdate()
	{
		return "if (\$this->isModified() && !\$this->isColumnModified(" . $this->getColumnForParameter('update_column')->getConstantName() . ")) {
	\$this->" . $this->getColumnSetter('update_column') . "(time());
}";
	}
	
	/**
	 * Add code in ObjectBuilder::preInsert
	 *
	 * @return    string The code to put at the hook
	 */
	public function preInsert()
	{
		return "if (!\$this->isColumnModified(" . $this->getColumnForParameter('create_column')->getConstantName() . ")) {
	\$this->" . $this->getColumnSetter('create_column') . "(time());
}
if (!\$this->isColumnModified(" . $this->getColumnForParameter('update_column')->getConstantName() . ")) {
	\$this->" . $this->getColumnSetter('update_column') . "(time());
}";
	}

	public function objectMethods($builder)
	{
		return "
/**
 * Mark the current object so that the update date doesn't get updated during next save
 *
 * @return     " . $builder->getStubObjectBuilder()->getClassname() . " The current object (for fluent API support)
 */
public function keepUpdateDateUnchanged()
{
	\$this->modifiedColumns[] = " . $this->getColumnForParameter('update_column')->getConstantName() . ";
	return \$this;
}
";
	}
	
	public function queryMethods($builder)
	{
		$queryClassName = $builder->getStubQueryBuilder()->getClassname();
		$updateColumnConstant = $this->getColumnForParameter('update_column')->getConstantName();
		$createColumnConstant = $this->getColumnForParameter('create_column')->getConstantName();
		return "
/**
 * Filter by the latest updated
 *
 * @param      int \$nbDays Maximum age of the latest update in days
 *
 * @return     $queryClassName The current query, for fuid interface
 */
public function recentlyUpdated(\$nbDays = 7)
{
	return \$this->addUsingAlias($updateColumnConstant, time() - \$nbDays * 24 * 60 * 60, Criteria::GREATER_EQUAL);
}

/**
 * Filter by the latest created
 *
 * @param      int \$nbDays Maximum age of in days
 *
 * @return     $queryClassName The current query, for fuid interface
 */
public function recentlyCreated(\$nbDays = 7)
{
	return \$this->addUsingAlias($createColumnConstant, time() - \$nbDays * 24 * 60 * 60, Criteria::GREATER_EQUAL);
}

/**
 * Order by update date desc
 *
 * @return     $queryClassName The current query, for fuid interface
 */
public function lastUpdatedFirst()
{
	return \$this->addDescendingOrderByColumn($updateColumnConstant);
}

/**
 * Order by update date asc
 *
 * @return     $queryClassName The current query, for fuid interface
 */
public function firstUpdatedFirst()
{
	return \$this->addAscendingOrderByColumn($updateColumnConstant);
}

/**
 * Order by create date desc
 *
 * @return     $queryClassName The current query, for fuid interface
 */
public function lastCreatedFirst()
{
	return \$this->addDescendingOrderByColumn($createColumnConstant);
}

/**
 * Order by create date asc
 *
 * @return     $queryClassName The current query, for fuid interface
 */
public function firstCreatedFirst()
{
	return \$this->addAscendingOrderByColumn($createColumnConstant);
}
";
	}
}