377 lines
9.7 KiB
Go
377 lines
9.7 KiB
Go
package usecase
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"encoding/xml"
|
||
"fmt"
|
||
"io"
|
||
"math/rand"
|
||
"net/http"
|
||
"strings"
|
||
"time"
|
||
"unicode/utf8"
|
||
|
||
"apps/backend/internal/module/studio/domain"
|
||
)
|
||
|
||
// PersonaPreview：依指紋一次 LLM 產主貼 + 回文（可抓新聞當話題靈感）。
|
||
func (s *Service) PersonaPreview(ctx context.Context, ownerUID int64, personaID, topic string, useNews bool) (*domain.PersonaPreview, error) {
|
||
if ownerUID <= 0 {
|
||
return nil, domain.ErrForbidden
|
||
}
|
||
personaID = strings.TrimSpace(personaID)
|
||
if personaID == "" {
|
||
return nil, fmt.Errorf("%w: empty persona_id", domain.ErrValidation)
|
||
}
|
||
p, err := s.GetPersona(ctx, ownerUID, personaID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
fp := strings.TrimSpace(p.Style.DraftText)
|
||
if fp == "" {
|
||
fp = serializeDraftText(p.Style.Draft)
|
||
}
|
||
if fp == "" && strings.TrimSpace(p.Style.Draft.Tone) == "" {
|
||
return nil, fmt.Errorf("%w: 人設尚無指紋，請先完成分析或手動填寫指紋", domain.ErrValidation)
|
||
}
|
||
|
||
if err := s.billAI(ctx, ownerUID, "persona preview", "compose.personaPreview"); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
topic = strings.TrimSpace(topic)
|
||
topicSource := "manual"
|
||
if topic == "" {
|
||
if useNews {
|
||
if t, ok := fetchNewsTopic(ctx); ok {
|
||
topic = t
|
||
topicSource = "news"
|
||
}
|
||
}
|
||
if topic == "" {
|
||
topic = pickFallbackTopic(p)
|
||
topicSource = "fallback"
|
||
}
|
||
}
|
||
|
||
fpBlock := buildFingerprintBlock(p, fp)
|
||
prompt := buildPreviewBothPrompt(fpBlock, topic, topicSource)
|
||
|
||
provider, model, apiKey, kerr := s.resolveUserAI(ctx, ownerUID)
|
||
notes := ""
|
||
var postText, replyText string
|
||
|
||
if kerr == nil && strings.TrimSpace(apiKey) != "" && !isSyntheticAIKey(apiKey) {
|
||
// 一次呼叫同時產主貼+回文，避免雙請求卡過 gateway 30s
|
||
raw, aerr := s.completeLLM(ctx, provider, model, apiKey, prompt)
|
||
if aerr != nil {
|
||
notes = "產文失敗，已用指紋規則試產：" + truncate(aerr.Error(), 100)
|
||
postText = rulePreviewPost(p, topic)
|
||
replyText = rulePreviewReply(p, postText)
|
||
} else {
|
||
postText, replyText = parsePreviewPair(raw)
|
||
if postText == "" {
|
||
postText = rulePreviewPost(p, topic)
|
||
}
|
||
if replyText == "" {
|
||
replyText = rulePreviewReply(p, postText)
|
||
}
|
||
if notes == "" && topicSource == "news" {
|
||
notes = "話題取自即時新聞，已改寫成人設口吻"
|
||
}
|
||
}
|
||
_ = model
|
||
_ = provider
|
||
} else {
|
||
postText = rulePreviewPost(p, topic)
|
||
replyText = rulePreviewReply(p, postText)
|
||
notes = "尚未設定 API Key，目前用指紋規則試產。請到設定填寫 Key 後再試。"
|
||
}
|
||
|
||
return &domain.PersonaPreview{
|
||
Topic: topic,
|
||
TopicSource: topicSource,
|
||
PostText: postText,
|
||
ReplyText: replyText,
|
||
Notes: notes,
|
||
}, nil
|
||
}
|
||
|
||
func buildFingerprintBlock(p *domain.Persona, fp string) string {
|
||
var b strings.Builder
|
||
if n := strings.TrimSpace(p.Name); n != "" {
|
||
b.WriteString("名稱：")
|
||
b.WriteString(n)
|
||
b.WriteString("\n")
|
||
}
|
||
if br := strings.TrimSpace(p.Brief); br != "" {
|
||
b.WriteString("定位：")
|
||
b.WriteString(br)
|
||
b.WriteString("\n")
|
||
}
|
||
b.WriteString("【語言指紋】\n")
|
||
b.WriteString(fp)
|
||
if len(p.Style.Dimensions) > 0 {
|
||
b.WriteString("\n【8D 摘要】\n")
|
||
for _, k := range dimKeys {
|
||
if d, ok := p.Style.Dimensions[k]; ok && strings.TrimSpace(d.Summary) != "" {
|
||
b.WriteString(k)
|
||
b.WriteString("：")
|
||
b.WriteString(d.Summary)
|
||
b.WriteString("\n")
|
||
}
|
||
}
|
||
}
|
||
if len(p.Guard.Avoid) > 0 {
|
||
b.WriteString("【禁止】")
|
||
b.WriteString(strings.Join(p.Guard.Avoid, "、"))
|
||
b.WriteString("\n")
|
||
}
|
||
return b.String()
|
||
}
|
||
|
||
func buildPreviewBothPrompt(fpBlock, topic, topicSource string) string {
|
||
srcHint := "自訂話題"
|
||
if topicSource == "news" {
|
||
srcHint = "今日新聞標題（請轉成個人觀察／情緒／提問，不要硬抄標題、不要寫成新聞稿）"
|
||
} else if topicSource == "fallback" {
|
||
srcHint = "生活靈感"
|
||
}
|
||
return strings.TrimSpace(fmt.Sprintf(`
|
||
你正在扮演一位 Threads 使用者本人（不是分析師、不是助手）。
|
||
任務：依語言指紋寫「一則主貼」與「一則短回文」。
|
||
|
||
規則（強制）：
|
||
- 只輸出一個 JSON 物件，不要 markdown 代碼塊、不要前言。
|
||
- 繁體中文（台灣用語）、口語、像真人滑手機打的。
|
||
- 嚴格遵守指紋的語氣／節奏／用字／CTA／禁忌。
|
||
- 主貼約 80～260 字，可分段空行；回文約 20～100 字。
|
||
- 話題僅作靈感，必須用「這個人會怎麼講」重寫。
|
||
|
||
【話題靈感】（來源：%s）
|
||
%s
|
||
|
||
%s
|
||
|
||
輸出格式：
|
||
{"post":"主貼正文","reply":"回文正文"}
|
||
`, srcHint, topic, fpBlock))
|
||
}
|
||
|
||
// parsePreviewPair 解析 {"post","reply"} 或寬鬆切割。
|
||
func parsePreviewPair(raw string) (post, reply string) {
|
||
s := strings.TrimSpace(raw)
|
||
if s == "" {
|
||
return "", ""
|
||
}
|
||
// strip fences
|
||
if i := strings.Index(s, "{"); i >= 0 {
|
||
if j := strings.LastIndex(s, "}"); j > i {
|
||
s = s[i : j+1]
|
||
}
|
||
}
|
||
var parsed struct {
|
||
Post string `json:"post"`
|
||
Reply string `json:"reply"`
|
||
// 別名
|
||
PostText string `json:"post_text"`
|
||
ReplyText string `json:"reply_text"`
|
||
}
|
||
if err := json.Unmarshal([]byte(s), &parsed); err == nil {
|
||
post = cleanGeneratedText(coalesceStr(parsed.Post, parsed.PostText))
|
||
reply = cleanGeneratedText(coalesceStr(parsed.Reply, parsed.ReplyText))
|
||
return post, reply
|
||
}
|
||
// 非 JSON：整段當主貼
|
||
return cleanGeneratedText(raw), ""
|
||
}
|
||
|
||
func coalesceStr(a, b string) string {
|
||
if strings.TrimSpace(a) != "" {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
func cleanGeneratedText(raw string) string {
|
||
return cleanGeneratedTextMax(raw, 500)
|
||
}
|
||
|
||
func cleanGeneratedTextMax(raw string, maxRunes int) string {
|
||
s := strings.TrimSpace(raw)
|
||
if s == "" {
|
||
return ""
|
||
}
|
||
// 去掉常見 markdown 圍欄
|
||
if strings.HasPrefix(s, "```") {
|
||
s = strings.TrimPrefix(s, "```json")
|
||
s = strings.TrimPrefix(s, "```")
|
||
s = strings.TrimSpace(s)
|
||
if i := strings.LastIndex(s, "```"); i >= 0 {
|
||
s = strings.TrimSpace(s[:i])
|
||
}
|
||
}
|
||
s = strings.TrimSpace(s)
|
||
for _, p := range []string{"主貼：", "主贴：", "貼文：", "仿寫：", "回覆：", "回文：", "Reply:", "Post:", "Mimic:"} {
|
||
s = strings.TrimPrefix(s, p)
|
||
s = strings.TrimSpace(s)
|
||
}
|
||
if maxRunes <= 0 {
|
||
maxRunes = 500
|
||
}
|
||
if utf8.RuneCountInString(s) > maxRunes {
|
||
r := []rune(s)
|
||
s = string(r[:maxRunes]) + "…"
|
||
}
|
||
return s
|
||
}
|
||
|
||
func rulePreviewPost(p *domain.Persona, topic string) string {
|
||
d := p.Style.Draft
|
||
hooks := d.Hooks
|
||
if hooks == "" {
|
||
hooks = "先丟自己的卡關"
|
||
}
|
||
cta := d.CtaStyle
|
||
if cta == "" {
|
||
cta = "你們呢？"
|
||
}
|
||
ex := d.Examples
|
||
lines := []string{
|
||
fmt.Sprintf("最近卡在「%s」。", truncate(topic, 40)),
|
||
"",
|
||
hooks + "——講真的，不是規格文那種。",
|
||
}
|
||
if ex != "" {
|
||
lines = append(lines, truncate(ex, 80))
|
||
}
|
||
lines = append(lines, "", cta)
|
||
return strings.Join(lines, "\n")
|
||
}
|
||
|
||
func rulePreviewReply(p *domain.Persona, postText string) string {
|
||
_ = p
|
||
return fmt.Sprintf("懂，我自己也遇過類似的。%s——你們後來怎麼解的？", truncate(postText, 24))
|
||
}
|
||
|
||
func pickFallbackTopic(p *domain.Persona) string {
|
||
seeds := []string{
|
||
"最近大家在煩的時間管理",
|
||
"網購踩雷又不能退的感覺",
|
||
"加班後只想滑手機不想動",
|
||
"突然很想整理房間但動不了",
|
||
"朋友安利的東西到底能不能信",
|
||
"週末計劃很多結果什麼都沒做",
|
||
"天氣一變心情就跟著飄",
|
||
"想學新東西但資訊量爆炸",
|
||
}
|
||
if a := strings.TrimSpace(p.Style.Draft.Audience); a != "" {
|
||
seeds = append(seeds, "跟「"+truncate(a, 20)+"」有關的日常卡關")
|
||
}
|
||
return seeds[rand.Intn(len(seeds))]
|
||
}
|
||
|
||
// fetchNewsTopic — Google News 繁中 RSS（短超時，失敗就走 fallback）
|
||
func fetchNewsTopic(ctx context.Context) (string, bool) {
|
||
urls := []string{
|
||
"https://news.google.com/rss?hl=zh-TW&gl=TW&ceid=TW:zh-Hant",
|
||
"https://news.google.com/rss/headlines/section/topic/TECHNOLOGY?hl=zh-TW&gl=TW&ceid=TW:zh-Hant",
|
||
}
|
||
u := urls[rand.Intn(len(urls))]
|
||
req, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
|
||
if err != nil {
|
||
return "", false
|
||
}
|
||
req.Header.Set("User-Agent", "HarborDesk/1.0 (+persona-preview)")
|
||
client := &http.Client{Timeout: 3 * time.Second}
|
||
res, err := client.Do(req)
|
||
if err != nil {
|
||
return "", false
|
||
}
|
||
defer res.Body.Close()
|
||
if res.StatusCode < 200 || res.StatusCode >= 300 {
|
||
return "", false
|
||
}
|
||
body, err := io.ReadAll(io.LimitReader(res.Body, 512<<10))
|
||
if err != nil || len(body) == 0 {
|
||
return "", false
|
||
}
|
||
titles := parseRSSTitles(body)
|
||
if len(titles) == 0 {
|
||
return "", false
|
||
}
|
||
t := cleanNewsTitle(titles[rand.Intn(len(titles))])
|
||
if t == "" {
|
||
return "", false
|
||
}
|
||
return t, true
|
||
}
|
||
|
||
type rssFeed struct {
|
||
XMLName xml.Name `xml:"rss"`
|
||
Items []rssItem `xml:"channel>item"`
|
||
}
|
||
|
||
type rssItem struct {
|
||
Title string `xml:"title"`
|
||
}
|
||
|
||
func parseRSSTitles(body []byte) []string {
|
||
var feed rssFeed
|
||
if err := xml.Unmarshal(body, &feed); err == nil {
|
||
var out []string
|
||
for _, it := range feed.Items {
|
||
if t := strings.TrimSpace(it.Title); t != "" {
|
||
out = append(out, t)
|
||
}
|
||
}
|
||
if len(out) > 0 {
|
||
return out
|
||
}
|
||
}
|
||
raw := string(body)
|
||
var out []string
|
||
for {
|
||
i := strings.Index(raw, "<title>")
|
||
if i < 0 {
|
||
break
|
||
}
|
||
raw = raw[i+7:]
|
||
j := strings.Index(raw, "</title>")
|
||
if j < 0 {
|
||
break
|
||
}
|
||
t := strings.TrimSpace(raw[:j])
|
||
raw = raw[j+8:]
|
||
if strings.Contains(strings.ToLower(t), "google news") {
|
||
continue
|
||
}
|
||
if t != "" {
|
||
out = append(out, t)
|
||
}
|
||
if len(out) >= 15 {
|
||
break
|
||
}
|
||
}
|
||
return out
|
||
}
|
||
|
||
func cleanNewsTitle(t string) string {
|
||
t = strings.TrimSpace(t)
|
||
if i := strings.LastIndex(t, " - "); i > 8 {
|
||
t = strings.TrimSpace(t[:i])
|
||
}
|
||
t = strings.TrimPrefix(t, "<![CDATA[")
|
||
t = strings.TrimSuffix(t, "]]>")
|
||
t = strings.TrimSpace(t)
|
||
if utf8.RuneCountInString(t) < 6 {
|
||
return ""
|
||
}
|
||
if utf8.RuneCountInString(t) > 80 {
|
||
r := []rune(t)
|
||
t = string(r[:80]) + "…"
|
||
}
|
||
return t
|
||
}
|