"""Pylons environment configuration"""

from mako.lookup import TemplateLookup

from pylons.configuration import PylonsConfig

from pylons.error import handle_mako_error

from pylons_app.config.routing import make_map

from pylons_app.lib.auth import set_available_permissions

from pylons_app.lib.utils import repo2db_mapper

from pylons_app.model import init_model

from sqlalchemy import engine_from_config

import logging

import os

import pylons_app.lib.app_globals as app_globals

import pylons_app.lib.helpers

log = logging.getLogger(__name__)

def load_environment(global_conf, app_conf):

    """Configure the Pylons environment via the ``pylons.config``

    object

    """

    config = PylonsConfig()

    # Pylons paths

    root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

    paths = dict(root=root,

                 controllers=os.path.join(root, 'controllers'),

                 static_files=os.path.join(root, 'public'),

                 templates=[os.path.join(root, 'templates')])

    # Initialize config with the basic options

    config.init_app(global_conf, app_conf, package='pylons_app', paths=paths)

    config['routes.map'] = make_map(config)

    config['pylons.app_globals'] = app_globals.Globals(config)

    config['pylons.h'] = pylons_app.lib.helpers

    # Setup cache object as early as possible

    import pylons

    pylons.cache._push_object(config['pylons.app_globals'].cache)

    # Create the Mako TemplateLookup, with the default auto-escaping

    config['pylons.app_globals'].mako_lookup = TemplateLookup(

        directories=paths['templates'],

        error_handler=handle_mako_error,

        module_directory=os.path.join(app_conf['cache_dir'], 'templates'),

        input_encoding='utf-8', default_filters=['escape'],

        imports=['from webhelpers.html import escape'])

    #sets the c attribute access when don't existing attribute are accessed

    config['pylons.strict_tmpl_context'] = True

    #MULTIPLE DB configs

    # Setup the SQLAlchemy database engine

    if config['debug']:

        #use query time debugging.

        from pylons_app.lib.timerproxy import TimerProxy

        sa_engine_db1 = engine_from_config(config, 'sqlalchemy.db1.',

                                                            proxy=TimerProxy())

    else:

        sa_engine_db1 = engine_from_config(config, 'sqlalchemy.db1.')

    init_model(sa_engine_db1)

    set_available_permissions(config)

    # CONFIGURATION OPTIONS HERE (note: all config options will override

    # any Pylons config options)

    return config

#!/usr/bin/env python

# encoding: utf-8

# repos controller for pylons

# Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>

# 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; version 2

# of the License or (at your opinion) any later version of the license.

# 

# 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, write to the Free Software

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

# MA  02110-1301, USA.

"""

Created on April 7, 2010

admin controller for pylons

@author: marcink

"""

from pylons import request, response, session, tmpl_context as c, url, \

    app_globals as g

from pylons.controllers.util import abort, redirect

from pylons.i18n.translation import _

from pylons_app.lib import helpers as h

from pylons_app.lib.base import BaseController, render

from pylons_app.model.hg_model import HgModel

import os

import shutil

log = logging.getLogger(__name__)

class ReposController(BaseController):

    """REST Controller styled on the Atom Publishing Protocol"""

    # To properly map this controller, ensure your config/routing.py

    # file has a resource setup:

    #     map.resource('repo', 'repos')

    @LoginRequired()

    def __before__(self):

        c.admin_user = session.get('admin_user')

        c.admin_username = session.get('admin_username')

        super(ReposController, self).__before__()

    def index(self, format='html'):

        """GET /repos: All items in the collection"""

        # url('repos')

        cached_repo_list = HgModel().get_repos()

        c.repos_list = sorted(cached_repo_list, key=itemgetter('name_sort'))

        return render('admin/repos/repos.html')

    def create(self):

        """POST /repos: Create a new item"""

        # url('repos')

        try:

            invalidate_cache('cached_repo_list')

        return redirect('repos')

    def new(self, format='html'):

        """GET /repos/new: Form to create a new item"""

        new_repo = request.GET.get('repo', '')

        return render('admin/repos/repo_add.html')

    def update(self, id):

        """PUT /repos/id: Update an existing item"""

        # Forms posted to this method should contain a hidden field:

        #    <input type="hidden" name="_method" value="PUT" />

        # Or using helpers:

        #    h.form(url('repo', id=ID),

        #           method='put')

        # url('repo', id=ID)

    def delete(self, id):

        """DELETE /repos/id: Delete an existing item"""

        # Forms posted to this method should contain a hidden field:

        #    <input type="hidden" name="_method" value="DELETE" />

        # Or using helpers:

        #    h.form(url('repo', id=ID),

        #           method='delete')

        # url('repo', id=ID)

        invalidate_cache('cached_repo_list')

        return redirect(url('repos'))

    def show(self, id, format='html'):

        """GET /repos/id: Show a specific item"""

        # url('repo', id=ID)

    def edit(self, id, format='html'):

        """GET /repos/id/edit: Form to edit an existing item"""

        # url('edit_repo', id=ID)

#!/usr/bin/env python

# encoding: utf-8

# users controller for pylons

# Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>

# 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; version 2

# of the License or (at your opinion) any later version of the license.

# 

# 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, write to the Free Software

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

# MA  02110-1301, USA.

"""

Created on April 4, 2010

users controller for pylons

@author: marcink

"""

import logging

from pylons import request, session, tmpl_context as c, url

from pylons.controllers.util import abort, redirect

from pylons.i18n.translation import _

from pylons_app.lib import helpers as h

from pylons_app.lib.auth import LoginRequired, CheckPermissionAll

from pylons_app.lib.base import BaseController, render

from pylons_app.model.db import User, UserLog

from pylons_app.model.forms import UserForm

from pylons_app.model.user_model import UserModel

import formencode

log = logging.getLogger(__name__)

class UsersController(BaseController):

    """REST Controller styled on the Atom Publishing Protocol"""

    # To properly map this controller, ensure your config/routing.py

    # file has a resource setup:

    #     map.resource('user', 'users')

    @LoginRequired()

    def __before__(self):

        c.admin_user = session.get('admin_user')

        c.admin_username = session.get('admin_username')

        super(UsersController, self).__before__()

    def index(self, format='html'):

        """GET /users: All items in the collection"""

        # url('users')

        c.users_list = self.sa.query(User).all()     

        return render('admin/users/users.html')

    def create(self):

        """POST /users: Create a new item"""

        # url('users')

        user_model = UserModel()

        login_form = UserForm()()

        try:

            form_result = login_form.to_python(dict(request.POST))

            user_model.create(form_result)

        except formencode.Invalid as errors:

            c.form_errors = errors.error_dict

            return htmlfill.render(

                 render('admin/users/user_add.html'),

                defaults=errors.value,

                encoding="UTF-8")

    def new(self, format='html'):

        """GET /users/new: Form to create a new item"""

        # url('new_user')

        return render('admin/users/user_add.html')

    def update(self, id):

        """PUT /users/id: Update an existing item"""

        # Forms posted to this method should contain a hidden field:

        #    <input type="hidden" name="_method" value="PUT" />

        # Or using helpers:

        #    h.form(url('user', id=ID),

        #           method='put')

        # url('user', id=ID)

        user_model = UserModel()

        try:

            user_model.update(id, form_result)

            h.flash(_('User updated succesfully'), category='success')

        except formencode.Invalid as errors:

            c.user = user_model.get_user(id)

            c.form_errors = errors.error_dict

            return htmlfill.render(

                 render('admin/users/user_edit.html'),

                defaults=errors.value,

                encoding="UTF-8")

    def delete(self, id):

        """DELETE /users/id: Delete an existing item"""

        # Forms posted to this method should contain a hidden field:

        #    <input type="hidden" name="_method" value="DELETE" />

        # Or using helpers:

        #    h.form(url('user', id=ID),

        #           method='delete')

        # url('user', id=ID)

        try:

            h.flash(_('sucessfully deleted user'), category='success')

        return redirect(url('users'))

    def show(self, id, format='html'):

        """GET /users/id: Show a specific item"""

        # url('user', id=ID)

    def edit(self, id, format='html'):

        """GET /users/id/edit: Form to edit an existing item"""

        # url('edit_user', id=ID)

        c.user = self.sa.query(User).get(id)

        defaults = c.user.__dict__

        return htmlfill.render(

            render('admin/users/user_edit.html'),

            defaults=defaults,

            encoding="UTF-8",

            force_defaults=False

        )    

#!/usr/bin/env python

# encoding: utf-8

# authentication and permission libraries

# Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>

# 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; version 2

# of the License or (at your opinion) any later version of the license.

# 

# 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, write to the Free Software

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

# MA  02110-1301, USA.

"""

Created on April 4, 2010

@author: marcink

"""

from functools import wraps

from pylons import session, url, app_globals as g

from pylons.controllers.util import abort, redirect

from pylons_app.model import meta

from pylons_app.model.db import User

from sqlalchemy.exc import OperationalError

from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound

import crypt

import logging

log = logging.getLogger(__name__)

def get_crypt_password(password):

    """

    Cryptographic function used for password hashing

    @param password: password to hash

    """

    return crypt.crypt(password, '6a')

def authfunc(environ, username, password):

    sa = meta.Session

    password_crypt = get_crypt_password(password)

    try:

        user = sa.query(User).filter(User.username == username).one()

    except (NoResultFound, MultipleResultsFound, OperationalError) as e:

        log.error(e)

        user = None

    if user:

        if user.active:

            if user.username == username and user.password == password_crypt:

                log.info('user %s authenticated correctly', username)

                return True

        else:

            log.error('user %s is disabled', username)

    return False

class  AuthUser(object):

    """

    A simple object that handles a mercurial username for authentication

    """

    username = 'None'

    is_authenticated = False

    is_admin = False

    permissions = set()

    group = set()

    def __init__(self):

        pass

def set_available_permissions(config):

    """

    This function will propagate pylons globals with all available defined

    permission given in db. We don't wannt to check each time from db for new 

    permissions since adding a new permission also requires application restart

    ie. to decorate new views with the newly created permission

    @param config:

    """

    from pylons_app.model.meta import Session

    from pylons_app.model.db import Permission

    logging.info('getting information about all available permissions')

    sa = Session()

    all_perms = sa.query(Permission).all()

    config['pylons.app_globals'].available_permissions = [x.permission_name for x in all_perms]

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

# DECORATORS

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

class LoginRequired(object):

    """

    Must be logged in to execute this function else redirect to login page

    """

    def __init__(self):

        pass

    def __call__(self, func):

        @wraps(func)

        def _wrapper(*fargs, **fkwargs):

            user = session.get('hg_app_user', AuthUser())

            log.info('Checking login required for user:%s', user.username)            

            if user.is_authenticated:

                    log.info('user %s is authenticated', user.username)

                    func(*fargs)

            else:

                logging.info('user %s not authenticated', user.username)

                logging.info('redirecting to login page')

                return redirect(url('login_home'))

        return _wrapper

class PermsDecorator(object):

    def __init__(self, *perms):

        available_perms = g.available_permissions

        for perm in perms:

            if perm not in available_perms:

                raise Exception("'%s' permission in not defined" % perm)

        self.required_perms = set(perms)

        self.user_perms = set([])#propagate this list from somewhere.

    def __call__(self, func):        

        @wraps(func)

        def _wrapper(*args, **kwargs):

            logging.info('checking %s permissions %s for %s',

               self.__class__.__name__[-3:], self.required_perms, func.__name__)            

            if self.check_permissions():

                logging.info('Permission granted for %s', func.__name__)

                return func(*args, **kwargs)

            else:

                logging.warning('Permission denied for %s', func.__name__)

                #redirect with forbidden ret code

                return redirect(url('access_denied'), 403) 

        return _wrapper

    def check_permissions(self):

        """

        Dummy function for overiding

        """

        raise Exception('You have to write this function in child class')

class CheckPermissionAll(PermsDecorator):

    """

    Checks for access permission for all given predicates. All of them have to

    be meet in order to fulfill the request

    """

    def check_permissions(self):

        if self.required_perms.issubset(self.user_perms):

            return True

        return False

from webhelpers.pylonslib.secure_form import secure_form

from webhelpers.text import chop_at, collapse, convert_accented_entities, \

    convert_misc_entities, lchop, plural, rchop, remove_formatting, \

    replace_whitespace, urlify, truncate

#Custom helper here :)

class _Link(object):

    '''

    Make a url based on label and url with help of url_for

    @param label:name of link    if not defined url is used

    @param url: the url for link

    '''

    def __call__(self, label='', *url_, **urlargs):

        if label is None or '':

            label = url

        link_fn = link_to(label, url(*url_, **urlargs))

        return link_fn

class _GetError(object):

    def __call__(self, field_name, form_errors):

        tmpl = """<span class="error_msg">%s</span>"""

        if form_errors and form_errors.has_key(field_name):

            return literal(tmpl % form_errors.get(field_name))

class _FilesBreadCrumbs(object):

    def __call__(self, repo_name, rev, paths):

        url_l = [link_to(repo_name, url('files_home', repo_name=repo_name, revision=rev, f_path=''))]

        paths_l = paths.split('/')

        for cnt, p in enumerate(paths_l, 1):

            if p != '':

                url_l.append(link_to(p, url('files_home', repo_name=repo_name, revision=rev, f_path='/'.join(paths_l[:cnt]))))

        return literal(' / '.join(url_l))

def pygmentize(filenode, **kwargs):

    """

    pygmentize function using pygments

    @param filenode:

    """

    return literal(code_highlight(filenode.content, filenode.lexer, HtmlFormatter(**kwargs)))

def pygmentize_annotation(filenode, **kwargs):

    """

    pygmentize function for annotation

    @param filenode:

    """

    color_dict = g.changeset_annotation_colors

    def gen_color():

        import random

        return [str(random.randrange(10, 235)) for _ in xrange(3)]

    def get_color_string(cs):

        if color_dict.has_key(cs):

            col = color_dict[cs]

        else:

            color_dict[cs] = gen_color()

            col = color_dict[cs]

        return "color: rgb(%s) ! important;" % (','.join(col))

    def url_func(changeset):

        return '%s\n' % (link_to(changeset.raw_id,

        url('changeset_home', repo_name='test', revision=changeset.raw_id),

        title=_('author') + ':%s rev:%s %s' % (changeset.author, changeset.revision,

                                                changeset.message,),

        style=get_color_string(changeset.raw_id)))

    return literal(annotate_highlight(filenode, url_func, **kwargs))

def recursive_replace(str, replace=' '):

    """

    Recursive replace of given sign to just one instance

    @param str: given string

    @param replace:char to find and replace multiple instances

    Examples::

    >>> recursive_replace("Mighty---Mighty-Bo--sstones",'-')

    'Mighty-Mighty-Bo-sstones'

    """

    if str.find(replace * 2) == -1:

        return str

    else:

        str = str.replace(replace * 2, replace)

        return recursive_replace(str, replace)  

def repo_name_slug(value):

    """

    Return slug of name of repository

    """

    slug = urlify(value)

        slug = slug.replace(c, '-')

    slug = recursive_replace(slug, '-')

    return slug

files_breadcrumbs = _FilesBreadCrumbs()

link = _Link()

flash = _Flash()

get_error = _GetError()

#!/usr/bin/env python

# encoding: utf-8

# Utilities for hg app

# Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>

# 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; version 2

# of the License or (at your opinion) any later version of the license.

# 

# 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, write to the Free Software

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

# MA  02110-1301, USA.

"""

Created on April 18, 2010

Utilities for hg app

@author: marcink

"""

import os

import logging

from mercurial import ui, config, hg

from mercurial.error import RepoError

log = logging.getLogger(__name__)

def get_repo_slug(request):

    return request.environ['pylons.routes_dict'].get('repo_name')

def is_mercurial(environ):

    """

    Returns True if request's target is mercurial server - header

    ``HTTP_ACCEPT`` of such request would start with ``application/mercurial``.

    """

    http_accept = environ.get('HTTP_ACCEPT')

    if http_accept and http_accept.startswith('application/mercurial'):

        return True

    return False

def check_repo_dir(paths):

    repos_path = paths[0][1].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 Exception('Not a valid repository in %s' % paths[0][1])

def check_repo(repo_name, base_path):

    repo_path = os.path.join(base_path, repo_name)

    try:

        r = hg.repository(ui.ui(), repo_path)

        hg.verify(r)

        #here we hnow that repo exists it was verified

        log.info('%s repo is already created', repo_name)

        return False

        #raise Exception('Repo exists')

    except RepoError:

        log.info('%s repo is free for creation', repo_name)

        #it means that there is no valid repo there...

        return True

def make_ui(path=None, checkpaths=True):        

    """

    A funcion that will read python rc files and make an ui from read options

    @param path: path to mercurial config file

    """

    if not path:

        log.error('repos config path is empty !')

    if not os.path.isfile(path):

        log.warning('Unable to read config file %s' % path)

        return False

    #propagated from mercurial documentation

    sections = [

                'alias',

                'auth',

                'decode/encode',

                'defaults',

                'diff',

                'email',

                'extensions',

                'format',

                'merge-patterns',

                'merge-tools',

                'hooks',

                'http_proxy',

                'smtp',

                'patch',

                'paths',

                'profiling',

                'server',

                'trusted',

                'ui',

                'web',

                ]

    baseui = ui.ui()

    cfg = config.config()

    cfg.read(path)

    if checkpaths:check_repo_dir(cfg.items('paths'))

    for section in sections:

        for k, v in cfg.items(section):

            baseui.setconfig(section, k, v)

    return baseui

def invalidate_cache(name, *args):

    """Invalidates given name cache"""

    from beaker.cache import region_invalidate

    log.info('INVALIDATING CACHE FOR %s', name)

    """propagate our arguments to make sure invalidation works. First

    argument has to be the name of cached func name give to cache decorator

    without that the invalidation would not work"""

    tmp = [name]

    tmp.extend(args)

    args = tuple(tmp)

    if name == 'cached_repo_list':

        from pylons_app.model.hg_model import _get_repos_cached

        region_invalidate(_get_repos_cached, None, *args)

    if name == 'full_changelog':

        from pylons_app.model.hg_model import _full_changelog_cached