#!/bin/sh

echo "========================================"
echo "  iptables-autoconf - Clear Rules      "
echo "========================================"
echo ""

delete_rule() {
    local pattern="$1"
    local count=$(iptables -t filter -L INPUT --line-numbers 2>/dev/null | grep "$pattern " | wc -l)
    
    if [ "$count" -eq 0 ]; then
        echo "  - $pattern: no rules found"
        return
    fi
    
    local rule_nums=$(iptables -t filter -L INPUT --line-numbers 2>/dev/null | grep "$pattern " | awk '{print $1}' | sort -rn)
    local deleted=0
    
    for num in $rule_nums; do
        if iptables -t filter -D INPUT $num 2>/dev/null; then
            deleted=$((deleted + 1))
        fi
    done
    
    echo "  - $pattern: removed $deleted rule(s)"
}

echo "[1/2] Removing iptables INPUT rules..."
echo "  [IPv4 single]:"
delete_rule "ustc_blacklist_v4"
echo "  [IPv4 CIDR]:"
delete_rule "ustc_blacklist_v4_net"
echo "  [IPv6 single]:"
delete_rule "ustc_blacklist_v6"
echo "  [IPv6 CIDR]:"
delete_rule "ustc_blacklist_v6_net"

echo ""
echo "[2/2] Current iptables rules:"
echo "----------------------------------------"
iptables -t filter -L INPUT --line-numbers -v 2>/dev/null | head -15
echo "----------------------------------------"

echo ""
echo "  Cleanup completed!"
echo "========================================"