filesize_guard.py

import fnmatch
import json
import os
import sys

DEFAULT_IGNORE_FILE = "./filesize_guard.ignore"

def load_ignore_patterns(ignore_file):
    default_patterns = [".git/*"]

    if not os.path.exists(ignore_file):
        if ignore_file == DEFAULT_IGNORE_FILE:
            return default_patterns

        print(f"Error: The specified ignore file '{ignore_file}' does not exist.", file=sys.stderr)
        sys.exit(1)

    with open(ignore_file, "r") as f:
        patterns = [line.strip() for line in f if line.strip() and not line.startswith("#")]
    return default_patterns + patterns

def should_ignore(file, patterns):
    return any(fnmatch.fnmatch(file, pattern) for pattern in patterns)

def load_files_from_json(file_paths):
    files = []
    for file_path in file_paths:
        if not os.path.exists(file_path):
            continue
        with open(file_path, "r") as f:
            files.extend(json.load(f))
    return files

ignore_patterns = load_ignore_patterns(sys.argv[1])
max_size_kb = int(sys.argv[2])
files = load_files_from_json(sys.argv[3:])

exit_code = 0

for file in files:
    if not file:
        continue

    if should_ignore(file, ignore_patterns):
        continue

    filesize_b = os.path.getsize(file)
    filesize_kb = filesize_b / 1024
    if filesize_kb >= max_size_kb:
        print(f"Error: '{file}' is {filesize_kb:.2f} kB, which exceeds the allowed limit of {max_size_kb} kB.")
        exit_code = 1

sys.exit(exit_code)