# coding: utf-8
#
# Copyright (C) 2016 Savoir-faire Linux Inc. (<www.savoirfairelinux.com>).
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

from __future__ import unicode_literals
from fabric.api import lcd, cd, task, roles, env, local, run, runs_once, execute
from fabric.contrib.console import confirm
from fabric.colors import red, green

# Import datetime
from datetime import datetime

import helpers as h

@task(alias='make')
@roles('local')
def make(action='install'):

    """
    Build the platform by running the Makefile specified in the local_vars.py configuration file.
    """

    # Update profile codebase
    h._update_profile()

    drush_opts = "--prepare-install " if action != 'update' else ''

    if (env.interactive_mode and confirm(red('Say [Y] to {} the site at {} with the French translation, if you say [n] '
                                         'the site will be installed in English only'.format(action, env.site_root)))
        ) or not env.interactive_mode:
        drush_opts += "--translations=fr "
        env.locale = True

    drush_opts += "--contrib-destination=profiles/{} ".format(env.site_profile)
    if env.interactive_mode:
        drush_opts += " --working-copy --no-gitinfofile"
    if not h.fab_exists('local', env.site_root):
        h.fab_run('local', "mkdir {}".format(env.site_root))
    with h.fab_cd('local', env.site_root):
      h.fab_run('local', 'drush make {} {} -y'.format(drush_opts, env.makefile))


@task
@roles('local')
def aliases():
    """
    Copy conf/aliases.drushrc.php in the site environment.
    """

    role = 'local'
    aliases = env.site_drush_aliases
    workspace = env.workspace

    if not h.fab_exists(role, aliases):
        h.fab_run(role, 'mkdir {}'.format(aliases))
    with h.fab_cd(role, aliases):
        # Create aliases
        if h.fab_exists(role, '{}/aliases.drushrc.php'.format(aliases)):
            h.fab_run(role, 'rm aliases.drushrc.php')
        h.fab_run(role, 'ln -s {}/conf/aliases.drushrc.php .'.format(workspace))

    print green('Drush aliases have been copied to {} directory.'.format(aliases))


@task
@roles('docker')
def updatedb():
    """
    Run the available database updates. Similar to drush updatedb.
    """

    role = 'docker'

    with h.fab_cd(role, env.docker_site_root):
      h.fab_run(role, 'drush updatedb -y')


@task
@roles('docker')
def site_install():

    """
    Run the site installation procedure.
    """

    role = 'docker'
    site_root = env.docker_site_root
    apache = env.apache_user
    profile = env.site_profile
    db_user = env.site_db_user
    db_pass = env.site_db_pass
    db_host = env.site_db_host
    db_name = env.site_db_name
    site_name = env.site_name
    site_admin_name = env.site_admin_user
    site_admin_pass = env.site_admin_pass
    site_subdir = env.site_subdir

    # Create first the database if necessary
    h._init_db('docker')

    with h.fab_cd(role, site_root):
        locale = '--locale="fr"' if env.locale else ''

        h.fab_run(role, 'sudo -u {} drush site-install {} {} --db-url=mysql://{}:{}@{}/{} --site-name={} '
                      '--account-name={} --account-pass={} --sites-subdir={} -y'.format(apache, profile, locale,
                                                                                        db_user, db_pass,
                                                                                        db_host, db_name, site_name,
                                                                                        site_admin_name,
                                                                                        site_admin_pass,
                                                                                        site_subdir))
    print green('Site installed successfully!')

    h.hook_execute('post_install', role)


@task
@roles('docker')
def archive_dump(role='docker'):
    """
    Archive the platform for release or deployment.
    :param role Default 'role' where to run the task
    """

    role = 'docker'

    with h.fab_cd(role, env.docker_site_root):
        platform = '{}-{}.tar.gz'.format(env.project_name, datetime.now().strftime('%Y%m%d_%H%M%S'))
        print(green('Cleaning previous archives'))
        h.fab_run(role, 'rm -f {}/*.tar.gz'.format(env.builddir))
        print(green('Archiving the platform'))
        h.fab_run(
            role,
            'drush archive-dump --destination={}/{} --tar-options="--exclude=.git"'.format(env.builddir, platform)
        )