wehub-resource-sync
60 lines
2.1 KiB
Python
Executable File
60 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Count the total number of dependencies of a file (recursively).
|
|
|
|
This produces the format for use in <https://github.com/benchmark-action/github-action-benchmark>.
|
|
|
|
Use the script:
|
|
python3 scripts/ci/count_dependencies.py -p rerun --all-features
|
|
python3 scripts/ci/count_dependencies.py -p rerun --no-default-features
|
|
|
|
Unfortunately, this script under-counts compared to what `cargo build` outputs.
|
|
There is also `cargo deps-list`, which also under-counts.
|
|
For instance:
|
|
|
|
* `scripts/ci/count_dependencies.py -p re_sdk --no-default-features` => 118
|
|
* `cargo deps-list -p re_sdk --no-default-features` => 165
|
|
* `cargo check -p re_sdk --no-default-features` => 213
|
|
|
|
So this script is more of a heurristic than an exact count.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Count crate dependencies")
|
|
|
|
parser.add_argument("-p", required=True, type=str, help="Crate name")
|
|
parser.add_argument("--all-features", default=False, action="store_true", help="Use all features")
|
|
parser.add_argument("--no-default-features", default=False, action="store_true", help="Use no default features")
|
|
parser.add_argument("-F", "--features", default="", type=str, help="Additional features to enable")
|
|
|
|
args = parser.parse_args()
|
|
|
|
crate = args.p
|
|
if args.all_features:
|
|
flags = "--all-features"
|
|
elif args.no_default_features:
|
|
flags = "--no-default-features"
|
|
else:
|
|
flags = ""
|
|
if args.features:
|
|
flags += f" --features {args.features}"
|
|
|
|
cmd = f'cargo tree --edges normal -p {crate} {flags} | tail -n +2 | grep -E "\\w+ v[0-9.]+" -o | sort -u | wc -l'
|
|
print(f"Running command: {cmd}", file=sys.stderr, flush=True)
|
|
count = int(os.popen(cmd).read().strip())
|
|
assert count > 0, f"Command failed. Maybe unknown crate? cmd: {cmd}"
|
|
print(json.dumps([{"name": f"{crate} {flags}", "value": count, "unit": "crates"}]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|