#!/usr/bin/env elixir

# IP Blocklist Example
#
# This example demonstrates how to use NFTables to create and manage
# an IP address blocklist using nftables sets.
#
# **Format**: This example uses JSON format for communication with libnftables.
#
# Requirements:
# - The NFTables port binary must have CAP_NET_ADMIN capability
# - Run: sudo setcap cap_net_admin=ep priv/port_nftables
#
# Usage:
#   mix run examples/04_ip_blocklist.exs

# Start NFTables (JSON-based port)
{:ok, pid} = NFTables.Port.start_link()
IO.puts("✓ NFTables started (JSON-based port)\n")

# Configuration
table = "filter"
blocklist_name = "banned_ips"

# Step 1: Create the blocklist set (if it doesn't exist)
IO.puts("Creating blocklist set...")

# Try to create the set (will fail if it already exists, which is fine)
# Use the high-level Set API instead of raw JSON
case NFTables.Set.add(pid, %{
  name: blocklist_name,
  table: table,
  family: :inet,
  key_type: :ipv4_addr,
  elements: []
}) do
  :ok ->
    IO.puts("✓ Created set '#{blocklist_name}' in table '#{table}'")

  {:error, reason} ->
    # Set might already exist
    IO.puts("Note: Set may already exist (#{reason})")
end

# Step 2: Check if the set exists
IO.puts("\nChecking if set exists...")
if NFTables.Set.exists?(pid, table, blocklist_name, :inet) do
  IO.puts("✓ Set '#{blocklist_name}' exists")
else
  IO.puts("✗ Set '#{blocklist_name}' does not exist")
  exit(1)
end

# Step 3: Add suspicious IPs to the blocklist
IO.puts("\nAdding IPs to blocklist...")

# Example: Block some IP addresses (now using string format)
# In a real scenario, these might come from an intrusion detection system
blocked_ips = [
  "192.168.1.100",  # Example attacker
  "10.0.0.50",      # Example scanner
  "203.0.113.42"    # TEST-NET-3 (example)
]

case NFTables.Set.add_elements(pid, table, blocklist_name, :inet, blocked_ips) do
  :ok ->
    IO.puts("✓ Added #{length(blocked_ips)} IPs to blocklist:")
    for ip_str <- blocked_ips do
      IO.puts("  - #{ip_str}")
    end

  {:error, reason} ->
    IO.puts("✗ Failed to add IPs: #{reason}")
end

# Step 4: List all blocked IPs
IO.puts("\nCurrent blocklist:")

case NFTables.Set.list_elements(pid, table, blocklist_name) do
  {:ok, elements} ->
    IO.puts("✓ Blocked IPs (#{length(elements)} total):")
    for elem <- Enum.sort_by(elements, & &1.key_ip) do
      IO.puts("  - #{elem.key_ip} (flags: #{elem.flags})")
    end

  {:error, reason} ->
    IO.puts("✗ Failed to list elements: #{reason}")
end

# Step 5: Remove an IP from the blocklist
# (Maybe it was a false positive)
IO.puts("\nRemoving false positive...")

ip_to_unblock = ["192.168.1.100"]

case NFTables.Set.delete_elements(pid, table, blocklist_name, :inet, ip_to_unblock) do
  :ok ->
    IO.puts("✓ Removed 192.168.1.100 from blocklist")

  {:error, reason} ->
    IO.puts("✗ Failed to remove IP: #{reason}")
end

# Step 6: Verify the IP was removed
IO.puts("\nUpdated blocklist:")

case NFTables.Set.list_elements(pid, table, blocklist_name) do
  {:ok, elements} ->
    IO.puts("✓ Blocked IPs (#{length(elements)} remaining):")
    for elem <- Enum.sort_by(elements, & &1.key_ip) do
      IO.puts("  - #{elem.key_ip}")
    end

  {:error, reason} ->
    IO.puts("✗ Failed to list elements: #{reason}")
end

# Step 7: Show all sets in the filter table
IO.puts("\nAll sets in '#{table}' table:")

case NFTables.Set.list(pid, family: :inet) do
  {:ok, sets} ->
    filter_sets = Enum.filter(sets, fn s -> s.table == table end)
    IO.puts("✓ Found #{length(filter_sets)} sets:")
    for set <- filter_sets do
      IO.puts("  - #{set.name} (key_len: #{set.key_len} bytes)")
    end

  {:error, reason} ->
    IO.puts("✗ Failed to list sets: #{reason}")
end

IO.puts("\n" <> String.duplicate("=", 60))
IO.puts("Example complete!")
IO.puts(String.duplicate("=", 60))
IO.puts("""

Next steps:
  1. Create nftables rules to use this set:
     nft add rule filter input ip saddr @banned_ips drop

  2. Integrate with your application to dynamically update the blocklist

  3. Use NFTables.Set.add_elements/5 to block new IPs in real-time

  4. Use NFTables.Set.delete_elements/5 to unblock IPs
""")