# List of modules to import when celery starts.

import sys

import os

import ConfigParser

root = os.getcwd()

sys.path.append(root)

config = ConfigParser.ConfigParser({'here':root})

config.read('%s/%s' % (root, PYLONS_CONFIG_NAME))

PYLONS_CONFIG = config

CELERY_IMPORTS = ("pylons_app.lib.celerylib.tasks",)

## Result store settings.

CELERY_RESULT_BACKEND = "database"

CELERY_RESULT_DBURI = dict(config.items('app:main'))['sqlalchemy.db1.url']

CELERY_RESULT_SERIALIZER = 'json'

BROKER_CONNECTION_MAX_RETRIES = 30

## Broker settings.

BROKER_HOST = "localhost"

BROKER_PORT = 5672

BROKER_VHOST = "rabbitmqhost"

BROKER_USER = "rabbitmq"

BROKER_PASSWORD = "qweqwe"

## Worker settings

    def index(self, revision):

        hg_model = HgModel()

        try:

            c.changeset = hg_model.get_repo(c.repo_name).get_changeset(revision)

        except RepositoryError:

            log.error(traceback.format_exc())

            return redirect(url('hg_home'))

        else:

            try:

                c.changeset_old = c.changeset.parents[0]

            except IndexError:

                c.changeset_old = None

            c.changes = []

            for node in c.changeset.added:

                filenode_old = FileNode(node.path, '')

                if filenode_old.is_binary or node.is_binary:

                    diff = 'binary file'

                else:    

                    f_udiff = differ.get_udiff(filenode_old, node)

                    diff = differ.DiffProcessor(f_udiff).as_html()

                cs1 = None

                c.changes.append(('added', node, diff, cs1, cs2))

            for node in c.changeset.changed:

                filenode_old = c.changeset_old.get_node(node.path)

                if filenode_old.is_binary or node.is_binary:

                    diff = 'binary file'

                else:    

                    f_udiff = differ.get_udiff(filenode_old, node)

                    diff = differ.DiffProcessor(f_udiff).as_html()

                c.changes.append(('changed', node, diff, cs1, cs2))

            for node in c.changeset.removed:

                c.changes.append(('removed', node, None, None, None))            

        return render('changeset/changeset.html')

    def raw_changeset(self,revision):

        hg_model = HgModel()

        method = request.GET.get('diff','show')

        try:

            c.changeset = hg_model.get_repo(c.repo_name).get_changeset(revision)

        except RepositoryError:

            log.error(traceback.format_exc())

            return redirect(url('hg_home'))

        else:

            try:

                c.changeset_old = c.changeset.parents[0]

            except IndexError:

                c.changeset_old = None

            c.changes = []

            for node in c.changeset.added:

                filenode_old = FileNode(node.path, '')

                if filenode_old.is_binary or node.is_binary:

                    diff = 'binary file'

                else:    

                    f_udiff = differ.get_udiff(filenode_old, node)

                    diff = differ.DiffProcessor(f_udiff).raw_diff()

                cs1 = None

                c.changes.append(('added', node, diff, cs1, cs2))

            for node in c.changeset.changed:

                filenode_old = c.changeset_old.get_node(node.path)

                if filenode_old.is_binary or node.is_binary:

                    diff = 'binary file'

                else:    

                    f_udiff = differ.get_udiff(filenode_old, node)

                    diff = differ.DiffProcessor(f_udiff).raw_diff()

                c.changes.append(('changed', node, diff, cs1, cs2))      

        response.content_type = 'text/plain'

        if method == 'download':

            response.content_disposition = 'attachment; filename=%s.patch' % revision 

        parent = True if len(c.changeset.parents) > 0 else False

        c.parent_tmpl = 'Parent  %s' % c.changeset.parents[0]._hex if parent else ''

        c.diffs = ''

        for x in c.changes:

            c.diffs += x[2]

        return render('changeset/raw_changeset.html')

    #secure it or not ?

    def __before__(self):

        super(FeedController, self).__before__()

        #common values for feeds

        self.description = 'Changes on %s repository'

        self.title = "%s feed"

        self.language = 'en-us'

        self.ttl = "5"

        self.feed_nr = 10

    def atom(self, repo_name):

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

        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)

        changesets = HgModel().get_repo(repo_name)

        for cs in changesets[:self.feed_nr]:

            feed.add_item(title=cs.message,

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

                                   description=str(cs.date))

        response.content_type = feed.mime_type

        return feed.writeString('utf-8')

    def rss(self, repo_name):

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

        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)

        changesets = HgModel().get_repo(repo_name)

        for cs in changesets[:self.feed_nr]:

            feed.add_item(title=cs.message,

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

                          description=str(cs.date))

        response.content_type = feed.mime_type

        return feed.writeString('utf-8')

        super(FilesController, self).__before__()

        c.file_size_limit = 250 * 1024 #limit of file size to display

    def index(self, repo_name, revision, f_path):

        hg_model = HgModel()

        c.repo = repo = hg_model.get_repo(c.repo_name)

        revision = request.POST.get('at_rev', None) or revision

        def get_next_rev(cur):

            max_rev = len(c.repo.revisions) - 1

            r = cur + 1

            if r > max_rev:

                r = max_rev

            return r

        def get_prev_rev(cur):

            r = cur - 1

            return r

        c.f_path = f_path

        try:

            cur_rev = repo.get_changeset(revision).revision

            c.url_prev = url('files_home', repo_name=c.repo_name,

                             revision=prev_rev, f_path=f_path) 

            c.url_next = url('files_home', repo_name=c.repo_name,

                             revision=next_rev, f_path=f_path)   

            c.changeset = repo.get_changeset(revision)

            c.rev_nr = c.changeset.revision

            c.files_list = c.changeset.get_node(f_path)

            c.file_history = self._get_history(repo, c.files_list, f_path)

        except (RepositoryError, ChangesetError):

            c.files_list = None

        return render('files/files.html')

    def rawfile(self, repo_name, revision, f_path):

        hg_model = HgModel()

        c.repo = hg_model.get_repo(c.repo_name)

        file_node = c.repo.get_changeset(revision).get_node(f_path)

        response.content_type = file_node.mimetype

        response.content_disposition = 'attachment; filename=%s' \

                                                    % f_path.split('/')[-1] 

        return file_node.content

    def raw(self, repo_name, revision, f_path):

        hg_model = HgModel()

        c.repo = hg_model.get_repo(c.repo_name)

        file_node = c.repo.get_changeset(revision).get_node(f_path)

        response.content_type = 'text/plain'

        return file_node.content

    def annotate(self, repo_name, revision, f_path):

        hg_model = HgModel()

        c.repo = hg_model.get_repo(c.repo_name)

        cs = c.repo.get_changeset(revision)

        c.file = cs.get_node(f_path)

        c.file_msg = cs.get_file_message(f_path)

        c.rev_nr = cs.revision        

        c.f_path = f_path

        return render('files/files_annotate.html')

    def archivefile(self, repo_name, revision, fileformat):

        archive_specs = {

          '.tar.bz2': ('application/x-tar', 'tbz2'),

          '.tar.gz': ('application/x-tar', 'tgz'),

          '.zip': ('application/zip', 'zip'),

        }

        if not archive_specs.has_key(fileformat):

            return 'Unknown archive type %s' % fileformat

        def read_in_chunks(file_object, chunk_size=1024 * 40):

            """Lazy function (generator) to read a file piece by piece.

            Default chunk size: 40k."""

            while True:

                data = file_object.read(chunk_size)

                if not data:

                    break

                yield data        

        archive = tempfile.TemporaryFile()

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

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

        c.action = request.GET.get('diff')

        c.no_changes = diff1 == diff2

        c.f_path = f_path

        c.repo = hg_model.get_repo(c.repo_name)

        try:

            if diff1 not in ['', None, 'None', '0' * 12]:

                c.changeset_1 = c.repo.get_changeset(diff1)

                node1 = c.changeset_1.get_node(f_path)

            else:

                c.changeset_1 = EmptyChangeset()

                node1 = FileNode('.', '')

            if diff2 not in ['', None, 'None', '0' * 12]:

                c.changeset_2 = c.repo.get_changeset(diff2)

                node2 = c.changeset_2.get_node(f_path)

            else:

                c.changeset_2 = EmptyChangeset()

                node2 = FileNode('.', '') 

        except RepositoryError:

            return redirect(url('files_home',

                                repo_name=c.repo_name, f_path=f_path))

        f_udiff = differ.get_udiff(node1, node2)

        diff = differ.DiffProcessor(f_udiff)

        if c.action == 'download':

            diff_name = '%s_vs_%s.diff' % (diff1, diff2)

            response.content_type = 'text/plain'

            response.content_disposition = 'attachment; filename=%s' \

                                                    % diff_name             

            return diff.raw_diff()

        elif c.action == 'raw':

            c.cur_diff = '<pre class="raw">%s</pre>' % h.escape(diff.raw_diff())

        elif c.action == 'diff':

            c.cur_diff = diff.as_html()

        else:

            #default option

            c.cur_diff = diff.as_html()

        if not c.cur_diff: c.no_changes = True    

        return render('files/file_diff.html')

    def _get_history(self, repo, node, f_path):

        from vcs.nodes import NodeKind

        if not node.kind is NodeKind.FILE:

            return []

        changesets = node.history

        hist_l = []

        for chs in changesets:

        return hist_l

        for c in xrange(n):

            h += golden_ratio

            h %= 1

            HSV_tuple = [h, 0.95, 0.95]

            RGB_tuple = colorsys.hsv_to_rgb(*HSV_tuple)

            yield map(lambda x:str(int(x * 256)), RGB_tuple)           

    cgenerator = gen_color()

    def get_color_string(cs):

        if color_dict.has_key(cs):

            col = color_dict[cs]

        else:

            col = color_dict[cs] = cgenerator.next()

        return "color: rgb(%s)! important;" % (', '.join(col))

    def url_func(changeset):

        tooltip_html = "<div style='font-size:0.8em'><b>Author:</b>" + \

        " %s<br/><b>Date:</b> %s</b><br/><b>Message:</b> %s<br/></div>" 

        tooltip_html = tooltip_html % (changeset.author,

                                               changeset.date,

                                               tooltip(changeset.message))

        lnk_format = 'r%-5s:%s' % (changeset.revision,

        uri = link_to(

                lnk_format,

                url('changeset_home', repo_name=changeset.repository.name,

                class_='tooltip',

                tooltip_title=tooltip_html

              )

        uri += '\n'

        return uri   

    return literal(annotate_highlight(filenode, url_func, **kwargs))

def repo_name_slug(value):

    """Return slug of name of repository

    This function is called on each creation/modification

    of repository to prevent bad names in repo

    """

    slug = remove_formatting(value)

    slug = strip_tags(slug)

    for c in """=[]\;'"<>,/~!@#$%^&*()+{}|: """:

        slug = slug.replace(c, '-')

    slug = recursive_replace(slug, '-')

    slug = collapse(slug, '-')

    return slug

def get_changeset_safe(repo, rev):

    from vcs.backends.base import BaseRepository

                        else:

                            repos_list[name].contact = sa.query(User)\

                            .filter(User.admin == True).first().full_contact

            except OSError:

                continue

        meta.Session.remove()

        return repos_list

    def get_repos(self):

        for name, repo in _get_repos_cached().items():

            if repo._get_hidden():

                #skip hidden web repository

                continue

            last_change = repo.last_change

            tip = h.get_changeset_safe(repo, 'tip')

            tmp_d = {}

            tmp_d['name'] = repo.name

            tmp_d['name_sort'] = tmp_d['name'].lower()

            tmp_d['description'] = repo.description

            tmp_d['description_sort'] = tmp_d['description']

            tmp_d['last_change'] = last_change

            tmp_d['last_change_sort'] = last_change[1] - last_change[0]

            tmp_d['tip_sort'] = tip.revision 

            tmp_d['rev'] = tip.revision

            tmp_d['contact'] = repo.contact

            tmp_d['contact_sort'] = tmp_d['contact']

            tmp_d['repo_archives'] = list(repo._get_archives())

            tmp_d['last_msg'] = tip.message

            tmp_d['repo'] = repo

            yield tmp_d

    def get_repo(self, repo_name):

        return _get_repos_cached()[repo_name]

% if c.repo_branches:

    <table class="table_disp">

        <tr>

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

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

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

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

        </tr>

		%for cnt,branch in enumerate(c.repo_branches.items()):

		<tr class="parity${cnt%2}">

			<td>${h.age(branch[1]._ctx.date())}</td>

			<td>

				<span class="logtags">

					<span class="branchtag">${h.link_to(branch[0],

				</span>			

			</td>

			<td class="nowrap">

			|

			</td>

		</tr>	

		%endfor

    </table>

%else:

	${_('There are no branches yet')}

%endif

        ${self.breadcrumbs()}

    </div>

    <div class="table">

		% if c.pagination:

			<div id="graph">

				<div id="graph_nodes">

					<canvas id="graph_canvas"></canvas>

				</div>

				<div id="graph_content">

					<div class="container_header">

        ${h.form(h.url.current(),method='get')}

        <div class="info_box">

          <span>${_('Show')}:</span>

          ${h.text('size',size=1,value=c.size)}

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

          ${h.submit('set',_('set'))}

        </div>

        ${h.end_form()}

					</div>

				%for cnt,cs in enumerate(c.pagination):

					<div id="chg_${cnt+1}" class="container">

						<div class="left">

								<span class="logtags">

									<span class="branchtag">${cs.branch}</span>

									%for tag in cs.tags:

										<span class="tagtag">${tag}</span>

									%endfor

								</span>					

							<div class="author">

								<div class="gravatar">

									<img alt="gravatar" src="${h.gravatar_url(h.email(cs.author),20)}"/>

								</div>

								<span>${h.person(cs.author)}</span><br/>

								<span><a href="mailto:${h.email_or_none(cs.author)}">${h.email_or_none(cs.author)}</a></span><br/>

							</div>

							<div class="message">

								${h.link_to(h.wrap_paragraphs(cs.message),

							</div>

						</div>	

						<div class="right">

									<div class="changes">

										<span class="removed" title="${_('removed')}">${len(cs.removed)}</span>

										<span class="changed" title="${_('changed')}">${len(cs.changed)}</span>

										<span class="added" title="${_('added')}">${len(cs.added)}</span>

									</div>					

										%if len(cs.parents)>1:

										<div class="merge">

											${_('merge')}<img alt="merge" src="/images/icons/arrow_join.png"/>

										</div>

										%endif						

									%for p_cs in reversed(cs.parents):

										</div>

									%endfor								

						</div>				

					</div>

				%endfor

				<div class="pagination-wh pagination-left">

					${c.pagination.pager('$link_previous ~2~ $link_next')}

				</div>			

				</div>

			</div>

			<script type="text/javascript" src="/js/graph.js"></script>

			<script type="text/javascript">

				YAHOO.util.Event.onDOMReady(function(){

					function set_canvas() {

						var c = document.getElementById('graph_nodes');

						var t = document.getElementById('graph_content');

						canvas = document.getElementById('graph_canvas');

						var div_h = t.clientHeight;

						c.style.height=div_h+'px';

						canvas.setAttribute('height',div_h);

						canvas.setAttribute('width',160);

					};

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

<%def name="title()">

    ${_('Changeset')}

</%def>

<%def name="breadcrumbs_links()">

    ${h.link_to(u'Home',h.url('/'))}

    »

    ${h.link_to(c.repo_name,h.url('summary_home',repo_name=c.repo_name))}

    »

</%def>

<%def name="page_nav()">

    ${self.menu('changelog')}     

</%def>

<%def name="main()">

<div class="box">

    <!-- box / title -->

    <div class="title">

        ${self.breadcrumbs()}

    </div>

    <div class="table">

		<div id="body" class="diffblock">

			<div class="code-header">

				<div>

				 &raquo; <span>${h.link_to(_('raw diff'),

				 &raquo; <span>${h.link_to(_('download diff'),

				</div>

			</div>

		</div>

    <div id="changeset_content">

		<div class="container">

			<div class="left">

				<div class="date">${_('Date')}: ${c.changeset.date}</div>

				<div class="author">${_('Author')}: ${c.changeset.author}</div>

				<div class="message">${h.wrap_paragraphs(c.changeset.message)}</div>

			</div>	

			<div class="right">

				<span class="logtags">

					<span class="branchtag">${c.changeset.branch}</span>

					%for tag in c.changeset.tags:

						<span class="tagtag">${tag}</span>

					%endfor

				</span>					

				%if len(c.changeset.parents)>1:

				<div class="merge">		

				${_('merge')}

				<img alt="merge" src="/images/icons/arrow_join.png">

				</div>

				%endif						

				%for p_cs in reversed(c.changeset.parents):

					</div>

				%endfor								

			</div>		

		</div>

		<span style="font-size:1.1em;font-weight: bold">${_('Files affected')}</span>

	    <div class="cs_files">

				%for change,filenode,diff,cs1,cs2 in c.changes:

					<div class="cs_${change}">${h.link_to(filenode.path,h.url.current(anchor='CHANGE-%s'%filenode.path))}</div>

				%endfor

		</div>		    

    </div>

	%for change,filenode,diff,cs1,cs2 in c.changes:

		%if change !='removed':

		<div style="clear:both;height:10px"></div>

		<div id="body" class="diffblock">

			<div id="${'CHANGE-%s'%filenode.path}" class="code-header">

				<div>

					<span>

						${h.link_to_if(change!='removed',filenode.path,h.url('files_home',repo_name=c.repo_name,

					</span>

					%if 1:

					&raquo; <span>${h.link_to(_('diff'),

					h.url('files_diff_home',repo_name=c.repo_name,f_path=filenode.path,diff2=cs2,diff1=cs1,diff='diff'))}</span>

					&raquo; <span>${h.link_to(_('raw diff'),

					h.url('files_diff_home',repo_name=c.repo_name,f_path=filenode.path,diff2=cs2,diff1=cs1,diff='raw'))}</span>

					&raquo; <span>${h.link_to(_('download diff'),

					h.url('files_diff_home',repo_name=c.repo_name,f_path=filenode.path,diff2=cs2,diff1=cs1,diff='download'))}</span>

					%endif

				</div>

			</div>

			<div class="code-body">        

					%if diff:

						${diff|n}

					%else:

						${_('No changes in this file')}

					%endif

			</div>

		</div>

		%endif

	%endfor 

    </div>

</div>    

</%def>

<%def name="breadcrumbs_links()">

    ${h.link_to(u'Home',h.url('/'))}

    »

    ${h.link_to(c.repo_name,h.url('summary_home',repo_name=c.repo_name))}

    »

    ${_('annotate')} @ R${c.rev_nr}:${c.cur_rev}

</%def>

<%def name="page_nav()">

		${self.menu('files')}     

</%def>

<%def name="main()">

<div class="box">

    <!-- box / title -->

    <div class="title">

        ${self.breadcrumbs()}

    </div>

    <div class="table">

		<div id="files_data">

			<h3 class="files_location">${_('Location')}: ${h.files_breadcrumbs(c.repo_name,c.cur_rev,c.file.path)}</h3>

			<dl class="overview">

				<dt>${_('Last revision')}</dt>

				<dt>${_('Size')}</dt>

				<dd>${h.format_byte_size(c.file.size,binary=True)}</dd>

    			<dt>${_('Mimetype')}</dt>

				<dd>${c.file.mimetype}</dd>				

				<dt>${_('Options')}</dt>

				<dd>${h.link_to(_('show source'),

						h.url('files_home',repo_name=c.repo_name,revision=c.cur_rev,f_path=c.f_path))}  

					/ ${h.link_to(_('show as raw'),

						h.url('files_raw_home',repo_name=c.repo_name,revision=c.cur_rev,f_path=c.f_path))}