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

"""

    rhodecode.controllers.feed

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

    Feed controller for rhodecode

    :created_on: Apr 23, 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

from pylons import url, response, tmpl_context as c

from pylons.i18n.translation import _

from beaker.cache import cache_region, region_invalidate

from webhelpers.feedgenerator import Atom1Feed, Rss201rev2Feed

from rhodecode.lib import helpers as h

from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator

from rhodecode.lib.base import BaseRepoController

from rhodecode.lib.diffs import DiffProcessor, LimitedDiffContainer

from rhodecode.model.db import CacheInvalidation

from rhodecode.lib.utils2 import safe_int, str2bool

log = logging.getLogger(__name__)

class FeedController(BaseRepoController):

    @LoginRequired(api_access=True)

    @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',

                                   'repository.admin')

    def __before__(self):

        super(FeedController, self).__before__()

        #common values for feeds

        self.description = _('Changes on %s repository')

        self.title = self.title = _('%s %s feed') % (c.rhodecode_name, '%s')

        self.language = 'en-us'

        self.ttl = "5"

        import rhodecode

        CONF = rhodecode.CONFIG

        self.include_diff = str2bool(CONF.get('rss_include_diff', False))

        self.feed_nr = safe_int(CONF.get('rss_items_per_page', 20))

        # we need to protect from parsing huge diffs here other way

        # we can kill the server

        self.feed_diff_limit = safe_int(CONF.get('rss_cut_off_limit'), 32 * 1024)

    def _get_title(self, cs):

        return "%s" % (

            h.shorter(cs.message, 160)

        )

    def __changes(self, cs):

        changes = []

        diff_processor = DiffProcessor(cs.diff(),

                                       diff_limit=self.feed_diff_limit)

        _parsed = diff_processor.prepare(inline_diff=False)

        limited_diff = False

        if isinstance(_parsed, LimitedDiffContainer):

            limited_diff = True

        for st in _parsed:

            st.update({'added': st['stats'][0],

                       'removed': st['stats'][1]})

            changes.append('\n %(operation)s %(filename)s '

                           '(%(added)s lines added, %(removed)s lines removed)'

                            % st)

        if limited_diff:

            changes = changes + ['\n ' +

                                 _('Changeset was too big and was cut off...')]

        return diff_processor, changes

    def __get_desc(self, cs):

        desc_msg = []

        desc_msg.append('%s %s %s<br/>' % (h.person(cs.author),

                                           _('commited on'),

                                           h.fmt_date(cs.date)))

        #branches, tags, bookmarks

        if cs.branch:

            desc_msg.append('branch: %s<br/>' % cs.branch)

        if h.is_hg(c.rhodecode_repo):

            for book in cs.bookmarks:

                desc_msg.append('bookmark: %s<br/>' % book)

        for tag in cs.tags:

            desc_msg.append('tag: %s<br/>' % tag)

        diff_processor, changes = self.__changes(cs)

        # rev link

        _url = url('changeset_home', repo_name=cs.repository.name,

                   revision=cs.raw_id, qualified=True)

        desc_msg.append('changesest: <a href="%s">%s</a>' % (_url, cs.raw_id[:8]))

        desc_msg.append('<pre>')

        desc_msg.append(cs.message)

        desc_msg.append('\n')

        desc_msg.extend(changes)

        if self.include_diff:

            desc_msg.append('\n\n')

            desc_msg.append(diff_processor.as_raw())

        desc_msg.append('</pre>')

        return desc_msg

    def atom(self, repo_name):

        """Produce an atom-1.0 feed via feedgenerator module"""

        @cache_region('long_term')

        def _get_feed_from_cache(key):

            feed = Atom1Feed(

                 title=self.title % repo_name,

                 link=url('summary_home', repo_name=repo_name,

                          qualified=True),

                 description=self.description % repo_name,

                 language=self.language,

                 ttl=self.ttl

            )

            for cs in reversed(list(c.rhodecode_repo[-self.feed_nr:])):

                feed.add_item(title=self._get_title(cs),

                              link=url('changeset_home', repo_name=repo_name,

                                       revision=cs.raw_id, qualified=True),

                              author_name=cs.author,

                              description=''.join(self.__get_desc(cs)),

                              pubdate=cs.date,

                              )

            response.content_type = feed.mime_type

            return feed.writeString('utf-8')

        key = repo_name + '_ATOM'

        inv = CacheInvalidation.invalidate(key)

        if inv is not None:

            region_invalidate(_get_feed_from_cache, None, key)

            CacheInvalidation.set_valid(inv.cache_key)

        return _get_feed_from_cache(key)

    def rss(self, repo_name):

        """Produce an rss2 feed via feedgenerator module"""

        @cache_region('long_term')

        def _get_feed_from_cache(key):

            feed = Rss201rev2Feed(

                title=self.title % repo_name,

                link=url('summary_home', repo_name=repo_name,

                         qualified=True),

                description=self.description % repo_name,

                language=self.language,

                ttl=self.ttl

            )

            for cs in reversed(list(c.rhodecode_repo[-self.feed_nr:])):

                feed.add_item(title=self._get_title(cs),

                              link=url('changeset_home', repo_name=repo_name,

                                       revision=cs.raw_id, qualified=True),

                              author_name=cs.author,

                              description=''.join(self.__get_desc(cs)),

                              pubdate=cs.date,

                             )

            response.content_type = feed.mime_type

            return feed.writeString('utf-8')

        key = repo_name + '_RSS'

        inv = CacheInvalidation.invalidate(key)

        if inv is not None:

            region_invalidate(_get_feed_from_cache, None, key)

            CacheInvalidation.set_valid(inv.cache_key)

        return _get_feed_from_cache(key)

    def as_raw(self, diff_lines=None):

        """

        Returns raw string diff

        """

        return self._diff

        #return u''.join(imap(self._line_counter, self._diff.splitlines(1)))

    def as_html(self, table_class='code-difftable', line_class='line',

                new_lineno_class='lineno old', old_lineno_class='lineno new',

                code_class='code', enable_comments=False, parsed_lines=None):

        """

        Return given diff as html table with customized css classes

        """

        def _link_to_if(condition, label, url):

            """

            Generates a link if condition is meet or just the label if not.

            """

            if condition:

                return '''<a href="%(url)s">%(label)s</a>''' % {

                    'url': url,

                    'label': label

                }

            else:

                return label

        if not self.parsed:

            self.prepare()

        diff_lines = self.parsed_diff

        if parsed_lines:

            diff_lines = parsed_lines

        _html_empty = True

        _html = []

        _html.append('''<table class="%(table_class)s">\n''' % {

            'table_class': table_class

        })

        for diff in diff_lines:

            for line in diff['chunks']:

                _html_empty = False

                for change in line:

                    _html.append('''<tr class="%(lc)s %(action)s">\n''' % {

                        'lc': line_class,

                        'action': change['action']

                    })

                    anchor_old_id = ''

                    anchor_new_id = ''

                    anchor_old = "%(filename)s_o%(oldline_no)s" % {

                        'filename': self._safe_id(diff['filename']),

                        'oldline_no': change['old_lineno']

                    }

                    anchor_new = "%(filename)s_n%(oldline_no)s" % {

                        'filename': self._safe_id(diff['filename']),

                        'oldline_no': change['new_lineno']

                    }

                    cond_old = (change['old_lineno'] != '...' and

                                change['old_lineno'])

                    cond_new = (change['new_lineno'] != '...' and

                                change['new_lineno'])

                    if cond_old:

                        anchor_old_id = 'id="%s"' % anchor_old

                    if cond_new:

                        anchor_new_id = 'id="%s"' % anchor_new

                    ###########################################################

                    # OLD LINE NUMBER

                    ###########################################################

                    _html.append('''\t<td %(a_id)s class="%(olc)s">''' % {

                        'a_id': anchor_old_id,

                        'olc': old_lineno_class

                    })

                    _html.append('''%(link)s''' % {

                        'link': _link_to_if(True, change['old_lineno'],

                                            '#%s' % anchor_old)

                    })

                    _html.append('''</td>\n''')

                    ###########################################################

                    # NEW LINE NUMBER

                    ###########################################################

                    _html.append('''\t<td %(a_id)s class="%(nlc)s">''' % {

                        'a_id': anchor_new_id,

                        'nlc': new_lineno_class

                    })

                    _html.append('''%(link)s''' % {

                        'link': _link_to_if(True, change['new_lineno'],

                                            '#%s' % anchor_new)

                    })

                    _html.append('''</td>\n''')

                    ###########################################################

                    # CODE

                    ###########################################################

                    comments = '' if enable_comments else 'no-comment'

                    _html.append('''\t<td class="%(cc)s %(inc)s">''' % {

                        'cc': code_class,

                        'inc': comments

                    })

                    _html.append('''\n\t\t<pre>%(code)s</pre>\n''' % {

                        'code': change['line']

                    })

                    _html.append('''\t</td>''')

                    _html.append('''\n</tr>\n''')

        _html.append('''</table>''')

        if _html_empty:

            return None

        return ''.join(_html)

    def stat(self):

        """

        Returns tuple of added, and removed lines for this instance

        """

        return self.adds, self.removes

class InMemoryBundleRepo(bundlerepository):

    def __init__(self, ui, path, bundlestream):

        self._tempparent = None

        localrepo.localrepository.__init__(self, ui, path)

        self.ui.setconfig('phases', 'publish', False)

        self.bundle = bundlestream

        # dict with the mapping 'filename' -> position in the bundle

        self.bundlefilespos = {}

def differ(org_repo, org_ref, other_repo, other_ref, discovery_data=None,

           remote_compare=False, context=3, ignore_whitespace=False):

    """

    General differ between branches, bookmarks, revisions of two remote or

    local but related repositories

    :param org_repo:

    :param org_ref:

    :param other_repo:

    :type other_repo:

    :type other_ref:

    """

    org_repo_scm = org_repo.scm_instance

    other_repo_scm = other_repo.scm_instance

    org_repo = org_repo_scm._repo

    other_repo = other_repo_scm._repo

    org_ref = org_ref[1]

    other_ref = other_ref[1]

    if org_repo == other_repo:

        log.debug('running diff between %s@%s and %s@%s'

                  % (org_repo.path, org_ref, other_repo.path, other_ref))

        _diff = org_repo_scm.get_diff(rev1=org_ref, rev2=other_ref,

            ignore_whitespace=ignore_whitespace, context=context)

        return _diff

    elif remote_compare:

        opts = diffopts(git=True, ignorews=ignore_whitespace, context=context)

        common, incoming, rheads = discovery_data

        org_repo_peer = localrepo.locallegacypeer(org_repo.local())

        # create a bundle (uncompressed if other repo is not local)

        if org_repo_peer.capable('getbundle'):

            # disable repo hooks here since it's just bundle !

            # patch and reset hooks section of UI config to not run any

            # hooks on fetching archives with subrepos

            for k, _ in org_repo.ui.configitems('hooks'):

                org_repo.ui.setconfig('hooks', k, None)

            unbundle = org_repo.getbundle('incoming', common=None,

                                          heads=None)

            buf = BytesIO()

            while True:

                chunk = unbundle._stream.read(1024 * 4)

                if not chunk:

                    break

                buf.write(chunk)

            buf.seek(0)

            # replace chunked _stream with data that can do tell() and seek()

            unbundle._stream = buf

            ui = make_ui('db')

            bundlerepo = InMemoryBundleRepo(ui, path=org_repo.root,

                                            bundlestream=unbundle)

            return ''.join(patch.diff(bundlerepo,

                                      node1=other_repo[other_ref].node(),

                                      node2=org_repo[org_ref].node(),

                                      opts=opts))

import datetime

import functools

import decimal

import imp

__all__ = ['json', 'simplejson', 'stdlibjson']

def _is_aware(value):

    """

    Determines if a given datetime.time is aware.

    The logic is described in Python's docs:

    http://docs.python.org/library/datetime.html#datetime.tzinfo

    """

    return (value.tzinfo is not None

            and value.tzinfo.utcoffset(value) is not None)

def _obj_dump(obj):

    """

    Custom function for dumping objects to JSON, if obj has __json__ attribute

    or method defined it will be used for serialization

    :param obj:

    """

    if isinstance(obj, complex):

        return [obj.real, obj.imag]

    # See "Date Time String Format" in the ECMA-262 specification.

    # some code borrowed from django 1.4

    elif isinstance(obj, datetime.datetime):

        r = obj.isoformat()

        if obj.microsecond:

            r = r[:23] + r[26:]

        if r.endswith('+00:00'):

            r = r[:-6] + 'Z'

        return r

    elif isinstance(obj, datetime.date):

        return obj.isoformat()

    elif isinstance(obj, decimal.Decimal):

        return str(obj)

    elif isinstance(obj, datetime.time):

        if _is_aware(obj):

            raise ValueError("JSON can't represent timezone-aware times.")

        r = obj.isoformat()

        if obj.microsecond:

            r = r[:12]

        return r

    elif isinstance(obj, set):

        return list(obj)

    elif hasattr(obj, '__json__'):

        if callable(obj.__json__):

            return obj.__json__()

        else:

            return obj.__json__

    else:

        raise NotImplementedError

# Import simplejson

try:

    # import simplejson initially

    _sj = imp.load_module('_sj', *imp.find_module('simplejson'))

    def extended_encode(obj):

        try:

            return _obj_dump(obj)

        except NotImplementedError:

            pass

        raise TypeError("%r is not JSON serializable" % (obj,))

    # we handle decimals our own it makes unified behavior of json vs

    # simplejson

    sj_version = [int(x) for x in _sj.__version__.split('.')]

    major, minor = sj_version[0], sj_version[1]

    if major < 2 or (major == 2 and minor < 1):

        # simplejson < 2.1 doesnt support use_decimal

        _sj.dumps = functools.partial(_sj.dumps,

                                             default=extended_encode)

        _sj.dump = functools.partial(_sj.dump,

                                            default=extended_encode)

    else:

        _sj.dumps = functools.partial(_sj.dumps,

                                             default=extended_encode,

                                             use_decimal=False)

        _sj.dump = functools.partial(_sj.dump,

                                            default=extended_encode,

                                            use_decimal=False)

    simplejson = _sj

except ImportError:

    # no simplejson set it to None

    simplejson = None

try:

    # simplejson not found try out regular json module

    _json = imp.load_module('_json', *imp.find_module('json'))

    # extended JSON encoder for json

    class ExtendedEncoder(_json.JSONEncoder):

        def default(self, obj):

            try:

                return _obj_dump(obj)

            except NotImplementedError:

                pass

            raise TypeError("%r is not JSON serializable" % (obj,))

    # monkey-patch JSON encoder to use extended version

    _json.dumps = functools.partial(_json.dumps, cls=ExtendedEncoder)

    _json.dump = functools.partial(_json.dump, cls=ExtendedEncoder)

    stdlibjson = _json

except ImportError:

    stdlibjson = None

# set all available json modules

if simplejson:

    json = _sj

elif stdlibjson:

    json = _json

else:

<dl>

    <dt class="file_history">${_('History')}</dt>

    <dd>

        <div>

            <div style="float:left">

            ${h.form(h.url('files_diff_home',repo_name=c.repo_name,f_path=c.f_path),method='get')}

            ${h.hidden('diff2',c.file_changeset.raw_id)}

            ${h.select('diff1',c.file_changeset.raw_id,c.file_history)}

            ${h.submit('diff',_('diff to revision'),class_="ui-btn")}

            ${h.submit('show_rev',_('show at revision'),class_="ui-btn")}

            ${h.hidden('annotate', c.annotate)}

            ${h.end_form()}

            </div>

            <div class="file_author">

                <div class="item">${h.literal(ungettext(u'%s author',u'%s authors',len(c.authors)) % ('<b>%s</b>' % len(c.authors))) }</div>

                %for email, user in c.authors:

                  <div class="contributor tooltip" style="float:left" title="${h.tooltip(user)}">

                    <div class="gravatar" style="margin:1px"><img alt="gravatar" src="${h.gravatar_url(email, 20)}"/> </div>

                  </div>

                %endfor

            </div>

        </div>

        <div style="clear:both"></div>

    </dd>