99 lines
2.6 KiB
Bash
Executable File
99 lines
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
|
|
set -e
|
|
|
|
# --- Help message ---
|
|
show_help() {
|
|
cat << EOF
|
|
Usage: $0 [OPTION]
|
|
|
|
Format all Rust code in the workspace, including excluded crates and special files.
|
|
|
|
OPTIONS:
|
|
--check Check if code is formatted (do not modify files)
|
|
--help Display this help message and exit
|
|
|
|
EXAMPLES:
|
|
$0 # Format all applicable Rust code
|
|
$0 --check # Check formatting without modifying files
|
|
|
|
EOF
|
|
}
|
|
|
|
WORKSPACE_ROOT="$(dirname "$(readlink -f "$0")")/.."
|
|
|
|
EXCLUDED_CRATES=$(sed -n -e 's/#.*//; /^\s*$/d' -e '/^\[workspace\]/,/^\[.*\]/{/exclude = \[/,/\]/p}' "$WORKSPACE_ROOT/Cargo.toml" | grep -v "exclude = \[" | tr -d '", \]')
|
|
|
|
CHECK_MODE=false
|
|
|
|
if [ "$#" -eq 1 ]; then
|
|
case "$1" in
|
|
--help)
|
|
show_help
|
|
exit 0
|
|
;;
|
|
--check)
|
|
CHECK_MODE=true
|
|
;;
|
|
*)
|
|
echo "Error: Invalid argument '$1'."
|
|
echo "Run '$0 --help' for usage."
|
|
exit 1
|
|
;;
|
|
esac
|
|
elif [ "$#" -gt 1 ]; then
|
|
echo "Error: Too many arguments."
|
|
echo "Run '$0 --help' for usage."
|
|
exit 1
|
|
fi
|
|
|
|
cd $WORKSPACE_ROOT
|
|
if [ "$CHECK_MODE" = true ]; then
|
|
cargo fmt --check
|
|
else
|
|
cargo fmt
|
|
fi
|
|
|
|
# Format the 100-line kernel demo as well
|
|
KERNEL_DEMO_FILE="$WORKSPACE_ROOT/osdk/tests/examples_in_book/write_a_kernel_in_100_lines_templates/lib.rs"
|
|
if [ "$CHECK_MODE" = true ]; then
|
|
rustfmt --check $KERNEL_DEMO_FILE
|
|
else
|
|
rustfmt $KERNEL_DEMO_FILE
|
|
fi
|
|
|
|
for CRATE in $EXCLUDED_CRATES; do
|
|
CRATE_DIR="$WORKSPACE_ROOT/$CRATE"
|
|
|
|
# Here temporarily skip processing this crate for now considering that this crate
|
|
# is not currently in use or under development.
|
|
case "$CRATE" in
|
|
# `cargo-component` crate currently is pinned to use Rust nightly-2023-02-05 version,
|
|
# and when using this script in the current Docker environment, it will
|
|
# additionally download this version of Rust.
|
|
*cargo-component*)
|
|
continue
|
|
;;
|
|
# `target/osdk/base` and `target/osdk/test-base` are generated by OSDK and does not need to be formatted.
|
|
# The directories do not exist before running `make kernel` or `make test`.
|
|
*target/osdk/base*)
|
|
continue
|
|
;;
|
|
*target/osdk/test-base*)
|
|
continue
|
|
;;
|
|
esac
|
|
|
|
if [ -d "$CRATE_DIR" ]; then
|
|
if [ "$CHECK_MODE" = true ]; then
|
|
(cd "$CRATE_DIR" && cargo fmt --check)
|
|
else
|
|
(cd "$CRATE_DIR" && cargo fmt)
|
|
fi
|
|
else
|
|
echo "Directory for crate $CRATE does not exist"
|
|
fi
|
|
done
|