mirror of
https://chromium.googlesource.com/chromium/tools/depot_tools.git
synced 2026-01-11 02:31:29 +00:00
3912091152
R=apolito@google.com, dpranke@google.com Bug: 1199102 Change-Id: I06ff344ac91464eb558e92598c87e43ecfa830dc Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/2994488 Auto-Submit: Josip Sokcevic <sokcevic@google.com> Commit-Queue: Dirk Pranke <dpranke@google.com> Reviewed-by: Anthony Polito <apolito@google.com> Reviewed-by: Dirk Pranke <dpranke@google.com>
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright (c) 2018 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 argparse
|
|
import io
|
|
import os
|
|
import re
|
|
import shutil
|
|
import sys
|
|
|
|
|
|
CIPD_SUBDIR_RE = '@Subdir (.*)'
|
|
|
|
|
|
def parse_cipd(root, contents):
|
|
tree = {}
|
|
current_subdir = None
|
|
for line in contents:
|
|
line = line.strip()
|
|
match = re.match(CIPD_SUBDIR_RE, line)
|
|
if match:
|
|
print('match')
|
|
current_subdir = os.path.join(root, *match.group(1).split('/'))
|
|
elif line and current_subdir:
|
|
print('no match')
|
|
tree.setdefault(current_subdir, []).append(line)
|
|
return tree
|
|
|
|
|
|
def main():
|
|
assert sys.argv[1] == 'ensure'
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-ensure-file')
|
|
parser.add_argument('-root')
|
|
args, _ = parser.parse_known_args()
|
|
|
|
with io.open(args.ensure_file, 'r', encoding='utf-8') as f:
|
|
new_content = parse_cipd(args.root, f.readlines())
|
|
|
|
# Install new packages
|
|
for path, packages in new_content.items():
|
|
if not os.path.exists(path):
|
|
os.makedirs(path)
|
|
with io.open(os.path.join(path, '_cipd'), 'w', encoding='utf-8') as f:
|
|
f.write('\n'.join(packages))
|
|
|
|
# Save the ensure file that we got
|
|
shutil.copy(args.ensure_file, os.path.join(args.root, '_cipd'))
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|