250 lines
8.0 KiB
YAML
250 lines
8.0 KiB
YAML
---
|
|
apiVersion: v1
|
|
kind: ConfigMap
|
|
metadata:
|
|
name: pasarguard-scripts
|
|
labels:
|
|
app: pasarguard-node
|
|
data:
|
|
init-uuid.sh: |
|
|
#!/bin/bash
|
|
set -e
|
|
echo "Started"
|
|
# NODE_NAME is already set via environment variable
|
|
NAMESPACE=$(cat /var/run/secrets/kubernetes.io/serviceaccount/namespace)
|
|
|
|
# Get DNS name from node label xray-node-address
|
|
DNS_NAME=$(kubectl get node "${NODE_NAME}" -o jsonpath='{.metadata.labels.xray-node-address}')
|
|
|
|
if [ -z "${DNS_NAME}" ]; then
|
|
echo "ERROR: Node ${NODE_NAME} does not have label 'xray-node-address'"
|
|
exit 1
|
|
fi
|
|
|
|
echo "Node: ${NODE_NAME}"
|
|
echo "DNS Name from label: ${DNS_NAME}"
|
|
|
|
# Use DNS name for ConfigMap name to ensure uniqueness
|
|
CONFIGMAP_NAME="node-uuid-${DNS_NAME//./-}"
|
|
|
|
echo "Checking ConfigMap: ${CONFIGMAP_NAME}"
|
|
|
|
# Check if ConfigMap exists and get UUID
|
|
if kubectl get configmap "${CONFIGMAP_NAME}" -n "${NAMESPACE}" &>/dev/null; then
|
|
echo "ConfigMap exists, reading UUID..."
|
|
API_KEY=$(kubectl get configmap "${CONFIGMAP_NAME}" -n "${NAMESPACE}" -o jsonpath='{.data.API_KEY}')
|
|
|
|
if [ -z "${API_KEY}" ]; then
|
|
echo "UUID not found in ConfigMap, generating new one..."
|
|
API_KEY=$(cat /proc/sys/kernel/random/uuid)
|
|
kubectl patch configmap "${CONFIGMAP_NAME}" -n "${NAMESPACE}" --type merge -p "{\"data\":{\"API_KEY\":\"${API_KEY}\"}}"
|
|
else
|
|
echo "Using existing UUID from ConfigMap"
|
|
fi
|
|
else
|
|
echo "ConfigMap does not exist, creating new one..."
|
|
API_KEY=$(cat /proc/sys/kernel/random/uuid)
|
|
kubectl create configmap "${CONFIGMAP_NAME}" -n "${NAMESPACE}" \
|
|
--from-literal=API_KEY="${API_KEY}" \
|
|
--from-literal=NODE_NAME="${NODE_NAME}"
|
|
fi
|
|
|
|
# Save UUID and node info to shared volume for the main container
|
|
echo -n "${API_KEY}" > /shared/api-key
|
|
echo -n "${NODE_NAME}" > /shared/node-name
|
|
echo -n "${CONFIGMAP_NAME}" > /shared/configmap-name
|
|
echo "UUID initialized: ${API_KEY}"
|
|
echo "Node name: ${NODE_NAME}"
|
|
echo "ConfigMap: ${CONFIGMAP_NAME}"
|
|
|
|
# Create Certificate for this node using DNS name from label
|
|
CERT_NAME="pasarguard-node-${DNS_NAME//./-}"
|
|
|
|
echo "Creating Certificate: ${CERT_NAME} for ${DNS_NAME}"
|
|
|
|
# Check if Certificate already exists
|
|
if ! kubectl get certificate "${CERT_NAME}" -n "${NAMESPACE}" &>/dev/null; then
|
|
echo "Certificate does not exist, creating..."
|
|
cat <<EOF | kubectl apply -f -
|
|
apiVersion: cert-manager.io/v1
|
|
kind: Certificate
|
|
metadata:
|
|
name: ${CERT_NAME}
|
|
namespace: ${NAMESPACE}
|
|
spec:
|
|
secretName: ${CERT_NAME}-tls
|
|
issuerRef:
|
|
name: letsencrypt
|
|
kind: ClusterIssuer
|
|
dnsNames:
|
|
- ${DNS_NAME}
|
|
EOF
|
|
else
|
|
echo "Certificate already exists"
|
|
fi
|
|
|
|
# Wait for certificate to be ready
|
|
|
|
echo "Waiting for certificate to be ready..."
|
|
for i in {1..600}; do
|
|
if kubectl get secret "${CERT_NAME}-tls" -n "${NAMESPACE}" &>/dev/null; then
|
|
echo "Certificate secret is ready!"
|
|
break
|
|
fi
|
|
echo "Waiting for certificate... ($i/600)"
|
|
sleep 1
|
|
done
|
|
|
|
if ! kubectl get secret "${CERT_NAME}-tls" -n "${NAMESPACE}" &>/dev/null; then
|
|
echo "WARNING: Certificate secret not ready after 600 seconds"
|
|
else
|
|
# Extract certificate and key from secret to shared volume
|
|
echo "Extracting certificate and key..."
|
|
kubectl get secret "${CERT_NAME}-tls" -n "${NAMESPACE}" -o jsonpath='{.data.tls\.crt}' | base64 -d > /shared/tls.crt
|
|
kubectl get secret "${CERT_NAME}-tls" -n "${NAMESPACE}" -o jsonpath='{.data.tls\.key}' | base64 -d > /shared/tls.key
|
|
echo "Certificate and key extracted successfully."
|
|
cat /shared/tls.crt
|
|
fi
|
|
|
|
# Create individual Service and Endpoints for this node
|
|
# Take only first part of node name before first dot
|
|
NODE_SHORT_NAME="${NODE_NAME%%.*}"
|
|
SERVICE_NAME="${NODE_SHORT_NAME}"
|
|
|
|
# Get node internal IP (take only first IP if multiple)
|
|
NODE_IP=$(kubectl get node "${NODE_NAME}" -o jsonpath='{.status.addresses[?(@.type=="InternalIP")].address}' | awk '{print $1}')
|
|
|
|
echo "Creating Service: ${SERVICE_NAME} for node ${NODE_NAME} (short: ${NODE_SHORT_NAME}) with IP ${NODE_IP}"
|
|
|
|
# Create Service without selector
|
|
cat <<EOF | kubectl apply -f -
|
|
apiVersion: v1
|
|
kind: Service
|
|
metadata:
|
|
name: ${SERVICE_NAME}
|
|
namespace: ${NAMESPACE}
|
|
labels:
|
|
app: pasarguard-node
|
|
node: ${NODE_NAME}
|
|
spec:
|
|
clusterIP: None
|
|
ports:
|
|
- name: api
|
|
port: 62050
|
|
protocol: TCP
|
|
targetPort: 62050
|
|
- name: metrics
|
|
port: 9550
|
|
protocol: TCP
|
|
targetPort: 9550
|
|
---
|
|
apiVersion: v1
|
|
kind: Endpoints
|
|
metadata:
|
|
name: ${SERVICE_NAME}
|
|
namespace: ${NAMESPACE}
|
|
labels:
|
|
app: pasarguard-node
|
|
node: ${NODE_NAME}
|
|
subsets:
|
|
- addresses:
|
|
- ip: ${NODE_IP}
|
|
nodeName: ${NODE_NAME}
|
|
ports:
|
|
- name: api
|
|
port: 62050
|
|
protocol: TCP
|
|
- name: metrics
|
|
port: 9550
|
|
protocol: TCP
|
|
EOF
|
|
|
|
echo "Service created: ${SERVICE_NAME}.${NAMESPACE}.svc.cluster.local -> ${NODE_IP}:62050"
|
|
|
|
exporter-start.sh: |
|
|
#!/bin/sh
|
|
# Install required tools
|
|
apk add --no-cache wget curl iproute2-ss bash
|
|
|
|
# Download v2ray-exporter
|
|
echo "Downloading v2ray-exporter..."
|
|
wget -O /tmp/v2ray-exporter https://github.com/wi1dcard/v2ray-exporter/releases/latest/download/v2ray-exporter_linux_amd64
|
|
mv /tmp/v2ray-exporter /usr/local/bin/v2ray-exporter
|
|
chmod +x /usr/local/bin/v2ray-exporter
|
|
|
|
# Wait for initial API port file
|
|
echo "Waiting for initial xray API port file..."
|
|
while [ ! -f /shared/xray-api-port ]; do
|
|
echo "Waiting for API port file..."
|
|
sleep 2
|
|
done
|
|
|
|
# Main loop - restart exporter if it crashes or port changes
|
|
while true; do
|
|
if [ -f /shared/xray-api-port ]; then
|
|
API_PORT=$(cat /shared/xray-api-port)
|
|
if [ -n "$API_PORT" ]; then
|
|
echo "Starting v2ray-exporter with endpoint 127.0.0.1:$API_PORT"
|
|
/usr/local/bin/v2ray-exporter --v2ray-endpoint "127.0.0.1:$API_PORT" --listen ":9550" &
|
|
EXPORTER_PID=$!
|
|
|
|
# Wait for exporter to exit or port file to change
|
|
while kill -0 $EXPORTER_PID 2>/dev/null; do
|
|
if [ -f /shared/xray-api-port ]; then
|
|
NEW_PORT=$(cat /shared/xray-api-port)
|
|
if [ "$NEW_PORT" != "$API_PORT" ]; then
|
|
echo "API port changed from $API_PORT to $NEW_PORT, restarting exporter"
|
|
kill $EXPORTER_PID 2>/dev/null
|
|
wait $EXPORTER_PID 2>/dev/null
|
|
break
|
|
fi
|
|
fi
|
|
sleep 5
|
|
done
|
|
|
|
echo "Exporter stopped, restarting..."
|
|
wait $EXPORTER_PID 2>/dev/null
|
|
fi
|
|
fi
|
|
sleep 2
|
|
done
|
|
|
|
pasarguard-start.sh: |
|
|
#!/bin/sh
|
|
# Read API_KEY from shared volume created by init container
|
|
if [ -f /shared/api-key ]; then
|
|
export API_KEY=$(cat /shared/api-key)
|
|
echo "Loaded API_KEY from shared volume"
|
|
else
|
|
echo "WARNING: API_KEY file not found, using default"
|
|
fi
|
|
|
|
cd /app
|
|
|
|
# Start main process in background
|
|
./main &
|
|
MAIN_PID=$!
|
|
|
|
# Start continuous port monitoring in background
|
|
{
|
|
sleep 10 # Wait for xray to start initially
|
|
LAST_PORT=""
|
|
|
|
while true; do
|
|
API_PORT=$(netstat -tlpn | grep xray | grep 127.0.0.1 | awk '{print $4}' | cut -d: -f2 | head -1)
|
|
if [ -n "$API_PORT" ] && [ "$API_PORT" != "$LAST_PORT" ]; then
|
|
echo "Found xray API port: $API_PORT"
|
|
echo -n "$API_PORT" > /shared/xray-api-port
|
|
LAST_PORT="$API_PORT"
|
|
fi
|
|
sleep 5 # Check every 5 seconds
|
|
done
|
|
} &
|
|
PORT_MONITOR_PID=$!
|
|
|
|
# Wait for main process to finish
|
|
wait $MAIN_PID
|
|
|
|
# Clean up port monitor
|
|
kill $PORT_MONITOR_PID 2>/dev/null
|