103 lines
2.9 KiB
Python
103 lines
2.9 KiB
Python
#!/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 = ""
|
|
|
|
interfaces = subprocess.check_output(["ip", "-o", "link", "show"]).decode()
|
|
interfaces = [
|
|
line.split(": ")[1]
|
|
for line in interfaces.strip().splitlines()
|
|
if not re.match(r"^(lo|wgcf|warp)$", line.split(": ")[1])
|
|
]
|
|
|
|
for iface in interfaces:
|
|
try:
|
|
ipv4 = subprocess.check_output(["ip", "-o", "-4", "addr", "show", iface]).decode()
|
|
for line in ipv4.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
|
|
ipv6 = subprocess.check_output(["ip", "-o", "-6", "addr", "show", iface]).decode()
|
|
for line in ipv6.strip().splitlines():
|
|
addr = line.split()[3].split("/")[0]
|
|
if not re.match(r"^(::1|fe80:)", addr):
|
|
ipv6_address = addr
|
|
break
|
|
except subprocess.CalledProcessError:
|
|
continue
|
|
|
|
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()
|