"""widget classes for form construction

:organization: Logilab

:copyright: 2009 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr

"""

__docformat__ = "restructuredtext en"

from datetime import date

from cubicweb.common import tags

from cubicweb.web import stdmsgs, INTERNAL_FIELD_VALUE

class FieldWidget(object):

    """abstract widget class"""

    # javascript / css files required by the widget

    needs_js = ()

    needs_css = ()

    # automatically set id and tabindex attributes ?

    setdomid = True

    settabindex = True

    def __init__(self, attrs=None, setdomid=None, settabindex=None):

        if attrs is None:

            attrs = {}

        self.attrs = attrs

        if setdomid is not None:

            # override class's default value

            self.setdomid = setdomid

        if settabindex is not None:

            # override class's default value

            self.settabindex = settabindex

    def add_media(self, form):

        """adds media (CSS & JS) required by this widget"""

        if self.needs_js:

            form.req.add_js(self.needs_js)

        if self.needs_css:

            form.req.add_css(self.needs_css)

    def render(self, form, field):

        """render the widget for the given `field` of `form`.

        To override in concrete class

        """

        raise NotImplementedError

    def _render_attrs(self, form, field):

        """return html tag name, attributes and a list of values for the field

        """

        name = form.context[field]['name']

        values = form.context[field]['value']

        if not isinstance(values, (tuple, list)):

            values = (values,)

        attrs = dict(self.attrs)

        if self.setdomid:

            attrs['id'] = form.context[field]['id']

        if self.settabindex and not 'tabindex' in attrs:

            attrs['tabindex'] = form.req.next_tabindex()

        return name, values, attrs

class Input(FieldWidget):

    """abstract widget class for <input> tag based widgets"""

    type = None

    def render(self, form, field):

        """render the widget for the given `field` of `form`.

        Generate one <input> tag for each field's value

        """

        self.add_media(form)

        name, values, attrs = self._render_attrs(form, field)

        # ensure something is rendered

        if not values:

            values = (INTERNAL_FIELD_VALUE,)

        inputs = [tags.input(name=name, value=value, type=self.type, **attrs)

                  for value in values]

        return u'\n'.join(inputs)

# basic html widgets ###########################################################

class TextInput(Input):

    """<input type='text'>"""

    type = 'text'

class PasswordInput(Input):

    """<input type='password'> and its confirmation field (using

    <field's name>-confirm as name)

    """

    type = 'password'

    def render(self, form, field):

        self.add_media(form)

        name, values, attrs = self._render_attrs(form, field)

        assert len(values) == 1

        id = attrs.pop('id')

        try:

            confirmname = '%s-confirm:%s' % tuple(name.rsplit(':', 1))

        except TypeError:

            confirmname = '%s-confirm' % name

        inputs = [tags.input(name=name, value=values[0], type=self.type, id=id, **attrs),

                  '<br/>',

                  tags.input(name=confirmname, value=values[0], type=self.type, **attrs),

                  ' ', tags.span(form.req._('confirm password'),

                                      **{'class': 'emphasis'})]

        return u'\n'.join(inputs)

class FileInput(Input):

    """<input type='file'>"""

    type = 'file'

    def _render_attrs(self, form, field):

        # ignore value which makes no sense here (XXX even on form validation error?)

        name, values, attrs = super(FileInput, self)._render_attrs(form, field)

        return name, ('',), attrs

class HiddenInput(Input):

    """<input type='hidden'>"""

    type = 'hidden'

    setdomid = False # by default, don't set id attribute on hidden input

    settabindex = False

class ButtonInput(Input):

    """<input type='button'>

    if you want a global form button, look at the Button, SubmitButton,

    ResetButton and ImgButton classes below.

    """

    type = 'button'

class TextArea(FieldWidget):

    """<textarea>"""

    def render(self, form, field):

        name, values, attrs = self._render_attrs(form, field)

        attrs.setdefault('onkeypress', 'autogrow(this)')

        attrs.setdefault('cols', 80)

        attrs.setdefault('rows', 20)

        if not values:

            value = u''

        elif len(values) == 1:

            value = values[0]

        else:

            raise ValueError('a textarea is not supposed to be multivalued')

        return tags.textarea(value, name=name, **attrs)

class FCKEditor(TextArea):

    """FCKEditor enabled <textarea>"""

    def __init__(self, *args, **kwargs):

        super(FCKEditor, self).__init__(*args, **kwargs)

        self.attrs['cubicweb:type'] = 'wysiwyg'

    def render(self, form, field):

        form.req.fckeditor_config()

        return super(FCKEditor, self).render(form, field)

class Select(FieldWidget):

    """<select>, for field having a specific vocabulary"""

    def __init__(self, attrs=None, multiple=False):

        super(Select, self).__init__(attrs)

        self._multiple = multiple

    def render(self, form, field):

        name, curvalues, attrs = self._render_attrs(form, field)

        if not 'size' in attrs and self._multiple:

            attrs['size'] = '5'

        options = []

        optgroup_opened = False

        for label, value in field.vocabulary(form):

            if value is None:

                # handle separator

                if optgroup_opened:

                    options.append(u'</optgroup>')

                options.append(u'<optgroup label="%s">' % (label or ''))

                optgroup_opened = True

            elif value in curvalues:

                options.append(tags.option(label, value=value, selected='selected'))

            else:

                options.append(tags.option(label, value=value))

        if optgroup_opened:

            options.append(u'</optgroup>')

        return tags.select(name=name, multiple=self._multiple,

                           options=options, **attrs)

class CheckBox(Input):

    """<input type='checkbox'>, for field having a specific vocabulary. One

    input will be generated for each possible value.

    """

    type = 'checkbox'

    def render(self, form, field):

        name, curvalues, attrs = self._render_attrs(form, field)

        domid = attrs.pop('id', None)

        options = []

        for i, (label, value) in enumerate(field.vocabulary(form)):

            iattrs = attrs.copy()

            if i == 0 and domid is not None:

                iattrs['id'] = domid

            if value in curvalues:

                iattrs['checked'] = u'checked'

            tag = tags.input(name=name, type=self.type, value=value, **iattrs)

            options.append(tag + label + '<br/>')

        return '\n'.join(options)

class Radio(CheckBox):

    """<input type='radio'>, for field having a specific vocabulary. One

    input will be generated for each possible value.

    """

    type = 'radio'

# javascript widgets ###########################################################

class DateTimePicker(TextInput):

    """<input type='text' + javascript date/time picker for date or datetime

    fields

    """

    monthnames = ('january', 'february', 'march', 'april',

                  'may', 'june', 'july', 'august',

                  'september', 'october', 'november', 'december')

    daynames = ('monday', 'tuesday', 'wednesday', 'thursday',

                'friday', 'saturday', 'sunday')

    needs_js = ('cubicweb.calendar.js',)

    needs_css = ('cubicweb.calendar_popup.css',)

    @classmethod

    def add_localized_infos(cls, req):

        """inserts JS variables defining localized months and days"""

        # import here to avoid dependancy from cubicweb-common to simplejson

        _ = req._

        monthnames = [_(mname) for mname in cls.monthnames]

        daynames = [_(dname) for dname in cls.daynames]

        req.html_headers.define_var('MONTHNAMES', monthnames)

        req.html_headers.define_var('DAYNAMES', daynames)

    def render(self, form, field):

        txtwidget = super(DateTimePicker, self).render(form, field)

        self.add_localized_infos(form.req)

        cal_button = self._render_calendar_popup(form, field)

        return txtwidget + cal_button

    def _render_calendar_popup(self, form, field):

        value = form.form_field_value(field)

        if not value:

            value = date.today()

        inputid = form.context[field]['id']

        helperid = '%shelper' % inputid

        year, month = value.year, value.month

        return (u"""<a onclick="toggleCalendar('%s', '%s', %s, %s);" class="calhelper">

<img src="%s" title="%s" alt="" /></a><div class="calpopup hidden" id="%s"></div>"""

                % (helperid, inputid, year, month,

                   form.req.external_resource('CALENDAR_ICON'),

                   form.req._('calendar'), helperid) )

# ajax widgets ################################################################

def init_ajax_attributes(attrs, wdgtype, loadtype=u'auto'):

    try:

        attrs['klass'] += u' widget'

    except KeyError:

        attrs['klass'] = u'widget'

    attrs.setdefault('cubicweb:wdgtype', wdgtype)

    attrs.setdefault('cubicweb:loadtype', loadtype)

class AjaxWidget(FieldWidget):

    """simple <div> based ajax widget"""

    def __init__(self, wdgtype, inputid=None, **kwargs):

        super(AjaxWidget, self).__init__(**kwargs)

        init_ajax_attributes(self.attrs, wdgtype)

        if inputid is not None:

            self.attrs['cubicweb:inputid'] = inputid

    def render(self, form, field):

        self.add_media(form)

        attrs = self._render_attrs(form, field)[-1]

        return tags.div(**attrs)

class AutoCompletionWidget(TextInput):

    """ajax widget for StringField, proposing matching existing values as you

    type.

    """

    needs_js = ('cubicweb.widgets.js', 'jquery.autocomplete.js')

    needs_css = ('jquery.autocomplete.css',)

    wdgtype = 'SuggestField'

    loadtype = 'auto'

    def _render_attrs(self, form, field):

        name, values, attrs = super(AutoCompletionWidget, self)._render_attrs(form, field)

        if not values[0]:

            values = (INTERNAL_FIELD_VALUE,)

        init_ajax_attributes(attrs, self.wdgtype, self.loadtype)

        # XXX entity form specific

        attrs['cubicweb:dataurl'] = self._get_url(form.edited_entity)

        return name, values, attrs

    def _get_url(self, entity):

        return entity.req.build_url('json', fname=entity.autocomplete_initfuncs[self.rschema],

                                pageid=entity.req.pageid, mode='remote')

class StaticFileAutoCompletionWidget(AutoCompletionWidget):

    """XXX describe me"""

    wdgtype = 'StaticFileSuggestField'

    def _get_url(self, entity):

        return entity.req.datadir_url + entity.autocomplete_initfuncs[self.rschema]

class RestrictedAutoCompletionWidget(AutoCompletionWidget):

    """XXX describe me"""

    wdgtype = 'RestrictedSuggestField'

class AddComboBoxWidget(Select):

    def _render_attrs(self, form, field):

        name, values, attrs = super(AddComboBoxWidget, self)._render_attrs(form, field)

        init_ajax_attributes(self.attrs, 'AddComboBox')

        # XXX entity form specific

        entity = form.edited_entity

        attrs['cubicweb:etype_to'] = entity.e_schema

        etype_from = entity.e_schema.subject_relation(self.name).objects(entity.e_schema)[0]

        attrs['cubicweb:etype_from'] = etype_from

    def render(self, form, field):

        return super(AddComboBoxWidget, self).render(form, field) + u'''

<div id="newvalue">

  <input type="text" id="newopt" />

  <a href="javascript:noop()" id="add_newopt">&nbsp;</a></div>

'''

# buttons ######################################################################

class Button(Input):

    """<input type='button'>, base class for global form buttons

    note label is a msgid which will be translated at form generation time, you

    should not give an already translated string.

    """

    type = 'button'

    def __init__(self, label=stdmsgs.BUTTON_OK, attrs=None,

                 setdomid=None, settabindex=None,

                 name='', value='', onclick=None, cwaction=None):

        super(Button, self).__init__(attrs, setdomid, settabindex)

        self.label = label

        self.name = name

        self.value = ''

        self.onclick = onclick

        self.cwaction = cwaction

        self.attrs.setdefault('klass', 'validateButton')

    def render(self, form, field=None):

        label = form.req._(self.label)

        attrs = self.attrs.copy()

        if self.cwaction:

            assert self.onclick is None

            attrs['onclick'] = "postForm('__action_%s', \'%s\', \'%s\')" % (