oasis_test_v2 2.2.111 → 2.2.113

package/dist/feishu/wrapper.sh

@@ -10,8 +10,16 @@
 # 环境变量：
 #   LARK_BIN          — 真实 lark-cli 绝对路径（避免递归调用自身）
 #   FEISHU_CREDS_FILE — 凭据文件路径（内含 export LARKSUITE_CLI_* 行）
+#
+# 按 Agent 限制可用模块（产品反馈「飞书连接器·按 Agent 的权限模型」）：只放行所选模块的
+# `lark-cli <domain> …` 与 `lark-cli api <METHOD> /open-apis/<service>/…`，机器人与用户身份一样拦
+# （bot 的企业 token 能调应用开通的全集，飞书侧分不开）。放行表由 inject() 按 FEISHU_AUTH_SELECTION
+# 算好写入凭据文件（feishuModulePolicyEnv），这里不认识任何模块，只按规则表（OASIS_FEISHU_RULES）查。
+# 共用的 drive 命令/接口按子命令、路径细分。**没有放行表、没有有效选择都按一个模块都没开放拦**，不会退成不限制。
+# 被拦时照 lark-cli 的错误形状往 stderr 打一段 JSON、退出码 3（与 lark-cli 自己的鉴权失败同码）。
 
 set -euo pipefail
+shopt -s extglob   # 放行规则用 +([!/]) / ?(/*) 表示「一段」「零到多段」
 
 OASIS_BIN_VAR="LARK_BIN"
 OASIS_CMD_NAME="lark-cli"
@@ -21,10 +29,164 @@ source "$(dirname "$(realpath "$0")")/../_base/wrapper-base.sh"
 oasis_resolve_bin
 oasis_check_available
 
+# 放行表只认凭据文件里的：先清掉 agent 环境里可能带进来的同名变量。
+unset OASIS_FEISHU_POLICY OASIS_FEISHU_ALLOW OASIS_FEISHU_RULES OASIS_FEISHU_LABELS OASIS_FEISHU_MODULES
+
 # Load feishu credentials into THIS process only (not visible to the agent).
 if [ -n "${FEISHU_CREDS_FILE:-}" ] && [ -f "${FEISHU_CREDS_FILE}" ]; then
   # shellcheck disable=SC1090
   source "${FEISHU_CREDS_FILE}"
 fi
 
+# 任何模块都放行的命令（帮助、查参数、内置技能、自检）；与 scopes.ts FEISHU_NEUTRAL_COMMANDS 同一份。
+FEISHU_NEUTRAL_COMMANDS=" help schema skills whoami doctor completion auth config profile update "
+# 任何模块都放行的接口 service；与 scopes.ts FEISHU_NEUTRAL_API_SERVICES 同一份。
+FEISHU_NEUTRAL_API=" authen auth "
+
+feishu_reply_blocked() {
+  printf '{\n  "ok": false,\n  "error": {\n    "type": "oasis_module_blocked",\n    "message": "%s",\n    "hint": "%s"\n  }\n}\n' "$1" "$2" >&2
+  exit 3
+}
+
+feishu_blocked() {
+  # $1 = 规则要求的模块（逗号分隔，空 = 不属于任何模块）；$2 = 给人看的命令写法（已过滤成安全字符）。
+  local mods="$1" shown="$2" need="" m entry label
+  local reconnect="请让配置人在 Oasis 里这个 agent 的飞书连接器详情点「重新连接」"
+  case "${OASIS_FEISHU_POLICY:-}" in
+    ok) ;;
+    none)
+      feishu_reply_blocked "本 agent 的飞书连接器还没有保存过可用模块的选择，Oasis 按「一个模块都没开放」处理，${shown} 已被拦下（机器人与用户身份都不可用）。" \
+        "${reconnect}，勾选需要的模块后完成授权。" ;;
+    invalid)
+      feishu_reply_blocked "本 agent 的飞书连接器保存的模块选择无法识别，Oasis 按「一个模块都没开放」处理，${shown} 已被拦下（机器人与用户身份都不可用）。" \
+        "${reconnect}，重新勾选模块后完成授权。" ;;
+    *)
+      feishu_reply_blocked "本次运行没有拿到飞书模块放行表，Oasis 不放行 ${shown}（机器人与用户身份都不可用）。" \
+        "请重新发起任务；仍出现请联系 Oasis 管理员检查飞书连接器。" ;;
+  esac
+  if [ -n "$mods" ]; then
+    for m in ${mods//,/ }; do
+      label="$m"
+      for entry in ${OASIS_FEISHU_LABELS:-}; do
+        if [ "${entry%%=*}" = "$m" ]; then label="${entry#*=}"; break; fi
+      done
+      need="${need:+${need}」或「}${label}"
+    done
+    feishu_reply_blocked "本 agent 的飞书连接器没有开放「${need}」模块，${shown} 已被 Oasis 拦下（机器人与用户身份都不可用）。当前开放的模块：${OASIS_FEISHU_MODULES:-无}。" \
+      "如需使用，${reconnect}，勾选「${need}」后完成授权。"
+  fi
+  feishu_reply_blocked "${shown} 不属于任何可勾选的飞书功能模块；本 agent 的飞书调用已按模块限制，Oasis 不放行它。当前开放的模块：${OASIS_FEISHU_MODULES:-无}。" \
+    "请改用已开放模块对应的 lark-cli 命令；共用的 drive 子命令须紧跟在 drive 之后写。确需此能力请联系 Oasis 管理员。"
+}
+
+# 只把「看起来像命令名 / service 名」的字符串回显出来，其余一律写成占位——它要进 JSON。
+feishu_safe() {
+  case "$1" in
+    ''|*[!A-Za-z0-9_.+-]*) printf '%s' "该命令" ;;
+    *) printf '%s' "$1" ;;
+  esac
+}
+
+# 按规则表判 $1（cmd/<domain>/… 或 api/<METHOD>/<service>/…）：命中且所需模块里有一个已放行 → 放行；
+# 命中但都没放行 → 拦并说缺哪个；一条都没命中 → 拦（不属于任何模块）。
+feishu_check_key() {
+  local key="$1" shown="$2" entry pat mods m
+  set -f
+  for entry in ${OASIS_FEISHU_RULES:-}; do
+    pat="${entry%=*}"; mods="${entry##*=}"
+    # shellcheck disable=SC2254
+    case "$key" in
+      $pat)
+        set +f
+        if [ "${OASIS_FEISHU_POLICY:-}" = "ok" ]; then
+          for m in ${mods//,/ }; do
+            case "${OASIS_FEISHU_ALLOW:-}" in *" $m "*) return 0 ;; esac
+          done
+        fi
+        feishu_blocked "$mods" "$shown" ;;
+    esac
+  done
+  set +f
+  feishu_blocked "" "$shown"
+}
+
+# 命令行里的一个词能不能原样进规则 key：只允许命令名字符，其余写成 ?（不会命中任何规则）。
+feishu_seg() {
+  case "$1" in
+    ''|*[!A-Za-z0-9_.+-]*) printf '?' ;;
+    *) printf '%s' "$1" ;;
+  esac
+}
+
+feishu_enforce_modules() {
+  local args=("$@") i=0 n=$# cmd=""
+  # 跳过命令前的全局参数（--profile X、-h/--help、-v/--version）。
+  while [ "$i" -lt "$n" ]; do
+    case "${args[$i]}" in
+      --profile) i=$((i + 2)) ;;
+      -*) i=$((i + 1)) ;;
+      *) cmd="${args[$i]}"; break ;;
+    esac
+  done
+  [ -n "$cmd" ] || return 0                      # 只有 --help / --version 之类
+  case "$FEISHU_NEUTRAL_COMMANDS" in *" $cmd "*) return 0 ;; esac
+
+  if [ "$cmd" = "api" ]; then
+    # lark-cli api <METHOD> <path> [flags]：api 之后第一、二个位置参数为 method、path（跳过带值的参数）。
+    local pos=0 method="" path="" a
+    i=$((i + 1))
+    while [ "$i" -lt "$n" ]; do
+      a="${args[$i]}"
+      case "$a" in
+        --as|--data|--file|--format|--jq|-q|--output|-o|--page-delay|--page-limit|--page-size|--params|--profile) i=$((i + 2)); continue ;;
+        -*) i=$((i + 1)); continue ;;
+      esac
+      pos=$((pos + 1))
+      if [ "$pos" -eq 1 ]; then method="$a"; fi
+      if [ "$pos" -eq 2 ]; then path="$a"; break; fi
+      i=$((i + 1))
+    done
+    [ -n "$path" ] || return 0                   # 没给 path：lark-cli 自己报用法错，不会发请求
+    # 归一成 <service>/…：lark-cli 接受完整 URL、/open-apis/ 前缀可省、前导斜杠可省；连续斜杠并成一个。
+    local p="$path"
+    p="${p#http://}"; p="${p#https://}"
+    case "$path" in http://*|https://*) p="${p#*/}" ;; esac
+    while [ "${p#/}" != "$p" ]; do p="${p#/}"; done
+    p="${p#open-apis/}"
+    while [ "${p#/}" != "$p" ]; do p="${p#/}"; done
+    p="${p%%[?#]*}"
+    while [ "${p//\/\//\/}" != "$p" ]; do p="${p//\/\//\/}"; done
+    # 路径里带 ..、. 段或百分号编码的，不去猜它最后落到哪个接口——直接拒。
+    case "/$p/" in *..*|*%*|*\\*|*/./*) feishu_blocked "" "lark-cli api（路径含 .. 或编码字符）" ;; esac
+    local svc="${p%%/*}"
+    case "$FEISHU_NEUTRAL_API" in *" $svc "*) return 0 ;; esac
+    case "$p" in *[!A-Za-z0-9_./:-]*) feishu_blocked "" "lark-cli api（路径含非法字符）" ;; esac
+    # 方法名转大写（macOS 自带 bash 3.2 没有 ${x^^}）。
+    method="$(feishu_seg "$method" | tr '[:lower:]' '[:upper:]')"
+    feishu_check_key "api/$method/$p" "lark-cli api …/open-apis/$(feishu_safe "$svc")/…"
+    return 0
+  fi
+
+  # lark-cli <domain> [子命令] [方法]：只取紧跟在 domain 后面的词（drive 这类共用 domain 按子命令分模块）；
+  # domain 后面先出现参数的，key 里记成「-」，共用 domain 的细分规则一条都不会命中。
+  local key="cmd/$(feishu_seg "$cmd")" depth=0 j=$((i + 1)) only_help=1 w
+  while [ "$j" -lt "$n" ]; do
+    case "${args[$j]}" in -h|--help) ;; *) only_help=0 ;; esac
+    j=$((j + 1))
+  done
+  [ "$only_help" -eq 1 ] && [ $((i + 1)) -lt "$n" ] && return 0   # lark-cli <domain> --help
+  j=$((i + 1))
+  while [ "$j" -lt "$n" ] && [ "$depth" -lt 2 ]; do
+    w="${args[$j]}"
+    case "$w" in -*) key="$key/-"; break ;; esac
+    key="$key/$(feishu_seg "$w")"
+    depth=$((depth + 1)); j=$((j + 1))
+  done
+  local shown="lark-cli $(feishu_safe "$cmd")"
+  [ "$depth" -ge 1 ] && shown="$shown $(feishu_safe "${args[$((i + 1))]}")"
+  feishu_check_key "$key" "$shown"
+}
+
+feishu_enforce_modules "$@"
+
 exec "$LARK_BIN" "$@"