# -*- coding: utf-8 -*-

import argparse

import kallithea

from kallithea.lib.paster_commands.common import BasePasterCommand

from kallithea.lib.utils import load_rcextensions

from kallithea.lib.utils2 import str2bool

__all__ = ['Command']

class Command(BasePasterCommand):

    """Kallithea: Celery worker for asynchronous tasks"""

    # Starts the celery worker using configuration from a paste.deploy

    # configuration file.

    requires_db_session = False # will start session on demand

    def take_action(self, args):

        from kallithea.lib import celerypylons

        try:

        except KeyError:

            CELERY_ON = False

        if not CELERY_ON:

            raise Exception('Please set use_celery = true in .ini config '

                            'file before running celeryd')

        kallithea.CELERY_ON = CELERY_ON

        cmd = celerypylons.worker.worker(celerypylons.app.app_or_default())

        celery_args = args.celery_args

        if '--' in celery_args:

            celery_args.remove('--')

        return cmd.run_from_argv('kallithea celery worker', celery_args)

    def get_parser(self, prog_name):

        parser = super(Command, self).get_parser(prog_name)

        parser.add_argument('celery_args', nargs=argparse.REMAINDER,

            help="Pass extra options to Celery after a '--' separator",

            )

        return parser

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

"""

kallithea.lib.paster_commands.common

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Common code for gearbox commands.

This file was forked by the Kallithea project in July 2014.

Original author and date, and relevant copyright and licensing information is below:

:created_on: Apr 18, 2010

:author: marcink

:copyright: (c) 2013 RhodeCode GmbH, and others.

:license: GPLv3, see LICENSE.md for more details.

"""

import os

import sys

import logging.config

import paste.deploy

import gearbox.command

import kallithea.config.middleware

import kallithea.model.base

import kallithea.lib.utils2

import kallithea.lib.utils

def ask_ok(prompt, retries=4, complaint='Yes or no please!'):

    while True:

        ok = raw_input(prompt)

        if ok in ('y', 'ye', 'yes'):

            return True

        if ok in ('n', 'no', 'nop', 'nope'):

            return False

        retries = retries - 1

        if retries < 0:

            raise IOError

        print complaint

class BasePasterCommand(gearbox.command.Command):

    """

    Abstract Base Class for gearbox commands.

    """

    # override to control how much get_parser and run should do:

    takes_config_file = True

    requires_db_session = True

    def run(self, args):

        """

        Overrides Command.run

        If needed by the command, read config file and initialize database before running.

        """

        if self.takes_config_file:

            path_to_ini_file = os.path.realpath(args.config_file)

            logging.config.fileConfig(path_to_ini_file)

            if self.requires_db_session:

                kallithea.model.base.init_model(engine)

        return super(BasePasterCommand, self).run(args)

    def get_parser(self, prog_name):

        parser = super(BasePasterCommand, self).get_parser(prog_name)

        if self.takes_config_file:

            parser.add_argument("-c", "--config",

                help='Kallithea .ini file with configuration of database etc',

                dest='config_file', required=True)

        return parser

    def error(self, msg, exitcode=1):

        """Write error message and exit"""

        sys.stderr.write('%s\n' % msg)

        raise SystemExit(exitcode)

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

"""

kallithea.lib.paster_commands.make_index

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

make-index gearbox command for Kallithea

This file was forked by the Kallithea project in July 2014.

Original author and date, and relevant copyright and licensing information is below:

:created_on: Aug 17, 2010

:author: marcink

:copyright: (c) 2013 RhodeCode GmbH, and others.

:license: GPLv3, see LICENSE.md for more details.

"""

import os

import sys

from os.path import dirname

from string import strip

from kallithea.model.repo import RepoModel

from kallithea.lib.paster_commands.common import BasePasterCommand

from kallithea.lib.utils import load_rcextensions

class Command(BasePasterCommand):

    "Kallithea: Create or update full text search index"

    def take_action(self, args):

        repo_location = args.repo_location \

            if args.repo_location else RepoModel().repos_path

        repo_list = map(strip, args.repo_list.split(',')) \

            if args.repo_list else None

        repo_update_list = map(strip, args.repo_update_list.split(',')) \

            if args.repo_update_list else None

        #======================================================================

        # WHOOSH DAEMON

        #======================================================================

        from kallithea.lib.pidlock import LockHeld, DaemonLock

        from kallithea.lib.indexers.daemon import WhooshIndexingDaemon

        try:

            l = DaemonLock(file_=os.path.join(dirname(dirname(index_location)),

                                              'make_index.lock'))

            WhooshIndexingDaemon(index_location=index_location,

                                 repo_location=repo_location,

                                 repo_list=repo_list,

                                 repo_update_list=repo_update_list) \

                .run(full_index=args.full_index)

            l.release()

        except LockHeld:

            sys.exit(1)

    def get_parser(self, prog_name):

        parser = super(Command, self).get_parser(prog_name)

        parser.add_argument('--repo-location',

                          action='store',

                          dest='repo_location',

                          help="Specifies repositories location to index OPTIONAL",

                          )

        parser.add_argument('--index-only',

                          action='store',

                          dest='repo_list',

                          help="Specifies a comma separated list of repositories "

                                "to build index on. If not given all repositories "

                                "are scanned for indexing. OPTIONAL",

                          )

        parser.add_argument('--update-only',

                          action='store',

                          dest='repo_update_list',

                          help="Specifies a comma separated list of repositories "

                                "to re-build index on. OPTIONAL",

                          )

        parser.add_argument('-f',

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

"""

kallithea.lib.paster_commands.make_rcextensions

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

make-rcext gearbox command for Kallithea

This file was forked by the Kallithea project in July 2014.

Original author and date, and relevant copyright and licensing information is below:

:created_on: Mar 6, 2012

:author: marcink

:copyright: (c) 2013 RhodeCode GmbH, and others.

:license: GPLv3, see LICENSE.md for more details.

"""

import os

import sys

import pkg_resources

from kallithea.lib.paster_commands.common import ask_ok, BasePasterCommand

class Command(BasePasterCommand):

    """Kallithea: Write template file for extending Kallithea in Python

    A rcextensions directory with a __init__.py file will be created next to

    the ini file. Local customizations in that file will survive upgrades.

    The file contains instructions on how it can be customized.

    """

    requires_db_session = False

    def take_action(self, args):

        content = pkg_resources.resource_string(

            'kallithea', os.path.join('config', 'rcextensions', '__init__.py')

        )

        ext_file = os.path.join(here, 'rcextensions', '__init__.py')

        if os.path.exists(ext_file):

            msg = ('Extension file %s already exists, do you want '

                   'to overwrite it ? [y/n]') % ext_file

            if not ask_ok(msg):

                print 'Nothing done, exiting...'

                return

        dirname = os.path.dirname(ext_file)

        if not os.path.isdir(dirname):

            os.makedirs(dirname)

        with open(ext_file, 'wb') as f:

            f.write(content)

            print 'Wrote new extensions file to %s' % ext_file

    def get_parser(self, prog_name):

        parser = super(Command, self).get_parser(prog_name)

        parser.add_argument('--user',

                          action='store',

                          dest='username',

                          default=None,

                          help='Admin Username')

        parser.add_argument('--email',

                          action='store',

                          dest='email',

                          default=None,

                          help='Admin Email')

        parser.add_argument('--password',

                          action='store',

                          dest='password',

                          default=None,

                          help='Admin password min 6 chars')

        parser.add_argument('--repos',

                          action='store',

                          dest='repos_location',

                          default=None,

                          help='Absolute path to repositories location')

        parser.add_argument('--force-yes',

                           action='store_true',

                           dest='force_ask',

                           default=None,

                           help='Force yes to every question')

        parser.add_argument('--force-no',

                           action='store_false',

                           dest='force_ask',

                           default=None,

                           help='Force no to every question')

        parser.add_argument('--public-access',

                           action='store_true',

                           dest='public_access',

                           default=None,

                           help='Enable public access on this installation (default)')

        parser.add_argument('--no-public-access',

                           action='store_false',

                           dest='public_access',

                           default=None,

                           help='Disable public access on this installation ')

        return parser

    def take_action(self, opts):

                            tests=False, cli_args=vars(opts))

        dbmanage.create_tables(override=True)

        opts = dbmanage.config_prompt(None)

        dbmanage.create_settings(opts)

        dbmanage.create_default_user()

        dbmanage.admin_prompt()

        dbmanage.create_permissions()

        dbmanage.populate_default_permissions()

        Session().commit()