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

"""

    rhodecode.controllers.admin.users

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

    Users crud controller for pylons

    :created_on: Apr 4, 2010

    :author: marcink

    :copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com>

    :license: GPLv3, see COPYING for more details.

"""

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

import logging

import traceback

import formencode

from pylons import response

from formencode import htmlfill

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

from pylons.controllers.util import redirect

from pylons.i18n.translation import _

from rhodecode.lib.exceptions import DefaultUserException, \

    UserOwnsReposException

from rhodecode.lib import helpers as h

from rhodecode.lib.auth import LoginRequired, HasPermissionAllDecorator, \

    AuthUser

from rhodecode.lib.base import BaseController, render

from rhodecode.model.db import User, Permission, UserEmailMap

from rhodecode.model.forms import UserForm

from rhodecode.model.user import UserModel

from rhodecode.model.meta import Session

from rhodecode.lib.utils import action_logger

from rhodecode.lib.compat import json

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

    @HasPermissionAllDecorator('hg.admin')

    def __before__(self):

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

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

        super(UsersController, self).__before__()

        c.available_permissions = config['available_permissions']

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

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

        # url('users')

        c.users_list = User.query().order_by(User.username).all()

        users_data = []

        total_records = len(c.users_list)

        for user in c.users_list:

            users_data.append({

                "raw_username": user.username,

                "firstname": user.name,

                "lastname": user.lastname,

                "last_login": h.fmt_date(user.last_login),

                "active": h.bool2icon(user.active),

                "admin": h.bool2icon(user.admin),

                "ldap": h.bool2icon(bool(user.ldap_dn)),

            })

        c.data = json.dumps({

            "totalRecords": total_records,

            "startIndex": 0,

            "sort": None,

            "dir": "asc",

            "records": users_data

        })

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

    def create(self):

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

        # url('users')

        user_model = UserModel()

        user_form = UserForm()()

        try:

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

            user_model.create(form_result)

            usr = form_result['username']

            action_logger(self.rhodecode_user, 'admin_created_user:%s' % usr,

                          None, self.ip_addr, self.sa)

            h.flash(_('created user %s') % usr,

                    category='success')

            Session().commit()

        except formencode.Invalid, errors:

            return htmlfill.render(

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

                defaults=errors.value,

                errors=errors.error_dict or {},

                prefix_error=False,

                encoding="UTF-8")

        except Exception:

            log.error(traceback.format_exc())

            h.flash(_('error occurred during creation of user %s') \

                    % request.POST.get('username'), category='error')

        return redirect(url('users'))

    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('update_user', id=ID),

        #           method='put')

        # url('user', id=ID)

        user_model = UserModel()

        c.user = user_model.get(id)

        c.perm_user = AuthUser(user_id=id)

        _form = UserForm(edit=True, old_data={'user_id': id,

                                              'email': c.user.email})()

        form_result = {}

        try:

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

            user_model.update(id, form_result)

            usr = form_result['username']

            action_logger(self.rhodecode_user, 'admin_updated_user:%s' % usr,

                          None, self.ip_addr, self.sa)

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

            Session().commit()

        except formencode.Invalid, errors:

            c.user_email_map = UserEmailMap.query()\

                            .filter(UserEmailMap.user == c.user).all()

            defaults = errors.value

            e = errors.error_dict or {}

            perm = Permission.get_by_key('hg.create.repository')

            defaults.update({'create_repo_perm': user_model.has_perm(id, perm)})

            defaults.update({'_method': 'put'})

            return htmlfill.render(

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

                defaults=defaults,

                errors=e,

                prefix_error=False,

                encoding="UTF-8")

        except Exception:

            log.error(traceback.format_exc())

            h.flash(_('error occurred during update of user %s') \

                    % form_result.get('username'), category='error')

        return redirect(url('users'))

    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('delete_user', id=ID),

        #           method='delete')

        # url('user', id=ID)

        user_model = UserModel()

        try:

            user_model.delete(id)

            Session().commit()

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

        except (UserOwnsReposException, DefaultUserException), e:

            h.flash(e, category='warning')

        except Exception:

            log.error(traceback.format_exc())

            h.flash(_('An error occurred during deletion of user'),

                    category='error')

        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 = User.get_or_404(id)

        if c.user.username == 'default':

            h.flash(_("You can't edit this user"), category='warning')

            return redirect(url('users'))

        c.perm_user = AuthUser(user_id=id)

        c.user.permissions = {}

        c.granted_permissions = UserModel().fill_perms(c.user)\

            .permissions['global']

        c.user_email_map = UserEmailMap.query()\

                        .filter(UserEmailMap.user == c.user).all()

        defaults = c.user.get_dict()

        perm = Permission.get_by_key('hg.create.repository')

        defaults.update({'create_repo_perm': UserModel().has_perm(id, perm)})

        return htmlfill.render(

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

            defaults=defaults,

            encoding="UTF-8",

            force_defaults=False

        )

    def update_perm(self, id):

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

        # url('user_perm', id=ID, method='put')

        grant_perm = request.POST.get('create_repo_perm', False)

        user_model = UserModel()

        if grant_perm:

            perm = Permission.get_by_key('hg.create.none')

            user_model.revoke_perm(id, perm)

            perm = Permission.get_by_key('hg.create.repository')

            user_model.grant_perm(id, perm)

            h.flash(_("Granted 'repository create' permission to user"),

                    category='success')

            Session().commit()

        else:

            perm = Permission.get_by_key('hg.create.repository')

            user_model.revoke_perm(id, perm)

            perm = Permission.get_by_key('hg.create.none')

            user_model.grant_perm(id, perm)

            h.flash(_("Revoked 'repository create' permission to user"),

                    category='success')

            Session().commit()

        return redirect(url('edit_user', id=id))

    def add_email(self, id):

        """POST /user_emails:Add an existing item"""

        # url('user_emails', id=ID, method='put')

        #TODO: validation and form !!!

        email = request.POST.get('new_email')

        user_model = UserModel()

        try:

            user_model.add_extra_email(id, email)

            Session().commit()

            h.flash(_("Added email %s to user") % email, category='success')

        except formencode.Invalid, error:

            msg = error.error_dict['email']

            h.flash(msg, category='error')

        except Exception:

            log.error(traceback.format_exc())

            h.flash(_('An error occurred during email saving'),

                    category='error')

        return redirect(url('edit_user', id=id))

    def delete_email(self, id):

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

        # url('user_emails_delete', id=ID, method='delete')

        user_model = UserModel()

        user_model.delete_extra_email(id, request.POST.get('del_email'))

        Session().commit()

        h.flash(_("Removed email from user"), category='success')

        return redirect(url('edit_user', id=id))

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

<%inherit file="/base/base.html"/>

<%def name="title()">

    ${_('Users administration')} - ${c.rhodecode_name}

</%def>

<%def name="breadcrumbs_links()">

    <input class="q_filter_box" id="q_filter" size="15" type="text" name="filter" value="${_('quick filter...')}"/> ${h.link_to(_('Admin'),h.url('admin_home'))} &raquo; <span id="user_count">0</span> ${_('users')}

</%def>

<%def name="page_nav()">

	${self.menu('admin')}

</%def>

<%def name="main()">

<div class="box">

    <!-- box / title -->

    <div class="title">

        ${self.breadcrumbs()} 

        <ul class="links">

          <li>

            <span>${h.link_to(_(u'ADD NEW USER'),h.url('new_user'))}</span>

          </li>

        </ul>

    </div>

    <!-- end box / title -->

    <div class="table yui-skin-sam" id="users_list_wrap"></div>

    <div id="user-paginator" style="padding: 0px 0px 0px 20px"></div>

</div>

<script>

  var url = "${h.url('formatted_users', format='json')}";

  var data = ${c.data|n};

  var myDataSource = new YAHOO.util.DataSource(data);

  myDataSource.responseType = YAHOO.util.DataSource.TYPE_JSON;

  myDataSource.responseSchema = {

	  resultsList: "records",

      fields: [

          {key: "gravatar"},

          {key: "raw_username"},

          {key: "username"},

          {key: "firstname"},

          {key: "lastname"},

          {key: "last_login"},

          {key: "active"},

          {key: "admin"},

          {key: "ldap"},

          {key: "action"},

      ]

   };

  myDataSource.doBeforeCallback = function(req,raw,res,cb) {

      // This is the filter function

      var data     = res.results || [],

          filtered = [],

          i,l;

      if (req) {

          req = req.toLowerCase();

          for (i = 0; i<data.length; i++) {

        	  var pos = data[i].raw_username.toLowerCase().indexOf(req)

              if (pos != -1) {

                  filtered.push(data[i]);

              }

          }

          res.results = filtered;

      }

      YUD.get('user_count').innerHTML = res.results.length;

      return res;

  }

  // main table sorting

  var myColumnDefs = [

      {key:"gravatar",label:"",sortable:false,},                      

      {key:"username",label:"${_('username')}",sortable:true,

    	  sortOptions: { sortFunction: linkSort }

      },

      {key:"firstname",label:"${_('firstname')}",sortable:true,},

      {key:"lastname",label:"${_('lastname')}",sortable:true,},

      {key:"last_login",label:"${_('last login')}",sortable:true,},

      {key:"active",label:"${_('active')}",sortable:true,},

      {key:"admin",label:"${_('admin')}",sortable:true,},

      {key:"ldap",label:"${_('ldap')}",sortable:true,},

      {key:"action",label:"${_('action')}",sortable:false},

  ];

  var myDataTable = new YAHOO.widget.DataTable("users_list_wrap", myColumnDefs, myDataSource,{

    sortedBy:{key:"username",dir:"asc"},

    paginator: new YAHOO.widget.Paginator({

        rowsPerPage: 15,

        alwaysVisible: false,

        template : "{PreviousPageLink} {FirstPageLink} {PageLinks} {LastPageLink} {NextPageLink}",

        pageLinks: 5,

        containerClass: 'pagination-wh',

        currentPageClass: 'pager_curpage',

        pageLinkClass: 'pager_link',

        nextPageLinkLabel: '>',

        previousPageLinkLabel: '<',

        firstPageLinkLabel: '<<',

        lastPageLinkLabel: '>>',

        containers:['user-paginator']

    }),

    MSG_SORTASC:"${_('Click to sort ascending')}",

    MSG_SORTDESC:"${_('Click to sort descending')}",

    MSG_EMPTY:"${_('No records found.')}",

    MSG_ERROR:"${_('Data error.')}",

    MSG_LOADING:"${_('Loading...')}",

  }

  );

  myDataTable.subscribe('postRenderEvent',function(oArgs) {

  });

  var filterTimeout = null;

  updateFilter  = function () {

      // Reset timeout

      filterTimeout = null;

      // Reset sort

      var state = myDataTable.getState();

          state.sortedBy = {key:'username', dir:YAHOO.widget.DataTable.CLASS_ASC};

      // Get filtered data

      myDataSource.sendRequest(YUD.get('q_filter').value,{

          success : myDataTable.onDataReturnInitializeTable,

          failure : myDataTable.onDataReturnInitializeTable,

          scope   : myDataTable,

          argument: state

      });

  };  

  YUE.on('q_filter','click',function(){

      YUD.get('q_filter').value = '';

   });

  YUE.on('q_filter','keyup',function (e) {

      clearTimeout(filterTimeout);

      filterTimeout = setTimeout(updateFilter,600);

  });  

</script>

</%def>

## DATA TABLE RE USABLE ELEMENTS

## usage:

## <%namespace name="dt" file="/data_table/_dt_elements.html"/>

<%def name="quick_menu(repo_name)">

  <ul class="menu_items hidden">

    <li style="border-top:1px solid #003367;margin-left:18px;padding-left:-99px"></li>

    <li>

       <a title="${_('Summary')}" href="${h.url('summary_home',repo_name=repo_name)}">

       <span class="icon">

           <img src="${h.url('/images/icons/clipboard_16.png')}" alt="${_('Summary')}" />

       </span>

       <span>${_('Summary')}</span>

       </a>

    </li>

    <li>

       <a title="${_('Changelog')}" href="${h.url('changelog_home',repo_name=repo_name)}">

       <span class="icon">

           <img src="${h.url('/images/icons/time.png')}" alt="${_('Changelog')}" />

       </span>

       <span>${_('Changelog')}</span>

       </a>

    </li>

    <li>

       <a title="${_('Files')}" href="${h.url('files_home',repo_name=repo_name)}">

       <span class="icon">

           <img src="${h.url('/images/icons/file.png')}" alt="${_('Files')}" />

       </span>

       <span>${_('Files')}</span>

       </a>

    </li>

    <li>

       <a title="${_('Fork')}" href="${h.url('repo_fork_home',repo_name=repo_name)}">

       <span class="icon">

           <img src="${h.url('/images/icons/arrow_divide.png')}" alt="${_('Fork')}" />

       </span>

       <span>${_('Fork')}</span>

       </a>

    </li>

  </ul>

</%def>

<%def name="repo_name(name,rtype,private,fork_of,short_name=False, admin=False)">

    <%

    def get_name(name,short_name=short_name):

      if short_name:

        return name.split('/')[-1]

      else:

        return name

    %>

  <div style="white-space: nowrap">

   ##TYPE OF REPO

   %if h.is_hg(rtype):

     <img class="icon" title="${_('Mercurial repository')}" alt="${_('Mercurial repository')}" src="${h.url('/images/icons/hgicon.png')}"/>

   %elif h.is_git(rtype):

     <img class="icon" title="${_('Git repository')}" alt="${_('Git repository')}" src="${h.url('/images/icons/giticon.png')}"/>

   %endif

   ##PRIVATE/PUBLIC

   %if private:

      <img class="icon" title="${_('private repository')}" alt="${_('private repository')}" src="${h.url('/images/icons/lock.png')}"/>

   %else:

      <img class="icon" title="${_('public repository')}" alt="${_('public repository')}" src="${h.url('/images/icons/lock_open.png')}"/>

   %endif

   ##NAME

   %if admin:

    ${h.link_to(get_name(name),h.url('edit_repo',repo_name=name),class_="repo_name")}

   %else:

    ${h.link_to(get_name(name),h.url('summary_home',repo_name=name),class_="repo_name")}

   %endif

   %if fork_of:

        <a href="${h.url('summary_home',repo_name=fork_of)}">

        <img class="icon" alt="${_('fork')}" title="${_('Fork of')} ${fork_of}" src="${h.url('/images/icons/arrow_divide.png')}"/></a>

   %endif

  </div>

</%def>

<%def name="revision(name,rev,tip,author,last_msg)">

  <div>

  %if rev >= 0:

      <pre><a title="${h.tooltip('%s:\n\n%s' % (author,last_msg))}" class="tooltip" href="${h.url('changeset_home',repo_name=name,revision=tip)}">${'r%s:%s' % (rev,h.short_id(tip))}</a></pre>

  %else:

      ${_('No changesets yet')}

  %endif

  </div>

</%def>