<?php
	/* Doctrine integration */
	require Kohana::find_file('classes', 'doctrine/Doctrine');
	spl_autoload_register(array('Doctrine', 'autoload'));
	// Getting kohana configurations for doctrine
	$db = Kohana::config('database');
	// initializing manager
	$manager = Doctrine_Manager::getInstance();
 
	// we load our database connections into Doctrine_Manager
	// this loop allows us to use multiple connections later on
	foreach ($db as $connection_name => $db_values) {
		// first we must convert to dsn format
		$dsn = $db[$connection_name]['type'] .
			'://' . $db[$connection_name]['connection']['username'] .
			':' . $db[$connection_name]['connection']['password'].
			'@' . $db[$connection_name]['connection']['hostname'] .
 			'/' . $db[$connection_name]['connection']['database'];
 
		Doctrine_Manager::connection($dsn, $connection_name);
	}
 
	// telling Doctrine where our models are located
	if ( is_dir(APPPATH.'doctrine/models') )
		Doctrine::loadModels(APPPATH.'doctrine/models');
 
	// (OPTIONAL) CONFIGURATION BELOW
 
	// this will allow us to use "mutators"
	$manager->setAttribute(Doctrine_Core::ATTR_AUTO_ACCESSOR_OVERRIDE, true );
	// Automatically free queries
	$manager->setAttribute(Doctrine_Core::ATTR_AUTO_FREE_QUERY_OBJECTS, true );
	// Enable validation
	//$manager->setAttribute(Doctrine::ATTR_VALIDATE, Doctrine::VALIDATE_ALL);
	// this sets all table columns to notnull and unsigned (for ints) by default
	$manager->setAttribute(
	Doctrine_Core::ATTR_DEFAULT_COLUMN_OPTIONS,
		array('notnull' => true, 'unsigned' => true)
	);
	// set the default primary key to be named 'id', integer, 4 bytes
	$manager->setAttribute(
	Doctrine_Core::ATTR_DEFAULT_IDENTIFIER_OPTIONS,
		array('name' => 'id', 'type' => 'integer', 'length' => 4)
	);

<?php defined('SYSPATH') or die('No direct script access.'); 
 
class Controller_Doctrine extends Controller
{
	public function before()
	{
		parent::before(); 
 
		//Restrict controller to localhost
		if ( !in_array($_SERVER['REMOTE_ADDR'], array('::1', '127.0.0.1')) )
		{
			echo "DENIED!";
			exit;
		}
	}
 
	function action_load_fixtures()
	{
		?>
		This will delete all existing data!<br />
		<form action="" method="POST">
			<input type="submit" name="schema" value="Load Schema"><br /><br />
		<?php 
 
		?>
		This will delete all existing data!<br />
		<form action="" method="POST">
			<input type="submit" name="data" value="Load Fixtures"><br /><br />
		<?php
 
		if ( !empty($_POST['schema']) )
		{
			Doctrine_Core::generateModelsFromYaml(
				APPPATH . DIRECTORY_SEPARATOR . 'doctrine/fixtures/schema',
				APPPATH . DIRECTORY_SEPARATOR . 'doctrine/models',
				array('generateBaseClasses'=>false)
			);
			echo "Done!";
		}
		elseif ( !empty($_POST['data']) )
		{
			Doctrine_Manager::connection()->execute("
				SET FOREIGN_KEY_CHECKS = 0
			"); 
 
			Doctrine::loadData(APPPATH . DIRECTORY_SEPARATOR . 'doctrine/fixtures/data');
			echo "Done!";
		}
	}
}

function action_create_tables()
{
	?>
	Reminder: Make sure the tables do not exist already.<br />
	<form action="" method="POST">
	<input type="submit" name="action" value="Create Tables"><br /><br />
	<?php 
 
	if ( !empty($_POST['action']) )
	{
		Doctrine::createTablesFromModels();
		echo "Done!";
	}
}