Function main

script/compressor.py:306–360

()

# --- Main CLI ---

def main() -> int:
    parser = argparse.ArgumentParser(description="Strict Hostlist Compressor")
    parser.add_argument("input", nargs="?", default="-", help="Input file (stdin default)")
    parser.add_argument("-o", "--output", help="Output file")
    parser.add_argument("-i", "--in-place", action="store_true", help="Modify input file")
    parser.add_argument("--filtered", help="Save filtered rules to file")
    parser.add_argument("--include-wildcards", action="store_true", help="Keep wildcard rules")
    
    args = parser.parse_args()

    # 1. Input Reading
    try:
        if args.input == "-":
            if args.in_place:
                logger.error("Cannot use --in-place with stdin")
                return 1
            content = sys.stdin.read().splitlines()
        else:
            content = Path(args.input).read_text(encoding="utf-8").splitlines()
    except Exception as e:
        logger.error(f"Read error: {e}")
        return 1

    # 2. Strict Validation (The JS Logic)
    validator = RuleValidator(allow_ip=False)
    valid_content, validation_filtered = validator.validate(content)
    
    logger.info(f"验证域名: {len(content)} -> {len(valid_content)} (移除 {len(validation_filtered)} 条)")

    # 3. Compression
    compressed, compression_filtered = compress_rules(valid_content, args.include_wildcards)
    
    all_filtered = validation_filtered + compression_filtered 

    logger.info(f"压缩完成 - {len(content)} -> {len(compressed)} (移除 {len(all_filtered)} 条)")

    # 4. Output
    output_lines = "\n".join(compressed) + "\n"
    
    try:
        if args.in_place and args.input != "-":
            Path(args.input).write_text(output_lines, encoding="utf-8", newline="\n")
        elif args.output:
            Path(args.output).write_text(output_lines, encoding="utf-8", newline="\n")
        else:
            sys.stdout.write(output_lines)

        if args.filtered:
            Path(args.filtered).write_text("\n".join(all_filtered) + "\n", encoding="utf-8", newline="\n")
            
    except Exception as e:
        logger.error(f"Write error: {e}")
        return 1
        
    return 0

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