You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
3 years ago
|
#!/usr/bin/env python3
|
||
|
|
# Copyright 2021 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.
|
||
|
|
"""Redirects to the version of rustfmt present in the Chrome tree.
|
||
|
|
|
||
|
|
Rust binaries are pulled down from Google Cloud Storage whenever you sync
|
||
|
|
Chrome. This script knows how to locate those tools, assuming the script is
|
||
|
|
invoked from inside a Chromium checkout."""
|
||
|
|
|
||
|
|
import gclient_paths
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
class NotFoundError(Exception):
|
||
|
|
"""A file could not be found."""
|
||
|
|
|
||
|
|
def __init__(self, e):
|
||
|
|
Exception.__init__(
|
||
|
|
self, 'Problem while looking for rustfmt in Chromium source tree:\n'
|
||
|
|
'%s' % e)
|
||
|
|
|
||
|
|
|
||
|
|
def FindRustfmtToolInChromiumTree():
|
||
|
|
"""Return a path to the rustfmt executable, or die trying."""
|
||
|
|
chromium_src_path = gclient_paths.GetPrimarySolutionPath()
|
||
|
|
if not chromium_src_path:
|
||
|
|
raise NotFoundError(
|
||
|
|
'Could not find checkout in any parent of the current path.\n'
|
||
|
|
'Set CHROMIUM_BUILDTOOLS_PATH to use outside of a chromium checkout.')
|
||
|
|
|
||
|
|
# TODO(lukasza): Deduplicate the is-Rust-supported and find-Rust-binaries code
|
||
|
|
# by somehow sharing the `rust_prefix` variable from //build/config/rust.gni
|
||
|
|
tool_path = os.path.join(chromium_src_path, 'third_party',
|
||
|
|
'android_rust_toolchain', 'toolchain', 'bin',
|
||
|
|
'rustfmt' + gclient_paths.GetExeSuffix())
|
||
|
|
if not os.path.exists(tool_path):
|
||
|
|
raise NotFoundError('File does not exist: %s' % tool_path)
|
||
|
|
return tool_path
|
||
|
|
|
||
|
|
|
||
|
|
def IsRustfmtSupported():
|
||
|
|
try:
|
||
|
|
FindRustfmtToolInChromiumTree()
|
||
|
|
return True
|
||
|
|
except NotFoundError:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def main(args):
|
||
|
|
try:
|
||
|
|
tool = FindRustfmtToolInChromiumTree()
|
||
|
|
except NotFoundError as e:
|
||
|
|
sys.stderr.write("%s\n" % str(e))
|
||
|
|
return 1
|
||
|
|
|
||
|
|
# Add some visibility to --help showing where the tool lives, since this
|
||
|
|
# redirection can be a little opaque.
|
||
|
|
help_syntax = ('-h', '--help', '-help', '-help-list', '--help-list')
|
||
|
|
if any(match in args for match in help_syntax):
|
||
|
|
print('\nDepot tools redirects you to the rustfmt at:\n %s\n' % tool)
|
||
|
|
|
||
|
|
return subprocess.call([tool] + args)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
try:
|
||
|
|
sys.exit(main(sys.argv[1:]))
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
sys.stderr.write('interrupted\n')
|
||
|
|
sys.exit(1)
|