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

"""

    rhodecode.controllers.summary

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

    Summary controller for Rhodecode

    :created_on: Apr 18, 2010

    :author: marcink

    :copyright: (C) 2009-2011 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; version 2

# of the License or (at your opinion) any later version of the license.

# 

# 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., 51 Franklin Street, Fifth Floor, Boston,

# MA  02110-1301, USA.

import calendar

import logging

from time import mktime

from datetime import datetime, timedelta, date

from vcs.exceptions import ChangesetError

from pylons import tmpl_context as c, request, url

from pylons.i18n.translation import _

from rhodecode.model.db import Statistics, Repository

from rhodecode.model.repo import RepoModel

from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator

from rhodecode.lib.base import BaseRepoController, render

from rhodecode.lib.utils import OrderedDict, EmptyChangeset

from rhodecode.lib.celerylib import run_task

from rhodecode.lib.celerylib.tasks import get_commits_stats

from rhodecode.lib.helpers import RepoPage

try:

    import json

except ImportError:

    #python 2.5 compatibility

    import simplejson as json

log = logging.getLogger(__name__)

class SummaryController(BaseRepoController):

    @LoginRequired()

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

                                   'repository.admin')

    def __before__(self):

        super(SummaryController, self).__before__()

    def index(self, repo_name):

        e = request.environ

        c.dbrepo = dbrepo = Repository.by_repo_name(repo_name)

        c.following = self.scm_model.is_following_repo(repo_name,

                                                       self.rhodecode_user.user_id)

        def url_generator(**kw):

            return url('shortlog_home', repo_name=repo_name, **kw)

        c.repo_changesets = RepoPage(c.rhodecode_repo, page=1, items_per_page=10,

                                 url=url_generator)

        if self.rhodecode_user.username == 'default':

            #for default(anonymous) user we don't need to pass credentials

            username = ''

            password = ''

        else:

            username = str(self.rhodecode_user.username)

            password = '@'

        uri = u'%(protocol)s://%(user)s%(password)s%(host)s%(prefix)s/%(repo_name)s' % {

                                        'protocol': e.get('wsgi.url_scheme'),

                                        'user':username,

                                        'password':password,

                                        'host':e.get('HTTP_HOST'),

                                        'prefix':e.get('SCRIPT_NAME'),

                                        'repo_name':repo_name, }

        c.clone_repo_url = uri

        c.repo_tags = OrderedDict()

        for name, hash in c.rhodecode_repo.tags.items()[:10]:

            try:

                c.repo_tags[name] = c.rhodecode_repo.get_changeset(hash)

            except ChangesetError:

                c.repo_tags[name] = EmptyChangeset(hash)

        c.repo_branches = OrderedDict()

        for name, hash in c.rhodecode_repo.branches.items()[:10]:

            try:

                c.repo_branches[name] = c.rhodecode_repo.get_changeset(hash)

            except ChangesetError:

                c.repo_branches[name] = EmptyChangeset(hash)

        td = date.today() + timedelta(days=1)

        td_1m = td - timedelta(days=calendar.mdays[td.month])

        td_1y = td - timedelta(days=365)

        ts_min_m = mktime(td_1m.timetuple())

        ts_min_y = mktime(td_1y.timetuple())

        ts_max_y = mktime(td.timetuple())

        if dbrepo.enable_statistics:

            c.no_data_msg = _('No data loaded yet')

            run_task(get_commits_stats, c.dbrepo.repo_name, ts_min_y, ts_max_y)

        else:

            c.no_data_msg = _('Statistics are disabled for this repository')

        c.ts_min = ts_min_m

        c.ts_max = ts_max_y

        stats = self.sa.query(Statistics)\

            .filter(Statistics.repository == dbrepo)\

            .scalar()

        if stats and stats.languages:

            c.no_data = False is dbrepo.enable_statistics

            lang_stats = json.loads(stats.languages)

            c.commit_data = stats.commit_activity

            c.overview_data = stats.commit_activity_combined

            c.trending_languages = json.dumps(OrderedDict(

                                       sorted(lang_stats.items(), reverse=True,

                                            key=lambda k: k[1])[:10]

                                        )

                                    )

        else:

            c.commit_data = json.dumps({})

            c.overview_data = json.dumps([[ts_min_y, 0], [ts_max_y, 10] ])

            c.trending_languages = json.dumps({})

            c.no_data = True

        c.enable_downloads = dbrepo.enable_downloads

        if c.enable_downloads:

            c.download_options = self._get_download_links(c.rhodecode_repo)

        return render('summary/summary.html')

    def _get_download_links(self, repo):

        download_l = []

        branches_group = ([], _("Branches"))

        tags_group = ([], _("Tags"))

        for name, chs in c.rhodecode_repo.branches.items():

            #chs = chs.split(':')[-1]

            branches_group[0].append((chs, name),)

        download_l.append(branches_group)

        for name, chs in c.rhodecode_repo.tags.items():

            #chs = chs.split(':')[-1]

            tags_group[0].append((chs, name),)

        download_l.append(tags_group)

        return download_l

			 </div>

			 <div class="field">

			  <div class="label">

			      <label>${_('Last change')}:</label>

			  </div>

			  <div class="input-short">

                  <b>${'r%s:%s' % (h.get_changeset_safe(c.rhodecode_repo,'tip').revision,

                            h.get_changeset_safe(c.rhodecode_repo,'tip').short_id)}</b> - 

			      <span class="tooltip" title="${c.rhodecode_repo.last_change}">

			      ${h.age(c.rhodecode_repo.last_change)}</span><br/>

			      ${_('by')} ${h.get_changeset_safe(c.rhodecode_repo,'tip').author} 

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

			      <label>${_('Clone url')}:</label>

			  </div>

			  <div class="input-short">

			      <input type="text" id="clone_url" readonly="readonly" value="hg clone ${c.clone_repo_url}" size="70"/>

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

			      <label>${_('Trending source files')}:</label>

			  </div>

			  <div class="input-short">

			    <div id="lang_stats"></div> 			   

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

			      <label>${_('Download')}:</label>

			  </div>

			  <div class="input-short">

		        %if len(c.rhodecode_repo.revisions) == 0:

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

		        %elif c.enable_downloads is False:

		          ${_('Downloads are disabled for this repository')}

                    %if h.HasPermissionAll('hg.admin')('enable stats on from summary'):

                        [${h.link_to(_('enable'),h.url('edit_repo',repo_name=c.repo_name))}]

                    %endif  		          

		        %else:

			        ${h.select('download_options',c.rhodecode_repo.get_changeset().raw_id,c.download_options)}

			        %for cnt,archive in enumerate(c.rhodecode_repo._get_archives()):

			             %if cnt >=1:

			             |

			             %endif

			             <span class="tooltip" title="${_('Download %s as %s') %('tip',archive['type'])}" 

			                  id="${archive['type']+'_link'}">${h.link_to(archive['type'],

			                h.url('files_archive_home',repo_name=c.dbrepo.repo_name,

			                fname='tip'+archive['extension']),class_="archive_icon")}</span>

			        %endfor

			    %endif

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

			      <label>${_('Feeds')}:</label>

			  </div>

			  <div class="input-short">

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

	            ${h.link_to(_('RSS'),h.url('rss_feed_home',repo_name=c.dbrepo.repo_name,api_key=c.rhodecode_user.api_key),class_='rss_icon')}

	            ${h.link_to(_('Atom'),h.url('atom_feed_home',repo_name=c.dbrepo.repo_name,api_key=c.rhodecode_user.api_key),class_='atom_icon')}

	           %else:

                ${h.link_to(_('RSS'),h.url('rss_feed_home',repo_name=c.dbrepo.repo_name),class_='rss_icon')}

                ${h.link_to(_('Atom'),h.url('atom_feed_home',repo_name=c.dbrepo.repo_name),class_='atom_icon')}	           

	           %endif 

			  </div>

			 </div>				 			 			 

	  </div>		 

	</div>

  	<script type="text/javascript">

	  	YUE.onDOMReady(function(e){

	  	    id = 'clone_url';

	  	    YUE.on(id,'click',function(e){

	  	        YUD.get('clone_url').select();

	  	    })

	  	})

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

  		var total = 0;

  		var no_data = true;

  		for (k in data){

  		    total += data[k];

  		    no_data = false;

  		} 

		var tbl = document.createElement('table');

		tbl.setAttribute('class','trending_language_tbl');

		var cnt =0;

  		for (k in data){

  			cnt+=1;

  			var hide = cnt>2;

	  		var tr = document.createElement('tr');

	  		if (hide){

	  			tr.setAttribute('style','display:none');

	  			tr.setAttribute('class','stats_hidden');

	  		}

	  		var percentage = Math.round((data[k]/total*100),2);

			var value = data[k];

	  		var td1 = document.createElement('td');

	  		td1.width=150;

	  		var trending_language_label = document.createElement('div');

	  		trending_language_label.innerHTML = k;

	  		td1.appendChild(trending_language_label);

	  		var td2 = document.createElement('td');

	  		td2.setAttribute('style','padding-right:14px !important');

  		    var trending_language = document.createElement('div');

  		    var nr_files = value+" ${_('files')}";

  		    trending_language.title = k+" "+nr_files;

  		    if (percentage>20){

  		    	trending_language.innerHTML = "<b style='font-size:0.8em'>"+percentage+"% "+nr_files+ "</b>";	

  		    }

  		    else{

  		    	trending_language.innerHTML = "<b style='font-size:0.8em'>"+percentage+"%</b>";

  		    }

  		    trending_language.setAttribute("class", 'trending_language top-right-rounded-corner bottom-right-rounded-corner');

  		    trending_language.style.width=percentage+"%";

			td2.appendChild(trending_language);

			tr.appendChild(td1);

			tr.appendChild(td2);

  		    tbl.appendChild(tr);

  		    if(cnt == 2){

  		    	var show_more = document.createElement('tr');

  		    	var td=document.createElement('td');

  		    	lnk = document.createElement('a');

  		    	lnk.href='#';

  		    	lnk.innerHTML = "${_("show more")}";

  		    	lnk.id='code_stats_show_more';

  		        td.appendChild(lnk);

  		    	show_more.appendChild(td);

  		    	show_more.appendChild(document.createElement('td'));

  		    	tbl.appendChild(show_more);

  		    }

  		}

  		if(no_data){

  			var tr = document.createElement('tr');

  			var td1 = document.createElement('td');

  			td1.innerHTML = "${c.no_data_msg}";

  			tr.appendChild(td1);

  			tbl.appendChild(tr);

		}

  		YUD.get('lang_stats').appendChild(tbl);

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

  			l = YUD.getElementsByClassName('stats_hidden')

  			for (e in l){

  			    YUD.setStyle(l[e],'display','');

  			};

  			YUD.setStyle(YUD.get('code_stats_show_more'),

  					'display','none');

  		})

             YUE.on('download_options','change',function(e){

                 var new_cs = e.target.options[e.target.selectedIndex];

                 var tmpl_links = {}

                 %for cnt,archive in enumerate(c.rhodecode_repo._get_archives()):

                	 tmpl_links['${archive['type']}'] = '${h.link_to(archive['type'],

                        h.url('files_archive_home',repo_name=c.dbrepo.repo_name,

                        fname='__CS__'+archive['extension']),class_="archive_icon")}';

                 %endfor

                 for(k in tmpl_links){

                	 var s = YUD.get(k+'_link')

                	 title_tmpl = "${_('Download %s as %s') % ('__CS_NAME__',archive['type'])}";

                	 s.title = title_tmpl.replace('__CS_NAME__',new_cs.text)

                	 s.innerHTML = tmpl_links[k].replace('__CS__',new_cs.value);

                 }

             })

  	</script>    				

</div>

<div class="box box-right"  style="min-height:455px">

    <!-- box / title -->

    <div class="title">

        <h5>${_('Commit activity by day / author')}</h5>

    </div>

    <div class="table">

         %if c.no_data:

           %if h.HasPermissionAll('hg.admin')('enable stats on from summary'):

                [${h.link_to(_('enable'),h.url('edit_repo',repo_name=c.repo_name))}]

           %endif         

           </div>

        <div id="commit_history" style="width:460px;height:300px;float:left"></div>

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

        <div id="overview" style="width:460px;height:100px;float:left"></div>

    	<div id="legend_data" style="clear:both;margin-top:10px;">

	    	<div id="legend_container"></div>

	    	<div id="legend_choices">

				<table id="legend_choices_tables" style="font-size:smaller;color:#545454"></table>

	    	</div>

    	</div>

		<script type="text/javascript">

		/**

		 * Plots summary graph

		 *

		 * @class SummaryPlot

		 * @param {from} initial from for detailed graph

		 * @param {to} initial to for detailed graph

		 * @param {dataset}

		 * @param {overview_dataset}

		 */

		function SummaryPlot(from,to,dataset,overview_dataset) {

			var initial_ranges = {

			    "xaxis":{

				    "from":from,

				   	"to":to,

				},

			};

		    var dataset = dataset;

		    var overview_dataset = [overview_dataset];

		    var choiceContainer = YUD.get("legend_choices");

		    var choiceContainerTable = YUD.get("legend_choices_tables");

		    var plotContainer = YUD.get('commit_history');

		    var overviewContainer = YUD.get('overview');

		    var plot_options = {

				bars: {show:true,align:'center',lineWidth:4},

				legend: {show:true, container:"legend_container"},

				points: {show:true,radius:0,fill:false},

				yaxis: {tickDecimals:0,},

				xaxis: {

					mode: "time", 

					timeformat: "%d/%m",

				    min:from,

				    max:to,	

				}, 

				grid: {

					hoverable: true, 

				    clickable: true,

				    autoHighlight:true,

				    color: "#999"

				},

				//selection: {mode: "x"}

		    };

		    var overview_options = {

				legend:{show:false},

			    bars: {show:true,barWidth: 2,},

			    shadowSize: 0,

			    xaxis: {mode: "time", timeformat: "%d/%m/%y",},

			    yaxis: {ticks: 3, min: 0,tickDecimals:0,},

			    grid: {color: "#999",},

			    selection: {mode: "x"}

			};

			/**

			*get dummy data needed in few places

			*/

		    function getDummyData(label){

		    	return {"label":label,

               	 "data":[{"time":0,

               		 "commits":0,

	                     "added":0,

	                     "changed":0,

	                     "removed":0,

                    }],

                    "schema":["commits"],

                    "color":'#ffffff',

           		}

			}

		    /**

		     * generate checkboxes accordindly to data

		     * @param keys

		     * @returns

		     */

		    function generateCheckboxes(data) {

			    //append checkboxes

			    var i = 0;

			    choiceContainerTable.innerHTML = '';

			    for(var pos in data) {

			    	data[pos].color = i;

			        i++;

			        if(data[pos].label != ''){

				        choiceContainerTable.innerHTML += '<tr><td>'+

				        '<input type="checkbox" name="' + data[pos].label +'" checked="checked" />'

				        +data[pos].label+

				        '</td></tr>';

			        }

			    }	

		    }

		    /**

		     * ToolTip show

		     */

		    function showTooltip(x, y, contents) {

		        var div=document.getElementById('tooltip');

		        if(!div) {

		            div = document.createElement('div');

		            div.id="tooltip";

		            div.style.position="absolute";

		            div.style.border='1px solid #fdd';

		            div.style.padding='2px';

		            div.style.backgroundColor='#fee';

		            document.body.appendChild(div);

		        }

		        YUD.setStyle(div, 'opacity', 0);

		        div.innerHTML = contents;

		        div.style.top=(y + 5) + "px";

		        div.style.left=(x + 5) + "px";

		        var anim = new YAHOO.util.Anim(div, {opacity: {to: 0.8}}, 0.2);

		        anim.animate();

		    }

			/**

			 * This function will detect if selected period has some changesets 

			   for this user if it does this data is then pushed for displaying

			   Additionally it will only display users that are selected by the checkbox

			*/

		    function getDataAccordingToRanges(ranges) {

		        var data = [];

		        var keys = [];

				for(var key in dataset){

					var push = false;

					//method1 slow !!

		            //*

		            for(var ds in dataset[key].data){

			            commit_data = dataset[key].data[ds];

			            if (commit_data.time >= ranges.xaxis.from && commit_data.time <= ranges.xaxis.to){

			            	push = true;

			            	break;

					    }

				    }

					//*/

				    /*//method2 sorted commit data !!!

				    var first_commit = dataset[key].data[0].time;

				    var last_commit = dataset[key].data[dataset[key].data.length-1].time;

				    if (first_commit >= ranges.xaxis.from && last_commit <= ranges.xaxis.to){

						push = true;

					}

				    //*/

				    if(push){			

				    	data.push(dataset[key]);

				    }

				}

				if(data.length >= 1){

					return data;

				} 

				else{

					//just return dummy data for graph to plot itself

					return [getDummyData('')];	

				}

		    }

			/**

			* redraw using new checkbox data

			*/

		    function plotchoiced(e,args){

			    var cur_data = args[0];

			    var cur_ranges = args[1];

				var new_data = [];

		    	var inputs = choiceContainer.getElementsByTagName("input");

		    	//show only checked labels

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

		            var checkbox_key = inputs[i].name;

	                if(inputs[i].checked){

						for(var d in cur_data){

							if(cur_data[d].label == checkbox_key){

								new_data.push(cur_data[d]);

							}

						}			                

	    	        }