# 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.

"""

Created on April 9, 2010

Model for RhodeCode

@author: marcink

"""

from beaker.cache import cache_region, region_invalidate

from mercurial import ui

from rhodecode.lib import helpers as h

from rhodecode.lib.auth import HasRepoPermissionAny

from rhodecode.lib.utils import get_repos

from rhodecode.model import meta

from rhodecode.model.db import Repository, User, RhodeCodeUi, CacheInvalidation

from rhodecode.model.caching_query import FromCache

from sqlalchemy.orm import joinedload

from sqlalchemy.orm.session import make_transient

from vcs import get_backend

from vcs.utils.helpers import get_scm

from vcs.exceptions import RepositoryError, VCSError

from vcs.utils.lazy import LazyProperty

import traceback

import logging

import os

import time

log = logging.getLogger(__name__)

class ScmModel(object):

    """

    Mercurial Model

    """

    def __init__(self):

        self.sa = meta.Session()

    @LazyProperty

    def repos_path(self):

        """

        Get's the repositories root path from database

        """

        q = self.sa.query(RhodeCodeUi).filter(RhodeCodeUi.ui_key == '/').one()

        return q.ui_value

    def repo_scan(self, repos_path, baseui, initial=False):

        """

        Listing of repositories in given path. This path should not be a 

        repository itself. Return a dictionary of repository objects

        :param repos_path: path to directory containing repositories

        :param baseui

        :param initial: initial scan

        """

        log.info('scanning for repositories in %s', repos_path)

        if not isinstance(baseui, ui.ui):

            baseui = ui.ui()

        repos_list = {}

        for name, path in get_repos(repos_path):

            try:

                if repos_list.has_key(name):

                    raise RepositoryError('Duplicate repository name %s '

                                    'found in %s' % (name, path))

                else:

                    klass = get_backend(path[0])

                    if path[0] == 'hg':

                        repos_list[name] = klass(path[1], baseui=baseui)

                    if path[0] == 'git':

                        repos_list[name] = klass(path[1])

            except OSError:

                continue

        return repos_list

    def get_repos(self, all_repos=None):

        """

        Get all repos from db and for each repo create it's backend instance.

        and fill that backed with information from database

        :param all_repos: give specific repositories list, good for filtering

        """

        if not all_repos:

        for r in all_repos:

            repo = self.get(r.repo_name)

            if repo is not None:

                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.dbrepo.description

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

                tmp_d['last_change'] = last_change

                tmp_d['last_change_sort'] = time.mktime(last_change.timetuple())

                tmp_d['tip'] = tip.raw_id

                tmp_d['tip_sort'] = tip.revision

                tmp_d['rev'] = tip.revision

                tmp_d['contact'] = repo.dbrepo.user.full_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 self.get(repo_name)

    def get(self, repo_name):

        """

        Get's repository from given name, creates BackendInstance and

        propagates it's data from database with all additional information

        :param repo_name:

        """

        if not HasRepoPermissionAny('repository.read', 'repository.write',

                            'repository.admin')(repo_name, 'get repo check'):

            return

        @cache_region('long_term')

        def _get_repo(repo_name):

            repo_path = os.path.join(self.repos_path, repo_name)

            alias = get_scm(repo_path)[0]

            log.debug('Creating instance of %s repository', alias)

            backend = get_backend(alias)

            if alias == 'hg':

                repo = backend(repo_path, create=False, baseui=None)

                #skip hidden web repository

                if repo._get_hidden():

                    return

            else:

                repo = backend(repo_path, create=False)

            dbrepo = self.sa.query(Repository)\

                .options(joinedload(Repository.fork))\

                .options(joinedload(Repository.user))\

                .filter(Repository.repo_name == repo_name)\

                .scalar()

            make_transient(dbrepo)

            repo.dbrepo = dbrepo

            return repo

        invalidate = self._should_invalidate(repo_name)

        if invalidate:

            log.info('invalidating cache for repository %s', repo_name)

            region_invalidate(_get_repo, None, repo_name)

            self._mark_invalidated(invalidate)

        return _get_repo(repo_name)

    def mark_for_invalidation(self, repo_name):

        """

        Puts cache invalidation task into db for 

        further global cache invalidation

        :param repo_name: this repo that should invalidation take place

        """

        log.debug('marking %s for invalidation', repo_name)

        cache = self.sa.query(CacheInvalidation)\

            .filter(CacheInvalidation.cache_key == repo_name).scalar()

        if cache:

            #mark this cache as inactive

            cache.cache_active = False

        else:

            log.debug('cache key not found in invalidation db -> creating one')

            cache = CacheInvalidation(repo_name)

        try:

            self.sa.add(cache)

            self.sa.commit()

		         %endif 

	             %if c.repo_info.dbrepo.private:

	                <img style="margin-bottom:2px" class="icon" title="${_('private repository')}" alt="${_('private repository')}" src="/images/icons/lock.png"/>

	             %else:

	                <img style="margin-bottom:2px" class="icon" title="${_('public repository')}" alt="${_('public repository')}" src="/images/icons/lock_open.png"/>

	             %endif

			      <span style="font-size: 1.6em;font-weight: bold;vertical-align: baseline;">${c.repo_info.name}</span>

			      <br/>

		            %if c.repo_info.dbrepo.fork:

		            	<span style="margin-top:5px">

		            	<a href="${h.url('summary_home',repo_name=c.repo_info.dbrepo.fork.repo_name)}">

		            	<img class="icon" alt="${_('public')}"

		            	title="${_('Fork of')} ${c.repo_info.dbrepo.fork.repo_name}" 

		            	src="/images/icons/arrow_divide.png"/>

		            	${_('Fork of')} ${c.repo_info.dbrepo.fork.repo_name}

		            	</a>

		            	</span>

		            %endif			      

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

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

			  </div>

			  <div class="input-short">

			      ${c.repo_info.dbrepo.description}

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

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

			  </div>

			  <div class="input-short">

			  	<div class="gravatar">

			  		<img alt="gravatar" src="${h.gravatar_url(c.repo_info.dbrepo.user.email)}"/>

			  	</div>

			  		${_('Username')}: ${c.repo_info.dbrepo.user.username}<br/>

			  		${_('Name')}: ${c.repo_info.dbrepo.user.name} ${c.repo_info.dbrepo.user.lastname}<br/>

			  		${_('Email')}: <a href="mailto:${c.repo_info.dbrepo.user.email}">${c.repo_info.dbrepo.user.email}</a>

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

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

			  </div>

			  <div class="input-short">

			      ${h.age(c.repo_info.last_change)} - ${c.repo_info.last_change} 

			      ${_('by')} ${h.get_changeset_safe(c.repo_info,'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 languages')}:</label>

			  </div>

			  <div class="input-short">

			    <div id="lang_stats">

			    </div> 			  

			  	<script type="text/javascript">

			  		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');

			  		for (k in data){

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

				  		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');

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

			  		    trending_language.title = k;

			  		    trending_language.innerHTML = "<b>"+percentage+"% "+value+" ${_('files')}</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(no_data){

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

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

			  			td1.innerHTML = "${_('No data loaded yet')}";

			  			tr.appendChild(td1);

			  			tbl.appendChild(tr);

					}

			  		YAHOO.util.Dom.get('lang_stats').appendChild(tbl);

			  	</script>

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

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

			  </div>

			  <div class="input-short">

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

		             %if cnt >=1:

		             |

		             %endif

		             ${h.link_to(c.repo_info.name+'.'+archive['type'],

		                h.url('files_archive_home',repo_name=c.repo_info.name,

		                revision='tip',fileformat=archive['extension']),class_="archive_icon")}

		        %endfor

			  </div>

			 </div>

			 <div class="field">

			  <div class="label">

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

			  </div>

			  <div class="input-short">

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

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

			  </div>

			 </div>				 			 			 

	  </div>		 

	</div>				

</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">

        <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 = YAHOO.util.Dom.get("legend_choices");

		    var choiceContainerTable = YAHOO.util.Dom.get("legend_choices_tables");

		    var plotContainer = YAHOO.util.Dom.get('commit_history');

		    var overviewContainer = YAHOO.util.Dom.get('overview');

		    var plot_options = {