ModelLoader

ModelLoader.php

<?php

namespace Proj;

use	Nette\Debug,
	Nette\Environment,
	PDO,
	NotORM,
	NotOrmPanel,
	NotORM_Structure_Convention;


class ModelLoader extends \Nette\Object
{

	private $dbConnection = null;
	private $models = array();


	public function __construct()
	{
		$this->dbConnect();
	}

	public function dbConnect(DbConfiguration $cfg)
	{
		$pdo = new PDO("{$cfg->driver}:host={$cfg->host};dbname={$cfg->database}", $cfg->username, $cfg->password);
        $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
		$pdo->query('SET NAMES utf8');

		$conn = new NotORM($pdo, new NotORM_Structure_Convention('id', '%s_id', '%ss'));

        if ($cfg->profiler) {
            $panel = NotOrmPanel::getInstance();
            $panel->setPlatform($cfg->driver);
            Debug::addPanel($panel);

            $conn->debug = function($query, $parameters) {
                NotOrmPanel::getInstance()->logQuery($query, $parameters);
            };
        }

		return $this->dbConnection = $conn;
	}

	public function getConnection()
	{
		return (null === $this->dbConnection) ? $this->dbConnect() : $this->dbConnection;
	}

	public function getModel($model)
	{
		if (!isset($this->models[$model])) {
			$class = 'Model\\' . ucfirst($model);
			$this->models[$model] = new $class($this->dbConnection);
		}
		
		return $this->models[$model];
	}

}