96 lines
2.9 KiB
Python
Executable File
96 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
# Import helper functions
|
|
sys.path.append(os.path.join(os.path.dirname(os.path.dirname(__file__))))
|
|
from helpers.functions import printfe, ensure_dependencies, command_exists
|
|
|
|
DOTFILES_ROOT = os.path.expanduser("~/.dotfiles")
|
|
|
|
def lint_ansible(fix=False):
|
|
"""Run ansible-lint on Ansible files"""
|
|
ansible_dir = os.path.join(DOTFILES_ROOT, "config/ansible")
|
|
|
|
if not os.path.isdir(ansible_dir):
|
|
printfe("yellow", "No ansible directory found at config/ansible")
|
|
return 0
|
|
|
|
# Find all YAML files in the ansible directory
|
|
yaml_files = []
|
|
for ext in [".yml", ".yaml"]:
|
|
yaml_files.extend(list(Path(ansible_dir).glob(f"**/*{ext}")))
|
|
|
|
if not yaml_files:
|
|
printfe("yellow", "No Ansible files found in config/ansible to lint")
|
|
return 0
|
|
|
|
if not command_exists("ansible-lint"):
|
|
printfe("red", "ansible-lint is not installed. Please install it with pip or your package manager.")
|
|
return 1
|
|
|
|
printfe("blue", f"Running ansible-lint{' with auto-fix' if fix else ''}...")
|
|
files_to_lint = [str(f) for f in yaml_files]
|
|
|
|
command = ["ansible-lint"]
|
|
if fix:
|
|
command.append("--fix")
|
|
command.extend(files_to_lint)
|
|
|
|
result = subprocess.run(command)
|
|
return result.returncode
|
|
|
|
def lint_nix():
|
|
"""Run nixfmt on Nix files"""
|
|
nix_files = list(Path(DOTFILES_ROOT).glob("**/*.nix"))
|
|
|
|
if not nix_files:
|
|
printfe("yellow", "No Nix files found to lint")
|
|
return 0
|
|
|
|
if not command_exists("nixfmt"):
|
|
printfe("red", "nixfmt is not installed. Please install it with nix-env or your package manager.")
|
|
return 1
|
|
|
|
printfe("blue", "Running nixfmt...")
|
|
exit_code = 0
|
|
for nix_file in nix_files:
|
|
printfe("cyan", f"Formatting {nix_file}")
|
|
result = subprocess.run(["nixfmt", str(nix_file)])
|
|
if result.returncode != 0:
|
|
exit_code = 1
|
|
|
|
return exit_code
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Run linters on dotfiles")
|
|
parser.add_argument("--ansible", action="store_true", help="Run only ansible-lint")
|
|
parser.add_argument("--nix", action="store_true", help="Run only nixfmt")
|
|
parser.add_argument("--fix", action="store_true", help="Auto-fix issues where possible (for ansible-lint)")
|
|
args = parser.parse_args()
|
|
|
|
# If no specific linter is specified, run both
|
|
run_ansible = args.ansible or not (args.ansible or args.nix)
|
|
run_nix = args.nix or not (args.ansible or args.nix)
|
|
|
|
exit_code = 0
|
|
|
|
if run_ansible:
|
|
ansible_result = lint_ansible(fix=args.fix)
|
|
if ansible_result != 0:
|
|
exit_code = ansible_result
|
|
|
|
if run_nix:
|
|
nix_result = lint_nix()
|
|
if nix_result != 0:
|
|
exit_code = nix_result
|
|
|
|
return exit_code
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|