Create removeclient.sh

awg/removeclient.sh

@@ -0,0 +1,92 @@
+#!/bin/bash
+
+set -e
+
+if [ "$#" -ne 4 ]; then
+    echo "Usage: $0 CLIENT_NAME CLIENT_PUBLIC_KEY WG_CONFIG_FILE DOCKER_CONTAINER"
+    exit 1
+fi
+
+CLIENT_NAME="$1"
+CLIENT_PUBLIC_KEY="$2"
+WG_CONFIG_FILE="$3"
+DOCKER_CONTAINER="$4"
+
+pwd=$(pwd)
+mkdir -p "$pwd/files"
+SERVER_CONF_PATH="$pwd/files/server.conf"
+
+docker exec -i "$DOCKER_CONTAINER" cat "$WG_CONFIG_FILE" > "$SERVER_CONF_PATH"
+
+awk -v pubkey="$CLIENT_PUBLIC_KEY" '
+BEGIN {in_peer=0; skip=0}
+/^\[Peer\]/ {
+    in_peer=1
+    peer_block = $0 "\n"
+    next
+}
+in_peer == 1 {
+    peer_block = peer_block $0 "\n"
+    if ($0 ~ /^PublicKey\s*=/) {
+        split($0, a, " = ")
+        if (a[2] == pubkey) {
+            skip=1
+        }
+    }
+    if ($0 ~ /^\[Peer\]/ || $0 ~ /^\[Interface\]/) {
+        if (skip == 1) {
+            skip=0
+            in_peer=0
+            next
+        } else {
+            print peer_block
+            in_peer=0
+        }
+    }
+    if ($0 == "") {
+        if (skip == 1) {
+            skip=0
+            in_peer=0
+            next
+        } else {
+            print peer_block
+            in_peer=0
+        }
+    }
+    next
+}
+{
+    print
+}
+END {
+    if (in_peer == 1 && skip == 1) {
+    } else if (in_peer ==1 ) {
+        print peer_block
+    }
+}
+' "$SERVER_CONF_PATH" > "$SERVER_CONF_PATH.tmp"
+
+mv "$SERVER_CONF_PATH.tmp" "$SERVER_CONF_PATH"
+
+docker exec -i "$DOCKER_CONTAINER" wg-quick strip "$WG_CONFIG_FILE" > /dev/null
+
+docker cp "$SERVER_CONF_PATH" "$DOCKER_CONTAINER":"$WG_CONFIG_FILE"
+
+docker exec -i "$DOCKER_CONTAINER" sh -c "wg-quick down '$WG_CONFIG_FILE' && wg-quick up '$WG_CONFIG_FILE'"
+
+rm -f "users/$CLIENT_NAME/$CLIENT_NAME.conf"
+rmdir "users/$CLIENT_NAME" 2>/dev/null || true
+
+CLIENTS_TABLE_PATH="$pwd/files/clientsTable"
+docker exec -i "$DOCKER_CONTAINER" cat /opt/amnezia/awg/clientsTable > "$CLIENTS_TABLE_PATH" || echo "[]" > "$CLIENTS_TABLE_PATH"
+
+if [ -f "$CLIENTS_TABLE_PATH" ]; then
+    jq --arg clientId "$CLIENT_PUBLIC_KEY" 'del(.[] | select(.clientId == $clientId))' "$CLIENTS_TABLE_PATH" > "$CLIENTS_TABLE_PATH.tmp"
+    mv "$CLIENTS_TABLE_PATH.tmp" "$CLIENTS_TABLE_PATH"
+    docker cp "$CLIENTS_TABLE_PATH" "$DOCKER_CONTAINER":/opt/amnezia/awg/clientsTable
+fi
+
+traffic_file="$pwd/users/$CLIENT_NAME/traffic.json"
+rm -f "$traffic_file"
+
+echo "Client $CLIENT_NAME успешно удален из WireGuard"