generate_cpp_frontend_dispatcher_implementation.py [plain text]
import logging
import string
from string import Template
try:
from .cpp_generator import CppGenerator
from .cpp_generator_templates import CppGeneratorTemplates as CppTemplates
from .generator import Generator, ucfirst
from .models import ObjectType, ArrayType, AliasedType, EnumType
except:
from cpp_generator import CppGenerator
from cpp_generator_templates import CppGeneratorTemplates as CppTemplates
from generator import Generator, ucfirst
from models import ObjectType, ArrayType, AliasedType, EnumType
log = logging.getLogger('global')
class CppFrontendDispatcherImplementationGenerator(CppGenerator):
def __init__(self, *args, **kwargs):
CppGenerator.__init__(self, *args, **kwargs)
def output_filename(self):
return "%sFrontendDispatchers.cpp" % self.protocol_name()
def domains_to_generate(self):
return [domain for domain in Generator.domains_to_generate(self) if len(self.events_for_domain(domain)) > 0]
def generate_output(self):
header_args = {
'primaryInclude': '"%sFrontendDispatchers.h"' % self.protocol_name(),
'secondaryIncludes': self._generate_secondary_header_includes(),
}
sections = []
sections.append(self.generate_license())
sections.append(Template(CppTemplates.ImplementationPrelude).substitute(None, **header_args))
sections.extend(list(map(self._generate_dispatcher_implementations_for_domain, self.domains_to_generate())))
sections.append(Template(CppTemplates.ImplementationPostlude).substitute(None, **header_args))
return "\n\n".join(sections)
def _generate_secondary_header_includes(self):
header_includes = [
(["JavaScriptCore", "WebKit"], ("JavaScriptCore", "inspector/InspectorFrontendRouter.h")),
]
return '\n'.join(self.generate_includes_from_entries(header_includes))
def _generate_dispatcher_implementations_for_domain(self, domain):
implementations = []
events = self.events_for_domain(domain)
for event in events:
implementations.append(self._generate_dispatcher_implementation_for_event(event, domain))
return self.wrap_with_guard_for_condition(domain.condition, '\n\n'.join(implementations))
def _generate_dispatcher_implementation_for_event(self, event, domain):
lines = []
parameter_assignments = []
formal_parameters = []
for parameter in event.event_parameters:
parameter_name = parameter.parameter_name
if parameter.is_optional:
parameter_name = 'opt_' + parameter_name
parameter_value = parameter_name
_type = parameter.type
if isinstance(_type, AliasedType):
_type = _type.aliased_type
if isinstance(_type, EnumType) and _type.is_anonymous:
_type = _type.primitive_type
if _type.is_enum():
if parameter.is_optional:
parameter_value = '*' + parameter_value
parameter_value = 'Protocol::%s::getEnumConstantValue(%s)' % (self.helpers_namespace(), parameter_value)
elif CppGenerator.should_release_argument(_type, parameter.is_optional):
parameter_value = parameter_value + '.releaseNonNull()'
elif CppGenerator.should_dereference_argument(_type, parameter.is_optional):
parameter_value = '*' + parameter_value
elif CppGenerator.should_move_argument(_type, parameter.is_optional):
parameter_value = 'WTFMove(%s)' % parameter_value
parameter_args = {
'parameterKey': parameter.parameter_name,
'parameterName': parameter_name,
'parameterValue': parameter_value,
'keyedSetMethod': CppGenerator.cpp_setter_method_for_type(_type),
}
if parameter.is_optional:
parameter_assignments.append(' if (!!%(parameterName)s)' % parameter_args)
parameter_assignments.append(' protocol_paramsObject->%(keyedSetMethod)s("%(parameterKey)s"_s, %(parameterValue)s);' % parameter_args)
else:
parameter_assignments.append(' protocol_paramsObject->%(keyedSetMethod)s("%(parameterKey)s"_s, %(parameterValue)s);' % parameter_args)
formal_parameters.append('%s %s' % (CppGenerator.cpp_type_for_event_parameter(_type, parameter.is_optional), parameter_name))
event_args = {
'domainName': domain.domain_name,
'eventName': event.event_name,
'formalParameters': ", ".join(formal_parameters)
}
lines.append('void %(domainName)sFrontendDispatcher::%(eventName)s(%(formalParameters)s)' % event_args)
lines.append('{')
lines.append(' auto protocol_jsonMessage = JSON::Object::create();')
lines.append(' protocol_jsonMessage->setString("method"_s, "%(domainName)s.%(eventName)s"_s);' % event_args)
if len(parameter_assignments) > 0:
lines.append(' auto protocol_paramsObject = JSON::Object::create();')
lines.extend(parameter_assignments)
lines.append(' protocol_jsonMessage->setObject("params"_s, WTFMove(protocol_paramsObject));')
lines.append('')
lines.append(' m_frontendRouter.sendEvent(protocol_jsonMessage->toJSONString());')
lines.append('}')
return self.wrap_with_guard_for_condition(event.condition, "\n".join(lines))