<?php defined('SYSPATH') or die('No direct access allowed.');

/**
 * This class overrides Kohana's DB
 *
 * @package    OSB/Modifications
 * @category   Classes
 * @category   Helpers
 * @author     Deon George
 * @copyright  (c) 2010 Deon George
 * @license    http://dev.leenooks.net/license.html
 */
class DB extends Kohana_DB {
	// Add the site_id to the delete query
	public static function delete($table = NULL)
	{
		$db = new Database_Query_Builder_Delete($table);

		if (! in_array($table,Config::$no_site_id_tables))
			return $db->where($table.'.site_id','=',Config::siteid());
		else
			return $db;
	}

	// Add the site_id to the update query
	final public static function update($table = NULL)
	{
		$db = new Database_Query_Builder_Update($table);

		if (! in_array($table,Config::$no_site_id_tables))
			return $db->where($table.'.site_id','=',Config::siteid());
		else
			return $db;
	}
}
?>