generate_objc_configuration_implementation.py [plain text]
import logging
import string
from string import Template
from generator import Generator
from objc_generator import ObjCGenerator
from objc_generator_templates import ObjCGeneratorTemplates as ObjCTemplates
log = logging.getLogger('global')
class ObjCConfigurationImplementationGenerator(ObjCGenerator):
def __init__(self, *args, **kwargs):
ObjCGenerator.__init__(self, *args, **kwargs)
def output_filename(self):
return '%sConfiguration.mm' % self.protocol_name()
def generate_output(self):
secondary_headers = [
'"%sInternal.h"' % self.protocol_name(),
'"%sBackendDispatchers.h"' % self.protocol_name(),
'<JavaScriptCore/AlternateDispatchableAgent.h>',
'<JavaScriptCore/AugmentableInspectorController.h>',
'<JavaScriptCore/InspectorAlternateBackendDispatchers.h>',
'<JavaScriptCore/InspectorBackendDispatchers.h>',
]
header_args = {
'primaryInclude': '"%sConfiguration.h"' % self.protocol_name(),
'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.append(self._generate_configuration_implementation_for_domains(domains))
sections.append(Template(ObjCTemplates.ImplementationPostlude).substitute(None, **header_args))
return '\n\n'.join(sections)
def _generate_configuration_implementation_for_domains(self, domains):
lines = []
lines.append('@implementation %sConfiguration' % self.protocol_name())
lines.append('{')
lines.append(' AugmentableInspectorController* _controller;')
lines.extend(self._generate_ivars(domains))
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('')
lines.extend(self._generate_dealloc(domains))
lines.append('')
for domain in domains:
if self.should_generate_commands_for_domain(domain):
lines.append(self._generate_handler_setter_for_domain(domain))
lines.append('')
if self.should_generate_events_for_domain(domain):
lines.append(self._generate_event_dispatcher_getter_for_domain(domain))
lines.append('')
lines.append('@end')
return '\n'.join(lines)
def _generate_ivars(self, domains):
lines = []
for domain in domains:
if self.should_generate_commands_for_domain(domain):
objc_class_name = '%s%sDomainHandler' % (self.objc_prefix(), domain.domain_name)
ivar_name = '_%sHandler' % ObjCGenerator.variable_name_prefix_for_domain(domain)
lines.append(' id<%s> %s;' % (objc_class_name, ivar_name))
if self.should_generate_events_for_domain(domain):
objc_class_name = '%s%sDomainEventDispatcher' % (self.objc_prefix(), domain.domain_name)
ivar_name = '_%sEventDispatcher' % ObjCGenerator.variable_name_prefix_for_domain(domain)
lines.append(' %s *%s;' % (objc_class_name, ivar_name))
return lines
def _generate_dealloc(self, domains):
lines = []
lines.append('- (void)dealloc')
lines.append('{')
for domain in domains:
if self.should_generate_commands_for_domain(domain):
lines.append(' [_%sHandler release];' % ObjCGenerator.variable_name_prefix_for_domain(domain))
if self.should_generate_events_for_domain(domain):
lines.append(' [_%sEventDispatcher release];' % ObjCGenerator.variable_name_prefix_for_domain(domain))
lines.append(' [super dealloc];')
lines.append('}')
return lines
def _generate_handler_setter_for_domain(self, domain):
setter_args = {
'objcPrefix': self.objc_prefix(),
'domainName': domain.domain_name,
'variableNamePrefix': ObjCGenerator.variable_name_prefix_for_domain(domain),
}
return Template(ObjCTemplates.ConfigurationCommandPropertyImplementation).substitute(None, **setter_args)
def _generate_event_dispatcher_getter_for_domain(self, domain):
getter_args = {
'objcPrefix': self.objc_prefix(),
'domainName': domain.domain_name,
'variableNamePrefix': ObjCGenerator.variable_name_prefix_for_domain(domain),
}
return Template(ObjCTemplates.ConfigurationGetterImplementation).substitute(None, **getter_args)
def _variable_name_prefix_for_domain(self, domain):
domain_name = domain.domain_name
if domain_name.startswith('DOM'):
return 'dom' + domain_name[3:]
if domain_name.startswith('CSS'):
return 'css' + domain_name[3:]
return domain_name[:1].lower() + domain_name[1:]