package kook // 本文件承载 KOOK 的**入站附件字节内联**(切片 IM-1):KOOK 的 s=0 消息把附件地址 // 直接放在 `content` 字段里(图片/视频/文件/音频同理),因此文件类消息拿到的是一个 // 绝对 CDN 地址,可以按来源白名单直接取回。 // // 两条边界: // - **只认平台资产主机**:白名单 = KOOK 附件 CDN(`img.kookapp.cn`)+ 已配置的 // API 根(`base_url`)。`content` 字段是发送方可影响的内容(一个伪造的文件消息 // 完全可以填任意字符串),因此**不是**「平台保证的地址」——白名单在这里是硬门: // 不在白名单里,一个字节都不发出去; // - 尺寸/超时/预筛/日志口径全在 `internal/channel/mediafetch`(四个适配器共用)。 import ( "context" "git.yeij.top/AskaEth/Cyrene_v2/internal/channel/adapter" "git.yeij.top/AskaEth/Cyrene_v2/internal/channel/mediafetch" ) // kookAssetOrigin 是 KOOK 附件的 CDN 来源(平台只在它上面提供附件地址)。 const kookAssetOrigin = "https://img.kookapp.cn" // inlineInboundMedia 把受支持格式的文件附件取回并内联到消息上。失败不抛出:该附件 // 保持无字节,由 channelbridge 的入站摄取照既有口径给模型一行诚实提示。 func (a *Adapter) inlineInboundMedia(ctx context.Context, msg *adapter.UnifiedMessage) { mediafetch.Inline(ctx, a.log, msg, kookFetcher{a: a}) } // kookFetcher 实现 mediafetch.Fetcher:KOOK 的媒体引用就是 `content` 里的绝对地址。 type kookFetcher struct{ a *Adapter } // Fetch 按来源白名单取回一份 KOOK 附件。 func (f kookFetcher) Fetch(ctx context.Context, m adapter.Media) ([]byte, error) { return mediafetch.Fetch(ctx, mediafetch.Request{ URL: m.URL, Allow: f.a.assetAllowlist(), Client: f.a.client, Timeout: mediafetch.DefaultTimeout, }) } // assetAllowlist 给出 KOOK 附件的来源白名单:附件 CDN + 已配置的 API 根。第二条是 // 为了让**自建/私有化部署**(把 `base_url` 指到自己的网关)也能走通,且它的可信度与 // 其余 API 调用一致(同一条由部署声明的地址)。两条都读不出来时给空名单 ⇒ 谁都不许。 func (a *Adapter) assetAllowlist() mediafetch.Allowlist { origins := []string{kookAssetOrigin} if origin, err := mediafetch.OriginOf(a.baseURL); err == nil { origins = append(origins, origin) } return mediafetch.NewAllowlist(origins...) }