#

# 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.cache_keys

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

cleanup-keys 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 27, 2013

:author: marcink

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

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

"""

from kallithea.lib.paster_commands.common import BasePasterCommand

from kallithea.model.meta import Session

from kallithea.lib.utils2 import safe_str

from kallithea.model.db import CacheInvalidation

class Command(BasePasterCommand):

    "Kallithea: Utilities for managing caching of database content"

    def take_action(self, args):

        _caches = CacheInvalidation.query().order_by(CacheInvalidation.cache_key).all()

        if args.show:

            for c_obj in _caches:

                print 'key:%s active:%s' % (safe_str(c_obj.cache_key), c_obj.cache_active)

        elif args.cleanup:

            for c_obj in _caches:

                Session().delete(c_obj)

                print 'Removing key: %s' % (safe_str(c_obj.cache_key))

            Session().commit()

        else:

            print 'Nothing done, exiting...'

    def get_parser(self, prog_name):

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

import argparse

import kallithea

from kallithea.lib.paster_commands.common import BasePasterCommand

__all__ = ['Command']

class Command(BasePasterCommand):

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

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

    # configuration file.

    def take_action(self, args):

        if not kallithea.CELERY_ON:

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

                            'file before running celeryd')

        from kallithea.lib import celerypylons

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

        celery_args = args.celery_args

        if '--' in celery_args:

            celery_args.remove('--')

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

# 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.cleanup

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

cleanup-repos 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: Jul 14, 2012

:author: marcink

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

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

"""

import os

import re

import shutil

import datetime

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

from kallithea.lib.utils import REMOVED_REPO_PAT

from kallithea.lib.utils2 import safe_str

from kallithea.model.db import Ui

class Command(BasePasterCommand):

    """Kallithea: Cleanup of backup files of deleted repositories"""

    def _parse_older_than(self, val):

        regex = re.compile(r'((?P<days>\d+?)d)?((?P<hours>\d+?)h)?((?P<minutes>\d+?)m)?((?P<seconds>\d+?)s)?')

        parts = regex.match(val)

        if not parts:

            return

        parts = parts.groupdict()

        time_params = {}

        for (name, param) in parts.iteritems():

            if param:

                time_params[name] = int(param)

        return datetime.timedelta(**time_params)

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

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

IIS installation tools for Kallithea

"""

import os

from kallithea.lib.paster_commands.common import BasePasterCommand

dispath_py_template = '''\

# Created by Kallithea 'gearbox install-iis'

import sys

if hasattr(sys, "isapidllhandle"):

    import win32traceutil

import isapi_wsgi

import os

def __ExtensionFactory__():

    from paste.deploy import loadapp

    from paste.script.util.logging_config import fileConfig

    fileConfig('%(inifile)s')

    application = loadapp('config:%(inifile)s')

    def app(environ, start_response):

        user = environ.get('REMOTE_USER', None)

        if user is not None:

            os.environ['REMOTE_USER'] = user

# (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.ishell

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

interactive shell 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: Apr 4, 2013

:author: marcink

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

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

"""

import sys

# imports, used in IPython shell

import time

import shutil

import datetime

from kallithea.model.db import *

from kallithea.lib.paster_commands.common import BasePasterCommand

class Command(BasePasterCommand):

    "Kallithea: Interactive Python shell"

    def take_action(self, args):

        try:

            from IPython import embed

            from traitlets.config.loader import Config

            cfg = Config()

            cfg.InteractiveShellEmbed.confirm_exit = False

            embed(config=cfg, banner1="Kallithea IShell.")

        except ImportError:

            print 'Kallithea ishell requires the IPython Python package'

            sys.exit(-1)

# 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 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):

        index_location = self.config['index_dir']

        load_rcextensions(self.config['here'])

        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

#

# 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 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):

        here = self.config['here']

        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

# (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.repo_scan

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

repo-scan 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: Feb 9, 2013

:author: marcink

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

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

"""

from kallithea.model.scm import ScmModel

from kallithea.lib.paster_commands.common import BasePasterCommand

from kallithea.lib.utils import repo2db_mapper

class Command(BasePasterCommand):

    """Kallithea: Scan file system for repositories

    Search under the repository root configured in the database,

    all new repositories, and report missing ones with an option of removing them.

    """

    def take_action(self, args):

        rm_obsolete = args.delete_obsolete

        print 'Now scanning root location for new repos ...'

        added, removed = repo2db_mapper(ScmModel().repo_scan(),

                                        remove_obsolete=rm_obsolete)

        print 'Scan completed.'

        if added:

            print 'Added: %s' % ', '.join(added)

        if removed:

            print '%s: %s' % ('Removed' if rm_obsolete else 'Missing',

                              ', '.join(removed))

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

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

Databaset setup gearbox command for Kallithea

"""

import kallithea

from kallithea.lib.db_manage import DbManage

from kallithea.lib.paster_commands.common import BasePasterCommand

from kallithea.model.meta import Session

# This is almost like SetupAppCommand ... but we have to pass options and it is

# thus simpler to drop websetup and reimplement everything

class Command(BasePasterCommand):

    """Kallithea: Configure the database specified in the .ini file

    Setup Kallithea according to its configuration file.  This is

    the second part of a two-phase web application installation

    process (the first phase is prepare-app). The setup process

    consist of things like setting up databases and creating the admin user

    """

    def get_description(self):

        return self.__doc__.splitlines()[0]

    requires_db_session = False # only available after this command has been run

    def get_parser(self, prog_name):

# (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.update_repoinfo

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

update-repoinfo 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: Jul 14, 2012

:author: marcink

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

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

"""

from kallithea.lib.paster_commands.common import BasePasterCommand

from kallithea.lib.utils2 import safe_unicode

from kallithea.model.db import Repository

from kallithea.model.meta import Session

class Command(BasePasterCommand):

    "Kallithea: Update database cache of repository data"

    def take_action(self, args):

        if args.repo_update_list is None:

            repo_list = Repository.query().all()

        else:

            repo_names = [safe_unicode(n.strip())

                          for n in args.repo_update_list.split(',')]

            repo_list = list(Repository.query()

                .filter(Repository.repo_name.in_(repo_names)))

        for repo in repo_list:

            repo.update_changeset_cache()

        Session().commit()

        if args.invalidate_cache:

            for r in repo_list:

                r.set_invalidate()

            print 'Updated repo info and invalidated cache for %s repositories' % (len(repo_list))

        else:

            print 'Updated repo info for %s repositories' % (len(repo_list))