"""workflow views:

* IWorkflowable views and forms

* workflow entities views (State, Transition, TrInfo)

:organization: Logilab

:copyright: 2001-2010 LOGILAB S.A. (Paris, FRANCE), license is LGPL v2.

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

:license: GNU Lesser General Public License, v2.1 - http://www.gnu.org/licenses

"""

__docformat__ = "restructuredtext en"

_ = unicode

from logilab.mtconverter import xml_escape

from logilab.common.graph import escape, GraphGenerator, DotBackend

from cubicweb import Unauthorized, view

from cubicweb.selectors import (implements, has_related_entities, one_line_rset,

                                relation_possible, match_form_params,

                                implements, score_entity)

from cubicweb.interfaces import IWorkflowable

from cubicweb.view import EntityView

from cubicweb.schema import display_name

from cubicweb.web import uicfg, stdmsgs, action, component, form, action

from cubicweb.web import formfields as ff, formwidgets as fwdgs

from cubicweb.web.views import TmpFileViewMixin, forms, primary, autoform

_pvs = uicfg.primaryview_section

_pvs.tag_subject_of(('Workflow', 'initial_state', '*'), 'hidden')

_pvs.tag_object_of(('*', 'state_of', 'Workflow'), 'hidden')

_pvs.tag_object_of(('*', 'transition_of', 'Workflow'), 'hidden')

_abaa = uicfg.actionbox_appearsin_addmenu

_abaa.tag_subject_of(('BaseTransition', 'condition', 'RQLExpression'), False)

_abaa.tag_subject_of(('State', 'allowed_transition', 'BaseTransition'), False)

_abaa.tag_object_of(('SubWorkflowExitPoint', 'destination_state', 'State'),

                    False)

_abaa.tag_object_of(('State', 'state_of', 'Workflow'), True)

_abaa.tag_object_of(('Transition', 'transition_of', 'Workflow'), True)

_abaa.tag_object_of(('WorkflowTransition', 'transition_of', 'Workflow'), True)

_afs = uicfg.autoform_section

_afs.tag_subject_of(('TrInfo', 'to_state', '*'), 'main', 'hidden')

_afs.tag_subject_of(('TrInfo', 'from_state', '*'), 'main', 'hidden')

_afs.tag_object_of(('State', 'allowed_transition', '*'), 'main', 'attributes')

# IWorkflowable views #########################################################

class ChangeStateForm(forms.CompositeEntityForm):

    __regid__ = 'changestate'

    form_renderer_id = 'base' # don't want EntityFormRenderer

    form_buttons = [fwdgs.SubmitButton(),

                    fwdgs.Button(stdmsgs.BUTTON_CANCEL, cwaction='cancel')]

class ChangeStateFormView(form.FormViewMixIn, view.EntityView):

    __regid__ = 'statuschange'

    title = _('status change')

    __select__ = (one_line_rset() & implements(IWorkflowable)

                  & match_form_params('treid'))

    def cell_call(self, row, col):

        entity = self.cw_rset.get_entity(row, col)

        transition = self._cw.entity_from_eid(self._cw.form['treid'])

        form = self.get_form(entity, transition)

        self.w(u'<h4>%s %s</h4>\n' % (self._cw._(transition.name),

                                      entity.view('oneline')))

        msg = self.req._('status will change from %(st1)s to %(st2)s') % {

            'st1': entity.printable_state,

            'st2': self._cw._(transition.destination(entity).name)}

        self.w(u'<p>%s</p>\n' % msg)

        self.w(form.render())

    def redirectpath(self, entity):

        return entity.rest_path()

    def get_form(self, entity, transition, **kwargs):

        # XXX used to specify both rset/row/col and entity in case implements

        # selector (and not implements) is used on custom form

        form = self._cw.vreg['forms'].select(

            'changestate', self._cw, entity=entity, transition=transition,

            redirect_path=self.redirectpath(entity), **kwargs)

        trinfo = self._cw.vreg['etypes'].etype_class('TrInfo')(self._cw)

        trinfo.eid = self._cw.varmaker.next()

        subform = self._cw.vreg['forms'].select('edition', self._cw, entity=trinfo,

                                                mainform=False)

        subform.field_by_name('wf_info_for', 'subject').value = entity.eid

        trfield = subform.field_by_name('by_transition', 'subject')

        trfield.widget = fwdgs.HiddenInput()

        trfield.value = transition.eid

        form.add_subform(subform)

        return form

class WFHistoryView(EntityView):

    __regid__ = 'wfhistory'

    __select__ = relation_possible('wf_info_for', role='object') & \

                 score_entity(lambda x: x.workflow_history)

    title = _('Workflow history')

    def cell_call(self, row, col, view=None):

        _ = self._cw._

        eid = self.cw_rset[row][col]

        sel = 'Any FS,TS,WF,D'

        rql = ' ORDERBY D DESC WHERE WF wf_info_for X,'\

              'WF from_state FS, WF to_state TS, WF comment C,'\

              'WF creation_date D'

        if self._cw.vreg.schema.eschema('CWUser').has_perm(self._cw, 'read'):

            sel += ',U,C'

            rql += ', WF owned_by U?'

            displaycols = range(5)

            headers = (_('from_state'), _('to_state'), _('comment'), _('date'),

                       _('CWUser'))

        else:

            sel += ',C'

            displaycols = range(4)

            headers = (_('from_state'), _('to_state'), _('comment'), _('date'))

        rql = '%s %s, X eid %%(x)s' % (sel, rql)

        try:

            rset = self._cw.execute(rql, {'x': eid}, 'x')

        except Unauthorized:

            return

        if rset:

            self.wview('table', rset, title=_(self.title), displayactions=False,

                       displaycols=displaycols, headers=headers)

class WFHistoryVComponent(component.EntityVComponent):

    """display the workflow history for entities supporting it"""

    __regid__ = 'wfhistory'

    __select__ = WFHistoryView.__select__ & component.EntityVComponent.__select__

    context = 'navcontentbottom'

    title = _('Workflow history')

    def cell_call(self, row, col, view=None):

        self.wview('wfhistory', self.cw_rset, row=row, col=col, view=view)

# workflow actions #############################################################

class WorkflowActions(action.Action):

    """fill 'workflow' sub-menu of the actions box"""

    __regid__ = 'workflow'

    __select__ = (action.Action.__select__ & one_line_rset() &

                  relation_possible('in_state'))

    submenu = _('workflow')

    order = 10

    def fill_menu(self, box, menu):

        entity = self.cw_rset.get_entity(self.cw_row or 0, self.cw_col or 0)

        menu.label = u'%s: %s' % (self._cw._('state'), entity.printable_state)

        menu.append_anyway = True

        super(WorkflowActions, self).fill_menu(box, menu)

    def actual_actions(self):

        entity = self.cw_rset.get_entity(self.cw_row or 0, self.cw_col or 0)

        hastr = False

        for tr in entity.possible_transitions():

            url = entity.absolute_url(vid='statuschange', treid=tr.eid)

            yield self.build_action(self._cw._(tr.name), url)

            hastr = True

        # don't propose to see wf if user can't pass any transition

        if hastr:

            wfurl = entity.current_workflow.absolute_url()

            yield self.build_action(self._cw._('view workflow'), wfurl)

        if entity.workflow_history:

            wfurl = entity.absolute_url(vid='wfhistory')

            yield self.build_action(self._cw._('view history'), wfurl)

# workflow entity types views ##################################################

_pvs = uicfg.primaryview_section

_pvs.tag_subject_of(('Workflow', 'initial_state', '*'), 'hidden')

_pvs.tag_object_of(('*', 'state_of', 'Workflow'), 'hidden')

_pvs.tag_object_of(('*', 'transition_of', 'Workflow'), 'hidden')

_abaa = uicfg.actionbox_appearsin_addmenu

_abaa.tag_subject_of(('BaseTransition', 'condition', 'RQLExpression'), False)

_abaa.tag_subject_of(('State', 'allowed_transition', 'BaseTransition'), False)

_abaa.tag_object_of(('SubWorkflowExitPoint', 'destination_state', 'State'),

                    False)

_abaa.tag_object_of(('State', 'state_of', 'Workflow'), True)

_abaa.tag_object_of(('BaseTransition', 'transition_of', 'Workflow'), False)

_abaa.tag_object_of(('Transition', 'transition_of', 'Workflow'), True)

_abaa.tag_object_of(('WorkflowTransition', 'transition_of', 'Workflow'), True)

class WorkflowPrimaryView(primary.PrimaryView):

    __select__ = implements('Workflow')

    def render_entity_attributes(self, entity):

        self.w(entity.view('reledit', rtype='description'))

        self.w(u'<img src="%s" alt="%s"/>' % (

            xml_escape(entity.absolute_url(vid='wfgraph')),

            xml_escape(self._cw._('graphical workflow for %s') % entity.name)))

class CellView(view.EntityView):

    __regid__ = 'cell'

    __select__ = implements('TrInfo')

    def cell_call(self, row, col, cellvid=None):

        self.w(self.cw_rset.get_entity(row, col).view('reledit', rtype='comment'))

class StateInContextView(view.EntityView):

    """convenience trick, State's incontext view should not be clickable"""

    __regid__ = 'incontext'

    __select__ = implements('State')

    def cell_call(self, row, col):

        self.w(xml_escape(self._cw.view('textincontext', self.cw_rset,

                                        row=row, col=col)))

# workflow entity types edition ################################################

_afs = uicfg.autoform_section

_afs.tag_subject_of(('TrInfo', 'to_state', '*'), 'main', 'hidden')

_afs.tag_subject_of(('TrInfo', 'from_state', '*'), 'main', 'hidden')

_afs.tag_object_of(('State', 'allowed_transition', '*'), 'main', 'attributes')

_afs.tag_subject_of(('State', 'allowed_transition', '*'), 'main', 'attributes')

def workflow_items_for_relation(req, wfeid, wfrelation, targetrelation):

    wf = req.entity_from_eid(wfeid)

    rschema = req.vreg.schema[targetrelation]

    return sorted((e.view('combobox'), e.eid)

                  for e in getattr(wf, 'reverse_%s' % wfrelation)

                  if rschema.has_perm(req, 'add', toeid=e.eid))

class TransitionEditionForm(autoform.AutomaticEntityForm):

    __select__ = implements('Transition')

    def workflow_states_for_relation(self, targetrelation):

        eids = self.edited_entity.linked_to('transition_of', 'subject')

        if eids:

            return workflow_items_for_relation(self._cw, eids[0], 'state_of',

                                               targetrelation)

        return []

    def subject_destination_state_vocabulary(self, rtype, limit=None):

        if not self.edited_entity.has_eid():

            return self.workflow_states_for_relation('destination_state')

        return self.subject_relation_vocabulary(rtype, limit)

    def object_allowed_transition_vocabulary(self, rtype, limit=None):

        if not self.edited_entity.has_eid():

            return self.workflow_states_for_relation('allowed_transition')

        return self.object_relation_vocabulary(rtype, limit)

class StateEditionForm(autoform.AutomaticEntityForm):

    __select__ = implements('State')

    def subject_allowed_transition_vocabulary(self, rtype, limit=None):

        if not self.edited_entity.has_eid():

            eids = self.edited_entity.linked_to('state_of', 'subject')

            if eids:

                return workflow_items_for_relation(self._cw, eids[0], 'transition_of',

                                                   'allowed_transition')

        return []

# workflow images ##############################################################

class WorkflowDotPropsHandler(object):

    def __init__(self, req):

        self._ = req._

    def node_properties(self, stateortransition):

        """return default DOT drawing options for a state or transition"""

        props = {'label': stateortransition.printable_value('name'),

                 'fontname': 'Courier'}

        if hasattr(stateortransition, 'state_of'):

            props['shape'] = 'box'

            props['style'] = 'filled'

            if stateortransition.reverse_initial_state:

                props['color'] = '#88CC88'

        else:

            props['shape'] = 'ellipse'

            descr = []

            tr = stateortransition

            if tr.require_group:

                descr.append('%s %s'% (

                    self._('groups:'),

                    ','.join(g.printable_value('name') for g in tr.require_group)))

            if tr.condition:

                descr.append('%s %s'% (

                    self._('condition:'),

                    ' | '.join(e.expression for e in tr.condition)))

            if descr:

                props['label'] += escape('\n'.join(descr))

        return props

    def edge_properties(self, transition, fromstate, tostate):

        return {'label': '', 'dir': 'forward',

                'color': 'black', 'style': 'filled'}

class WorkflowVisitor:

    def __init__(self, entity):

        self.entity = entity

    def nodes(self):

        for state in self.entity.reverse_state_of:

            state.complete()

            yield state.eid, state

        for transition in self.entity.reverse_transition_of:

            transition.complete()

            yield transition.eid, transition

    def edges(self):

        for transition in self.entity.reverse_transition_of:

            for incomingstate in transition.reverse_allowed_transition:

                yield incomingstate.eid, transition.eid, transition

            for outgoingstate in transition.potential_destinations():

                yield transition.eid, outgoingstate.eid, transition

class WorkflowImageView(TmpFileViewMixin, view.EntityView):

    __regid__ = 'wfgraph'

    __select__ = implements('Workflow')

    content_type = 'image/png'

    def _generate(self, tmpfile):

        """display schema information for an entity"""

        entity = self.cw_rset.get_entity(self.cw_row, self.cw_col)

        visitor = WorkflowVisitor(entity)

        prophdlr = WorkflowDotPropsHandler(self._cw)

        generator = GraphGenerator(DotBackend('workflow', 'LR',

                                              ratio='compress', size='30,12'))

        return generator.generate(visitor, prophdlr, tmpfile)