# -*- coding: utf-8 -*-
# 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 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
import click

import kallithea
import kallithea.bin.kallithea_cli_base as cli_base
import kallithea.lib.utils
import kallithea.model.scm
from kallithea.lib.db_manage import DbManage
from kallithea.model import meta


@cli_base.register_command(needs_config_file=True, config_file_initialize_app=True)
@click.option('--reuse/--no-reuse', default=False,
        help='Reuse and clean existing database instead of dropping and creating (default: no reuse)')
@click.option('--user', help='Username of administrator account.')
@click.option('--password', help='Password for administrator account.')
@click.option('--email', help='Email address of administrator account.')
@click.option('--repos', help='Absolute path to repositories location.')
@click.option('--force-yes', is_flag=True, help='Answer yes to every question.')
@click.option('--force-no', is_flag=True, help='Answer no to every question.')
@click.option('--public-access/--no-public-access', default=True,
        help='Enable/disable public access on this installation (default: enable)')
def db_create(user, password, email, repos, force_yes, force_no, public_access, reuse, config=None):
    """Initialize the database.

    Create all required tables in the database specified in the configuration
    file. Create the administrator account. Set certain settings based on
    values you provide.

    You can pass the answers to all questions as options to this command.
    """
    if config is not None:  # first called with config, before app initialization
        dbconf = config['sqlalchemy.url']

        # force_ask should be True (yes), False (no), or None (ask)
        if force_yes:
            force_ask = True
        elif force_no:
            force_ask = False
        else:
            force_ask = None

        cli_args = dict(
                username=user,
                password=password,
                email=email,
                repos_location=repos,
                force_ask=force_ask,
                public_access=public_access,
        )
        dbmanage = DbManage(dbconf=dbconf, root=config['here'],
                            cli_args=cli_args)
        dbmanage.create_tables(reuse_database=reuse)
        repo_root_path = dbmanage.prompt_repo_root_path(None)
        dbmanage.create_settings(repo_root_path)
        dbmanage.create_default_user()
        dbmanage.create_admin_user()
        dbmanage.create_permissions()
        dbmanage.populate_default_permissions()
        meta.Session().commit()

    else:  # then called again after app initialization
        added, _ = kallithea.lib.utils.repo2db_mapper(kallithea.model.scm.ScmModel().repo_scan())
        if added:
            click.echo('Initial repository scan: added following repositories:')
            click.echo('\t%s' % '\n\t'.join(added))
        else:
            click.echo('Initial repository scan: no repositories found.')

        click.echo('Database set up successfully.')
        click.echo("Don't forget to build the front-end using 'kallithea-cli front-end-build'.")