#
# 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 2 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, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#

import cgi
import os

import pkg_resources

import bzrlib

import loggerhead
from loggerhead.zptsupport import zpt


templatefunctions = {}


def templatefunc(func):
    templatefunctions[func.__name__] = func
    return func


_base = os.path.dirname(__file__)


def _pt(name):
    return zpt(os.path.join(_base, 'templates', name + '.pt'))


templatefunctions['macros'] = _pt('macros').macros
templatefunctions['breadcrumbs'] = _pt('breadcrumbs').macros


@templatefunc
def file_change_summary(url, entry, file_changes, style='normal', currently_showing=None):
    if style == 'fragment':
        def file_link(filename):
            if currently_showing and filename == currently_showing:
                return '<b><a href="#%s">%s</a></b>' % (
                    cgi.escape(filename), cgi.escape(filename))
            else:
                return revision_link(
                    url, entry.revno, filename, '#' + filename)
    else:
        def file_link(filename):
            return '<a href="%s%s" title="View changes to %s in revision %s">%s</a>' % (
                url(['/revision', entry.revno]), '#' + filename, cgi.escape(filename),
                cgi.escape(entry.revno), cgi.escape(filename))
    return _pt('revisionfilechanges').expand(
        entry=entry, file_changes=file_changes, file_link=file_link, **templatefunctions)


@templatefunc
def revisioninfo(url, branch, entry, file_changes=None, currently_showing=None, merged_in=None):
    from loggerhead import util
    return _pt('revisioninfo').expand(
        url=url, change=entry, branch=branch, util=util,
        file_changes=file_changes, currently_showing=currently_showing,
        merged_in=merged_in, **templatefunctions)


@templatefunc
def branchinfo(branch):
    if branch.served_url is not None:
        return _pt('branchinfo').expand(branch=branch, **templatefunctions)
    else:
        return ''


@templatefunc
def collapse_button(group, name, branch, normal='block'):
    return _pt('collapse-button').expand(
        group=group, name=name, normal=normal, branch=branch,
        **templatefunctions)


@templatefunc
def collapse_all_button(group, branch, normal='block'):
    return _pt('collapse-all-button').expand(
        group=group, normal=normal, branch=branch,
        **templatefunctions)


@templatefunc
def revno_with_nick(entry):
    if entry.branch_nick:
        extra = ' ' + entry.branch_nick
    else:
        extra = ''
    return '(%s%s)' % (entry.revno, extra)


@templatefunc
def search_box(branch, navigation):
    return _pt('search-box').expand(branch=branch, navigation=navigation,
        **templatefunctions)


@templatefunc
def feed_link(branch, url):
    return _pt('feed-link').expand(branch=branch, url=url, **templatefunctions)


@templatefunc
def menu(branch, url, fileview_active=False):
    return _pt('menu').expand(branch=branch, url=url,
        fileview_active=fileview_active, **templatefunctions)


@templatefunc
def view_link(url, revno, path):
    return '<a href="%s" title="Annotate %s">%s</a>' % (
        url(['/view', revno, path]), cgi.escape(path), cgi.escape(path))

@templatefunc
def revision_link(url, revno, path, frag=''):
    return '<a href="%s%s" title="View changes to %s in revision %s">%s</a>' % (
        url(['/revision', revno, path]), frag, cgi.escape(path),
        cgi.escape(revno), cgi.escape(path))


@templatefunc
def loggerhead_version():
    return loggerhead.__version__

_cached_generator_string = None

@templatefunc
def generator_string():
    global _cached_generator_string
    if _cached_generator_string is None:
        versions = []

        # TODO: Errors -- e.g. from a missing/invalid __version__ attribute, or
        # ValueError accessing Distribution.version -- should be non-fatal.

        versions.append(('Loggerhead', loggerhead.__version__))

        import sys
        python_version = bzrlib._format_version_tuple(sys.version_info)
        versions.append(('Python', python_version))

        versions.append(('Bazaar', bzrlib.__version__))

        Paste = pkg_resources.get_distribution('Paste')
        versions.append(('Paste', Paste.version))

        try:
            PasteDeploy = pkg_resources.get_distribution('PasteDeploy')
        except pkg_resources.DistributionNotFound:
            pass
        else:
            versions.append(('PasteDeploy', PasteDeploy.version))

        import simpletal
        versions.append(('SimpleTAL', simpletal.__version__))

        try:
            import pygments
        except ImportError:
            pass
        else:
            versions.append(('Pygments', pygments.__version__))

        try:
            from bzrlib.plugins import search
        except ImportError:
            pass
        else:
            bzr_search_version = bzrlib._format_version_tuple(
                search.version_info)
            versions.append(('bzr-search', bzr_search_version))

        # TODO: On old Python versions, elementtree may be used.

        try:
            import simplejson
        except ImportError:
            pass
        else:
            versions.append(('simplejson', simplejson.__version__))

        try:
            Dozer = pkg_resources.get_distribution('Dozer')
        except pkg_resources.DistributionNotFound:
            pass
        else:
            versions.append(('Dozer', Dozer.version))

        version_strings = ("%s/%s" % t for t in versions)
        _cached_generator_string = ' '.join(version_strings)
    return _cached_generator_string