You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.7 KiB
87 lines
2.7 KiB
4 months ago
|
#!/usr/bin/env python2.7
|
||
|
|
# Copyright 2015 gRPC authors.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
# produces cleaner build.yaml files
|
||
|
|
|
||
|
|
import collections
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
TEST = (os.environ.get('TEST', 'false') == 'true')
|
||
|
|
|
||
|
|
_TOP_LEVEL_KEYS = [
|
||
|
|
'settings', 'proto_deps', 'filegroups', 'libs', 'targets', 'vspackages'
|
||
|
|
]
|
||
|
|
_ELEM_KEYS = [
|
||
|
|
'name', 'gtest', 'cpu_cost', 'flaky', 'build', 'run', 'language',
|
||
|
|
'public_headers', 'headers', 'src', 'deps'
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def repr_ordered_dict(dumper, odict):
|
||
|
|
return dumper.represent_mapping(u'tag:yaml.org,2002:map', odict.items())
|
||
|
|
|
||
|
|
|
||
|
|
yaml.add_representer(collections.OrderedDict, repr_ordered_dict)
|
||
|
|
|
||
|
|
|
||
|
|
def rebuild_as_ordered_dict(indict, special_keys):
|
||
|
|
outdict = collections.OrderedDict()
|
||
|
|
for key in sorted(indict.keys()):
|
||
|
|
if '#' in key:
|
||
|
|
outdict[key] = indict[key]
|
||
|
|
for key in special_keys:
|
||
|
|
if key in indict:
|
||
|
|
outdict[key] = indict[key]
|
||
|
|
for key in sorted(indict.keys()):
|
||
|
|
if key in special_keys: continue
|
||
|
|
if '#' in key: continue
|
||
|
|
outdict[key] = indict[key]
|
||
|
|
return outdict
|
||
|
|
|
||
|
|
|
||
|
|
def clean_elem(indict):
|
||
|
|
for name in ['public_headers', 'headers', 'src']:
|
||
|
|
if name not in indict: continue
|
||
|
|
inlist = indict[name]
|
||
|
|
protos = list(x for x in inlist if os.path.splitext(x)[1] == '.proto')
|
||
|
|
others = set(x for x in inlist if x not in protos)
|
||
|
|
indict[name] = protos + sorted(others)
|
||
|
|
return rebuild_as_ordered_dict(indict, _ELEM_KEYS)
|
||
|
|
|
||
|
|
|
||
|
|
for filename in sys.argv[1:]:
|
||
|
|
with open(filename) as f:
|
||
|
|
js = yaml.load(f)
|
||
|
|
js = rebuild_as_ordered_dict(js, _TOP_LEVEL_KEYS)
|
||
|
|
for grp in ['filegroups', 'libs', 'targets']:
|
||
|
|
if grp not in js: continue
|
||
|
|
js[grp] = sorted(
|
||
|
|
[clean_elem(x) for x in js[grp]],
|
||
|
|
key=lambda x: (x.get('language', '_'), x['name']))
|
||
|
|
output = yaml.dump(js, indent=2, width=80, default_flow_style=False)
|
||
|
|
# massage out trailing whitespace
|
||
|
|
lines = []
|
||
|
|
for line in output.splitlines():
|
||
|
|
lines.append(line.rstrip() + '\n')
|
||
|
|
output = ''.join(lines)
|
||
|
|
if TEST:
|
||
|
|
with open(filename) as f:
|
||
|
|
assert f.read() == output
|
||
|
|
else:
|
||
|
|
with open(filename, 'w') as f:
|
||
|
|
f.write(output)
|