# -*- 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 os

import sys

import uuid

from collections import defaultdict

import click

import mako.exceptions

import kallithea.bin.kallithea_cli_base as cli_base

from kallithea.lib import inifile

def show_defaults(ctx, param, value):

    # Following construct is taken from the Click documentation:

    # https://click.palletsprojects.com/en/7.x/options/#callbacks-and-eager-options

    # "The resilient_parsing flag is applied to the context if Click wants to

    # parse the command line without any destructive behavior that would change

    # the execution flow. In this case, because we would exit the program, we

    # instead do nothing."

    if not value or ctx.resilient_parsing:

        return

    for key, value in inifile.default_variables.items():

        click.echo('%s=%s' % (key, value))

    ctx.exit()

@cli_base.register_command()

@click.option('--show-defaults', callback=show_defaults,

              is_flag=True, expose_value=False, is_eager=True,

              help='Show the default values that can be overridden')

@click.argument('config_file', type=click.Path(dir_okay=False, writable=True), required=True)

@click.argument('key_value_pairs', nargs=-1)

def config_create(config_file, key_value_pairs):

    """Create a new configuration file.

    This command creates a default configuration file, possibly adding/updating

    settings you specify.

    The primary high level configuration keys and their default values are

    shown with --show-defaults . Custom values for these keys can be specified

    on the command line as key=value arguments.

    Additional key=value arguments will be patched/inserted in the [app:main]

    section ... until another section name specifies where any following values

    should go.

    """

    mako_variable_values = {

        'version': kallithea.__version__,

        'git_hook_interpreter': sys.executable,

        'user_home_path': os.path.expanduser('~'),

        'kallithea_cli_path': cli_base.kallithea_cli_path,

    }

    ini_settings = defaultdict(dict)

    section_name = None

    for parameter in key_value_pairs:

        parts = parameter.split('=', 1)

        if len(parts) == 1 and parameter.startswith('[') and parameter.endswith(']'):

            section_name = parameter

        elif len(parts) == 2:

            key, value = parts

            if section_name is None and key in inifile.default_variables:

                mako_variable_values[key] = value

            else:

                if section_name is None:

                    section_name = '[app:main]'

                ini_settings[section_name][key] = value

        else:

            raise ValueError("Invalid name=value parameter %r" % parameter)

    # use default that cannot be replaced

    mako_variable_values.update({

        'uuid': lambda: uuid.uuid4().hex,

    })

    click.echo('Creating config file using:')

    for key, value in inifile.default_variables.items():

        if isinstance(value, str):

            options = inifile.variable_options.get(key)

            if options:

                click.echo('  %s=%s  (options: %s)' % (key, mako_variable_values.get(key, value), ', '.join(options)))

            else:

                click.echo('  %s=%s' % (key, mako_variable_values.get(key, value)))

    try:

        config_file_abs = os.path.abspath(config_file)

        inifile.create(config_file_abs, mako_variable_values, ini_settings)

        click.echo('Wrote new config file in %s' % config_file_abs)

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

    except Exception:

        click.echo(mako.exceptions.text_error_template().render())

# -*- 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/>.

"""

Global configuration file for TurboGears2 specific settings in Kallithea.

This file complements the .ini file.

"""

import logging

import os

import platform

import sys

import alembic.config

import mercurial

import tg

from alembic.migration import MigrationContext

from alembic.script.base import ScriptDirectory

from sqlalchemy import create_engine

from tg import FullStackApplicationConfigurator

import kallithea.model.base

import kallithea.model.meta

from kallithea.lib import celerypylons

from kallithea.lib.utils import check_git_version, load_extensions, set_app_settings, set_indexer_config, set_vcs_config

from kallithea.lib.utils2 import asbool

from kallithea.model import db

log = logging.getLogger(__name__)

base_config = FullStackApplicationConfigurator()

base_config.update_blueprint({

    'package': kallithea,

    # Rendering Engines Configuration

    'renderers': [

        'json',

        'mako',

    ],

    'default_renderer': 'mako',

    'use_dotted_templatenames': False,

    # Configure Sessions, store data as JSON to avoid pickle security issues

    'session.enabled': True,

    'session.data_serializer': 'json',

    # Configure the base SQLALchemy Setup

    'use_sqlalchemy': True,

    'model': kallithea.model.base,

    'DBSession': kallithea.model.meta.Session,

    # Configure App without an authentication backend.

    'auth_backend': None,

    # Use custom error page for these errors. By default, Turbogears2 does not add

    # 400 in this list.

    # Explicitly listing all is considered more robust than appending to defaults,

    # in light of possible future framework changes.

    'errorpage.status_codes': [400, 401, 403, 404],

    # Disable transaction manager -- currently Kallithea takes care of transactions itself

    'tm.enabled': False,

    # Set the default i18n source language so TG doesn't search beyond 'en' in Accept-Language.

    'i18n.lang': 'en',

})

# DebugBar, a debug toolbar for TurboGears2.

# (https://github.com/TurboGears/tgext.debugbar)

# To enable it, install 'tgext.debugbar' and 'kajiki', and run Kallithea with

# 'debug = true' (not in production!)

# See the Kallithea documentation for more information.

try:

    import kajiki  # only to check its existence

    from tgext.debugbar import enable_debugbar

    assert kajiki

except ImportError:

    pass

else:

    base_config.get_blueprint_value('renderers').append('kajiki')

    enable_debugbar(base_config)

def setup_configuration(app):

    config = app.config

        log.error("Terminating ...")

        sys.exit(1)

    # Mercurial sets encoding at module import time, so we have to monkey patch it

    hgencoding = config.get('hgencoding')

    if hgencoding:

        mercurial.encoding.encoding = hgencoding

    if config.get('ignore_alembic_revision', False):

        log.warning('database alembic revision checking is disabled')

    else:

        dbconf = config['sqlalchemy.url']

        alembic_cfg = alembic.config.Config()

        alembic_cfg.set_main_option('script_location', 'kallithea:alembic')

        alembic_cfg.set_main_option('sqlalchemy.url', dbconf)

        script_dir = ScriptDirectory.from_config(alembic_cfg)

        available_heads = sorted(script_dir.get_heads())

        engine = create_engine(dbconf)

        with engine.connect() as conn:

            context = MigrationContext.configure(conn)

            current_heads = sorted(str(s) for s in context.get_current_heads())

        if current_heads != available_heads:

            log.error('Failed to run Kallithea:\n\n'

                      'The database version does not match the Kallithea version.\n'

                      'Please read the documentation on how to upgrade or downgrade the database.\n'

                      'Current database version id(s): %s\n'

                      'Expected database version id(s): %s\n'

                      'If you are a developer and you know what you are doing, you can add `ignore_alembic_revision = True` '

                      'to your .ini file to skip the check.\n' % (' '.join(current_heads), ' '.join(available_heads)))

            sys.exit(1)

    # store some globals into kallithea

    kallithea.DEFAULT_USER_ID = db.User.get_default_user().user_id

    if asbool(config.get('use_celery')):

        kallithea.CELERY_APP = celerypylons.make_app()

    kallithea.CONFIG = config

    load_extensions(root_path=config['here'])

    set_app_settings(config)

    instance_id = kallithea.CONFIG.get('instance_id', '*')

    if instance_id == '*':

        instance_id = '%s-%s' % (platform.uname()[1], os.getpid())

        kallithea.CONFIG['instance_id'] = instance_id

    # update kallithea.CONFIG with the meanwhile changed 'config'

    kallithea.CONFIG.update(config)

    # configure vcs and indexer libraries (they are supposed to be independent

    # as much as possible and thus avoid importing tg.config or

    # kallithea.CONFIG).

    set_vcs_config(kallithea.CONFIG)

    set_indexer_config(kallithea.CONFIG)

    check_git_version()

    kallithea.model.meta.Session.remove()

tg.hooks.register('configure_new_app', setup_configuration)