40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#!/usr/bin/env nix-shell
|
|
#!nix-shell -p nginx-config-formatter python3 -i python3
|
|
|
|
import os
|
|
import re
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from tempfile import TemporaryDirectory
|
|
|
|
|
|
def nginx_config() -> str:
|
|
reload_config = "/etc/nginx/nginx.conf"
|
|
if os.path.exists(reload_config):
|
|
return reload_config
|
|
out = subprocess.check_output(["systemctl", "cat", "nginx"])
|
|
match = re.search(r"-c '(\S+-nginx\.conf)", out.decode("utf-8"))
|
|
if not match:
|
|
print("Could not find nginx.conf in nginx.service", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
return match.group(1)
|
|
|
|
|
|
def main():
|
|
config_path = nginx_config()
|
|
with TemporaryDirectory() as temp_dir:
|
|
temp_path = os.path.join(temp_dir, "nginx.conf")
|
|
with open(temp_path, "wb+") as temp_file, open(
|
|
config_path, "rb"
|
|
) as config_file:
|
|
shutil.copyfileobj(config_file, temp_file)
|
|
temp_file.flush()
|
|
subprocess.check_call(["nginxfmt", temp_file.name])
|
|
editor = os.environ.get("EDITOR", "cat")
|
|
subprocess.check_call([editor, temp_file.name] + sys.argv[1:])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|