# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from codegen_util import FileInfo, Util
from code_generator import (EventTemplateBase, ProjectInfoBase, EventInfoBase,
MetricInfoBase)
class ProjectInfoTs(ProjectInfoBase):
"""Codegen-related info about a project in Typescript."""
def __init__(self, project):
super().__init__(project, "webui")
class EventInfoTs(EventInfoBase):
"""Codegen-related info about an event in Typescript."""
def __init__(self, event, project_info):
super().__init__(event, project_info)
if self.is_event_sequence == 'true':
self.systemUptime = (
'{microseconds: BigInt(Math.floor(Date.now() * 1000))}')
else:
self.systemUptime = 'null'
class MetricInfoTs(MetricInfoBase):
"""Codegen-related info about a metric in Typescript."""
def __init__(self, metric, project_info, event_info):
super().__init__(metric)
self.project_name = project_info.name
self.event_name = event_info.name
if metric.type == 'hmac-string':
self.ts_type = 'number'
self.type_enum = 'hmacValue'
elif metric.type == 'int':
self.ts_type = 'bigint'
self.type_enum = 'longValue'
elif metric.type == 'raw-string':
self.ts_type = 'string'
self.type_enum = 'rawStrValue'
elif metric.type == 'double':
self.ts_type = 'number'
self.type_enum = 'doubleValue'
elif metric.type == 'int-array':
# todo(b/341807121): support int array in Typescript.
self.type_enum = ''
else:
if self.is_enum:
self.ts_type = project_info.name + "_" + metric.type
self.type_enum = 'intValue'
else:
raise ValueError('Invalid metric type.')
class TemplateTypescript(EventTemplateBase):
"""Template for producing Typescript code from structured.xml, to
be used in WebUI pages."""
def __init__(self, model, dirname, basename, file_template, project_template,
enum_template, event_template, metric_template,
metric_field_template, metric_build_code_template):
super().__init__(model, dirname, basename, file_template, project_template,
enum_template, event_template, metric_template)
self.metric_field_template = metric_field_template
self.metric_build_code_template = metric_build_code_template
def write_file(self):
file_info = FileInfo(self.dirname, self.basename)
with open(file_info.filepath, 'w') as f:
f.write(self._stamp_file())
def _stamp_file(self):
project_code = "".join(
self._stamp_project(project) for project in self.model.projects)
return self.file_template.format(project_code=project_code)
def _stamp_project(self, project):
project_info = ProjectInfoTs(project)
if project_info.should_codegen:
event_code = ''.join(
self._stamp_event(project_info, event) for event in project.events)
enum_code = '\n\n'.join(
self._stamp_enum(project_info, enum) for enum in project.enums)
return self.project_template.format(project=project,
enum_code=enum_code,
event_code=event_code)
return ""
def _stamp_event(self, project_info, event):
event_info = EventInfoTs(event, project_info)
metric_code = ''.join(
self._stamp_metric(project_info, event_info, metric)
for metric in event.metrics)
metric_fields = ''.join(
self._stamp_metric_fields(project_info, event_info, metric)
for metric in event.metrics)
metric_build_code = ''.join(
self._stamp_metric_build_code(project_info, event_info, metric)
for metric in event.metrics)
return self.event_template.format(event=event_info,
metric_fields=metric_fields,
metric_code=metric_code,
metric_build_code=metric_build_code)
def _stamp_metric(self, project_info, event_info, metric):
metric_info = MetricInfoTs(metric, project_info, event_info)
return self.metric_template.format(metric=metric_info)
def _stamp_metric_fields(self, project_info, event_info, metric):
metric_info = MetricInfoTs(metric, project_info, event_info)
return self.metric_field_template.format(metric=metric_info)
def _stamp_metric_build_code(self, project_info, event_info, metric):
metric_info = MetricInfoTs(metric, project_info, event_info)
return self.metric_build_code_template.format(metric=metric_info)
def _stamp_enum(self, project_info, enum):
variants = ',\n'.join(
['{v.name} = {v.value}'.format(v=v) for v in enum.variants])
return self.enum_template.format(project_info=project_info,
enum=enum,
variants=variants)
Codebase Browser
chromium