110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
|
package placement
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type CtaType string
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
CtaNone CtaType = "none"
|
|||
|
|
CtaLink CtaType = "link"
|
|||
|
|
CtaDM CtaType = "dm"
|
|||
|
|
CtaFollow CtaType = "follow"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type ProductContextFields struct {
|
|||
|
|
Brand string `json:"brand"`
|
|||
|
|
Product string `json:"product"`
|
|||
|
|
Features string `json:"features"`
|
|||
|
|
PlacementTone string `json:"placementTone,omitempty"`
|
|||
|
|
CtaType CtaType `json:"ctaType,omitempty"`
|
|||
|
|
CtaUrl string `json:"ctaUrl,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func ParseProductContext(raw string) ProductContextFields {
|
|||
|
|
raw = strings.TrimSpace(raw)
|
|||
|
|
if raw == "" {
|
|||
|
|
return ProductContextFields{CtaType: CtaNone}
|
|||
|
|
}
|
|||
|
|
var parsed ProductContextFields
|
|||
|
|
if err := json.Unmarshal([]byte(raw), &parsed); err != nil {
|
|||
|
|
return ProductContextFields{CtaType: CtaNone, Features: raw}
|
|||
|
|
}
|
|||
|
|
parsed.Brand = strings.TrimSpace(parsed.Brand)
|
|||
|
|
parsed.Product = strings.TrimSpace(parsed.Product)
|
|||
|
|
parsed.Features = strings.TrimSpace(parsed.Features)
|
|||
|
|
parsed.PlacementTone = strings.TrimSpace(parsed.PlacementTone)
|
|||
|
|
parsed.CtaUrl = strings.TrimSpace(parsed.CtaUrl)
|
|||
|
|
if parsed.CtaType == "" {
|
|||
|
|
parsed.CtaType = CtaNone
|
|||
|
|
}
|
|||
|
|
return parsed
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func SerializeProductContext(fields ProductContextFields) string {
|
|||
|
|
brand := strings.TrimSpace(fields.Brand)
|
|||
|
|
product := strings.TrimSpace(fields.Product)
|
|||
|
|
features := strings.TrimSpace(fields.Features)
|
|||
|
|
tone := strings.TrimSpace(fields.PlacementTone)
|
|||
|
|
ctaType := fields.CtaType
|
|||
|
|
if ctaType == "" {
|
|||
|
|
ctaType = CtaNone
|
|||
|
|
}
|
|||
|
|
ctaUrl := strings.TrimSpace(fields.CtaUrl)
|
|||
|
|
if brand == "" && product == "" && features == "" && tone == "" && ctaType == CtaNone && ctaUrl == "" {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
payload, _ := json.Marshal(ProductContextFields{
|
|||
|
|
Brand: brand,
|
|||
|
|
Product: product,
|
|||
|
|
Features: features,
|
|||
|
|
PlacementTone: tone,
|
|||
|
|
CtaType: ctaType,
|
|||
|
|
CtaUrl: ctaUrl,
|
|||
|
|
})
|
|||
|
|
return string(payload)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func FormatProductContextForPrompt(raw string) string {
|
|||
|
|
fields := ParseProductContext(raw)
|
|||
|
|
lines := []string{}
|
|||
|
|
if fields.Brand != "" {
|
|||
|
|
lines = append(lines, "品牌："+fields.Brand)
|
|||
|
|
}
|
|||
|
|
if fields.Product != "" {
|
|||
|
|
lines = append(lines, "產品："+fields.Product)
|
|||
|
|
}
|
|||
|
|
if fields.Features != "" {
|
|||
|
|
lines = append(lines, "特色／能幫上忙的地方："+fields.Features)
|
|||
|
|
}
|
|||
|
|
if fields.PlacementTone != "" {
|
|||
|
|
lines = append(lines, "置入語氣偏好："+fields.PlacementTone)
|
|||
|
|
}
|
|||
|
|
if fields.CtaType == CtaLink && fields.CtaUrl != "" {
|
|||
|
|
lines = append(lines, "留言 CTA 連結："+fields.CtaUrl)
|
|||
|
|
} else if fields.CtaType == CtaDM {
|
|||
|
|
lines = append(lines, "留言 CTA：引導私訊")
|
|||
|
|
} else if fields.CtaType == CtaFollow {
|
|||
|
|
lines = append(lines, "留言 CTA：引導追蹤")
|
|||
|
|
}
|
|||
|
|
if len(lines) == 0 {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
return strings.Join(lines, "\n")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func ProductBriefFromContext(raw string) string {
|
|||
|
|
formatted := FormatProductContextForPrompt(raw)
|
|||
|
|
if formatted != "" {
|
|||
|
|
return formatted
|
|||
|
|
}
|
|||
|
|
return strings.TrimSpace(raw)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func HasProductContext(raw string) bool {
|
|||
|
|
fields := ParseProductContext(raw)
|
|||
|
|
return fields.Brand != "" || fields.Product != "" || fields.Features != ""
|
|||
|
|
}
|