Template
Add Amnezia Web Panel source with PostgreSQL 17 storage.
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,800 @@
|
||||
import json
|
||||
import os
|
||||
import uuid
|
||||
import logging
|
||||
import base64
|
||||
import shlex
|
||||
from datetime import datetime
|
||||
import urllib.parse
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class XrayManager:
|
||||
"""Manages Xray (VLESS-Reality) protocol installation and client management."""
|
||||
|
||||
PROTOCOL = 'xray'
|
||||
CONTAINER_NAME = 'amnezia-xray'
|
||||
IMAGE_NAME = 'amneziavpn/amnezia-xray' # or we can build it
|
||||
|
||||
def __init__(self, ssh_manager, protocol='xray'):
|
||||
self.ssh = ssh_manager
|
||||
self.protocol = protocol or self.PROTOCOL
|
||||
self.base_protocol = str(self.protocol).split('__', 1)[0]
|
||||
self.instance = self._instance_index(self.protocol)
|
||||
self.container_name = self._container_name(self.protocol)
|
||||
self.image_name = self._image_name(self.protocol)
|
||||
|
||||
def _instance_index(self, protocol):
|
||||
parts = str(protocol or '').split('__', 1)
|
||||
if len(parts) == 2:
|
||||
try:
|
||||
return max(1, int(parts[1]))
|
||||
except ValueError:
|
||||
return 1
|
||||
return 1
|
||||
|
||||
def _container_name(self, protocol=None):
|
||||
idx = self._instance_index(protocol or self.protocol)
|
||||
base_name = self.CONTAINER_NAME
|
||||
return base_name if idx <= 1 else f'{base_name}-{idx}'
|
||||
|
||||
def _image_name(self, protocol=None):
|
||||
idx = self._instance_index(protocol or self.protocol)
|
||||
base_name = self.IMAGE_NAME
|
||||
return base_name if idx <= 1 else f'{base_name}-{idx}'
|
||||
|
||||
def _config_dir(self):
|
||||
return '/opt/amnezia/xray' if self.instance <= 1 else f'/opt/amnezia/xray-{self.instance}'
|
||||
|
||||
def _config_path(self):
|
||||
return f'{self._config_dir()}/server.json'
|
||||
|
||||
def _list_xray_files(self):
|
||||
"""List filenames in the on-disk Xray config directory."""
|
||||
out, _, code = self.ssh.run_sudo_command(f"ls -1 {self._config_dir()} 2>/dev/null")
|
||||
if code != 0:
|
||||
return []
|
||||
return [f for f in out.strip().split('\n') if f]
|
||||
|
||||
def _detect_layout(self):
|
||||
"""Pick which on-disk layout this installation uses.
|
||||
|
||||
'native' — official Amnezia client layout: xray_private.key,
|
||||
xray_public.key, xray_short_id.key, xray_uuid.key plus a clientsTable
|
||||
file without an extension.
|
||||
'panel' — legacy web-panel layout: meta.json + clientsTable.json.
|
||||
|
||||
On a fresh node with no Xray files yet, defaults to 'native' so new
|
||||
installs produce the same artifacts as the official client.
|
||||
"""
|
||||
if hasattr(self, '_cached_layout'):
|
||||
return self._cached_layout
|
||||
files = set(self._list_xray_files())
|
||||
if {'xray_private.key', 'xray_public.key'} & files:
|
||||
layout = 'native'
|
||||
elif 'meta.json' in files:
|
||||
layout = 'panel'
|
||||
else:
|
||||
layout = 'native'
|
||||
self._cached_layout = layout
|
||||
return layout
|
||||
|
||||
def _clients_table_filename(self):
|
||||
return 'clientsTable' if self._detect_layout() == 'native' else 'clientsTable.json'
|
||||
|
||||
def _clients_table_path(self):
|
||||
return f'{self._config_dir()}/{self._clients_table_filename()}'
|
||||
|
||||
def _read_remote_file(self, path):
|
||||
"""Read a remote text file, preferring the running container's view."""
|
||||
out, _, code = self.ssh.run_sudo_command(
|
||||
f"docker exec {self.container_name} cat {path} 2>/dev/null"
|
||||
)
|
||||
if code != 0 or not out:
|
||||
out, _, code = self.ssh.run_sudo_command(f"cat {path} 2>/dev/null")
|
||||
if code != 0 or not out.strip():
|
||||
return None
|
||||
return out
|
||||
|
||||
def _derive_pubkey_from_priv(self, priv_key):
|
||||
"""Derive the Reality public key from a private key via the xray binary.
|
||||
Used as a fallback when xray_public.key is missing or unreadable.
|
||||
"""
|
||||
if not priv_key:
|
||||
return ''
|
||||
out, _, code = self.ssh.run_sudo_command(
|
||||
f"docker exec {self.container_name} /usr/bin/xray x25519 -i {priv_key}"
|
||||
)
|
||||
if code != 0 or not out.strip():
|
||||
out, _, code = self.ssh.run_sudo_command(
|
||||
f"docker run --rm --entrypoint=\"\" {self.image_name} /usr/bin/xray x25519 -i {priv_key}"
|
||||
)
|
||||
if code != 0 or not out:
|
||||
return ''
|
||||
for line in out.split('\n'):
|
||||
if 'Public' in line and ':' in line:
|
||||
return line.split(':', 1)[1].strip()
|
||||
return ''
|
||||
|
||||
def _get_default_xray_uuid(self):
|
||||
"""UUID of the install-time default client (xray_uuid.key) — meaningful only
|
||||
for native-layout installs. Imports skip this UUID, mirroring the official
|
||||
Amnezia client behaviour (see usersController.cpp::getXrayClients).
|
||||
"""
|
||||
if self._detect_layout() != 'native':
|
||||
return ''
|
||||
out = self._read_remote_file(f"{self._config_dir()}/xray_uuid.key")
|
||||
return (out or '').strip()
|
||||
|
||||
# ===================== INSTALLATION =====================
|
||||
|
||||
def check_docker_installed(self):
|
||||
out, err, code = self.ssh.run_command("docker --version 2>/dev/null")
|
||||
if code != 0: return False
|
||||
out2, _, code2 = self.ssh.run_command("systemctl is-active docker 2>/dev/null || service docker status 2>/dev/null")
|
||||
return 'active' in out2 or 'running' in out2.lower()
|
||||
|
||||
def check_container_running(self):
|
||||
out, _, _ = self.ssh.run_sudo_command(
|
||||
f"docker ps --filter name=^{self.container_name}$ --format '{{{{.Status}}}}'"
|
||||
)
|
||||
return 'Up' in out
|
||||
|
||||
def check_protocol_installed(self):
|
||||
out, _, _ = self.ssh.run_sudo_command(
|
||||
f"docker ps -a --filter name=^{self.container_name}$ --format '{{{{.Names}}}}'"
|
||||
)
|
||||
return self.container_name in out.strip().split('\n')
|
||||
|
||||
def get_server_status(self, protocol):
|
||||
exists = self.check_protocol_installed()
|
||||
running = self.check_container_running()
|
||||
clients = self.get_clients() if exists else []
|
||||
meta = self._get_meta_json() if exists else {}
|
||||
return {
|
||||
'container_exists': exists,
|
||||
'container_running': running,
|
||||
'clients_count': len(clients),
|
||||
'port': meta.get('port')
|
||||
}
|
||||
|
||||
def install_protocol(self, port=443, site_name='yahoo.com'):
|
||||
"""Full installation of Xray."""
|
||||
results = []
|
||||
|
||||
if not self.check_docker_installed():
|
||||
results.append("Installing Docker...")
|
||||
# Using same install method as AWGManager or assume it's installed
|
||||
pass
|
||||
|
||||
results.append("Removing old container if exists...")
|
||||
if self.check_protocol_installed():
|
||||
self.remove_container()
|
||||
|
||||
results.append("Building Docker image...")
|
||||
config_dir = self._config_dir()
|
||||
dockerfile_folder = f"/opt/amnezia/{self.container_name}"
|
||||
dockerfile_content = f"""FROM alpine:3.15
|
||||
RUN apk add --no-cache curl unzip bash openssl netcat-openbsd dumb-init rng-tools xz iptables ip6tables
|
||||
RUN apk --update upgrade --no-cache
|
||||
RUN mkdir -p {config_dir}
|
||||
RUN curl -L -H "Cache-Control: no-cache" -o /root/xray.zip "https://github.com/XTLS/Xray-core/releases/download/v26.3.27/Xray-linux-64.zip" && \\
|
||||
unzip /root/xray.zip -d /usr/bin/ && \\
|
||||
chmod a+x /usr/bin/xray && \\
|
||||
rm /root/xray.zip
|
||||
|
||||
# Tune network
|
||||
RUN echo "fs.file-max = 51200" >> /etc/sysctl.conf && \\
|
||||
echo "net.core.rmem_max = 67108864" >> /etc/sysctl.conf && \\
|
||||
echo "net.core.wmem_max = 67108864" >> /etc/sysctl.conf && \\
|
||||
echo "net.core.netdev_max_backlog = 250000" >> /etc/sysctl.conf && \\
|
||||
echo "net.core.somaxconn = 4096" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_syncookies = 1" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_tw_reuse = 1" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_tw_recycle = 0" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_fin_timeout = 30" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_keepalive_time = 1200" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.ip_local_port_range = 10000 65000" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_max_syn_backlog = 8192" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_max_tw_buckets = 5000" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_fastopen = 3" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_mem = 25600 51200 102400" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_rmem = 4096 87380 67108864" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_wmem = 4096 65536 67108864" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_mtu_probing = 1" >> /etc/sysctl.conf && \\
|
||||
echo "net.ipv4.tcp_congestion_control = hybla" >> /etc/sysctl.conf
|
||||
|
||||
RUN mkdir -p /etc/security && \\
|
||||
echo "* soft nofile 51200" >> /etc/security/limits.conf && \\
|
||||
echo "* hard nofile 51200" >> /etc/security/limits.conf
|
||||
|
||||
RUN echo '#!/bin/bash' > /opt/amnezia/start.sh && \\
|
||||
echo 'sysctl -p /etc/sysctl.conf 2>/dev/null' >> /opt/amnezia/start.sh && \\
|
||||
echo '/usr/bin/xray -config {config_dir}/server.json' >> /opt/amnezia/start.sh && \\
|
||||
chmod a+x /opt/amnezia/start.sh
|
||||
|
||||
ENTRYPOINT [ "dumb-init", "/opt/amnezia/start.sh" ]
|
||||
"""
|
||||
self.ssh.run_sudo_command(f"mkdir -p {dockerfile_folder}")
|
||||
self.ssh.upload_file_sudo(dockerfile_content, f"{dockerfile_folder}/Dockerfile")
|
||||
|
||||
_, err, code = self.ssh.run_sudo_command(
|
||||
f"docker build --no-cache -t {self.image_name} {dockerfile_folder}", timeout=300
|
||||
)
|
||||
if code != 0: raise RuntimeError(f"Failed to build container: {err}")
|
||||
|
||||
results.append("Generating keys and config...")
|
||||
# We generate a base config using a temp container or directly if host has openssl
|
||||
|
||||
# Xray keypair generation using a temporary run overriding the entrypoint
|
||||
keypair_cmd = f"docker run --rm --entrypoint=\"\" {self.image_name} /usr/bin/xray x25519"
|
||||
out_kp, err_kp, code_kp = self.ssh.run_sudo_command(keypair_cmd)
|
||||
if code_kp != 0: raise RuntimeError(f"Failed to generate x25519 keys: {err_kp}")
|
||||
|
||||
priv_key = ""
|
||||
pub_key = ""
|
||||
for line in out_kp.split('\n'):
|
||||
if "Private" in line: priv_key = line.split(":", 1)[1].strip()
|
||||
if "Public" in line: pub_key = line.split(":", 1)[1].strip()
|
||||
|
||||
short_id_cmd = f"docker run --rm --entrypoint=\"\" {self.image_name} openssl rand -hex 8"
|
||||
out_sid, _, _ = self.ssh.run_sudo_command(short_id_cmd)
|
||||
short_id = out_sid.strip()
|
||||
|
||||
# Generate initial server.json with Stats and API enabled
|
||||
server_json = {
|
||||
"log": {"loglevel": "error"},
|
||||
"stats": {},
|
||||
"api": {
|
||||
"services": ["StatsService", "LoggerService", "HandlerService"],
|
||||
"tag": "api"
|
||||
},
|
||||
"policy": {
|
||||
"levels": {
|
||||
"0": {"statsUserUplink": True, "statsUserDownlink": True}
|
||||
},
|
||||
"system": {
|
||||
"statsInboundUplink": True, "statsInboundDownlink": True,
|
||||
"statsOutboundUplink": True, "statsOutboundDownlink": True
|
||||
}
|
||||
},
|
||||
"inbounds": [
|
||||
{
|
||||
"port": int(port),
|
||||
"protocol": "vless",
|
||||
"tag": "proxy",
|
||||
"settings": {
|
||||
"clients": [],
|
||||
"decryption": "none"
|
||||
},
|
||||
"streamSettings": {
|
||||
"network": "tcp",
|
||||
"security": "reality",
|
||||
"realitySettings": {
|
||||
"dest": f"{site_name}:443",
|
||||
"serverNames": [site_name],
|
||||
"privateKey": priv_key,
|
||||
"shortIds": [short_id]
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"listen": "127.0.0.1",
|
||||
"port": 10085,
|
||||
"protocol": "dokodemo-door",
|
||||
"settings": {"address": "127.0.0.1"},
|
||||
"tag": "api"
|
||||
}
|
||||
],
|
||||
"outbounds": [{"protocol": "freedom"}],
|
||||
"routing": {
|
||||
"rules": [
|
||||
{
|
||||
"inboundTag": ["api"],
|
||||
"outboundTag": "api",
|
||||
"type": "field"
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
self.ssh.run_sudo_command(f"mkdir -p {config_dir}")
|
||||
self.ssh.upload_file_sudo(json.dumps(server_json, indent=2), f"{config_dir}/server.json")
|
||||
# Native layout — separate key files matching the official Amnezia client install.
|
||||
# See client/server_scripts/xray/configure_container.sh for the canonical layout.
|
||||
self.ssh.upload_file_sudo(priv_key + '\n', "/opt/amnezia/xray/xray_private.key")
|
||||
self.ssh.upload_file_sudo(pub_key + '\n', "/opt/amnezia/xray/xray_public.key")
|
||||
self.ssh.upload_file_sudo(short_id + '\n', "/opt/amnezia/xray/xray_short_id.key")
|
||||
# xray_uuid.key marks the install-time "default" client whose ID gets skipped on
|
||||
# auto-import. Panel installs do not reserve such a client, so we leave it empty.
|
||||
self.ssh.upload_file_sudo('\n', "/opt/amnezia/xray/xray_uuid.key")
|
||||
self.ssh.upload_file_sudo("[]", "/opt/amnezia/xray/clientsTable")
|
||||
|
||||
results.append("Starting container...")
|
||||
run_cmd = f"""docker run -d \\
|
||||
--restart always \\
|
||||
--privileged \\
|
||||
--cap-add=NET_ADMIN \\
|
||||
-p {port}:{port}/tcp \\
|
||||
-p {port}:{port}/udp \\
|
||||
-v {config_dir}:{config_dir} \\
|
||||
--name {self.container_name} \\
|
||||
{self.image_name}"""
|
||||
|
||||
_, err, code = self.ssh.run_sudo_command(run_cmd)
|
||||
if code != 0: raise RuntimeError(f"Failed to run container: {err}")
|
||||
|
||||
# Try to connect to network if needed
|
||||
self.ssh.run_sudo_command(f"docker network connect amnezia-dns-net {self.container_name} || true")
|
||||
|
||||
results.append("Xray configured and running")
|
||||
return {'status': 'success', 'protocol': self.protocol, 'port': port, 'log': results}
|
||||
|
||||
def remove_container(self):
|
||||
self.ssh.run_sudo_command(f"docker stop {self.container_name}")
|
||||
self.ssh.run_sudo_command(f"docker rm -fv {self.container_name}")
|
||||
if self.instance <= 1:
|
||||
self.ssh.run_sudo_command(f"docker rmi {self.image_name}")
|
||||
return True
|
||||
|
||||
# ===================== CLIENT MANAGEMENT =====================
|
||||
|
||||
def _get_server_json(self):
|
||||
"""Read server.json — tries inside container first, falls back to host path."""
|
||||
out, _, code = self.ssh.run_sudo_command(
|
||||
f"docker exec {self.container_name} cat {self._config_path()}"
|
||||
)
|
||||
if code != 0:
|
||||
out, _, code = self.ssh.run_sudo_command(f"cat {self._config_path()}")
|
||||
if code != 0 or not out.strip():
|
||||
return None
|
||||
return json.loads(out)
|
||||
|
||||
def _save_server_json(self, data):
|
||||
return self._write_server_json(data, restart=True)
|
||||
|
||||
def _write_server_json(self, data, restart=True):
|
||||
"""Write server.json into container via docker cp AND sync to host path."""
|
||||
tmp_file = "/tmp/_xray_server.json"
|
||||
self.ssh.upload_file_sudo(json.dumps(data, indent=2), tmp_file)
|
||||
self.ssh.run_sudo_command(
|
||||
f"docker cp {tmp_file} {self.container_name}:{self._config_path()}"
|
||||
)
|
||||
# Also keep host copy in sync (handles both volume-mount and no-mount installs)
|
||||
self.ssh.run_sudo_command(
|
||||
f"cp {tmp_file} {self._config_path()} 2>/dev/null || true"
|
||||
)
|
||||
if restart:
|
||||
self.ssh.run_sudo_command(f"docker restart {self.container_name}")
|
||||
|
||||
def _get_vless_inbound(self, config):
|
||||
for inbound in config.get('inbounds', []):
|
||||
if inbound.get('protocol') == 'vless':
|
||||
return inbound
|
||||
return None
|
||||
|
||||
def _get_vless_inbound_tag(self, config):
|
||||
inbound = self._get_vless_inbound(config)
|
||||
return inbound.get('tag') if inbound else None
|
||||
|
||||
def _run_xray_api_json(self, subcommand, payload):
|
||||
tmp_name = f"/tmp/_xray_api_{uuid.uuid4().hex}.json"
|
||||
container_tmp = tmp_name
|
||||
try:
|
||||
self.ssh.upload_file_sudo(json.dumps(payload, indent=2), tmp_name)
|
||||
_, err, code = self.ssh.run_sudo_command(
|
||||
f"docker cp {tmp_name} {self.container_name}:{container_tmp}"
|
||||
)
|
||||
if code != 0:
|
||||
return False, err
|
||||
out, err, code = self.ssh.run_sudo_command(
|
||||
f"docker exec -i {self.container_name} /usr/bin/xray api {subcommand} "
|
||||
f"-server=127.0.0.1:10085 {container_tmp}"
|
||||
)
|
||||
return code == 0, err or out
|
||||
finally:
|
||||
self.ssh.run_sudo_command(f"rm -f {tmp_name}")
|
||||
self.ssh.run_sudo_command(f"docker exec -i {self.container_name} rm -f {container_tmp} 2>/dev/null || true")
|
||||
|
||||
def _xray_api_add_user(self, config, client):
|
||||
tag = self._get_vless_inbound_tag(config)
|
||||
if not tag:
|
||||
return False
|
||||
payload = {
|
||||
"inbounds": [{
|
||||
"tag": tag,
|
||||
"protocol": "vless",
|
||||
"settings": {
|
||||
"clients": [client],
|
||||
"decryption": "none",
|
||||
}
|
||||
}]
|
||||
}
|
||||
ok, message = self._run_xray_api_json('adu', payload)
|
||||
if not ok:
|
||||
logger.warning(f"Xray API add user failed: {message}")
|
||||
return False
|
||||
return True
|
||||
|
||||
def _xray_api_remove_user(self, config, client_id):
|
||||
tag = self._get_vless_inbound_tag(config)
|
||||
if not tag:
|
||||
return False
|
||||
cmd = (
|
||||
f"docker exec -i {self.container_name} /usr/bin/xray api rmu "
|
||||
f"-server=127.0.0.1:10085 "
|
||||
f"-tag={shlex.quote(tag)} {shlex.quote(client_id)}"
|
||||
)
|
||||
out, err, code = self.ssh.run_sudo_command(cmd)
|
||||
if code != 0:
|
||||
logger.warning(f"Xray API remove user failed: {err or out}")
|
||||
return False
|
||||
return True
|
||||
|
||||
def _get_meta_json(self):
|
||||
"""Read protocol metadata. Supports both layouts.
|
||||
|
||||
Native layout pulls keys from xray_*.key files. Panel layout reads
|
||||
meta.json. Either way, port and site_name come from server.json since
|
||||
that is the authoritative runtime config — meta.json may go stale if
|
||||
the user edits server.json directly via the panel.
|
||||
"""
|
||||
config = self._get_server_json() or {}
|
||||
|
||||
port = None
|
||||
site_name = None
|
||||
rs = {}
|
||||
try:
|
||||
ib = next(b for b in config.get('inbounds', []) if b.get('protocol') == 'vless')
|
||||
port = ib.get('port')
|
||||
rs = ib.get('streamSettings', {}).get('realitySettings', {}) or {}
|
||||
names = rs.get('serverNames') or []
|
||||
if names:
|
||||
site_name = names[0]
|
||||
except StopIteration:
|
||||
pass
|
||||
|
||||
if self._detect_layout() == 'native':
|
||||
priv = (self._read_remote_file(f"{self._config_dir()}/xray_private.key") or '').strip()
|
||||
pub = (self._read_remote_file(f"{self._config_dir()}/xray_public.key") or '').strip()
|
||||
sid = (self._read_remote_file(f"{self._config_dir()}/xray_short_id.key") or '').strip()
|
||||
if not priv:
|
||||
priv = rs.get('privateKey', '')
|
||||
if not sid:
|
||||
sids = rs.get('shortIds') or []
|
||||
sid = sids[0] if sids else ''
|
||||
if not pub:
|
||||
pub = self._derive_pubkey_from_priv(priv)
|
||||
return {
|
||||
'private_key': priv,
|
||||
'public_key': pub,
|
||||
'short_id': sid,
|
||||
'port': port,
|
||||
'site_name': site_name or 'yahoo.com',
|
||||
}
|
||||
|
||||
# Panel (legacy) layout
|
||||
out = self._read_remote_file(f"{self._config_dir()}/meta.json")
|
||||
meta = {}
|
||||
if out:
|
||||
try:
|
||||
meta = json.loads(out)
|
||||
except Exception:
|
||||
meta = {}
|
||||
if port is not None:
|
||||
meta['port'] = port
|
||||
if site_name:
|
||||
meta['site_name'] = site_name
|
||||
if not meta.get('private_key'):
|
||||
meta['private_key'] = rs.get('privateKey', '')
|
||||
if not meta.get('short_id'):
|
||||
sids = rs.get('shortIds') or []
|
||||
if sids:
|
||||
meta['short_id'] = sids[0]
|
||||
if not meta.get('public_key') and meta.get('private_key'):
|
||||
meta['public_key'] = self._derive_pubkey_from_priv(meta['private_key'])
|
||||
return meta
|
||||
|
||||
def _get_clients_table(self):
|
||||
"""Read clientsTable, trying both layout filenames."""
|
||||
layout = self._detect_layout()
|
||||
primary = 'clientsTable' if layout == 'native' else 'clientsTable.json'
|
||||
fallback = 'clientsTable.json' if layout == 'native' else 'clientsTable'
|
||||
for fname in (primary, fallback):
|
||||
out = self._read_remote_file(f"{self._config_dir()}/{fname}")
|
||||
if not out or not out.strip():
|
||||
continue
|
||||
try:
|
||||
return json.loads(out)
|
||||
except Exception:
|
||||
continue
|
||||
return []
|
||||
|
||||
def _save_clients_table(self, data):
|
||||
"""Write clientsTable to the file matching the current layout, in both
|
||||
the container and the host bind-mount.
|
||||
"""
|
||||
path = self._clients_table_path()
|
||||
tmp_file = "/tmp/_xray_clients.json"
|
||||
self.ssh.upload_file_sudo(json.dumps(data, indent=2), tmp_file)
|
||||
self.ssh.run_sudo_command(
|
||||
f"docker cp {tmp_file} {self.container_name}:{path}"
|
||||
)
|
||||
self.ssh.run_sudo_command(
|
||||
f"cp {tmp_file} {path} 2>/dev/null || true"
|
||||
)
|
||||
|
||||
def _upgrade_config_for_stats(self, config, restart=True):
|
||||
"""Injects API and Stats blocks into older Xray configs transparently."""
|
||||
dirty = False
|
||||
if 'stats' not in config:
|
||||
config['stats'] = {}
|
||||
dirty = True
|
||||
if 'api' not in config:
|
||||
config['api'] = {"services": ["StatsService", "LoggerService", "HandlerService"], "tag": "api"}
|
||||
dirty = True
|
||||
else:
|
||||
services = config['api'].setdefault('services', [])
|
||||
if 'HandlerService' not in services:
|
||||
services.append('HandlerService')
|
||||
dirty = True
|
||||
if 'policy' not in config:
|
||||
config['policy'] = {
|
||||
"levels": {"0": {"statsUserUplink": True, "statsUserDownlink": True}},
|
||||
"system": {"statsInboundUplink": True, "statsInboundDownlink": True, "statsOutboundUplink": True, "statsOutboundDownlink": True}
|
||||
}
|
||||
dirty = True
|
||||
if 'routing' not in config:
|
||||
config['routing'] = {"rules": [{"inboundTag": ["api"], "outboundTag": "api", "type": "field"}]}
|
||||
dirty = True
|
||||
|
||||
has_api_inbound = any(ib.get('tag') == 'api' for ib in config.get('inbounds', []))
|
||||
if not has_api_inbound:
|
||||
config.setdefault('inbounds', []).append({
|
||||
"listen": "127.0.0.1",
|
||||
"port": 10085,
|
||||
"protocol": "dokodemo-door",
|
||||
"settings": {"address": "127.0.0.1"},
|
||||
"tag": "api"
|
||||
})
|
||||
dirty = True
|
||||
|
||||
for ib in config.get('inbounds', []):
|
||||
if ib.get('protocol') == 'vless':
|
||||
if not ib.get('tag'):
|
||||
ib['tag'] = 'proxy'
|
||||
dirty = True
|
||||
for c in ib.get('settings', {}).get('clients', []):
|
||||
if 'email' not in c:
|
||||
c['email'] = c['id']
|
||||
dirty = True
|
||||
|
||||
if dirty:
|
||||
self._write_server_json(config, restart=restart)
|
||||
return dirty
|
||||
|
||||
def _query_xray_stats(self):
|
||||
"""Query Xray API for traffic stats using xray api command."""
|
||||
out, _, code = self.ssh.run_sudo_command(
|
||||
f"docker exec -i {self.container_name} /usr/bin/xray api statsquery -server=127.0.0.1:10085"
|
||||
)
|
||||
if code != 0 or not out.strip():
|
||||
return {}
|
||||
|
||||
try:
|
||||
stats_raw = json.loads(out)
|
||||
except Exception:
|
||||
return {}
|
||||
|
||||
results = {}
|
||||
# Output format: {"stat": [{"name": "user>>>uid>>>traffic>>>downlink", "value": "123"}, ...]}
|
||||
for item in stats_raw.get('stat', []):
|
||||
name_parts = item.get('name', '').split('>>>')
|
||||
if len(name_parts) == 4 and name_parts[0] == 'user':
|
||||
uid = name_parts[1]
|
||||
t_type = name_parts[3] # uplink or downlink
|
||||
val = int(item.get('value', 0))
|
||||
|
||||
if uid not in results:
|
||||
results[uid] = {'rx': 0, 'tx': 0}
|
||||
|
||||
if t_type == 'downlink':
|
||||
results[uid]['rx'] = val
|
||||
elif t_type == 'uplink':
|
||||
results[uid]['tx'] = val
|
||||
|
||||
return results
|
||||
|
||||
def _format_bytes(self, size):
|
||||
# Format bytes to string like AWG (e.g., 1.50 MiB)
|
||||
power = 2**10
|
||||
n = 0
|
||||
powers = {0: 'B', 1: 'KiB', 2: 'MiB', 3: 'GiB', 4: 'TiB'}
|
||||
while size > power:
|
||||
size /= power
|
||||
n += 1
|
||||
v = round(size, 2)
|
||||
if v == int(v):
|
||||
v = int(v)
|
||||
return f"{v} {powers.get(n, 'B')}"
|
||||
|
||||
def get_clients(self, protocol=None):
|
||||
config = self._get_server_json()
|
||||
if not config:
|
||||
return []
|
||||
|
||||
self._upgrade_config_for_stats(config, restart=False)
|
||||
|
||||
# Collect all client IDs currently registered in the Xray server config
|
||||
xray_clients = []
|
||||
for ib in config.get('inbounds', []):
|
||||
if ib.get('protocol') == 'vless':
|
||||
xray_clients.extend(ib.get('settings', {}).get('clients', []))
|
||||
|
||||
clients_table = self._get_clients_table()
|
||||
table_ids = {c['clientId'] for c in clients_table}
|
||||
|
||||
# Auto-import clients present in server.json but missing from clientsTable
|
||||
# (e.g. added via the native Amnezia phone/desktop app). Skip the install-time
|
||||
# default UUID for native-layout installs — the official client treats it as
|
||||
# the device of the user who installed the server, not a manageable client.
|
||||
default_uuid = self._get_default_xray_uuid()
|
||||
updated = False
|
||||
for xc in xray_clients:
|
||||
uid = xc.get('id')
|
||||
if not uid or uid in table_ids or uid == default_uuid:
|
||||
continue
|
||||
clients_table.append({
|
||||
'clientId': uid,
|
||||
'userData': {
|
||||
'clientName': f'Imported_{uid[:8]}',
|
||||
'creationDate': datetime.now().isoformat(),
|
||||
'enabled': True
|
||||
}
|
||||
})
|
||||
table_ids.add(uid)
|
||||
updated = True
|
||||
logger.info(f"Auto-imported Xray client {uid[:8]} from server.json")
|
||||
|
||||
if updated:
|
||||
self._save_clients_table(clients_table)
|
||||
|
||||
stats = self._query_xray_stats()
|
||||
|
||||
for c in clients_table:
|
||||
uid = c.get('clientId', '')
|
||||
if uid in stats:
|
||||
user_data = c.setdefault('userData', {})
|
||||
rx = stats[uid]['rx']
|
||||
tx = stats[uid]['tx']
|
||||
if rx > 0 or tx > 0:
|
||||
user_data['dataReceived'] = self._format_bytes(rx)
|
||||
user_data['dataSent'] = self._format_bytes(tx)
|
||||
user_data['dataReceivedBytes'] = rx
|
||||
user_data['dataSentBytes'] = tx
|
||||
|
||||
return clients_table
|
||||
|
||||
def get_client_config(self, protocol, client_id, server_host, port):
|
||||
clients = self._get_clients_table()
|
||||
client = next((c for c in clients if c['clientId'] == client_id), None)
|
||||
if not client: return None
|
||||
|
||||
meta = self._get_meta_json()
|
||||
if not meta: return None
|
||||
|
||||
config = self._get_server_json()
|
||||
sni = meta.get('site_name', 'yahoo.com')
|
||||
if config:
|
||||
try:
|
||||
sni = config['inbounds'][0]['streamSettings']['realitySettings']['serverNames'][0]
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Format URL
|
||||
# vless://{id}@{host}:{port}?type=tcp&security=reality&pbk={public_key}&sni={sni}&fp=chrome&sid={short_id}&spx=%2F&flow=xtls-rprx-vision#{name}
|
||||
|
||||
name = client.get('userData', {}).get('clientName', 'vpn')
|
||||
encoded_name = urllib.parse.quote(name)
|
||||
|
||||
url = (
|
||||
f"vless://{client_id}@{server_host}:{meta.get('port', port)}"
|
||||
f"?type=tcp&security=reality&pbk={meta['public_key']}"
|
||||
f"&sni={sni}&fp=chrome&sid={meta['short_id']}"
|
||||
f"&spx=%2F&flow=xtls-rprx-vision#{encoded_name}"
|
||||
)
|
||||
return url
|
||||
|
||||
def add_client(self, protocol, client_name, server_host, port):
|
||||
client_id = str(uuid.uuid4())
|
||||
|
||||
config = self._get_server_json()
|
||||
if not config: raise RuntimeError("Xray server config not found.")
|
||||
|
||||
self._upgrade_config_for_stats(config, restart=False)
|
||||
|
||||
inbound = self._get_vless_inbound(config)
|
||||
if not inbound:
|
||||
raise RuntimeError("Xray VLESS inbound not found.")
|
||||
|
||||
# Ensure clients structure exists
|
||||
clients_node = inbound.setdefault('settings', {}).setdefault('clients', [])
|
||||
client = {
|
||||
"id": client_id,
|
||||
"flow": "xtls-rprx-vision",
|
||||
"email": client_id
|
||||
}
|
||||
if not self._xray_api_add_user(config, client):
|
||||
raise RuntimeError(
|
||||
"Xray runtime API is not available for hot user updates. "
|
||||
"The server config was upgraded, but the container must be restarted once to enable HandlerService. "
|
||||
"Restart the Xray container manually and try again."
|
||||
)
|
||||
clients_node.append(client)
|
||||
self._write_server_json(config, restart=False)
|
||||
|
||||
# Update table
|
||||
clients_table = self._get_clients_table()
|
||||
clients_table.append({
|
||||
'clientId': client_id,
|
||||
'userData': {
|
||||
'clientName': client_name,
|
||||
'creationDate': datetime.now().isoformat(),
|
||||
'enabled': True
|
||||
}
|
||||
})
|
||||
self._save_clients_table(clients_table)
|
||||
|
||||
return {
|
||||
'client_id': client_id,
|
||||
'config': self.get_client_config(protocol, client_id, server_host, port)
|
||||
}
|
||||
|
||||
def toggle_client(self, protocol, client_id, enable):
|
||||
config = self._get_server_json()
|
||||
self._upgrade_config_for_stats(config, restart=False)
|
||||
inbound = self._get_vless_inbound(config)
|
||||
if not inbound:
|
||||
raise RuntimeError("Xray VLESS inbound not found.")
|
||||
clients_node = inbound.setdefault('settings', {}).setdefault('clients', [])
|
||||
|
||||
# If toggling on and not present, we can re-add it from clientsTable
|
||||
if enable:
|
||||
if not any(c['id'] == client_id for c in clients_node):
|
||||
client = {
|
||||
"id": client_id,
|
||||
"flow": "xtls-rprx-vision",
|
||||
"email": client_id
|
||||
}
|
||||
if not self._xray_api_add_user(config, client):
|
||||
raise RuntimeError("Xray runtime API failed to enable the client without restarting the container.")
|
||||
clients_node.append(client)
|
||||
else:
|
||||
if not self._xray_api_remove_user(config, client_id):
|
||||
raise RuntimeError("Xray runtime API failed to disable the client without restarting the container.")
|
||||
inbound['settings']['clients'] = [c for c in clients_node if c['id'] != client_id]
|
||||
|
||||
self._write_server_json(config, restart=False)
|
||||
|
||||
clients_table = self._get_clients_table()
|
||||
for c in clients_table:
|
||||
if c['clientId'] == client_id:
|
||||
c.setdefault('userData', {})['enabled'] = enable
|
||||
self._save_clients_table(clients_table)
|
||||
|
||||
def remove_client(self, protocol, client_id):
|
||||
config = self._get_server_json()
|
||||
self._upgrade_config_for_stats(config, restart=False)
|
||||
inbound = self._get_vless_inbound(config)
|
||||
if not inbound:
|
||||
raise RuntimeError("Xray VLESS inbound not found.")
|
||||
clients_node = inbound.setdefault('settings', {}).setdefault('clients', [])
|
||||
if not self._xray_api_remove_user(config, client_id):
|
||||
raise RuntimeError("Xray runtime API failed to remove the client without restarting the container.")
|
||||
inbound['settings']['clients'] = [c for c in clients_node if c['id'] != client_id]
|
||||
self._write_server_json(config, restart=False)
|
||||
|
||||
clients_table = self._get_clients_table()
|
||||
clients_table = [c for c in clients_table if c['clientId'] != client_id]
|
||||
self._save_clients_table(clients_table)
|
||||
return True
|
||||
Reference in New Issue
Block a user