# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

# MA  02110-1301, USA.

"""

Created on April 9, 2010

Model for hg app

@author: marcink

"""

from beaker.cache import cache_region

from mercurial import ui

from mercurial.hgweb.hgwebdir_mod import findrepos

from pylons.i18n.translation import _

from pylons_app.lib.auth import HasRepoPermissionAny

from pylons_app.model import meta

from pylons_app.model.db import Repository, User

from pylons_app.lib import helpers as h

from vcs.exceptions import RepositoryError, VCSError

import logging

import os

import sys

log = logging.getLogger(__name__)

try:

    from vcs.backends.hg import MercurialRepository

except ImportError:

    sys.stderr.write('You have to import vcs module')

    raise Exception('Unable to import vcs')

def _get_repos_cached_initial(app_globals, initial):

    """return cached dict with repos

    """

    g = app_globals

    return HgModel.repo_scan(g.paths[0][0], g.paths[0][1], g.baseui, initial)

@cache_region('long_term', 'cached_repo_list')

def _get_repos_cached():

    """return cached dict with repos

    """

    log.info('getting all repositories list')

    from pylons import app_globals as g

    return HgModel.repo_scan(g.paths[0][0], g.paths[0][1], g.baseui)

@cache_region('super_short_term', 'cached_repos_switcher_list')

def _get_repos_switcher_cached(cached_repo_list):

    repos_lst = []

    for repo in [x for x in cached_repo_list.values()]:

        if HasRepoPermissionAny('repository.write', 'repository.read',

            repos_lst.append((repo.name, repo.dbrepo.private,))

@cache_region('long_term', 'full_changelog')

def _full_changelog_cached(repo_name):

    log.info('getting full changelog for %s', repo_name)

    return list(reversed(list(HgModel().get_repo(repo_name))))

class HgModel(object):

    """Mercurial Model

    """

    def __init__(self):

        pass

    @staticmethod

    def repo_scan(repos_prefix, repos_path, baseui, initial=False):

        """

        Listing of repositories in given path. This path should not be a 

        repository itself. Return a dictionary of repository objects

        :param repos_path: path to directory it could take syntax with 

        * or ** for deep recursive displaying repositories

        """

        sa = meta.Session()

        def check_repo_dir(path):

            """Checks the repository

            :param path:

            """

            repos_path = path.split('/')

            if repos_path[-1] in ['*', '**']:

                repos_path = repos_path[:-1]

            if repos_path[0] != '/':

                repos_path[0] = '/'

            if not os.path.isdir(os.path.join(*repos_path)):

                raise RepositoryError('Not a valid repository in %s' % path)        

        if not repos_path.endswith('*'):

            raise VCSError('You need to specify * or ** at the end of path '

                            'for recursive scanning')

        check_repo_dir(repos_path)

        log.info('scanning for repositories in %s', repos_path)

        repos = findrepos([(repos_prefix, repos_path)])

        if not isinstance(baseui, ui.ui):

            baseui = ui.ui()

        repos_list = {}

        for name, path in repos:

            try:

                #name = name.split('/')[-1]

                if repos_list.has_key(name):