#!/usr/bin/env python3

import sys
import re
import subprocess
from pathlib import Path

core_scripts_dir = Path(__file__).resolve().parents[1]
if str(core_scripts_dir) not in sys.path:
    sys.path.append(str(core_scripts_dir))

from paths import CONFIG_ENV


def ensure_env_file_exists():
    if not CONFIG_ENV.exists():
        print("CONFIG_ENV not found. Creating a new one...")
        CONFIG_ENV.touch()


def update_config(key: str, value: str):
    content = []
    if CONFIG_ENV.exists():
        with CONFIG_ENV.open("r") as f:
            content = [line for line in f if not line.startswith(f"{key}=")]
    content.append(f"{key}={value}\n")
    with CONFIG_ENV.open("w") as f:
        f.writelines(content)


def get_interface_addresses():
    ipv4_address = ""
    ipv6_address = ""

    try:
        interfaces_output = subprocess.check_output(["ip", "-o", "link", "show"], stderr=subprocess.DEVNULL).decode()
        interface_lines = interfaces_output.strip().splitlines()
        candidate_interfaces = []
        for line in interface_lines:
            parts = line.split(': ')
            if len(parts) > 1:
                iface_name = parts[1].split('@')[0]
                if iface_name not in ["lo", "wgcf", "warp"]:
                    candidate_interfaces.append(iface_name)

        for iface in candidate_interfaces:
            try:
                if not ipv4_address:
                    ipv4_output = subprocess.check_output(["ip", "-o", "-4", "addr", "show", iface], stderr=subprocess.DEVNULL).decode()
                    for line in ipv4_output.strip().splitlines():
                        addr = line.split()[3].split("/")[0]
                        if not re.match(r"^(127\.|10\.|192\.168\.|172\.(1[6-9]|2[0-9]|3[0-1]))", addr):
                            ipv4_address = addr
                            break
                if not ipv6_address:
                    ipv6_output = subprocess.check_output(["ip", "-o", "-6", "addr", "show", iface], stderr=subprocess.DEVNULL).decode()
                    for line in ipv6_output.strip().splitlines():
                        addr = line.split()[3].split("/")[0]
                        if not re.match(r"^(::1|fe80:)", addr):
                            ipv6_address = addr
                            break
            except subprocess.CalledProcessError:
                continue
    except (subprocess.CalledProcessError, FileNotFoundError):
        pass

    if not ipv4_address:
        try:
            ipv4_address = subprocess.check_output(["curl", "-s", "-4", "ip.sb"], timeout=5).decode().strip()
            if not re.match(r"^\d{1,3}(\.\d{1,3}){3}$", ipv4_address):
                ipv4_address = ""
        except (subprocess.CalledProcessError, FileNotFoundError, subprocess.TimeoutExpired):
            ipv4_address = ""

    if not ipv6_address:
        try:
            ipv6_address = subprocess.check_output(["curl", "-s", "-6", "ip.sb"], timeout=5).decode().strip()
            if ":" not in ipv6_address:
                ipv6_address = ""
        except (subprocess.CalledProcessError, FileNotFoundError, subprocess.TimeoutExpired):
            ipv6_address = ""

    return ipv4_address, ipv6_address


def add_ips():
    ensure_env_file_exists()
    ipv4, ipv6 = get_interface_addresses()

    update_config("IP4", ipv4 or "")
    update_config("IP6", ipv6 or "")

    print(f"Updated IP4={ipv4 or 'Not Found'}")
    print(f"Updated IP6={ipv6 or 'Not Found'}")


def edit_ip(option: str, new_ip: str):
    ensure_env_file_exists()
    if option == "-4":
        update_config("IP4", new_ip)
        print(f"IP4 has been updated to {new_ip}.")
    elif option == "-6":
        update_config("IP6", new_ip)
        print(f"IP6 has been updated to {new_ip}.")
    else:
        print("Invalid option. Use -4 for IPv4 or -6 for IPv6.")


def main():
    if len(sys.argv) < 2:
        print("Usage: {add|edit -4|-6 <new_ip>}")
        sys.exit(1)

    action = sys.argv[1]

    if action == "add":
        add_ips()
    elif action == "edit" and len(sys.argv) == 4:
        edit_ip(sys.argv[2], sys.argv[3])
    else:
        print("Usage: {add|edit -4|-6 <new_ip>}")
        sys.exit(1)


if __name__ == "__main__":
    main()