#!/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 operator import itemgetter

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.auth import LoginRequired

from pylons_app.lib.base import BaseController, render

from pylons_app.lib.utils import invalidate_cache

from pylons_app.model.repo_model import RepoModel

from pylons_app.model.hg_model import HgModel

from pylons_app.model.forms import RepoForm

from pylons_app.model.meta import Session

from datetime import datetime

import formencode

from formencode import htmlfill

import logging

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')

        repo_model = RepoModel()

        _form = RepoForm()()

        try:

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

            repo_model.create(form_result, c.hg_app_user)

            invalidate_cache('cached_repo_list')

            h.flash(_('created repository %s') % form_result['repo_name'],

                    category='success')

        except formencode.Invalid as errors:

            c.form_errors = errors.error_dict

            c.new_repo = errors.value['repo_name']

            return htmlfill.render(

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

                defaults=errors.value,

                encoding="UTF-8")        

        except Exception:

            h.flash(_('error occured during creation of repository %s') \

                    % form_result['repo_name'], category='error')

        return redirect('repos')

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

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

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

        c.new_repo = h.repo_name_slug(new_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)

        repo_model = RepoModel()

        _form = RepoForm(edit=True)()

        try:

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

            repo_model.update(id, form_result)

            invalidate_cache('cached_repo_list')

            h.flash(_('Repository %s updated succesfully' % id), category='success')

        except formencode.Invalid as errors:

            c.repo_info = repo_model.get(id)

            errors.value.update({'user':c.repo_info.user.username})

            c.form_errors = errors.error_dict

            return htmlfill.render(

                 render('admin/repos/repo_edit.html'),

                defaults=errors.value,

                encoding="UTF-8")

        except Exception:

            h.flash(_('error occured during update of repository %s') \

                    % form_result['repo_name'], category='error')

        return redirect(url('repos'))

    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)

        repo_model = RepoModel()

        repo = repo_model.get(id)

        if not repo:

            h.flash(_('%s repository is not mapped to db perhaps' 

                      ' it was moved or renamed  from the filesystem'

                      ' please run the application again'

                      ' in order to rescan repositories') % id, category='error')

            return redirect(url('repos'))

        try:

            repo_model.delete(repo)            

            invalidate_cache('cached_repo_list')

            h.flash(_('deleted repository %s') % id, category='success')

        except Exception:

            h.flash(_('An error occured during deletion of %s') % id,

                    category='error')

        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)

        repo_model = RepoModel()

        c.repo_info = repo = repo_model.get(id)

        if not repo:

            h.flash(_('%s repository is not mapped to db perhaps' 

                      ' it was created or renamed from the filesystem'

                      ' please run the application again'

                      ' in order to rescan repositories') % id, category='error')

            return redirect(url('repos'))        

        defaults = c.repo_info.__dict__

        defaults.update({'user':c.repo_info.user.username})

        for p in c.repo_info.repo2perm:

            defaults.update({'perm_%s' % p.user.username: 

                             p.permission.permission_name})

        return htmlfill.render(

            render('admin/repos/repo_edit.html'),

            defaults=defaults,

            encoding="UTF-8",

            force_defaults=False

        )          

#!/usr/bin/env python

# encoding: utf-8

# model for handling repositories actions

# 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 Jun 5, 2010

model for handling repositories actions

@author: marcink

"""

from pylons_app.model.meta import Session

from pylons_app.model.db import Repository, Repo2Perm, User, Permission

import shutil

import os

from datetime import datetime

from pylons_app.lib.utils import check_repo

from pylons import app_globals as g

import traceback

import logging

log = logging.getLogger(__name__)

class RepoModel(object):

    def __init__(self):

        self.sa = Session()

    def get(self, id):

        return self.sa.query(Repository).get(id)

    def update(self, repo_id, form_data):

        try:

            if repo_id != form_data['repo_name']:

                self.__rename_repo(repo_id, form_data['repo_name'])

            cur_repo = self.sa.query(Repository).get(repo_id)

            for k, v in form_data.items():

                if k == 'user':

                    cur_repo.user_id = v

                else:

                    setattr(cur_repo, k, v)

            #update permissions

            for username, perm in form_data['perms_updates']:

                r2p = self.sa.query(Repo2Perm)\

                        .filter(Repo2Perm.user == self.sa.query(User)\

                                .filter(User.username == username).one())\

                        .filter(Repo2Perm.repository == repo_id).one()

                r2p.permission_id = self.sa.query(Permission).filter(

                                                Permission.permission_name == 

                                                perm).one().permission_id

                self.sa.add(r2p)

            for username, perm in form_data['perms_new']:

                r2p = Repo2Perm()

                r2p.repository = repo_id

                r2p.user = self.sa.query(User)\

                                .filter(User.username == username).one()

                r2p.permission_id = self.sa.query(Permission).filter(

                                                Permission.permission_name == 

                                                perm).one().permission_id

                self.sa.add(r2p)

            self.sa.add(cur_repo)

            self.sa.commit()

        except:

            log.error(traceback.format_exc())

            self.sa.rollback()

            raise    

    def create(self, form_data, cur_user, just_db=False):

        try:

            repo_name = form_data['repo_name']

            new_repo = Repository()

            for k, v in form_data.items():

                setattr(new_repo, k, v)

            new_repo.user_id = cur_user.user_id

            self.sa.add(new_repo)

            #create default permission

            repo2perm = Repo2Perm()

            repo2perm.permission_id = self.sa.query(Permission)\

                    .filter(Permission.permission_name == 'repository.read')\

                    .one().permission_id

            repo2perm.repository = repo_name

            repo2perm.user_id = self.sa.query(User)\

                    .filter(User.username == 'default').one().user_id 

            self.sa.add(repo2perm)

            self.sa.commit()

            if not just_db:

                self.__create_repo(repo_name)

        except:

            log.error(traceback.format_exc())

            self.sa.rollback()

            raise    

    def delete(self, repo):

        try:

            self.sa.delete(repo)

            self.sa.commit()

            self.__delete_repo(repo.repo_name)

        except:

            log.error(traceback.format_exc())

            self.sa.rollback()

            raise

    def __create_repo(self, repo_name):        

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

        if check_repo(repo_name, g.base_path):

            log.info('creating repo %s in %s', repo_name, repo_path)

            from vcs.backends.hg import MercurialRepository

            MercurialRepository(repo_path, create=True)

    def __rename_repo(self, old, new):

        log.info('renaming repoo from %s to %s', old, new)

        old_path = os.path.join(g.base_path, old)

        new_path = os.path.join(g.base_path, new)

        shutil.move(old_path, new_path)

    def __delete_repo(self, name):

        rm_path = os.path.join(g.base_path, name)

        log.info("Removing %s", rm_path)

        #disable hg 

        shutil.move(os.path.join(rm_path, '.hg'), os.path.join(rm_path, 'rm__.hg'))

        #disable repo

        shutil.move(rm_path, os.path.join(g.base_path, 'rm__%s__%s' \

                                          % (datetime.today(), name)))

/*** Initial Settings ***/

	margin: 15px 50px;

	background: #DBD4C6;

	font-family: sans-serif;

}

}

a {

	color: #556CB5;

	text-decoration: none;

}

	text-decoration: underline;

}

/*** end of Initial Settings ***/ 

/*** ***/

.table_disp {

}

.table_disp .header {

}

.table_disp .header td {

}

.table_disp td {

}

table tr.parity0:hover,table tr.parity1:hover {

}

table tr.parity0 {

}

table tr.parity1 {

}

}

}

}

}

.warning_msg {

	background-image: url("/images/icons/warning_msg.png");

}

.success_msg {

	background-image: url("/images/icons/success_msg.png");

}

.notice_msg {

	background-image: url("/images/icons/notice_msg.png");

}

	overflow: auto;

#msg_close {

}

}

}

.yui-tt {

}

.yui-tt-shadow {

}

div#main {

	padding: 5px;

}

div#container {

	background: #FFFFFF;

	position: relative;

	color: #666;

}

div.page-header {

	padding: 50px 20px 0;

	background: #556cb5 top left repeat-x;

	position: relative;

}

div.page-header h1 {

	margin: 10px 0 30px;

	font-size: 1.8em;

	font-weight: bold;

	font-family: sans-serif;

	letter-spacing: 1px;

	color: #FFFFFF;

}

div.page-header h1 a {

	font-weight: bold;

	color: #FFFFFF;

}

div.page-header a {

	text-decoration: none;

}

div.page-header form {

	position: absolute;

	margin-bottom: 2px;

	bottom: 0;

	right: 20px;

}

div.page-header form label {

	color: #DDD;

}

div.page-header form input {

	padding: 2px;

	border: solid 1px #DDD;

}

div.page-header form dl {

	overflow: hidden;

}

div.page-header form dl dt {

	font-size: 1.2em;

}

div.page-header form dl dt,div.page-header form dl dd {

	margin: 0 0 0 5px;

	float: left;

	height: 24px;

	line-height: 20px;

}

ul.page-nav {

	margin: 10px 0 0 0;

	list-style-type: none;

	overflow: hidden;

	width: 800px;

	padding: 0;

}

ul.page-nav li {

	margin: 0 4px 0 0;

	float: left;

	height: 24px;

	font-size: 1.1em;

	line-height: 24px;

	text-align: center;

	background: #556CB5;

}

ul.page-nav li.current {

	background: #FFF;

	padding-right: 5px;

	padding-left: 5px;

}

ul.page-nav li.current a {

	color: #556CB5;

}

ul.page-nav li a {

	height: 24px;

	color: #FFF;

	padding-right: 5px;

	padding-left: 5px;

	display: block;

	text-decoration: none;

	font-weight: bold;

}

ul.page-nav li.logout a {