import itertools
from collections import Counter, defaultdict
BUILTIN_ATTRIBUTE = "Builtin"
ASYNC_ATTRIBUTE = "Async"
SYNCHRONOUS_ATTRIBUTE = 'Synchronous'
class MessageReceiver(object):
def __init__(self, name, superclass, attributes, messages, condition):
self.name = name
self.superclass = superclass
self.attributes = frozenset(attributes or [])
self.messages = messages
self.condition = condition
def iterparameters(self):
return itertools.chain((parameter for message in self.messages for parameter in message.parameters),
(reply_parameter for message in self.messages if message.reply_parameters for reply_parameter in message.reply_parameters))
def has_attribute(self, attribute):
return attribute in self.attributes
class Message(object):
def __init__(self, name, parameters, reply_parameters, attributes, condition):
self.name = name
self.parameters = parameters
self.reply_parameters = reply_parameters
self.attributes = frozenset(attributes or [])
self.condition = condition
def has_attribute(self, attribute):
return attribute in self.attributes
class Parameter(object):
def __init__(self, kind, type, name, attributes=None, condition=None):
self.kind = kind
self.type = type
self.name = name
self.attributes = frozenset(attributes or [])
self.condition = condition
def has_attribute(self, attribute):
return attribute in self.attributes
ipc_receiver = MessageReceiver(name="IPC", superclass=None, attributes=[BUILTIN_ATTRIBUTE], messages=[
Message('WrappedAsyncMessageForTesting', [], [], attributes=[BUILTIN_ATTRIBUTE, SYNCHRONOUS_ATTRIBUTE], condition=None),
Message('SyncMessageReply', [], [], attributes=[BUILTIN_ATTRIBUTE], condition=None),
Message('InitializeConnection', [], [], attributes=[BUILTIN_ATTRIBUTE], condition="PLATFORM(COCOA)"),
Message('LegacySessionState', [], [], attributes=[BUILTIN_ATTRIBUTE], condition=None)
], condition=None)
def check_global_model_inputs(receivers):
errors = []
receiver_counts = Counter([r.name for r in receivers])
receiver_duplicates = [n for n, c in receiver_counts.items() if c > 1]
if receiver_duplicates:
errors.append('Duplicate message receiver names: %s' % (', '.join(receiver_duplicates)))
matching_attributes = [SYNCHRONOUS_ATTRIBUTE]
for receiver in receivers:
receiver_messages = defaultdict(list)
for message in receiver.messages:
receiver_messages[message.name].append(message)
for messages in receiver_messages.values():
m0 = messages[0]
for i in range(1, len(messages)):
mi = messages[i]
if any(m0.has_attribute(a) != mi.has_attribute(a) for a in matching_attributes):
errors.append('Receiver %s message %s attribute mismatch: %s (%s) != %s (%s))' % (receiver.name, message.name,
m0.attributes, m0.condition, mi.attributes, mi.condition))
return errors
def generate_global_model(receivers):
async_reply_messages = []
for receiver in receivers:
for message in receiver.messages:
if message.has_attribute(ASYNC_ATTRIBUTE):
async_reply_messages.append(Message(name='%s_%sReply' % (receiver.name, message.name), parameters=message.reply_parameters, reply_parameters=[], attributes=None, condition=message.condition))
async_reply_receiver = MessageReceiver(name='AsyncReply', superclass='None', attributes=[BUILTIN_ATTRIBUTE], messages=async_reply_messages, condition=None)
return [ipc_receiver, async_reply_receiver] + receivers