from kallithea.model.forms import RepoGroupForm, RepoGroupPermsForm

from kallithea.model.repo import RepoModel

from kallithea.model.repo_group import RepoGroupModel

from kallithea.model.scm import AvailableRepoGroupChoices, RepoGroupList

log = logging.getLogger(__name__)

class RepoGroupsController(base.BaseController):

    @LoginRequired(allow_default_user=True)

    def _before(self, *args, **kwargs):

        super(RepoGroupsController, self)._before(*args, **kwargs)

    def __load_defaults(self, extras=(), exclude=()):

        """extras is used for keeping current parent ignoring permissions

        exclude is used for not moving group to itself TODO: also exclude descendants

        Note: only admin can create top level groups

        """

        repo_groups = AvailableRepoGroupChoices('admin', extras)

        exclude_group_ids = set(rg.group_id for rg in exclude)

        c.repo_groups = [rg for rg in repo_groups

                         if rg[0] not in exclude_group_ids]

    def __load_data(self, group_id):

        """

        Load defaults settings for edit, and update

        :param group_id:

        """

        repo_group = db.RepoGroup.get_or_404(group_id)

        data = repo_group.get_dict()

        data['group_name'] = repo_group.name

        data['owner'] = repo_group.owner.username

        # fill repository group users

        for p in repo_group.repo_group_to_perm:

            data.update({'u_perm_%s' % p.user.username:

                             p.permission.permission_name})

        # fill repository group groups

        for p in repo_group.users_group_to_perm:

            data.update({'g_perm_%s' % p.users_group.users_group_name:

                             p.permission.permission_name})

        return data

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

        _list = db.RepoGroup.query(sorted=True).all()

        group_iter = RepoGroupList(_list, perm_level='admin')

        repo_groups_data = []

        _tmpl_lookup = app_globals.mako_lookup

        template = _tmpl_lookup.get_template('data_table/_dt_elements.html')

        def repo_group_name(repo_group_name, children_groups):

            return template.get_def("repo_group_name") \

                .render_unicode(repo_group_name, children_groups, _=_, webutils=webutils, c=c)

        def repo_group_actions(repo_group_id, repo_group_name, gr_count):

            return template.get_def("repo_group_actions") \

                .render_unicode(repo_group_id, repo_group_name, gr_count, _=_, webutils=webutils, c=c,

                        ungettext=ungettext)

        for repo_gr in group_iter:

            children_groups = [g.name for g in repo_gr.parents] + [repo_gr.name]

            repo_count = repo_gr.repositories.count()

            repo_groups_data.append({

                "raw_name": webutils.escape(repo_gr.group_name),

                "group_name": repo_group_name(repo_gr.group_name, children_groups),

                "desc": webutils.escape(repo_gr.group_description),

                "repos": repo_count,

                "owner": repo_gr.owner.username,

                "action": repo_group_actions(repo_gr.group_id, repo_gr.group_name,

                                             repo_count)

            })

        c.data = {

            "sort": None,

            "dir": "asc",

            "records": repo_groups_data

        }

        return base.render('admin/repo_groups/repo_groups.html')

    def create(self):

        self.__load_defaults()

        # permissions for can create group based on parent_id are checked

        # here in the Form

        repo_group_form = RepoGroupForm(repo_groups=c.repo_groups)

        form_result = None

        try:

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

            gr = RepoGroupModel().create(

                group_name=form_result['group_name'],

    @HasRepoGroupPermissionLevelDecorator('admin')

    def edit(self, group_name):

        c.active = 'settings'

        c.repo_group = db.RepoGroup.guess_instance(group_name)

        self.__load_defaults(extras=[c.repo_group.parent_group],

                             exclude=[c.repo_group])

        defaults = self.__load_data(c.repo_group.group_id)

        return htmlfill.render(

            base.render('admin/repo_groups/repo_group_edit.html'),

            defaults=defaults,

            encoding="UTF-8",

            force_defaults=False

        )

    @HasRepoGroupPermissionLevelDecorator('admin')

    def edit_repo_group_advanced(self, group_name):

        c.active = 'advanced'

        c.repo_group = db.RepoGroup.guess_instance(group_name)

        return base.render('admin/repo_groups/repo_group_edit.html')

    @HasRepoGroupPermissionLevelDecorator('admin')

    def edit_repo_group_perms(self, group_name):

        c.active = 'perms'

        c.repo_group = db.RepoGroup.guess_instance(group_name)

        self.__load_defaults()

        defaults = self.__load_data(c.repo_group.group_id)

        return htmlfill.render(

            base.render('admin/repo_groups/repo_group_edit.html'),

            defaults=defaults,

            encoding="UTF-8",

            force_defaults=False

        )

    @HasRepoGroupPermissionLevelDecorator('admin')

    def update_perms(self, group_name):

        """

        Update permissions for given repository group

        :param group_name:

        """

        c.repo_group = db.RepoGroup.guess_instance(group_name)

        valid_recursive_choices = ['none', 'repos', 'groups', 'all']

        form_result = RepoGroupPermsForm(valid_recursive_choices)().to_python(request.POST)

        recursive = form_result['recursive']

        # iterate over all members(if in recursive mode) of this groups and

        # set the permissions !

        # this can be potentially heavy operation

        RepoGroupModel()._update_permissions(c.repo_group,

                                             form_result['perms_new'],

                                             form_result['perms_updates'],

                                             recursive)

        # TODO: implement this

        #action_logger(request.authuser, 'admin_changed_repo_permissions',

        #              repo_name, request.ip_addr)

        meta.Session().commit()

        webutils.flash(_('Repository group permissions updated'), category='success')

        raise HTTPFound(location=url('edit_repo_group_perms', group_name=group_name))

    @HasRepoGroupPermissionLevelDecorator('admin')

    def delete_perms(self, group_name):

        try:

            obj_type = request.POST.get('obj_type')

            obj_id = None

            if obj_type == 'user':

                obj_id = safe_int(request.POST.get('user_id'))

            elif obj_type == 'user_group':

                obj_id = safe_int(request.POST.get('user_group_id'))

            recursive = request.POST.get('recursive', 'none')

            if obj_type == 'user':

                RepoGroupModel().delete_permission(repo_group=group_name,

                                                   obj=obj_id, obj_type='user',

                                                   recursive=recursive)

            elif obj_type == 'user_group':

                RepoGroupModel().delete_permission(repo_group=group_name,

                                                   obj=obj_id,

                                                   obj_type='user_group',

                                                   recursive=recursive)

            meta.Session().commit()

        except Exception:

            log.error(traceback.format_exc())

            webutils.flash(_('An error occurred during revoking of permission'),

                    category='error')

            raise HTTPInternalServerError()

${h.form(url('edit_repo_group_perms', group_name=c.repo_group.group_name))}

<div class="form">

        <div>

            <table id="permissions_manage" class="table">

                <tr>

                    <td>${_('None')}<br />(${_('Not visible')})</td>

                    <td>${_('Read')}<br />(${_('Visible')})</td>

                    <td>${_('Write')}<br />(${_('Add repos')})</td>

                    <td>${_('Admin')}<br />(${_('Add/Edit groups')})</td>

                    <td>${_('User/User Group')}</td>

                    <td></td>

                </tr>

                ## USERS

                %for r2p in c.repo_group.repo_group_to_perm:

                    <tr id="id${id(r2p.user.username)}">

                        <td>${h.radio('u_perm_%s' % r2p.user.username,'group.none')}</td>

                        <td>${h.radio('u_perm_%s' % r2p.user.username,'group.read')}</td>

                        <td>${h.radio('u_perm_%s' % r2p.user.username,'group.write')}</td>

                        <td>${h.radio('u_perm_%s' % r2p.user.username,'group.admin')}</td>

                        <td>

                            ${h.gravatar(r2p.user.email, cls="perm-gravatar", size=14)}

                            %if h.HasPermissionAny('hg.admin')() and r2p.user.username != 'default':

                             <a href="${h.url('edit_user',id=r2p.user.user_id)}">${r2p.user.username}</a>

                            %else:

                             ${r2p.user.username if r2p.user.username != 'default' else _('Default')}

                            %endif

                        </td>

                        <td>

                          %if r2p.user.username !='default':

                            <button type="button" class="btn btn-default btn-xs" onclick="ajaxActionRevoke(${r2p.user.user_id}, 'user', '${'id%s'%id(r2p.user.username)}', '${r2p.user.username}')">

                             <i class="icon-minus-circled"></i>${_('Revoke')}

                            </button>

                          %endif

                        </td>

                    </tr>

                %endfor

                ## USER GROUPS

                %for g2p in c.repo_group.users_group_to_perm:

                    <tr id="id${id(g2p.users_group.users_group_name)}">

                        <td>${h.radio('g_perm_%s' % g2p.users_group.users_group_name,'group.none')}</td>

                        <td>${h.radio('g_perm_%s' % g2p.users_group.users_group_name,'group.read')}</td>

                        <td>${h.radio('g_perm_%s' % g2p.users_group.users_group_name,'group.write')}</td>

                        <td>${h.radio('g_perm_%s' % g2p.users_group.users_group_name,'group.admin')}</td>

                        <td>

                            <i class="icon-users"></i>

                            %if h.HasPermissionAny('hg.admin')():

                             <a href="${h.url('edit_users_group',id=g2p.users_group.users_group_id)}">

                                 ${g2p.users_group.users_group_name}

                             </a>

                            %else:

                             ${g2p.users_group.users_group_name}

                            %endif

                        </td>

                        <td>

                            <button type="button" class="btn btn-default btn-xs" onclick="ajaxActionRevoke(${g2p.users_group.users_group_id}, 'user_group', '${'id%s'%id(g2p.users_group.users_group_name)}', '${g2p.users_group.users_group_name}')">

                            <i class="icon-minus-circled"></i>${_('Revoke')}

                            </button>

                        </td>

                    </tr>

                %endfor

                ## New entries added by addPermAction here.

                <tr class="new_members last_new_member" id="add_perm_input"><td colspan="6"></td></tr>

                <tr>

                    <td colspan="6">

                        <button type="button" id="add_perm" class="btn btn-link btn-xs">

                            <i class="icon-plus"></i>${_('Add new')}

                        </button>

                    </td>

                </tr>

                <tr>

                    <td colspan="6">

                       ${_('Apply to children')}:

                       ${h.radio('recursive', 'none', label=_('None'), checked="checked")}

                       ${h.radio('recursive', 'groups', label=_('Repository Groups'))}

                       ${h.radio('recursive', 'repos', label=_('Repositories'))}

                       ${h.radio('recursive', 'all', label=_('Both'))}

                       <span class="help-block">${_('Set or revoke permission to all children of that group, including non-private repositories and other groups if selected.')}</span>

                    </td>

                </tr>

            </table>

        </div>