|
|
#!/usr/bin/env bash
|
|
|
set -euo pipefail
|
|
|
|
|
|
# 测试 open-im-server 手机号 is_snd 接口（/phone/get_sn_info、/phone/set_sn_info）
|
|
|
#
|
|
|
# 用法：
|
|
|
# 查询（无需 token，已加入 GinParseToken 白名单）：
|
|
|
# ./scripts/phone_sn_api.sh get <phone>
|
|
|
#
|
|
|
# 写入（需要用户 token）：
|
|
|
# OPENIM_TOKEN="<用户 token>" ./scripts/phone_sn_api.sh set <phone> <userID整数> <is_snd:0|1|true|false>
|
|
|
#
|
|
|
# 环境变量（可覆盖）：
|
|
|
# OPENIM_API_ADDR 默认 http://127.0.0.1:10002
|
|
|
# OPENIM_TOKEN set 时必填（Header: token）
|
|
|
# OPERATION_ID 默认自动生成
|
|
|
|
|
|
OPENIM_API_ADDR="${OPENIM_API_ADDR:-http://127.0.0.1:10002}"
|
|
|
OPENIM_TOKEN="${OPENIM_TOKEN:-}"
|
|
|
OPERATION_ID="${OPERATION_ID:-phone_sn_$(date +%s)_$RANDOM}"
|
|
|
|
|
|
ACTION="${1:-}"
|
|
|
PHONE="${2:-}"
|
|
|
USER_ID="${3:-}"
|
|
|
IS_SND_RAW="${4:-}"
|
|
|
|
|
|
die() {
|
|
|
echo "ERROR: $*" >&2
|
|
|
exit 1
|
|
|
}
|
|
|
|
|
|
usage() {
|
|
|
cat <<'EOF'
|
|
|
用法：
|
|
|
查询（无需 token）：
|
|
|
./scripts/phone_sn_api.sh get <phone>
|
|
|
|
|
|
写入（需要 OPENIM_TOKEN）：
|
|
|
OPENIM_TOKEN="<用户token>" ./scripts/phone_sn_api.sh set <phone> <userID整数> <is_snd:0|1|true|false>
|
|
|
|
|
|
环境变量：
|
|
|
OPENIM_API_ADDR 默认 http://127.0.0.1:10002
|
|
|
OPENIM_TOKEN set 时必填
|
|
|
OPERATION_ID 可选，默认自动生成
|
|
|
EOF
|
|
|
}
|
|
|
|
|
|
curl_post() {
|
|
|
local path=$1
|
|
|
local json_body=$2
|
|
|
local with_token=${3:-0}
|
|
|
local -a hdrs=(
|
|
|
-H "Content-Type: application/json"
|
|
|
-H "operationID: ${OPERATION_ID}"
|
|
|
)
|
|
|
if [[ "$with_token" == "1" ]]; then
|
|
|
[[ -n "$OPENIM_TOKEN" ]] || die "set 接口需要环境变量 OPENIM_TOKEN（用户 token）"
|
|
|
hdrs+=(-H "token: ${OPENIM_TOKEN}")
|
|
|
fi
|
|
|
curl -sS "${hdrs[@]}" -X POST "${OPENIM_API_ADDR}${path}" -d "${json_body}"
|
|
|
}
|
|
|
|
|
|
pretty_json() {
|
|
|
if command -v jq >/dev/null 2>&1; then
|
|
|
jq .
|
|
|
else
|
|
|
cat
|
|
|
fi
|
|
|
}
|
|
|
|
|
|
case "$ACTION" in
|
|
|
get)
|
|
|
[[ -n "$PHONE" ]] || die "用法: $0 get <phone>"
|
|
|
body=$(printf '{"phone":"%s"}' "$PHONE")
|
|
|
echo "==> POST ${OPENIM_API_ADDR}/phone/get_sn_info"
|
|
|
echo " body: ${body}"
|
|
|
resp=$(curl_post "/phone/get_sn_info" "$body" 0)
|
|
|
echo "$resp" | pretty_json
|
|
|
;;
|
|
|
set)
|
|
|
[[ -n "$PHONE" && -n "$USER_ID" && -n "$IS_SND_RAW" ]] || die "用法: $0 set <phone> <userID> <is_snd:0|1|true|false>"
|
|
|
case "$IS_SND_RAW" in
|
|
|
1|true|True|TRUE) is_snd=true ;;
|
|
|
0|false|False|FALSE) is_snd=false ;;
|
|
|
*) die "is_snd 必须是 0、1、true 或 false" ;;
|
|
|
esac
|
|
|
# userID 必须为 JSON 数字
|
|
|
if ! [[ "$USER_ID" =~ ^-?[0-9]+$ ]]; then
|
|
|
die "userID 必须为整数"
|
|
|
fi
|
|
|
body=$(printf '{"phone":"%s","userID":%s,"is_snd":%s}' "$PHONE" "$USER_ID" "$is_snd")
|
|
|
echo "==> POST ${OPENIM_API_ADDR}/phone/set_sn_info"
|
|
|
echo " body: ${body}"
|
|
|
resp=$(curl_post "/phone/set_sn_info" "$body" 1)
|
|
|
echo "$resp" | pretty_json
|
|
|
;;
|
|
|
""|-h|--help|help)
|
|
|
usage
|
|
|
exit 0
|
|
|
;;
|
|
|
*)
|
|
|
die "未知动作: $ACTION，使用 -h 查看帮助"
|
|
|
;;
|
|
|
esac
|
