| # Copyright 2019 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import json5_generator |
| import template_expander |
| from collections import defaultdict |
| from make_runtime_features_utilities import origin_trials |
| |
| |
| class FeaturePolicyFeatureWriter(json5_generator.Writer): |
| file_basename = 'policy_helper' |
| |
| def __init__(self, json5_file_path, output_dir): |
| super(FeaturePolicyFeatureWriter, self).__init__( |
| json5_file_path, output_dir) |
| runtime_features = [] |
| feature_policy_features = [] |
| # Note: there can be feature with same 'name' attribute in |
| # document_policy_features and in feature_policy_features. |
| # They are supposed to have the same 'depends_on' attribute. |
| # However, their feature_policy_name and document_policy_name |
| # might be different. |
| document_policy_features = [] |
| |
| def to_devtools_enum_format(feature_policy_name): |
| """ Convert '-' separated feature_policy_name to cammel case devtool enum name """ |
| return ''.join( |
| [name.capitalize() for name in feature_policy_name.split('-')]) |
| |
| for feature in self.json5_file.name_dictionaries: |
| if feature['feature_policy_name']: |
| feature['devtools_enum_name'] = to_devtools_enum_format( |
| feature['feature_policy_name']) |
| feature_policy_features.append(feature) |
| elif feature['document_policy_name']: |
| document_policy_features.append(feature) |
| else: |
| runtime_features.append(feature) |
| |
| origin_trials_set = origin_trials(runtime_features) |
| fp_origin_trial_dependency_map = defaultdict(list) |
| dp_origin_trial_dependency_map = defaultdict(list) |
| runtime_to_feature_policy_map = defaultdict(list) |
| runtime_to_document_policy_map = defaultdict(list) |
| for feature in feature_policy_features + document_policy_features: |
| for dependency in feature['depends_on']: |
| if str(dependency) in origin_trials_set: |
| if feature['feature_policy_name']: |
| fp_origin_trial_dependency_map[feature['name']].append( |
| dependency) |
| else: |
| dp_origin_trial_dependency_map[feature['name']].append( |
| dependency) |
| else: |
| if feature['feature_policy_name']: |
| runtime_to_feature_policy_map[dependency].append( |
| feature['name']) |
| else: |
| runtime_to_document_policy_map[dependency].append( |
| feature['name']) |
| |
| self._outputs = { |
| self.file_basename + '.cc': |
| template_expander.use_jinja('templates/' + |
| self.file_basename + '.cc.tmpl')(lambda: { |
| 'header_guard': self.make_header_guard( |
| self._relative_output_dir + |
| self.file_basename + '.h'), |
| 'input_files': self._input_files, |
| 'feature_policy_features': feature_policy_features, |
| 'document_policy_features': document_policy_features, |
| 'fp_origin_trial_dependency_map': |
| fp_origin_trial_dependency_map, |
| 'dp_origin_trial_dependency_map': |
| dp_origin_trial_dependency_map, |
| 'runtime_to_feature_policy_map': |
| runtime_to_feature_policy_map, |
| 'runtime_to_document_policy_map': |
| runtime_to_document_policy_map |
| }), |
| } |
| |
| |
| if __name__ == '__main__': |
| json5_generator.Maker(FeaturePolicyFeatureWriter).main() |
