13262a5698
新增功能： - 新增 Sora 账号管理和 OAuth 认证 - 新增 Sora 视频/图片生成 API 网关 - 新增 Sora 任务调度和缓存机制 - 新增 Sora 使用统计和计费支持 - 前端增加 Sora 平台配置界面 安全修复（代码审核）： - [SEC-001] 限制媒体下载响应体大小（图片 20MB、视频 200MB），防止 DoS 攻击 - [SEC-002] 限制 SDK API 响应大小（1MB），防止内存耗尽 - [SEC-003] 修复 SSRF 风险，添加 URL 验证并强制使用代理配置 BUG 修复（代码审核）： - [BUG-001] 修复 for 循环内 defer 累积导致的资源泄漏 - [BUG-002] 修复图片并发槽位获取失败时已持有锁未释放的永久泄漏 性能优化（代码审核）： - [PERF-001] 添加 Sentinel Token 缓存（3 分钟有效期），减少 PoW 计算开销 技术细节： - 使用 io.LimitReader 限制所有外部输入的大小 - 添加 urlvalidator 验证防止 SSRF 攻击 - 使用 sync.Map 实现线程安全的包级缓存 - 优化并发槽位管理，添加 releaseAll 模式防止泄漏 影响范围： - 后端：新增 Sora 相关数据模型、服务、网关和管理接口 - 前端：新增 Sora 平台配置、账号管理和监控界面 - 配置：新增 Sora 相关配置项和环境变量 Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package sora
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
var storyboardRe = regexp.MustCompile(`\[(\d+(?:\.\d+)?)s\]`)
|
|
|
|
// IsStoryboardPrompt 检测是否为分镜提示词。
|
|
func IsStoryboardPrompt(prompt string) bool {
|
|
if strings.TrimSpace(prompt) == "" {
|
|
return false
|
|
}
|
|
return storyboardRe.MatchString(prompt)
|
|
}
|
|
|
|
// FormatStoryboardPrompt 将分镜提示词转换为 API 需要的格式。
|
|
func FormatStoryboardPrompt(prompt string) string {
|
|
prompt = strings.TrimSpace(prompt)
|
|
if prompt == "" {
|
|
return prompt
|
|
}
|
|
matches := storyboardRe.FindAllStringSubmatchIndex(prompt, -1)
|
|
if len(matches) == 0 {
|
|
return prompt
|
|
}
|
|
firstIdx := matches[0][0]
|
|
instructions := strings.TrimSpace(prompt[:firstIdx])
|
|
|
|
shotPattern := regexp.MustCompile(`\[(\d+(?:\.\d+)?)s\]\s*([^\[]+)`)
|
|
shotMatches := shotPattern.FindAllStringSubmatch(prompt, -1)
|
|
if len(shotMatches) == 0 {
|
|
return prompt
|
|
}
|
|
|
|
shots := make([]string, 0, len(shotMatches))
|
|
for i, sm := range shotMatches {
|
|
if len(sm) < 3 {
|
|
continue
|
|
}
|
|
duration := strings.TrimSpace(sm[1])
|
|
scene := strings.TrimSpace(sm[2])
|
|
shots = append(shots, "Shot "+itoa(i+1)+":\nduration: "+duration+"sec\nScene: "+scene)
|
|
}
|
|
|
|
timeline := strings.Join(shots, "\n\n")
|
|
if instructions != "" {
|
|
return "current timeline:\n" + timeline + "\n\ninstructions:\n" + instructions
|
|
}
|
|
return timeline
|
|
}
|
|
|
|
// ExtractRemixID 提取分享链接中的 remix ID。
|
|
func ExtractRemixID(text string) string {
|
|
text = strings.TrimSpace(text)
|
|
if text == "" {
|
|
return ""
|
|
}
|
|
re := regexp.MustCompile(`s_[a-f0-9]{32}`)
|
|
match := re.FindString(text)
|
|
return match
|
|
}
|