<?php

/**
 * sysPass
 *
 * @author    nuxsmin
 * @link      http://syspass.org
 * @copyright 2012-2015 Rubén Domínguez nuxsmin@syspass.org
 *
 * This file is part of sysPass.
 *
 * sysPass 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 3 of the License, or
 * (at your option) any later version.
 *
 * sysPass 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 sysPass.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

namespace SP;

defined('APP_ROOT') || die(_('No es posible acceder directamente a este archivo'));

define('IS_INSTALLER', 1);

/**
 * Esta clase es la encargada de instalar sysPass.
 */
class Installer
{
    /**
     * @var string Usuario de la BD
     */
    private static $_dbuser;
    /**
     * @var string Clave de la BD
     */
    private static $_dbpass;
    /**
     * @var string Nombre de la BD
     */
    private static $_dbname;
    /**
     * @var string Host de la BD
     */
    private static $_dbhost;
    /**
     * @var \PDO Instancia a de conexión a la BD
     */
    private static $_dbc;
    /**
     * @var string Usuario 'admin' de sysPass
     */
    private static $_username;
    /**
     * @var string Clave del usuario 'admin' de sysPass
     */
    private static $_password;
    /**
     * @var string Clave maestra de sysPass
     */
    private static $_masterPassword;
    /**
     * @var bool Activar/desactivar Modo hosting
     */
    private static $_isHostingMode;

    /**
     * @param string $dbuser
     */
    public static function setDbuser($dbuser)
    {
        self::$_dbuser = $dbuser;
    }

    /**
     * @param string $dbpass
     */
    public static function setDbpass($dbpass)
    {
        self::$_dbpass = $dbpass;
    }

    /**
     * @param string $dbname
     */
    public static function setDbname($dbname)
    {
        self::$_dbname = $dbname;
    }

    /**
     * @param string $dbhost
     */
    public static function setDbhost($dbhost)
    {
        self::$_dbhost = $dbhost;
    }

    /**
     * @param string $username
     */
    public static function setUsername($username)
    {
        self::$_username = $username;
    }

    /**
     * @param string $password
     */
    public static function setPassword($password)
    {
        self::$_password = $password;
    }

    /**
     * @param string $masterPassword
     */
    public static function setMasterPassword($masterPassword)
    {
        self::$_masterPassword = $masterPassword;
    }

    /**
     * @param boolean $isHostingMode
     */
    public static function setIsHostingMode($isHostingMode)
    {
        self::$_isHostingMode = $isHostingMode;
    }

    /**
     * Iniciar instalación.
     *
     * @return array resultado del proceso
     */
    public static function install()
    {
        $error = array();

        if (!self::$_username) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar nombre de usuario admin'),
                'hint' => _('Usuario admin para acceso a la aplicación'));
        } elseif (!self::$_password) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar la clave de admin'),
                'hint' => _('Clave del usuario admin de la aplicación'));
        } elseif (!self::$_masterPassword) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar la clave maestra'),
                'hint' => _('Clave maestra para encriptar las claves'));
        } elseif (strlen(self::$_masterPassword) < 11) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Clave maestra muy corta'),
                'hint' => _('La longitud de la clave maestra ha de ser mayor de 11 caracteres'));
        } elseif (!self::$_dbuser) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar el usuario de la BBDD'),
                'hint' => _('Usuario con permisos de administrador de la Base de Datos'));
        } elseif (!self::$_dbpass) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar la clave de la BBDD'),
                'hint' => _('Clave del usuario administrador de la Base de Datos'));
        } elseif (!self::$_dbname) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar el nombre de la BBDD'),
                'hint' => _('Nombre para la BBDD de la aplicación pej. syspass'));
        } elseif (substr_count(self::$_dbname, '.') >= 1) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('El nombre de la BBDD no puede contener "."'),
                'hint' => _('Elimine los puntos del nombre de la Base de Datos'));
        } elseif (!self::$_dbhost) {
            $error[] = array(
                'type' => SPException::SP_CRITICAL,
                'description' => _('Indicar el servidor de la BBDD'),
                'hint' => _('Servidor donde se instalará la Base de Datos'));
        }

        if (count($error) === 0) { //no errors, good
            // Generate a random salt that is used to salt the local user passwords
            Config::setValue('passwordsalt', Util::generate_random_bytes(30));
            Config::setValue('version', implode(Util::getVersion(true)));

            if (preg_match('/(.*):(\d{1,5})/', self::$_dbhost, $match)) {
                self::setDbhost($match[1]);
                $dbport = $match[2];
            } else {
                $dbport = 3306;
            }

            // Save DB connection info
            Config::setValue('dbhost', self::$_dbhost);
            Config::setValue('dbname', self::$_dbname);

            // Set some basic configuration options
            Config::setDefaultValues();

            try {
                self::checkDatabaseAdmin(self::$_dbhost, self::$_dbuser, self::$_dbpass, $dbport);
                self::setupMySQLDatabase();
                self::createAdminAccount();
            } catch (SPException $e) {
                $error[] = array(
                    'type' => $e->getType(),
                    'description' => $e->getMessage(),
                    'hint' => $e->getHint());
                return $error;
            }

            Config::setConfigDbValue('version', implode(Util::getVersion(true)));
            Config::setValue('installed', 1);
        }

        return $error;
    }

    /**
     * Comprobar la conexión con la BBDD.
     * Comprobar si la conexión con la base de datos para sysPass es posible con
     * los datos facilitados.
     *
     * @param string $dbhost  host de conexión
     * @param string $dbadmin usuario de conexión
     * @param string $dbpass  clave de conexión
     * @param string $dbport  puerto de conexión
     * @throws SPException
     */
    private static function checkDatabaseAdmin($dbhost, $dbadmin, $dbpass, $dbport)
    {
        try {
            $dsn = 'mysql:host=' . $dbhost . ';dbport=' . $dbport . ';charset=utf8';
            self::$_dbc = new \PDO($dsn, $dbadmin, $dbpass);
        } catch (\PDOException $e) {
            throw new SPException(SPException::SP_CRITICAL
                , _('No es posible conectar con la BD')
                , _('Compruebe los datos de conexión') . '<br>' . $e->getMessage());
        }
    }

    /**
     * Configurar la base de datos.
     * Esta función crea la base de datos y el usuario necesario para sysPass.
     *
     * @throws SPException
     */
    private static function setupMySQLDatabase()
    {
        // Comprobar si el usuario sumistrado existe
        $query = "SELECT COUNT(*) "
            . "FROM mysql.user "
            . "WHERE user='" . self::$_username . "' AND host='" . self::$_dbhost . "'";

        // Si no es modo hosting se crea un hash para la clave y un usuario con prefijo "sp_" para la DB
        if (!self::$_isHostingMode){
            self::setDbpass(md5(time() . self::$_password));
            self::setDbuser(substr('sp_' . self::$_username, 0, 16));
        }

        // Si no existe el usuario, se intenta crear
        if (intval(self::$_dbc->query($query)->fetchColumn()) === 0) {
            // Se comprueba si el nuevo usuario es distinto del creado en otra instalación
            if (self::$_dbuser != Config::getValue('dbuser')) {
                self::createDBUser();
            }

            // Guardar el nuevo usuario/clave de conexión a la BD
            Config::setValue('dbuser', self::$_dbuser);
            Config::setValue('dbpass', self::$_dbpass);
        }

        try {
            self::createMySQLDatabase(self::$_dbpass);
            self::createDBStructure();
        } catch (SPException $e){
            throw $e;
        }
    }

    /**
     * Crear el usuario para conectar con la base de datos.
     * Esta función crea el usuario para conectar con la base de datos.
     * Si se marca en modo hosting, no se crea el usuario.
     *
     * @throws SPException
     */
    private static function createDBUser()
    {
        if (self::$_isHostingMode) {
            return;
        }

        $query = "CREATE USER '" . self::$_dbuser . "'@'localhost' IDENTIFIED BY '" . self::$_dbpass . "'";

        try {
            self::$_dbc->query($query);
        } catch (\PDOException $e) {
            throw new SPException(SPException::SP_CRITICAL
                , _('El usuario de MySQL ya existe') . " (" . self::$_dbuser . ")"
                , _('Indique un nuevo usuario o elimine el existente'));
        }
    }

    /**
     * Crear la base de datos en MySQL.
     *
     * @throws SPException
     */
    private static function createMySQLDatabase()
    {
        if (self::checkDatabaseExist()) {
            throw new SPException(SPException::SP_CRITICAL
                , _('La BBDD ya existe')
                , _('Indique una nueva Base de Datos o elimine la existente'));
        }

        $query = "CREATE DATABASE IF NOT EXISTS `" . self::$_dbname . "`";

        try {
            self::$_dbc->query($query);
        } catch (\PDOException $e) {
            throw new SPException(SPException::SP_CRITICAL
                , _('Error al crear la BBDD') . " (" . $e->getMessage() . ")"
                , _('Verifique los permisos del usuario de la Base de Datos'));
        }

        if (!self::$_isHostingMode) {
            $query = "GRANT ALL PRIVILEGES ON `" . self::$_dbname . "`.* TO '" . self::$_dbuser . "'@'" . self::$_dbhost . "' IDENTIFIED BY '" . self::$_dbpass . "';";

            try {
                self::$_dbc->query($query);
            } catch (\PDOException $e) {
                throw new SPException(SPException::SP_CRITICAL
                    , _('Error al establecer permisos de la BBDD') . " (" . $e->getMessage() . ")"
                    , _('Verifique los permisos del usuario de la Base de Datos'));
            }
        }
    }

    /**
     * Comprobar si la base de datos indicada existe.
     *
     * @return bool
     */
    private static function checkDatabaseExist()
    {
        $query = "SELECT COUNT(*) "
            . "FROM information_schema.tables "
            . "WHERE table_schema = '" . self::$_dbname . "' "
            . "AND table_name = 'usrData' LIMIT 1";

        return (intval(self::$_dbc->query($query)->fetchColumn()) > 0);
    }

    /**
     * Crear la estructura de la base de datos.
     * Esta función crea la estructura de la base de datos a partir del archivo dbsctructure.sql.
     *
     * @throws SPException
     */
    private static function createDBStructure()
    {
        $fileName = dirname(__FILE__) . '/dbstructure.sql';

        if (!file_exists($fileName)) {
            throw new SPException(SPException::SP_CRITICAL
                , _('El archivo de estructura de la BBDD no existe')
                , _('No es posible crear la BBDD de la aplicación. Descárguela de nuevo.'));
        }

        // Usar la base de datos de sysPass
        try {
            self::$_dbc->query('USE ' . self::$_dbname);
        } catch (\PDOException $e) {
            throw new SPException(SPException::SP_CRITICAL
                , _('Error al seleccionar la BBDD') . " '" . self::$_dbname . "' (" . $e->getMessage() . ")"
                , _('No es posible usar la Base de Datos para crear la estructura. Compruebe los permisos y que no exista.'));
        }

        // Leemos el archivo SQL para crear las tablas de la BBDD
        $handle = fopen($fileName, 'rb');

        if ($handle) {
            while (!feof($handle)) {
                $buffer = stream_get_line($handle, 1000000, ";\n");
                if (strlen(trim($buffer)) > 0) {
                    try {
                        self::$_dbc->query($buffer);
                    } catch (\PDOException $e) {
                        // drop database on error
                        self::$_dbc->query("DROP DATABASE " . self::$_dbname . ";");

                        throw new SPException(SPException::SP_CRITICAL
                            , _('Error al crear la BBDD') . ' (' . $e->getMessage() . ')'
                            , _('Error al crear la estructura de la Base de Datos.'));
                    }
                }
            }
        }
    }

    /**
     * Crear el usuario admin de sysPass.
     * Esta función crea el grupo, perfil y usuario 'admin' para utilizar sysPass.
     *
     * @throws SPException
     */
    private static function createAdminAccount()
    {
        // Datos del grupo
        Groups::$groupName = "Admins";
        Groups::$groupDescription = "Admins";

        if (!Groups::addGroup()) {
            self::rollback();

            throw new SPException(SPException::SP_CRITICAL
                , _('Error al crear el grupo "admin"')
                , _('Informe al desarrollador'));
        }

        $user = new Users;

        // Establecer el id de grupo del usuario al recién creado
        $user->userGroupId = Groups::$queryLastId;

        $profile = new Profile();

        $profile->setName('Admin');
        $profile->setAccAdd(true);
        $profile->setAccView(true);
        $profile->setAccViewPass(true);
        $profile->setAccViewHistory(true);
        $profile->setAccEdit(true);
        $profile->setAccEditPass(true);
        $profile->setAccDelete(true);
        $profile->setConfigGeneral(true);
        $profile->setConfigEncryption(true);
        $profile->setConfigBackup(true);
        $profile->setMgmCategories(true);
        $profile->setMgmCustomers(true);
        $profile->setMgmUsers(true);
        $profile->setMgmGroups(true);
        $profile->setMgmProfiles(true);
        $profile->setEvl(true);

        if (!$profile->profileAdd()) {
            self::rollback();

            throw new SPException(SPException::SP_CRITICAL
                , _('Error al crear el perfil "admin"')
                , _('Informe al desarrollador'));
        }

        // Datos del usuario
        $user->userLogin = self::$_username;
        $user->userPass = self::$_password;
        $user->userName = 'Admin';
        $user->userProfileId = $profile->getId();
        $user->userIsAdminApp = 1;
        $user->userIsAdminAcc = 0;
        $user->userIsDisabled = 0;

        if (!$user->addUser()) {
            self::rollback();

            throw new SPException(SPException::SP_CRITICAL
                , _('Error al crear el usuario "admin"')
                , _('Informe al desarrollador'));
        }

        // Guardar el hash de la clave maestra
        Config::setArrConfigValue('masterPwd', Crypt::mkHashPassword(self::$_masterPassword));
        Config::setArrConfigValue('lastupdatempass', time());
        Config::writeConfigDb(true);

        if (!$user->updateUserMPass(self::$_masterPassword, false)) {
            self::rollback();

            throw new SPException(SPException::SP_CRITICAL
                , _('Error al actualizar la clave maestra del usuario "admin"')
                , _('Informe al desarrollador'));
        }
    }

    /**
     * Deshacer la instalación en caso de fallo.
     * Esta función elimina la base de datos y el usuario de sysPass
     */
    private static function rollback()
    {
        try {
            self::$_dbc->query("DROP DATABASE IF EXISTS " . self::$_dbname . ";");
            self::$_dbc->query("DROP USER '" . self::$_dbuser . "'@'" . self::$_dbhost . "';");
            self::$_dbc->query("DROP USER '" . self::$_dbuser . "'@'%';");
        } catch (\PDOException $e) {
            Config::deleteKey('dbuser');
            Config::deleteKey('dbpass');
        }
    }

}