#!/usr/bin/env python3
# Copyright 2022 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# Run `rustfmt` on all Rust code contained in the crosvm workspace, including
# all commmon/* crates as well.
#
# Usage:
#
#    $ bin/fmt
#
# To print a diff and exit 1 if code is not formatted, but without changing any
# files, use:
#
#    $ bin/fmt --check
#

from impl.common import (
    CROSVM_ROOT,
    run_main,
    cmd,
    chdir,
)


def main(check: bool = False, nightly: bool = False):
    chdir(CROSVM_ROOT)
    cmd(
        "./tools/health-check --all rust_format python_format markdown_format",
        "--fix" if not check else None,
        "--nightly" if nightly else None,
    ).fg()


if __name__ == "__main__":
    run_main(main)