import logging

from pylons import tmpl_context as c, request

from pylons.i18n.translation import _

from webob.exc import HTTPBadRequest

import rhodecode

from rhodecode.lib import helpers as h

from rhodecode.lib.ext_json import json

from rhodecode.lib.auth import LoginRequired

from rhodecode.lib.base import BaseController, render

from rhodecode.model.db import Repository

from sqlalchemy.sql.expression import func

log = logging.getLogger(__name__)

class HomeController(BaseController):

    @LoginRequired()

    def __before__(self):

        super(HomeController, self).__before__()

    def index(self):

        c.groups = self.scm_model.get_repos_groups()

        c.group = None

        if c.visual.lightweight_dashboard is False:

            c.repos_list = self.scm_model.get_repos()

        ## lightweight version of dashboard

        else:

            c.repos_list = Repository.query()\

                            .filter(Repository.group_id == None)\

                            .order_by(func.lower(Repository.repo_name))\

                            .all()

            repos_data = []

            total_records = len(c.repos_list)

            _tmpl_lookup = rhodecode.CONFIG['pylons.app_globals'].mako_lookup

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

            quick_menu = lambda repo_name: (template.get_def("quick_menu")

                                            .render(repo_name, _=_, h=h, c=c))

            repo_lnk = lambda name, rtype, private, fork_of: (

                template.get_def("repo_name")

                .render(name, rtype, private, fork_of, short_name=False,

                        admin=False, _=_, h=h, c=c))

            rss_lnk = lambda repo_name: (template.get_def("rss")

                                           .render(repo_name, _=_, h=h, c=c))

            atom_lnk = lambda repo_name: (template.get_def("atom")

                                           .render(repo_name, _=_, h=h, c=c))

            for repo in c.repos_list:

                repos_data.append({

                    "menu": quick_menu(repo.repo_name),

                    "raw_name": repo.repo_name.lower(),

                    "name": repo_lnk(repo.repo_name, repo.repo_type,

                                     repo.private, repo.fork),

                    "desc": repo.description,

                    "owner": h.person(repo.user.username),

                    "rss": rss_lnk(repo.repo_name),

                    "atom": atom_lnk(repo.repo_name),

                })

            c.data = json.dumps({

                "totalRecords": total_records,

                "startIndex": 0,

                "sort": "name",

                "dir": "asc",

                "records": repos_data

            })

        return render('/index.html')

    def repo_switcher(self):

        if request.is_xhr:

            all_repos = Repository.query().order_by(Repository.repo_name).all()

            c.repos_list = self.scm_model.get_repos(all_repos,

                                                    sort_key='name_sort',

                                                    simple=True)

            return render('/repo_switcher_list.html')

        else:

            return HTTPBadRequest()

    def branch_tag_switcher(self, repo_name):

        if request.is_xhr:

            c.rhodecode_db_repo = Repository.get_by_repo_name(c.repo_name)

            c.rhodecode_repo = c.rhodecode_db_repo.scm_instance

            return render('/switch_to_list.html')

        else:

            return HTTPBadRequest()

        # WSGIController.__call__ dispatches to the Controller method

        # the request is routed to. This routing information is

        # available in environ['pylons.routes_dict']

        start = time.time()

        try:

            self.ip_addr = _get_ip_addr(environ)

            # make sure that we update permissions each time we call controller

            api_key = request.GET.get('api_key')

            cookie_store = CookieStoreWrapper(session.get('rhodecode_user'))

            user_id = cookie_store.get('user_id', None)

            username = get_container_username(environ, config)

            auth_user = AuthUser(user_id, api_key, username)

            request.user = auth_user

            self.rhodecode_user = c.rhodecode_user = auth_user

            if not self.rhodecode_user.is_authenticated and \

                       self.rhodecode_user.user_id is not None:

                self.rhodecode_user.set_authenticated(

                    cookie_store.get('is_authenticated')

                )

            log.info('IP: %s User: %s accessed %s' % (

               self.ip_addr, auth_user, safe_unicode(_get_access_path(environ)))

            )

            return WSGIController.__call__(self, environ, start_response)

        finally:

            log.info('IP: %s Request to %s time: %.3fs' % (

                _get_ip_addr(environ),

                safe_unicode(_get_access_path(environ)), time.time() - start)

            )

            meta.Session.remove()

class BaseRepoController(BaseController):

    """

    Base class for controllers responsible for loading all needed data for

    repository loaded items are

    c.rhodecode_repo: instance of scm repository

    c.rhodecode_db_repo: instance of db

    c.repository_followers: number of followers

    c.repository_forks: number of forks

    """

    def __before__(self):

        super(BaseRepoController, self).__before__()

        if c.repo_name:

            dbr = c.rhodecode_db_repo = Repository.get_by_repo_name(c.repo_name)

            c.rhodecode_repo = c.rhodecode_db_repo.scm_instance

            if c.rhodecode_repo is None:

                log.error('%s this repository is present in database but it '

                          'cannot be created as an scm instance', c.repo_name)

                redirect(url('home'))

            # some globals counter for menu

            c.repository_followers = self.scm_model.get_followers(dbr)

            c.repository_forks = self.scm_model.get_forks(dbr)

            c.repository_pull_requests = self.scm_model.get_pull_requests(dbr)

        """

        repo = self

        data = dict(

            repo_id=repo.repo_id,

            repo_name=repo.repo_name,

            repo_type=repo.repo_type,

            clone_uri=repo.clone_uri,

            private=repo.private,

            created_on=repo.created_on,

            description=repo.description,

            landing_rev=repo.landing_rev,

            owner=repo.user.username,

            fork_of=repo.fork.repo_name if repo.fork else None

        )

        return data

    @classmethod

    def lock(cls, repo, user_id):

        repo.locked = [user_id, time.time()]

        Session().add(repo)

        Session().commit()

    @classmethod

    def unlock(cls, repo):

        repo.locked = None

        Session().add(repo)

        Session().commit()

    @property

    def last_db_change(self):

        return self.updated_on

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

    # SCM PROPERTIES

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

    def get_changeset(self, rev=None):

        return get_changeset_safe(self.scm_instance, rev)

    def get_landing_changeset(self):

        """

        Returns landing changeset, or if that doesn't exist returns the tip

        """

        cs = self.get_changeset(self.landing_rev) or self.get_changeset()

        return cs

    @property

    def tip(self):

        return self.get_changeset('tip')

    @property

    def author(self):

        return self.tip.author

    @property

    def last_change(self):

        return self.scm_instance.last_change

    def get_comments(self, revisions=None):

        """

        Returns comments for this repository grouped by revisions

        :param revisions: filter query by revisions only

        """

        cmts = ChangesetComment.query()\

            .filter(ChangesetComment.repo == self)

        if revisions:

            cmts = cmts.filter(ChangesetComment.revision.in_(revisions))

        grouped = defaultdict(list)

        for cmt in cmts.all():

            grouped[cmt.revision].append(cmt)

        return grouped

    def statuses(self, revisions=None):

        """

        Returns statuses for this repository

        :param revisions: list of revisions to get statuses for

        :type revisions: list

        """

        statuses = ChangesetStatus.query()\

            .filter(ChangesetStatus.repo == self)\

            .filter(ChangesetStatus.version == 0)

        if revisions:

            statuses = statuses.filter(ChangesetStatus.revision.in_(revisions))

        grouped = {}

        #maybe we have open new pullrequest without a status ?

        stat = ChangesetStatus.STATUS_UNDER_REVIEW

        status_lbl = ChangesetStatus.get_status_lbl(stat)

        for pr in PullRequest.query().filter(PullRequest.org_repo == self).all():

            for rev in pr.revisions:

                pr_id = pr.pull_request_id

    </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 and c.visual.show_private_icon:

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

   %elif not private and c.visual.show_public_icon:

     <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.repo_name)}">

        <img class="icon" alt="${_('fork')}" title="${_('Fork of')} ${fork_of.repo_name}" 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>

<%def name="rss(name)">

  %if c.rhodecode_user.username != 'default':

    <a title="${_('Subscribe to %s rss feed')% name}" class="rss_icon"  href="${h.url('rss_feed_home',repo_name=name,api_key=c.rhodecode_user.api_key)}"></a>

  %else:

    <a title="${_('Subscribe to %s rss feed')% name}" class="rss_icon"  href="${h.url('rss_feed_home',repo_name=name)}"></a>

  %endif

</%def>

<%def name="atom(name)">

  %if c.rhodecode_user.username != 'default':

    <a title="${_('Subscribe to %s atom feed')% name}"  class="atom_icon" href="${h.url('atom_feed_home',repo_name=name,api_key=c.rhodecode_user.api_key)}"></a>

  %else:

    <a title="${_('Subscribe to %s atom feed')% name}"  class="atom_icon" href="${h.url('atom_feed_home',repo_name=name)}"></a>

  %endif

</%def>

<%def name="user_gravatar(email, size=24)">

    <div class="gravatar"><img alt="gravatar" src="${h.gravatar_url(email, size)}"/> </div>

</%def>

<%def name="user_actions(user_id, username)">

  ${h.form(h.url('delete_user', id=user_id),method='delete')}

      ${h.submit('remove_',_('delete'),id="remove_user_%s" % user_id,

      class_="delete_icon action_button",onclick="return confirm('"+_('Confirm to delete this user: %s') % username+"');")}

  ${h.end_form()}

</%def>

<%def name="user_name(user_id, username)">

    ${h.link_to(username,h.url('edit_user', id=user_id))}

</%def>

                  % endfor

              </table>

            </div>

            <div style="height: 20px"></div>

            % endif

            <div id="welcome" style="display:none;text-align:center">

                <h1><a href="${h.url('home')}">${c.rhodecode_name} ${c.rhodecode_version}</a></h1>

            </div>

             <%cnt=0%>

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

            % if c.visual.lightweight_dashboard is False:

              ## old full detailed version

            <div id='repos_list_wrap' class="yui-skin-sam">

            <table id="repos_list">

            <thead>

                <tr>

                    <th class="left"></th>

                    <th class="left">${_('Name')}</th>

                    <th class="left">${_('Description')}</th>

                    <th class="left">${_('Last change')}</th>

                    <th class="left">${_('Tip')}</th>

                    <th class="left">${_('Owner')}</th>

                    <th class="left">${_('RSS')}</th>

                    <th class="left">${_('Atom')}</th>

                </tr>

            </thead>

            <tbody>

            %for cnt,repo in enumerate(c.repos_list):

                <tr class="parity${(cnt+1)%2}">

                    ##QUICK MENU

                    <td class="quick_repo_menu">

                      ${dt.quick_menu(repo['name'])}

                    </td>

                    ##REPO NAME AND ICONS

                    <td class="reponame">

                      ${dt.repo_name(repo['name'],repo['dbrepo']['repo_type'],repo['dbrepo']['private'],h.AttributeDict(repo['dbrepo_fork']),pageargs.get('short_repo_names'))}

                    </td>

                    ##DESCRIPTION

                    <td><span class="tooltip" title="${h.tooltip(repo['description'])}">

                       %if c.visual.stylify_metatags:

                       ${h.urlify_text(h.desc_stylize(h.truncate(repo['description'],60)))}</span>

                       %else:

                       ${h.truncate(repo['description'],60)}</span>

                       %endif

                    </td>

                    ##LAST CHANGE DATE

                    <td>

                    </td>

                    ##LAST REVISION

                    <td>

                        ${dt.revision(repo['name'],repo['rev'],repo['tip'],repo['author'],repo['last_msg'])}

                    </td>

                    ##

                    <td title="${repo['contact']}">${h.person(repo['contact'])}</td>

                    <td>

                      ${dt.rss(repo['name'])}

                    </td>

                    <td>

                      ${dt.atom(repo['name'])}

                    </td>

                </tr>

            %endfor

            </tbody>

            </table>

            </div>

            % else:

              ## lightweight version

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

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

            % endif

        </div>

    </div>

    % if c.visual.lightweight_dashboard is False:

    <script>

      YUD.get('repo_count').innerHTML = ${cnt+1 if cnt else 0};

      var func = function(node){

          return node.parentNode.parentNode.parentNode.parentNode;

      }

      // groups table sorting

      var myColumnDefs = [

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

              sortOptions: { sortFunction: groupNameSort }},

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

      ];

      var myDataSource = new YAHOO.util.DataSource(YUD.get("groups_list"));

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

      myDataSource.responseSchema = {

          fields: [

              {key:"name"},

              {key:"desc"},

          ]

      };

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

      myDataSource.responseSchema = {

          fields: [

              {key:"menu"},

              //{key:"raw_name"},

              {key:"name"},

              {key:"desc"},

              {key:"last_change"},

              {key:"tip"},

              {key:"owner"},

              {key:"rss"},

              {key:"atom"},

          ]

      };

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

              {

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

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

          tooltip_activate();

          quick_repo_menu();

          q_filter('q_filter',YUQ('div.table tr td a.repo_name'),func);

      });

    </script>

    % else:

      <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:"menu"},

               {key:"raw_name"},

               {key:"name"},

               {key:"desc"},

               {key:"owner"},

               {key:"rss"},

               {key:"atom"},

            ]

         };

        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_name.toLowerCase().indexOf(req)

                    if (pos != -1) {

                        filtered.push(data[i]);

                    }

                }

                res.results = filtered;

            }

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

            return res;

        }

        // main table sorting

        var myColumnDefs = [

            {key:"menu",label:"",sortable:false,className:"quick_repo_menu hidden"},

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

                sortOptions: { sortFunction: nameSort }},

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

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

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

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

        ];

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

          sortedBy:{key:"name",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) {

            tooltip_activate();

            quick_repo_menu();

        });

        var filterTimeout = null;

        updateFilter  = function () {

            // Reset timeout

            filterTimeout = null;

            // Reset sort

            var state = myDataTable.getState();

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

            // Get filtered data

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

                success : myDataTable.onDataReturnInitializeTable,

                failure : myDataTable.onDataReturnInitializeTable,