Kris
f943c31ce4
新增了完整的Confluence集成插件，包含以下核心功能： 1. 基础认证与配置管理，支持API Token和OAuth2两种认证方式 2. 评论去重、权限控制与提及解析 3. 知识库检索与页面内容处理 4. 评论收发、编辑删除与流式回复支持 5. 附件与页面标签管理 6. 内容属性存储与AI元数据管理 7. Webhook事件接收与处理 8. 完整的插件配置与状态检查
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from yuxi.channel.extensions.confluence.types import ConfluenceAccount
|
|
|
|
ENV_MAP = {
|
|
"url": "CONFLUENCE_URL",
|
|
"email": "CONFLUENCE_EMAIL",
|
|
"api_token": "CONFLUENCE_API_TOKEN",
|
|
"oauth_client_id": "CONFLUENCE_OAUTH_CLIENT_ID",
|
|
"oauth_client_secret": "CONFLUENCE_OAUTH_CLIENT_SECRET",
|
|
"auth_mode": "CONFLUENCE_AUTH_MODE",
|
|
"deployment_type": "CONFLUENCE_DEPLOYMENT_TYPE",
|
|
"space_keys": "CONFLUENCE_SPACE_KEYS",
|
|
"comment_poll_enabled": "CONFLUENCE_COMMENT_POLL_ENABLED",
|
|
"poll_interval": "CONFLUENCE_POLL_INTERVAL",
|
|
"comment_policy": "CONFLUENCE_COMMENT_POLICY",
|
|
"allow_from": "CONFLUENCE_ALLOW_FROM",
|
|
}
|
|
|
|
|
|
def _env_or_config(key: str) -> str | None:
|
|
env_key = ENV_MAP.get(key, "")
|
|
if env_key:
|
|
return os.getenv(env_key)
|
|
return None
|
|
|
|
|
|
def list_confluence_account_ids(config: dict) -> list[str]:
|
|
accounts = config.get("channels", {}).get("confluence", {}).get("accounts", {})
|
|
if accounts:
|
|
return list(accounts.keys())
|
|
return ["default"]
|
|
|
|
|
|
def resolve_confluence_account(config: dict, account_id: str = "default") -> ConfluenceAccount:
|
|
from yuxi.channel.extensions.confluence.types import ConfluenceAccount
|
|
|
|
confluence_cfg = config.get("channels", {}).get("confluence", {})
|
|
account_cfg = confluence_cfg.get("accounts", {}).get(account_id, confluence_cfg)
|
|
|
|
def _get(key: str) -> str | None:
|
|
val = account_cfg.get(key)
|
|
if val is not None:
|
|
return str(val)
|
|
return _env_or_config(key)
|
|
|
|
space_keys_str = _get("space_keys") or ""
|
|
allow_from_str = _get("allow_from") or ""
|
|
poll_enabled_str = _get("comment_poll_enabled") or "true"
|
|
|
|
return ConfluenceAccount(
|
|
account_id=account_id,
|
|
url=_get("url") or "",
|
|
email=_get("email") or "",
|
|
api_token=_get("api_token") or "",
|
|
oauth_client_id=_get("oauth_client_id") or "",
|
|
oauth_client_secret=_get("oauth_client_secret") or "",
|
|
auth_mode=_get("auth_mode") or "api_token",
|
|
deployment_type=_get("deployment_type") or "cloud",
|
|
space_keys=[k.strip() for k in space_keys_str.split(",") if k.strip()],
|
|
comment_poll_enabled=poll_enabled_str.lower() != "false",
|
|
poll_interval=float(_get("poll_interval") or "30"),
|
|
comment_policy=_get("comment_policy") or "open",
|
|
allow_from=[a.strip() for a in allow_from_str.split(",") if a.strip()],
|
|
name=account_cfg.get("name", "Confluence Bot"),
|
|
)
|