mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-06-17 13:17:46 +08:00
105 lines
2.9 KiB
Bash
Executable File
105 lines
2.9 KiB
Bash
Executable File
#!/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
|