82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
|
package placement
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// DiscoverChannel identifies which backend fulfilled a placement discover query.
|
|||
|
|
type DiscoverChannel string
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
DiscoverThreadsAPI DiscoverChannel = "threads_api"
|
|||
|
|
DiscoverBrave DiscoverChannel = "brave"
|
|||
|
|
DiscoverCrawler DiscoverChannel = "crawler"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// DiscoverRequest is used by scan jobs; expand-graph only uses Brave knowledge_expand.
|
|||
|
|
type DiscoverRequest struct {
|
|||
|
|
Query string
|
|||
|
|
Keyword string // plain tag for crawler; optional
|
|||
|
|
Recency bool
|
|||
|
|
Limit int
|
|||
|
|
Member MemberContext
|
|||
|
|
Crawler CrawlerSearchFn
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type DiscoverPost struct {
|
|||
|
|
Text string
|
|||
|
|
Permalink string
|
|||
|
|
ExternalID string
|
|||
|
|
Author string
|
|||
|
|
PostedAt string
|
|||
|
|
LikeCount int
|
|||
|
|
ReplyCount int
|
|||
|
|
Source DiscoverChannel
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Discover runs keyword discovery respecting the member's connection prefs.
|
|||
|
|
// Formal mode (dev_mode=false) never falls back to crawler.
|
|||
|
|
func Discover(ctx context.Context, req DiscoverRequest) ([]DiscoverPost, DiscoverChannel, error) {
|
|||
|
|
m := req.Member
|
|||
|
|
if m.DevMode {
|
|||
|
|
if !m.BrowserConnected {
|
|||
|
|
return nil, "", fmt.Errorf("開發模式需先同步 Chrome Session")
|
|||
|
|
}
|
|||
|
|
if req.Crawler == nil {
|
|||
|
|
return nil, DiscoverCrawler, fmt.Errorf("crawler search not configured")
|
|||
|
|
}
|
|||
|
|
keyword := CrawlerKeywordFromQuery(req.Query, req.Keyword)
|
|||
|
|
if keyword == "" {
|
|||
|
|
return nil, DiscoverCrawler, fmt.Errorf("crawler keyword is empty")
|
|||
|
|
}
|
|||
|
|
posts, err := req.Crawler(ctx, m, keyword, req.Limit)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, DiscoverCrawler, err
|
|||
|
|
}
|
|||
|
|
return posts, DiscoverCrawler, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if m.AllowsThreadsAPI {
|
|||
|
|
if !m.ApiConnected {
|
|||
|
|
return nil, "", fmt.Errorf("正式模式需先完成 Threads API 連線")
|
|||
|
|
}
|
|||
|
|
posts, err := keywordSearchViaThreadsAPI(ctx, req)
|
|||
|
|
if err == nil && len(posts) > 0 {
|
|||
|
|
return posts, DiscoverThreadsAPI, nil
|
|||
|
|
}
|
|||
|
|
if err != nil && !m.AllowsBrave {
|
|||
|
|
return nil, "", err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if m.AllowsBrave {
|
|||
|
|
if m.BraveAPIKey == "" {
|
|||
|
|
return nil, "", fmt.Errorf("請在設定頁設定 Brave Search API key（跟隨此登入帳號）")
|
|||
|
|
}
|
|||
|
|
return nil, DiscoverBrave, fmt.Errorf("brave threads discover delegated to worker")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil, "", fmt.Errorf("目前搜尋來源模式無可用管道：%s", m.SearchSourceMode)
|
|||
|
|
}
|