#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2010-2011 Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
# Copyright (C) 2007  skam <massimo dot scamarcia at gmail.com>
#                          (http://djangosnippets.org/snippets/233/)

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero 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 Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# See the file COPYING for details.
from django import forms
from django.utils.safestring import mark_safe
from django.forms.widgets import flatatt
from django.utils.encoding import smart_unicode
from django.utils.html import escape
from django.utils.simplejson import JSONEncoder
from django.core.urlresolvers import resolve, reverse
from django.utils.translation import ugettext_lazy as _

from ishtar import settings
import models

class DeleteWidget(forms.CheckboxInput):
    def render(self, name, value, attrs=None):
        final_attrs = flatatt(self.build_attrs(attrs, name=name,
                                               value='1'))
        output = ['<tr class="delete"><td colspan="2">']
        output.append(u"<button%s>%s</button>" % (final_attrs, _("Delete")))
        output.append('</td></tr>')
        return mark_safe('\n'.join(output))

class SquareMeterWidget(forms.TextInput):
    def render(self, name, value, attrs=None):
        if not value:
            value = u""
        final_attrs = flatatt(self.build_attrs(attrs, name=name, value=value))
        output = u'<input class="area_widget" type="text"%s> %s '\
                 u'(<span id="ha_%s">0</span> ha)' % (final_attrs,
                 settings.SURFACE_UNIT_LABEL, attrs['id'])
        output += """
<script type="text/javascript"><!--//
    function evaluate_%(safe_id)s(){
        value = parseFloat($("#%(id)s").val());
        if(!isNaN(value)){
            value = value/10000;
        } else {
            value = 0;
        }
        $("#ha_%(id)s").html(value);
    }
    $("#%(id)s").keyup(evaluate_%(safe_id)s);
    $(document).ready(evaluate_%(safe_id)s());
//--></script>
""" % {"id":attrs['id'], "safe_id":attrs['id'].replace('-', '_')}
        return mark_safe(output)

AreaWidget = forms.TextInput
if settings.SURFACE_UNIT == 'square-metre':
    global AreaWidget
    AreaWidget = SquareMeterWidget

class JQueryDate(forms.TextInput):
    def render(self, name, value=None, attrs=None):
        rendered = super(JQueryDate, self).render(name, value, attrs)
        rendered += """
<script type="text/javascript"><!--//
    $(function() {
        $("#id_%(name)s").datepicker($.datepicker.regional["%(country)s"]);
        var val = $("#id_%(name)s").val();
        if(val){
            var dtp = $.datepicker.parseDate('yy-mm-dd', val);
            val = $.datepicker.formatDate(
                    $.datepicker.regional["%(country)s"]['dateFormat'],
                    dtp);
            $("#id_%(name)s").val(val);
        }
    });
//--></script>
""" % {"name":name, "country":settings.COUNTRY}
        return rendered

class JQueryDate(forms.TextInput):
    def render(self, name, value=None, attrs=None):
        rendered = super(JQueryDate, self).render(name, value, attrs)
        rendered += """
<script type="text/javascript"><!--//
    $(function() {
        $("#id_%(name)s").datepicker($.datepicker.regional["%(country)s"]);
        var val = $("#id_%(name)s").val();
        if(val){
            var dt = $.datepicker.parseDate('yy-mm-dd', val);
            val = $.datepicker.formatDate(
                    $.datepicker.regional["%(country)s"]['dateFormat'],
                    dt);
            $("#id_%(name)s").val(val);
        }
    });
//--></script>
""" % {"name":name, "country":settings.COUNTRY}
        return rendered

class JQueryAutoComplete(forms.TextInput):
    def __init__(self, source, associated_model=None, options={}, attrs={},
                 new=False):
        """
        Source can be a list containing the autocomplete values or a
        string containing the url used for the request.
        """
        self.options = None
        self.attrs = {}
        self.source = source
        self.associated_model = associated_model
        if len(options) > 0:
            self.options = JSONEncoder().encode(options)
        self.attrs.update(attrs)
        self.new = new

    def render_js(self, field_id):
        if isinstance(self.source, list):
            source = JSONEncoder().encode(self.source)
        elif isinstance(self.source, str) or isinstance(self.source, unicode):
            source = "'%s'" % escape(self.source)
        else:
            try:
                source = "'" + unicode(self.source) + "'"
            except:
                raise ValueError('source type is not valid')
        options = 'source : ' + source
        options += ''', select: function( event, ui ) {
            if(ui.item){
                $('#id_%s').val(ui.item.id);
            } else {
                $('#id_%s').val(null);
            }
        }, minLength: 2
        ''' % (field_id, field_id)
        if self.options:
            options += ',%s' % self.options

        js = u'$(\'#id_select_%s\').autocomplete({%s});\n' % (field_id, options)
        js += u'''$(\'#id_select_%s\').live('click', function(){
                $('#id_%s').val(null);
                $('#id_select_%s').val(null);
});''' % (field_id, field_id, field_id)
        return js

    def render(self, name, value=None, attrs=None):
        attrs_hidden = self.build_attrs(attrs, name=name)
        attrs_select = self.build_attrs(attrs)

        if value:
            val =  escape(smart_unicode(value))
            attrs_hidden['value'] = val
            attrs_select['value'] = val
            if self.associated_model:
                try:
                    attrs_select['value'] = unicode(self.associated_model.\
objects.get(pk=value))
                except:
                    attrs_select['value'] = ""
        if not self.attrs.has_key('id'):
            attrs_hidden['id'] = 'id_%s' % name
            attrs_select['id'] = 'id_select_%s' % name
        if 'class' not in attrs_select:
            attrs_select['class'] = 'autocomplete'
        new = ''
        if self.new:
            model_name = self.associated_model._meta.object_name.lower()
            new = u'  <a href="#" onclick="open_window(\'%s\');">%s</a>' % \
   (reverse('new-' + model_name, args=[attrs_select['id']]),
   unicode(_(u"Add")))
        html = u'''<input%(attrs_select)s/>%(new)s\
<input type="hidden"%(attrs_hidden)s/>\
        <script type="text/javascript"><!--//
        $(function() {%(js)s});//--></script>
        ''' % {
            'attrs_select' : flatatt(attrs_select),
            'attrs_hidden' : flatatt(attrs_hidden),
            'js' : self.render_js(name),
            'new':new
        }
        return html

class JQueryJqGrid(forms.RadioSelect):
    COL_TPL = "{name:'%(idx)s', index:'%(idx)s', sortable:true}"
    class Media:
        js = ['%s/js/i18n/grid.locale-%s.js' % (settings.MEDIA_URL,
                                                settings.COUNTRY),
              '%s/js/jquery.jqGrid.min.js' % settings.MEDIA_URL,
             ]
        css = {'all':['%s/media/ui.jqgrid.css' % settings.MEDIA_URL,
                     ]}

    def __init__(self, source, form, associated_model, attrs={},
                 table_cols='TABLE_COLS', multiple=False):
        self.source = source
        self.form = form
        self.attrs = attrs
        self.associated_model = associated_model
        self.table_cols = table_cols
        self.multiple = multiple

    def render(self, name, value=None, attrs=None):
        rendered = unicode(self.form)
        rendered += """
</table>
<button id='search_%s' class='submit'>%s</button>
<h4>%s</h4>
""" % (name, unicode(_("Search")), unicode(_("Search and select an item")))
        extra_cols = []
        col_names, col_idx = [], []
        for k in self.form.fields:
            field = self.form.fields[k]
            col_idx.append(u'"%s"' % k)
        for field_name in getattr(self.associated_model, self.table_cols):
            field = self.associated_model
            keys = field_name.split('.')
            try:
                for key in keys:
                    if hasattr(field, 'rel'):
                        field = field.rel.to
                    field = field._meta.get_field(key)
            except:
                continue
            col_names.append(u'"%s"' % field.verbose_name)
            extra_cols.append(self.COL_TPL % {'idx':field.name})
        col_names = col_names and ",\n".join(col_names) or ""
        col_idx = col_idx and ",\n".join(col_idx) or ""
        extra_cols = extra_cols and ",\n".join(extra_cols) or ""
        rendered += """<table id="grid_%s" class='jqgrid'></table>
<div id="pager_%s"></div>
""" % (name, name)
        encoding = settings.ENCODING or 'utf-8'
        rendered += """
<div id="foot_%s" class="gridfooter"><a href="%scsv" target="_blank">%s (%s)</a></div>
""" % (name, unicode(self.source), unicode(_("Export as CSV")), encoding)
        if self.multiple:
            rendered += '''
<input type="button" id="add_button_%s" value="%s"/>
<ul id='selectmulti_%s' class='selectmulti'>
</ul>
''' % (name, unicode(_("Add")), name)
        rendered += '<input type="hidden" id="hidden_%s" name="%s"/>' % (name,
                                                                        name)
        dct = {'name':name, 'col_names':col_names, 'extra_cols':extra_cols,
       'source':unicode(self.source), 'col_idx':col_idx,
       'no_result':unicode(_("No results")), 'loading':unicode(_("Loading...")),
       'remove':unicode(_("Remove")), 'sname':name.replace('-', '')}

        rendered += """
<script type="text/javascript">
var query_vars = new Array(%(col_idx)s);
jQuery(document).ready(function(){
  jQuery("#search_%(name)s").click(function (){
    var data = "";
    for (idx in query_vars)
    {
        var key = query_vars[idx];
        var val = jQuery("#id_"+key).val();
        if (val){
            if (data) data += "&";
            data += key + "=" + val;
        }
    }
    var mygrid = jQuery("#grid_%(name)s");
    var url = "%(source)s?submited=1&amp;" + data;
    mygrid.setGridParam({url:url});
    mygrid.trigger("reloadGrid");
    return false;
  });

  jQuery("#grid_%(name)s").jqGrid({
    url:'%(source)s',
    datatype: "json",
    mtype: 'GET',
    colNames:['id', '', %(col_names)s],
    colModel:[
      {name:'id', index:'id', hidden:true},
      {name:'link', index:'link', width:80},
      %(extra_cols)s
    ],
    sortname: 'value',
    viewrecords: true,
    sortorder: "asc",
    emptyrecords: "%(no_result)s",
    loadtext: "%(loading)s",
    pager: '#pager_%(name)s',
    pgbuttons: false,
    pginput: false,
    width:740,
    jsonReader : {repeatitems: false}
  });
""" % dct
        if self.multiple:
            rendered += """
  var selItems_%(sname)s = new Array();
  jQuery("#add_button_%(name)s").click(function (){
    var mygrid = jQuery("#grid_%(name)s");
    var idx = mygrid.getGridParam('selrow');
    var label = mygrid.getCell(idx, 2);
    for (id in selItems_%(sname)s){
        if(selItems_%(sname)s[id] == idx){
            return false;
        }
    }
    selItems_%(sname)s.push(idx);
    jQuery("#selectmulti_%(name)s").append(
               "<li id='selected_%(name)s_"+idx+"'>"+label+" <a href='#' class='remove' onclick=\\"multiRemoveItem('selItems_%(sname)s', '%(name)s', "+ idx +");return false;\\">%(remove)s</a></li>");
    return true;
  });
  jQuery("#submit_form").click(function (){
    jQuery("#hidden_%(name)s").val(selItems_%(sname)s);
    return true;
  });
""" % dct
        else:
            rendered += """
  jQuery("#submit_form").click(function (){
    var mygrid = jQuery("#grid_%(name)s");
    jQuery("#hidden_%(name)s").val(mygrid.getGridParam('selrow'));
    return true;
  });
""" % dct
        rendered += """
});
</script>
"""
        return mark_safe(rendered)