Aska
2283383b9e
- proactive.go(新): parseProactive 从结论解析 [我决定:push|no_push][渠道:...]<正文>;
兼容大小写/全半角冒号; 无标记回退(无/nothing=不推); 畸形/孤标记 fail-safe 不推
- context.go: guidancePrompt 引导结构化结尾+保留P1温暖引导
- push.go: PushRequest{Message,ChannelHint}; Pusher.Push 路由(auto→active优先/IM兜底,
web→仅active, im→仅IM, 具体平台→IM+platform打标+缺IM回退默认记detail); buildProactiveMessage
- backthink.go: think() 用 parseProactive→ShouldPush=push&&disturbAllowed(不再被tr.Import卡死)
- decide.go: disturbAllowed(asleep/working/quiet 硬性不打扰门) 取代旧import门(保留旧Decide兼容)
- 测试: parseProactive 6组+think端到端(真实pusher:kook→IM/无标记→active/no_push→不推/睡眠→门拦截)
+buildProactiveMessage+具体平台无IM回退; 后端 build/vet/test 全绿
- 注: 生产 Active/IM sink 未注入 wire.go backthink.Deps(另一步); 多平台IM受单IMAdapter接口限制
140 lines
3.2 KiB
Go
140 lines
3.2 KiB
Go
package backthink
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"io"
|
|
"log/slog"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/channel/adapter"
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/db"
|
|
)
|
|
|
|
// mockThinker records every Think invocation and returns a fixed conclusion. It
|
|
// respects ctx cancellation so Stop can drain a hung pass.
|
|
type mockThinker struct {
|
|
mu sync.Mutex
|
|
text string
|
|
err error
|
|
sleep time.Duration
|
|
calls int
|
|
last string
|
|
}
|
|
|
|
func (m *mockThinker) Think(ctx context.Context, sessionID, trigger string) (string, error) {
|
|
if err := ctx.Err(); err != nil {
|
|
return "", err
|
|
}
|
|
if m.sleep > 0 {
|
|
select {
|
|
case <-ctx.Done():
|
|
return "", ctx.Err()
|
|
case <-time.After(m.sleep):
|
|
}
|
|
}
|
|
m.mu.Lock()
|
|
m.calls++
|
|
m.last = trigger
|
|
m.mu.Unlock()
|
|
return m.text, m.err
|
|
}
|
|
|
|
func (m *mockThinker) count() int { m.mu.Lock(); defer m.mu.Unlock(); return m.calls }
|
|
func (m *mockThinker) lastTrigger() string { m.mu.Lock(); defer m.mu.Unlock(); return m.last }
|
|
|
|
// mockState returns a fixed UserState.
|
|
type mockState struct{ state UserState }
|
|
|
|
func (m *mockState) Current(context.Context) (UserState, error) { return m.state, nil }
|
|
|
|
// mockPusher records Push calls and returns a fixed result.
|
|
type mockPusher struct {
|
|
mu sync.Mutex
|
|
result PushResult
|
|
calls int
|
|
lastMsg adapter.UnifiedMessage
|
|
lastHint string
|
|
}
|
|
|
|
func (p *mockPusher) Push(ctx context.Context, req PushRequest) PushResult {
|
|
p.mu.Lock()
|
|
p.calls++
|
|
p.lastMsg = req.Message
|
|
p.lastHint = req.ChannelHint
|
|
p.mu.Unlock()
|
|
return p.result
|
|
}
|
|
|
|
func (p *mockPusher) count() int { p.mu.Lock(); defer p.mu.Unlock(); return p.calls }
|
|
|
|
// mockActive implements ActiveClient.
|
|
type mockActive struct {
|
|
mu sync.Mutex
|
|
n int
|
|
err error
|
|
calls int
|
|
}
|
|
|
|
func (m *mockActive) SendProactive(ctx context.Context, msg adapter.UnifiedMessage) (int, error) {
|
|
m.mu.Lock()
|
|
m.calls++
|
|
m.mu.Unlock()
|
|
return m.n, m.err
|
|
}
|
|
|
|
// mockIMAdapter implements IMAdapter.
|
|
type mockIMAdapter struct {
|
|
mu sync.Mutex
|
|
err error
|
|
calls int
|
|
last adapter.UnifiedMessage
|
|
}
|
|
|
|
func (m *mockIMAdapter) SendProactive(ctx context.Context, msg adapter.UnifiedMessage) error {
|
|
m.mu.Lock()
|
|
m.calls++
|
|
m.last = msg
|
|
m.mu.Unlock()
|
|
return m.err
|
|
}
|
|
|
|
func (m *mockIMAdapter) count() int { m.mu.Lock(); defer m.mu.Unlock(); return m.calls }
|
|
|
|
// mockGate implements PushGate.
|
|
type mockGate struct{ allow bool }
|
|
|
|
func (g *mockGate) Allow(context.Context) bool { return g.allow }
|
|
|
|
// newTestDB opens an isolated in-memory SQLite handle for one test and closes it
|
|
// on cleanup.
|
|
func newTestDB(t *testing.T) *sql.DB {
|
|
t.Helper()
|
|
h, err := db.Open(context.Background(), ":memory:", nil)
|
|
if err != nil {
|
|
t.Fatalf("open test db: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = h.Close() })
|
|
return h
|
|
}
|
|
|
|
// discardLog returns a logger that writes nowhere.
|
|
func discardLog() *slog.Logger {
|
|
return slog.New(slog.NewTextHandler(io.Discard, nil))
|
|
}
|
|
|
|
// eventually polls cond until it is true or the deadline passes.
|
|
func eventually(t *testing.T, timeout time.Duration, cond func() bool) {
|
|
t.Helper()
|
|
deadline := time.Now().Add(timeout)
|
|
for time.Now().Before(deadline) {
|
|
if cond() {
|
|
return
|
|
}
|
|
time.Sleep(5 * time.Millisecond)
|
|
}
|
|
t.Fatalf("condition not met within %v", timeout)
|
|
}
|