<?php
// {{{ICINGA_LICENSE_HEADER}}}
/**
 * This file is part of Icinga Web 2.
 *
 * Icinga Web 2 - Head for multiple monitoring backends.
 * Copyright (C) 2013 Icinga Development Team
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * @copyright  2013 Icinga Development Team <info@icinga.org>
 * @license    http://www.gnu.org/licenses/gpl-2.0.txt GPL, version 2
 * @author     Icinga Development Team <info@icinga.org>
 *
 */
// {{{ICINGA_LICENSE_HEADER}}}

namespace Test\Monitoring\Testlib\Datasource\Strategies;

/**
 * SetupStrategy implementation for MySQL based IDO-Backends
 *
 * This strategy creates a new MySQL Database and removes old ones
 * if necessary. Per default the database user is icinga_unittest:icinga_unittest
 * and the database to be created is also icinga_unittest. The user must
 * have all Privileges on the test database, so its best to create him
 * with:
 *
 * mysql -u root
 * # CREATE USER `icinga_unittest`@`localhost` IDENTIFIED BY 'icinga_unittest';
 * # CREATE DATABASE icinga_unittest;
 * # GRANT ALL PRIVILEGES ON icinga_unittest.* TO `icinga_unittest`@`localhost`
 *
 **/
class MySQLSetupStrategy implements SetupStrategy {

    /**
     * Tears down any existing databases and creates a new blank IDO scheme.
     *
     * The database is created according to the passed version (or using the newest version if no version is provided),
     * using the idoMySQL-%VERSION%.sql underneath the schemes folder.
     * A \PDO Connection can be provided, if not the icinga_unittest default
     * connection will be established and used.
     *
     * @param String $version   An optional version to use as the db scheme
     * @param \PDO $connection  An optional connection to use instead of icinga_unittest
     * @return \PDO             The connection that has been created
     *
     * @throws \PDOException    In case connecting to or creating the database fails
     * @throws \Exception       In case of an invalid/non-existing DB scheme
     */
    public function setup($version = null, $connection = null)
    {
        if ($connection === null) {
            $connection = new \PDO("mysql:dbname=icinga_unittest", "icinga_unittest", "icinga_unittest");
        }

        $connection->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
        $this->teardown($connection);

        // the latest schema doesn't have a suffix, so if no version is given this one is used
        $sqlFileName = 'idoMySQL'.($version !== null ? '-'.$version : '' ).'.sql';
        $path = realpath(dirname(__FILE__).'/../schemes/'.$sqlFileName);
        if (!file_exists($path)) {
            throw new \Exception('File '.$path.' not found: Could not create scheme for IDO mysql backend '.($version ? '(version : '.$version.')' :''));
        }

        $connection->setAttribute(\PDO::ATTR_EMULATE_PREPARES, true);

        if ($connection->query(file_get_contents($path)) === false) {
            $error = $connection->errorInfo();;
            throw new \PDOException($error[0].' : '.$error[2]);
        }
        return $connection;
    }

    /**
     * Drops all tables from the connection via DROP TABLE
     *
     * @param \PDO $connection  An optional connection to use, if none is
     *                          given the icinga_unittest default will be used
     *
     */
    public function teardown($connection = null)
    {
        if ($connection === null) {
            $connection = new \PDO("mysql:dbname=icinga_unittest", "icinga_unittest", "icinga_unittest");
        }

        $tables = $connection->query("SHOW TABLES")->fetchAll();
        foreach($tables as $table) {
            $connection->query("DROP TABLE ".$table[0]);
        }
    }
}