#!/usr/bin/env python3

# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Invokes `rustc --print=cfg --target=...` and writes results to a file.

This script is typically invoked using the
`//build/rust/gni_impl:rustc_print_cfg` target.
"""

import argparse
import io
import os
import platform
import re
import subprocess
import sys
import tempfile


def rustc_name():
    if platform.system() == 'Windows':
        return "rustc.exe"
    else:
        return "rustc"


def capture_rustc_cfg(rust_prefix, target, output_path):
    """Invokes `rustc --print=cfg --target=<target>` and saves results.

    Results are saved to a file at `output_path`."""

    rustc_path = os.path.join(rust_prefix, rustc_name())

    # TODO(lukasza): Check if command-line flags other `--target` may affect the
    # output of `--print-cfg`.  If so, then consider also passing extra `args`
    # (derived from `rustflags` maybe?).
    args = [rustc_path, "--print=cfg", f"--target={target}"]

    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    with open(output_path, 'w') as output_file:
        subprocess.run(args, stdout=output_file, check=True)


def main():
    parser = argparse.ArgumentParser("rustc_print_cfg.py")
    parser.add_argument('--rust-prefix', required=True, help='rust path prefix')
    parser.add_argument('--target', required=True, help='rust target triple')
    parser.add_argument('--output-path', required=True, help='output file')
    args = parser.parse_args()
    capture_rustc_cfg(args.rust_prefix, args.target, args.output_path)
    return 0


if __name__ == '__main__':
    sys.exit(main())