e5a0fc467e
Integrates the new `warp.py` script into the installation and uninstallation workflows. This refactors the `install.py` and `uninstall.py` scripts to call the local Python-based WARP manager instead of downloading and executing an external bash script.
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
import subprocess
|
|
import sys
|
|
import json
|
|
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 *
|
|
|
|
WARP_SCRIPT_PATH = Path(__file__).resolve().parent / "warp.py"
|
|
WARP_DEVICE = "wgcf"
|
|
|
|
def is_service_active(service_name: str) -> bool:
|
|
return subprocess.run(["systemctl", "is-active", "--quiet", service_name]).returncode == 0
|
|
|
|
|
|
def install_warp():
|
|
print("Installing WARP...")
|
|
result = subprocess.run([sys.executable, str(WARP_SCRIPT_PATH), "install"])
|
|
return result.returncode == 0
|
|
|
|
|
|
def add_warp_outbound_to_config():
|
|
if not CONFIG_FILE.exists():
|
|
print(f"Error: Config file {CONFIG_FILE} not found.")
|
|
return
|
|
|
|
with open(CONFIG_FILE, "r") as f:
|
|
config = json.load(f)
|
|
|
|
outbounds = config.get("outbounds", [])
|
|
if any(outbound.get("name") == "warps" for outbound in outbounds):
|
|
print("WARP outbound already exists in the configuration.")
|
|
return
|
|
|
|
outbounds.append({
|
|
"name": "warps",
|
|
"type": "direct",
|
|
"direct": {
|
|
"mode": 4,
|
|
"bindDevice": WARP_DEVICE
|
|
}
|
|
})
|
|
config["outbounds"] = outbounds
|
|
|
|
with open(CONFIG_FILE, "w") as f:
|
|
json.dump(config, f, indent=2)
|
|
|
|
print("WARP outbound added to config.json.")
|
|
|
|
|
|
def restart_hysteria():
|
|
subprocess.run(["python3", str(CLI_PATH), "restart-hysteria2"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
print("Hysteria2 restarted with updated configuration.")
|
|
|
|
|
|
def main():
|
|
warp_service = f"wg-quick@{WARP_DEVICE}.service"
|
|
|
|
if is_service_active(warp_service):
|
|
print("WARP is already active. Checking configuration...")
|
|
add_warp_outbound_to_config()
|
|
restart_hysteria()
|
|
else:
|
|
if install_warp() and is_service_active(warp_service):
|
|
print("WARP installation successful.")
|
|
add_warp_outbound_to_config()
|
|
restart_hysteria()
|
|
else:
|
|
print("WARP installation failed.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |