oasis_test_v2 2.2.110 → 2.2.112

package/dist/feishu/wrapper.sh

@@ -10,6 +10,12 @@
 # 环境变量：
 #   LARK_BIN          — 真实 lark-cli 绝对路径（避免递归调用自身）
 #   FEISHU_CREDS_FILE — 凭据文件路径（内含 export LARKSUITE_CLI_* 行）
+#
+# 按 Agent 限制可用模块（产品反馈「飞书连接器·按 Agent 的权限模型」）：凭据文件里有
+# OASIS_FEISHU_ALLOW 时，只放行所选模块的 `lark-cli <domain>` 与 `lark-cli api <METHOD> /open-apis/<service>/…`，
+# 机器人与用户身份一样拦（bot 的企业 token 能调应用开通的全集，飞书侧分不开）。放行表由 inject() 按
+# FEISHU_AUTH_SELECTION 算好写入（feishuModulePolicyEnv），这里不认识任何模块，只查表。
+# 被拦时照 lark-cli 的错误形状往 stderr 打一段 JSON、退出码 3（与 lark-cli 自己的鉴权失败同码）。
 
 set -euo pipefail
 
@@ -21,10 +27,94 @@ source "$(dirname "$(realpath "$0")")/../_base/wrapper-base.sh"
 oasis_resolve_bin
 oasis_check_available
 
+# 放行表只认凭据文件里的：先清掉 agent 环境里可能带进来的同名变量。
+unset OASIS_FEISHU_ALLOW OASIS_FEISHU_NEED 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_blocked() {
+  # $1 = 放行表里的键（cmd:<domain> / api:<service>）；$2 = 给人看的命令写法（已过滤成安全字符）。
+  local key="$1" shown="$2" need="" message hint
+  local entry
+  for entry in ${OASIS_FEISHU_NEED:-}; do
+    if [ "${entry%%=*}" = "$key" ]; then need="${entry#*=}"; break; fi
+  done
+  if [ -n "$need" ]; then
+    message="本 agent 的飞书连接器没有开放「${need}」模块，${shown} 已被 Oasis 拦下（机器人与用户身份都不可用）。当前开放的模块：${OASIS_FEISHU_MODULES:-无}。"
+    hint="如需使用，请让配置人在 Oasis 里这个 agent 的飞书连接器详情点「重新连接」，勾选「${need}」后完成授权。"
+  else
+    message="${shown} 不属于任何可勾选的飞书功能模块；本 agent 的飞书调用已按模块限制，Oasis 不放行它。当前开放的模块：${OASIS_FEISHU_MODULES:-无}。"
+    hint="请改用已开放模块对应的 lark-cli 命令；确需此能力请联系 Oasis 管理员。"
+  fi
+  printf '{\n  "ok": false,\n  "error": {\n    "type": "oasis_module_blocked",\n    "message": "%s",\n    "hint": "%s"\n  }\n}\n' "$message" "$hint" >&2
+  exit 3
+}
+
+# 只把「看起来像命令名 / service 名」的字符串回显出来，其余一律写成占位——它要进 JSON。
+feishu_safe() {
+  case "$1" in
+    ''|*[!A-Za-z0-9_.-]*) printf '%s' "该命令" ;;
+    *) printf '%s' "$1" ;;
+  esac
+}
+
+feishu_enforce_modules() {
+  [ -n "${OASIS_FEISHU_ALLOW:-}" ] || return 0   # 没保存过模块选择：不限制
+  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 之后第二个位置参数为 path（跳过带值的参数）。
+    local pos=0 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 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
+    # 路径里带 .. 或百分号编码的，不去猜它最后落到哪个 service——直接拒。
+    case "$p" in *..*|*%*|*\\*) feishu_blocked "api:?" "lark-cli api（路径含 .. 或编码字符）" ;; esac
+    local svc="${p%%[/?#]*}"
+    case "$FEISHU_NEUTRAL_API" in *" $svc "*) return 0 ;; esac
+    case "$OASIS_FEISHU_ALLOW" in *" api:$svc "*) return 0 ;; esac
+    feishu_blocked "api:$svc" "lark-cli api …/open-apis/$(feishu_safe "$svc")/…"
+  fi
+
+  case "$OASIS_FEISHU_ALLOW" in *" cmd:$cmd "*) return 0 ;; esac
+  feishu_blocked "cmd:$cmd" "lark-cli $(feishu_safe "$cmd")"
+}
+
+feishu_enforce_modules "$@"
+
 exec "$LARK_BIN" "$@"