##############################################################################
#
# Copyright (c) 2002 Nexedi SARL and Contributors. All Rights Reserved.
#                    Jean-Paul Smets-Solanes <jp@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# 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; either version 2
# 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 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

from Products.CMFActivity.ActivityTool import registerActivity, MESSAGE_EXECUTED
from Products.CMFActivity.Errors import ActivityFlushError
from Queue import Queue, VALID

from zLOG import LOG

try:
  from transaction import get as get_transaction
except ImportError:
  pass

class RAMDict(Queue):
  """
    A simple RAM based queue. It is not compatible with transactions which
    means methods can be called before an object even exists or before
    it is modified. This also means there is no garantee on any kind of sequenciality.

    Dictionnary is global.
  """

  def __init__(self):
    Queue.__init__(self)
    self.queue_dict = {}

  def getDict(self, activity_tool_path):
    return self.queue_dict.setdefault(activity_tool_path, {})

  def finishQueueMessage(self, activity_tool_path, m):
    if m.is_registered:
      self.getDict(activity_tool_path)[(tuple(m.object_path), m.method_id)] = m

  def finishDeleteMessage(self, activity_tool_path, message):
    for key, m in self.getDict(activity_tool_path).items():
      if m.object_path == message.object_path and m.method_id == message.method_id:
        del self.getDict(activity_tool_path)[(tuple(m.object_path), m.method_id)]

  def registerActivityBuffer(self, activity_buffer):
    pass

  def generateMessageUID(self, m):
    return (tuple(m.object_path), m.method_id)

  def isMessageRegistered(self, activity_buffer, activity_tool, m):
    uid_set = activity_buffer.getUidSet(self)
    return self.generateMessageUID(m) in uid_set

  def registerMessage(self, activity_buffer, activity_tool, m):
    message_list = activity_buffer.getMessageList(self)
    message_list.append(m)
    uid_set = activity_buffer.getUidSet(self)
    uid_set.add(self.generateMessageUID(m))
    m.is_registered = 1

  def dequeueMessage(self, activity_tool, processing_node):
    path = activity_tool.getPhysicalPath()
    if len(self.getDict(path).keys()) is 0:
      return 1  # Go to sleep
    for key, m in self.getDict(path).items():
      if m.validate(self, activity_tool) is VALID:
        activity_tool.invoke(m)
        if m.getExecutionState() == MESSAGE_EXECUTED:
          del self.getDict(path)[key]
          get_transaction().commit()
          return 0
        else:
          # Start a new transaction and keep on to next message
          get_transaction().commit()
    return 1

  def countMessage(self, activity_tool,path=None,method_id=None,**kw):
    tool_path = activity_tool.getPhysicalPath()
    count = 0
    for (key,m) in self.getDict(tool_path).items():
      add = 1
      if path is not None:
        object_path = '/'.join(m.object_path)
        if object_path != path:
          add = 0
      if method_id is not None:
        if m.method_id != method_id:
          add = 0
      count += add
    return count

  def hasActivity(self, activity_tool, object, **kw):
    if object is not None:
      object_path = object.getPhysicalPath()
    else:
      object_path = None
    active_process = kw.get('active_process', None)
    path = activity_tool.getPhysicalPath()
    for m in self.getDict(path).values():
      # Filter active process and path if defined
      if active_process is None or m.active_process == active_process:
        if object_path is None or m.object_path == object_path:
          return 1
    return 0

  def flush(self, activity_tool, object_path, invoke=0, method_id=None, **kw):
    path = '/'.join(object_path)
    # LOG('Flush', 0, str((path, invoke, method_id)))
    method_dict = {}
    # Parse each message in registered
    for m in activity_tool.getRegisteredMessageList(self):
      if object_path == m.object_path and (method_id is None or method_id == m.method_id):
        if not method_dict.has_key(m.method_id):
          if invoke:
            # First Validate
            if m.validate(self, activity_tool) is VALID:
              activity_tool.invoke(m) # Try to invoke the message - what happens if invoke calls flushActivity ??
              if m.getExecutionState() != MESSAGE_EXECUTED:                                                 # Make sure message could be invoked
                # The message no longer exists
                raise ActivityFlushError, (
                    'Could not evaluate %s on %s' % (method_id , path))
              else:
                method_dict[m.method_id] = 1
                activity_tool.unregisterMessage(self, m) 
            else:
              # The message no longer exists
              raise ActivityFlushError, (
                  'The document %s does not exist' % path) 
          else:
            method_dict[m.method_id] = 1
            activity_tool.unregisterMessage(self, m)
        else:
          method_dict[m.method_id] = 1
          activity_tool.unregisterMessage(self, m)
    # Parse each message in RAM dict
    path = activity_tool.getPhysicalPath()
    for key, m in self.getDict(path).items():
      if object_path == m.object_path and (method_id is None or method_id == m.method_id):
        if not method_dict.has_key(m.method_id):
          LOG('CMFActivity RAMDict: ', 0, 'flushing object %s' % '/'.join(m.object_path))
          if invoke:
            activity_tool.invoke(m)
            if m.getExecutionState() == MESSAGE_EXECUTED:
              method_dict[m.method_id] = 1
              self.deleteMessage(activity_tool, m)
          else:
            method_dict[m.method_id] = 1
            self.deleteMessage(activity_tool, m)
        else:
          self.deleteMessage(activity_tool, m)

  def getMessageList(self, activity_tool, processing_node=None,**kw):
    new_queue = []
    path = activity_tool.getPhysicalPath()
    for m in self.getDict(path).values():
      m.processing_node = 1
      m.priority = 0
      new_queue.append(m)
    return new_queue

registerActivity(RAMDict)