blob: 0eef849bf4e5bacc1e8113b515dda7ce4fb9d2e3 [file] [log] [blame]
#!/usr/bin/env python
# Copyright 2018 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
### runs rustfmt on a Rust target
import argparse
import os
import platform
import subprocess
import sys
import rust
from rust import ROOT_PATH
sys.path += [os.path.join(ROOT_PATH, "third_party", "pytoml")]
import pytoml as toml
def main():
parser = argparse.ArgumentParser("Format a rust target")
help="GN target to format. \
Use '.[:target]' to discover the cargo target \
for the current directory or use the \
absolute path to the target \
(relative to $FUCHSIA_DIR). \
For example: //garnet/bin/foo/bar:baz.")
parser.add_argument("-v", "--verbose",
help="Show verbose output")
parser.add_argument("-s", "--print-sources",
help="Only print sources; do not format")
args = parser.parse_args()
if args.print_sources and not all(os.path.exists(gn_target.manifest_path()) for gn_target in args.gn_targets):
return 0
cargo_tomls = [] # List of (gn_target, cargo_toml)
for gn_target in args.gn_targets:
with open(gn_target.manifest_path(), "r") as fin:
cargo_tomls.append((gn_target, toml.load(fin)))
main_files = [] # List of (gn_target, main_file)
for gn_target, cargo_toml in cargo_tomls:
if 'bin' in cargo_toml:
bins = cargo_toml['bin']
if len(bins) != 1:
print("Expected a single bin target for {gn_target}, found {n}".format(
gn_target = gn_target,
n = len(bins)))
return 1
main_files.append((gn_target, bins[0]['path']))
elif 'lib' in cargo_toml:
main_files.append((gn_target, cargo_toml['lib']['path']))
if args.print_sources:
if main_files:
print('\n'.join(main_file[1] for main_file in main_files))
return 0
for gn_target, main_file in main_files:
if not main_file or not os.path.exists(main_file):
print("No source root (typically or found for {gn_target}".format(
return 1
host_platform = "%s-%s" % (
platform.system().lower().replace("darwin", "mac"),
"x86_64": "x64",
"aarch64": "arm64",
buildtools_dir = os.path.join(ROOT_PATH, "buildtools", host_platform)
rustfmt = os.path.join(buildtools_dir, "rust", "bin", "rustfmt")
call_args = [
] + [main_file[1] for main_file in main_files]
if args.verbose:
if __name__ == '__main__':