go-edge-shell/goedge_add_node.sh

#!/usr/bin/env bash

set -e

###############################################################################
# 函数：检查并安装依赖
###############################################################################
function install_deps_if_needed() {
  local packages=("curl" "jq" "base64" "unzip")

  for pkg in "${packages[@]}"; do
    if ! command -v "$pkg" &> /dev/null; then
      echo "[INFO] 检测到命令 '$pkg' 不存在，需要安装..."

      if command -v apt-get &> /dev/null; then
        echo "[INFO] 检测到系统使用 apt-get，开始安装 $pkg"
        sudo apt-get update -y
        sudo apt-get install -y curl jq coreutils unzip
      elif command -v yum &> /dev/null; then
        echo "[INFO] 检测到系统使用 yum，开始安装 $pkg"
        sudo yum install -y curl jq coreutils unzip
      elif command -v dnf &> /dev/null; then
        echo "[INFO] 检测到系统使用 dnf，开始安装 $pkg"
        sudo dnf install -y curl jq coreutils unzip
      elif command -v zypper &> /dev/null; then
        echo "[INFO] 检测到系统使用 zypper，开始安装 $pkg"
        sudo zypper install -y curl jq coreutils unzip
      elif command -v apk &> /dev/null; then
        echo "[INFO] 检测到系统使用 apk (Alpine)，开始安装 $pkg"
        sudo apk add --no-cache curl jq coreutils unzip
      else
        echo "[ERROR] 未识别的包管理器，无法自动安装 $pkg，请手动安装。"
        exit 1
      fi

      if ! command -v "$pkg" &> /dev/null; then
        echo "[ERROR] 安装 $pkg 失败，请手动安装后再试。"
        exit 1
      fi
    fi
  done
}

###############################################################################
# 主体脚本
###############################################################################
install_deps_if_needed  # 1. 安装依赖

###############################################################################
# 2. 解析参数
###############################################################################
ACID=""
ACKEY=""
NAME=""
CLUSTER_ID=""
GROUP_ID=0
REGION_ID=0
DNS_ROUTES=""

# HTTP_API: 必填，用于脚本调用HTTP接口
HTTP_API=""

# RPC endpoints: 用于写入 api_node.yaml，可多地址
API_HOSTS=()

GRANT_ID=1       # 默认
WEB_INSTALL=0    # 是否使用 --webinstall

while [[ $# -gt 0 ]]; do
  case $1 in
    --acid)
      ACID="$2"
      shift 2
      ;;
    --ackey)
      ACKEY="$2"
      shift 2
      ;;
    --name)
      NAME="$2"
      shift 2
      ;;
    --clusterid)
      CLUSTER_ID="$2"
      shift 2
      ;;
    --grantid)
      GRANT_ID="$2"
      shift 2
      ;;
    --groupid)
      GROUP_ID="$2"
      shift 2
      ;;
    --regionid)
      REGION_ID="$2"
      shift 2
      ;;
    --dnsroutes)
      DNS_ROUTES="$2"
      shift 2
      ;;
    --http-api)
      HTTP_API="$2"
      shift 2
      ;;
    --api-host)
      # 将后续所有非开头带 "--" 的参数视为RPC地址(可多个)
      shift
      while [[ $# -gt 0 && ! $1 =~ ^-- ]]; do
        API_HOSTS+=("$1")
        shift
      done
      ;;
    --webinstall)
      WEB_INSTALL=1
      shift
      ;;
    *)
      echo "[ERROR] 未知参数: $1"
      exit 1
      ;;
  esac
done

# 校验必填: acid, ackey, clusterid, http-api
if [[ -z "$ACID" || -z "$ACKEY" || -z "$CLUSTER_ID" ]]; then
  echo "[ERROR] 用法: $0 --acid <ACCESS_KEY_ID> --ackey <ACCESS_KEY> --clusterid <CLUSTER_ID> --http-api <HTTP_API> [--grantid <GRANT_ID>] [--name <NAME>] [--groupid <GROUP_ID>] [--regionid <REGION_ID>] [--dnsroutes <DNS_ROUTES>] [--api-host <RPC1> [RPC2]...] [--webinstall]"
  exit 1
fi

if [[ -z "$HTTP_API" ]]; then
  echo "[ERROR] 必须指定 --http-api <HTTP_API>"
  exit 1
fi

###############################################################################
# 3. 获取本机 IPv4, 拼装默认名称
###############################################################################
echo "[INFO] 获取本机 IPv4..."
LOCAL_IPV4=$(curl -s 4.ipw.cn || true)
if [[ -z "$LOCAL_IPV4" ]]; then
  LOCAL_IPV4="0.0.0.0"
  echo "[WARN] 无法获取IPv4，使用默认: $LOCAL_IPV4"
fi
echo "[INFO] 本机 IPv4: $LOCAL_IPV4"

if [[ -z "$NAME" ]]; then
  DATE_STR=$(date +%Y-%m-%d)
  NAME="${LOCAL_IPV4}(${DATE_STR})"
  echo "[INFO] 未指定 --name，自动设置 NAME=$NAME"
fi

###############################################################################
# 4. 获取 Token (使用 HTTP_API)
###############################################################################
echo "[INFO] 获取 Token..."
get_token_resp=$(curl -s -X POST \
  -H "Content-Type: application/json" \
  "$HTTP_API/APIAccessTokenService/getAPIAccessToken" \
  -d '{
    "type": "admin",
    "accessKeyId": "'"$ACID"'",
    "accessKey": "'"$ACKEY"'"
}')
get_token_code=$(echo "$get_token_resp" | jq -r '.code')
if [ "$get_token_code" != "200" ]; then
  echo "[ERROR] 获取Token失败: $get_token_resp"
  exit 1
fi
TOKEN=$(echo "$get_token_resp" | jq -r '.data.token')
echo "[INFO] 成功获取 Token"

###############################################################################
# 5. 创建节点 (createNode)
###############################################################################
echo "[INFO] 正在创建节点..."

if [[ -n "$DNS_ROUTES" ]]; then
  dns_routes_json_array=$(echo "$DNS_ROUTES" | sed 's/ /","/g')
  dns_routes_json_array="[\"$dns_routes_json_array\"]"
else
  dns_routes_json_array="[]"
fi

node_login_json=$(cat <<EOF
{
  "host": "$LOCAL_IPV4",
  "port": 22,
  "grantId": $GRANT_ID
}
EOF
)
BASE64_PARAMS=$(echo -n "$node_login_json" | base64 | tr -d '\n')

create_node_payload=$(cat <<EOF
{
  "name": "$NAME",
  "nodeClusterId": $CLUSTER_ID,
  "nodeLogin": {
    "id": 3,
    "name": "SSH",
    "type": "ssh",
    "params": "$BASE64_PARAMS"
  },
  "nodeGroupId": $GROUP_ID,
  "dnsRoutes": $dns_routes_json_array,
  "nodeRegionId": $REGION_ID
}
EOF
)

create_node_resp=$(curl -s -X POST \
  -H "Content-Type: application/json" \
  -H "X-Edge-Access-Token: $TOKEN" \
  "$HTTP_API/NodeService/createNode" \
  -d "$create_node_payload"
)
create_node_code=$(echo "$create_node_resp" | jq -r '.code')
if [ "$create_node_code" != "200" ]; then
  echo "[ERROR] 创建节点失败: $create_node_resp"
  exit 1
fi
NODE_ID=$(echo "$create_node_resp" | jq -r '.data.nodeId')
echo "[INFO] 节点创建成功, nodeId=$NODE_ID"

###############################################################################
# 6. 批量创建 IP (createNodeIPAddresses)
###############################################################################
echo "[INFO] 添加节点IP地址..."
LOCAL_IPV6=$(curl -s 6.ipw.cn || true)
if [[ -z "$LOCAL_IPV6" ]]; then
  echo "[WARN] 获取 IPv6 失败"
fi
echo "[INFO] IPv6: $LOCAL_IPV6"

ip_array='["'"$LOCAL_IPV4"'"]'
if [[ -n "$LOCAL_IPV6" ]]; then
  ip_array='["'"$LOCAL_IPV4"'","'"$LOCAL_IPV6"'"]'
fi

create_ips_payload=$(cat <<EOF
{
  "nodeId": $NODE_ID,
  "role": "node",
  "name": "auto-ips",
  "ipList": $ip_array,
  "canAccess": true,
  "isUp": true,
  "groupValue": "",
  "nodeClusterIds": []
}
EOF
)

create_ips_resp=$(curl -s -X POST \
  -H "Content-Type: application/json" \
  -H "X-Edge-Access-Token: $TOKEN" \
  "$HTTP_API/NodeIPAddressService/createNodeIPAddresses" \
  -d "$create_ips_payload"
)
create_ips_code=$(echo "$create_ips_resp" | jq -r '.code')
if [ "$create_ips_code" != "200" ]; then
  echo "[ERROR] 创建 IP 失败: $create_ips_resp"
  exit 1
fi
ADDR_IDS=$(echo "$create_ips_resp" | jq -r '.data.nodeIPAddressIds')
echo "[INFO] 批量添加 IP 完成"

###############################################################################
# 7. 获取节点信息 (findEnabledNode)
###############################################################################
echo "[INFO] 获取节点信息..."
find_node_resp=$(curl -s -X POST \
  -H "Content-Type: application/json" \
  -H "X-Edge-Access-Token: $TOKEN" \
  "$HTTP_API/NodeService/findEnabledNode" \
  -d '{
    "nodeId": '"$NODE_ID"'
  }'
)
find_node_code=$(echo "$find_node_resp" | jq -r '.code')
if [ "$find_node_code" != "200" ]; then
  echo "[ERROR] 获取节点信息失败: $find_node_resp"
  exit 1
fi

node_info=$(echo "$find_node_resp" | jq -r '.data.node')
NODE_UNIQUE_ID=$(echo "$node_info" | jq -r '.uniqueId')
NODE_SECRET=$(echo "$node_info" | jq -r '.secret')

###############################################################################
# 8. 根据 --webinstall 判断自动/手动安装
###############################################################################
if [[ "$WEB_INSTALL" -eq 1 ]]; then
  # -------------------- 自动(网页)安装 -------------------- #
  echo "[INFO] --webinstall 已指定, 开始调用 /NodeService/installNode..."

  install_node_payload=$(cat <<EOF
{
  "nodeId": $NODE_ID
}
EOF
)
  install_node_resp=$(curl -s -X POST \
    -H "Content-Type: application/json" \
    -H "X-Edge-Access-Token: $TOKEN" \
    "$HTTP_API/NodeService/installNode" \
    -d "$install_node_payload"
  )
  install_node_code=$(echo "$install_node_resp" | jq -r '.code')
  if [ "$install_node_code" != "200" ]; then
    echo "[ERROR] installNode 失败: $install_node_resp"
    exit 1
  fi

else
  # -------------------- 手动安装模式 -------------------- #
  echo "[INFO] 未指定 --webinstall，进入手动安装流程..."

  if [[ ! -d /usr/local/goedge ]]; then
    sudo mkdir -p /usr/local/goedge
  fi

  cd /usr/local/goedge

  ARCH=$(uname -m)
  if [[ "$ARCH" == "x86_64" ]]; then
    echo "[INFO] x86_64 -> 下载 amd64"
    sudo curl -L -o edge-node.zip "https://git.randallanjie.com/Randall/go-edge-shell/releases/download/edge-node/edge-node-linux-amd64-plus-v1.3.9.zip"
  elif [[ "$ARCH" == "aarch64" ]]; then
    echo "[INFO] aarch64 -> 下载 arm64"
    sudo curl -L -o edge-node.zip "https://git.randallanjie.com/Randall/go-edge-shell/releases/download/edge-node/edge-node-linux-arm64-plus-v1.3.9.zip"
  else
    echo "[WARN] 未知架构 $ARCH -> 默认amd64"
    sudo curl -L -o edge-node.zip "https://git.randallanjie.com/Randall/go-edge-shell/releases/download/edge-node/edge-node-linux-amd64-plus-v1.3.9.zip"
  fi

  sudo unzip -o edge-node.zip

  RPC_ENDPOINTS_JSON='[]'
  if [[ ${#API_HOSTS[@]} -gt 0 ]]; then
    RPC_ENDPOINTS_JSON='['
    for ((i=0; i<${#API_HOSTS[@]}; i++)); do
      RPC_ENDPOINTS_JSON+="\"${API_HOSTS[i]}\""
      if [[ $i -lt $(( ${#API_HOSTS[@]} - 1 )) ]]; then
        RPC_ENDPOINTS_JSON+=","
      fi
    done
    RPC_ENDPOINTS_JSON+=']'
  fi

  if [[ ! -f /usr/local/goedge/edge-node/configs/api_node.yaml ]]; then
    sudo tee /usr/local/goedge/edge-node/configs/api_node.yaml >/dev/null <<EOT
rpc.endpoints: [ "" ]
nodeId: ""
secret: ""
EOT
  fi

  sudo sed -i "s|^rpc.endpoints:.*|rpc.endpoints: $RPC_ENDPOINTS_JSON|g" /usr/local/goedge/edge-node/configs/api_node.yaml
  sudo sed -i "s|^nodeId:.*|nodeId: \"$NODE_UNIQUE_ID\"|g" /usr/local/goedge/edge-node/configs/api_node.yaml
  sudo sed -i "s|^secret:.*|secret: \"$NODE_SECRET\"|g" /usr/local/goedge/edge-node/configs/api_node.yaml

  cd /usr/local/goedge/edge-node/bin
  sudo ./edge-node service
  sudo systemctl daemon-reload
  sudo systemctl start edge-node.service
  sudo systemctl enable edge-node.service

  update_payload=$(cat <<EOF
{
  "nodeId": $NODE_ID,
  "isInstalled": true
}
EOF
)
  update_resp=$(curl -s -X POST \
    -H "Content-Type: application/json" \
    -H "X-Edge-Access-Token: $TOKEN" \
    "$HTTP_API/NodeService/updateNodeIsInstalled" \
    -d "$update_payload"
  )
  update_code=$(echo "$update_resp" | jq -r '.code')
  if [ "$update_code" != "200" ]; then
    echo "[ERROR] updateNodeIsInstalled 失败: $update_resp"
    exit 1
  fi
fi

echo "[INFO] 脚本执行完成。"
exit 0