terraform_fmt.py

from __future__ import annotations

import logging
import os
import shlex
import sys
from subprocess import PIPE, run
from typing import Sequence

from .common import parse_cmdline, setup_logging

logger = logging.getLogger(__name__)


def main(argv: Sequence[str] | None = None) -> int:
    setup_logging()
    logger.debug(sys.version_info)
    args, hook_config, files, tf_init_args, env_vars = parse_cmdline(argv)
    if os.environ.get("PRE_COMMIT_COLOR") == "never":
        args.append("-no-color")
    cmd = ["terraform", "fmt", *args, *files]
    logger.info("calling %s", shlex.join(cmd))
    logger.debug("env_vars: %r", env_vars)
    logger.debug("args: %r", args)
    completed_process = run(cmd, env={**os.environ, **env_vars}, text=True, stdout=PIPE)
    if completed_process.stdout:
        print(completed_process.stdout)
    return completed_process.returncode


if __name__ == "__main__":
    raise SystemExit(main())