generate_objc_frontend_dispatcher_implementation.py [plain text]
import logging
import string
from string import Template
from cpp_generator import CppGenerator
from generator import Generator, ucfirst
from objc_generator import ObjCGenerator
from objc_generator_templates import ObjCGeneratorTemplates as ObjCTemplates
log = logging.getLogger('global')
class ObjCFrontendDispatcherImplementationGenerator(Generator):
def __init__(self, model, input_filepath):
Generator.__init__(self, model, input_filepath)
def output_filename(self):
return '%sEventDispatchers.mm' % ObjCGenerator.OBJC_PREFIX
def domains_to_generate(self):
return filter(ObjCGenerator.should_generate_domain_event_dispatcher_filter(self.model()), Generator.domains_to_generate(self))
def generate_output(self):
secondary_headers = [
'"%sEnumConversionHelpers.h"' % ObjCGenerator.OBJC_PREFIX,
'<JavaScriptCore/InspectorFrontendChannel.h>',
'<JavaScriptCore/InspectorValues.h>',
]
header_args = {
'primaryInclude': '"%sInternal.h"' % ObjCGenerator.OBJC_PREFIX,
'secondaryIncludes': '\n'.join(['#import %s' % header for header in secondary_headers]),
}
domains = self.domains_to_generate()
sections = []
sections.append(self.generate_license())
sections.append(Template(ObjCTemplates.ImplementationPrelude).substitute(None, **header_args))
sections.extend(map(self._generate_event_dispatcher_implementations, domains))
sections.append(Template(ObjCTemplates.ImplementationPostlude).substitute(None, **header_args))
return '\n\n'.join(sections)
def _generate_event_dispatcher_implementations(self, domain):
if not domain.events:
return ''
lines = []
objc_name = '%s%sDomainEventDispatcher' % (ObjCGenerator.OBJC_PREFIX, domain.domain_name)
lines.append('@implementation %s' % objc_name)
lines.append('{')
lines.append(' AugmentableInspectorController* _controller;')
lines.append('}')
lines.append('')
lines.append('- (instancetype)initWithController:(AugmentableInspectorController*)controller;')
lines.append('{')
lines.append(' self = [super init];')
lines.append(' if (!self)')
lines.append(' return nil;')
lines.append(' ASSERT(controller);')
lines.append(' _controller = controller;')
lines.append(' return self;')
lines.append('}')
lines.append('')
for event in domain.events:
lines.append(self._generate_event(domain, event))
lines.append('')
lines.append('@end')
return '\n'.join(lines)
def _generate_event(self, domain, event):
lines = []
lines.append(self._generate_event_signature(domain, event))
lines.append('{')
lines.append(' FrontendChannel* frontendChannel = _controller->frontendChannel();')
lines.append(' if (!frontendChannel)')
lines.append(' return;')
lines.append('')
required_pointer_parameters = filter(lambda parameter: not parameter.is_optional and ObjCGenerator.is_type_objc_pointer_type(parameter.type), event.event_parameters)
for parameter in required_pointer_parameters:
var_name = ObjCGenerator.identifier_to_objc_identifier(parameter.parameter_name)
lines.append(' THROW_EXCEPTION_FOR_REQUIRED_PARAMETER(%s, @"%s");' % (var_name, var_name))
objc_array_class = ObjCGenerator.objc_class_for_array_type(parameter.type)
if objc_array_class and objc_array_class.startswith(ObjCGenerator.OBJC_PREFIX):
lines.append(' THROW_EXCEPTION_FOR_BAD_TYPE_IN_ARRAY(%s, [%s class]);' % (var_name, objc_array_class))
optional_pointer_parameters = filter(lambda parameter: parameter.is_optional and ObjCGenerator.is_type_objc_pointer_type(parameter.type), event.event_parameters)
for parameter in optional_pointer_parameters:
var_name = ObjCGenerator.identifier_to_objc_identifier(parameter.parameter_name)
lines.append(' THROW_EXCEPTION_FOR_BAD_OPTIONAL_PARAMETER(%s, @"%s");' % (var_name, var_name))
objc_array_class = ObjCGenerator.objc_class_for_array_type(parameter.type)
if objc_array_class and objc_array_class.startswith(ObjCGenerator.OBJC_PREFIX):
lines.append(' THROW_EXCEPTION_FOR_BAD_TYPE_IN_OPTIONAL_ARRAY(%s, [%s class]);' % (var_name, objc_array_class))
if required_pointer_parameters or optional_pointer_parameters:
lines.append('')
lines.append(' Ref<InspectorObject> jsonMessage = InspectorObject::create();')
lines.append(' jsonMessage->setString(ASCIILiteral("method"), ASCIILiteral("%s.%s"));' % (domain.domain_name, event.event_name))
if event.event_parameters:
lines.extend(self._generate_event_out_parameters(domain, event))
lines.append(' frontendChannel->sendMessageToFrontend(jsonMessage->toJSONString());')
lines.append('}')
return '\n'.join(lines)
def _generate_event_signature(self, domain, event):
if not event.event_parameters:
return '- (void)%s' % event.event_name
pairs = []
for parameter in event.event_parameters:
param_name = parameter.parameter_name
pairs.append('%s:(%s)%s' % (param_name, ObjCGenerator.objc_type_for_param(domain, event.event_name, parameter), param_name))
pairs[0] = ucfirst(pairs[0])
return '- (void)%sWith%s' % (event.event_name, ' '.join(pairs))
def _generate_event_out_parameters(self, domain, event):
lines = []
lines.append(' Ref<InspectorObject> paramsObject = InspectorObject::create();')
for parameter in event.event_parameters:
keyed_set_method = CppGenerator.cpp_setter_method_for_type(parameter.type)
var_name = parameter.parameter_name
safe_var_name = '(*%s)' % var_name if parameter.is_optional else var_name
export_expression = ObjCGenerator.objc_protocol_export_expression_for_variable(parameter.type, safe_var_name)
if not parameter.is_optional:
lines.append(' paramsObject->%s(ASCIILiteral("%s"), %s);' % (keyed_set_method, parameter.parameter_name, export_expression))
else:
lines.append(' if (%s)' % (parameter.parameter_name))
lines.append(' paramsObject->%s(ASCIILiteral("%s"), %s);' % (keyed_set_method, parameter.parameter_name, export_expression))
lines.append(' jsonMessage->setObject(ASCIILiteral("params"), WTF::move(paramsObject));')
return lines