mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
239a22899b
模型模式: - 新增 LLMConfig/Source.ThinkingEnabled 配置，通过 ExtraBody 控制 DeepSeek thinking mode，默认关闭 - SeedDefaults/ToConfig 读写 core.llm.thinking_enabled - deepseek.lua 移除硬编码 temperature=0 Unicode 截断: - truncateStr 改按 rune 计数，修复中文截断乱码 审计修复 (Critical): - graph.go: defer rows.Close 在 for 循环 → 显式 Close (连接池泄漏) - cli/openclaw/plugin.go: bare type assertion → comma-ok (panic) - channel.go: payload["type"].(string) → comma-ok (panic) - webui/handler.go: .(string) → fmt.Sprint (panic) - agent.go: 添加 nil provider 错误返回 审计修复 (High): - events/bus.go: copy handler slice under RLock (data race) - webui/handler.go: SSE 通过 channel 串行化写入 (data race) - timer/plugin.go: time.Sleep → select with stopCh (Stop 阻塞) - provider.go: stream ch <- 添加 select ctx.Done (goroutine 泄漏) - main.go: outputCh goroutine 添加 ctx.Done 退出路径
105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
package timer
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"gitcode.com/JianFeeeee/HomeAgent/internal/plugin"
|
|
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
|
)
|
|
|
|
func init() {
|
|
plugin.RegisterFactory("timer", func(name string, config map[string]interface{}) (sdk.Plugin, error) {
|
|
return New(name), nil
|
|
})
|
|
}
|
|
|
|
type Plugin struct {
|
|
name string
|
|
mu sync.Mutex
|
|
wg sync.WaitGroup
|
|
stopCh chan struct{}
|
|
}
|
|
|
|
type timerTask struct {
|
|
id int
|
|
dur time.Duration
|
|
message string
|
|
doneAt time.Time
|
|
s *sdk.PluginSDK
|
|
}
|
|
|
|
func New(name string) *Plugin {
|
|
return &Plugin{name: name, stopCh: make(chan struct{})}
|
|
}
|
|
|
|
func (p *Plugin) Name() string { return p.name }
|
|
|
|
func (p *Plugin) Start(s *sdk.PluginSDK) error {
|
|
s.RegisterTool("timer_set", sdk.ToolDef{
|
|
Name: "timer_set",
|
|
Description: "设置一个定时提醒。倒计时结束后通过中断通道通知 agent。",
|
|
Parameters: map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"duration": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "持续时间，例如 5s, 2m, 1h",
|
|
},
|
|
"message": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "提醒内容",
|
|
},
|
|
},
|
|
"required": []string{"duration", "message"},
|
|
},
|
|
}, func(args map[string]interface{}) (interface{}, error) {
|
|
durStr, _ := args["duration"].(string)
|
|
message, _ := args["message"].(string)
|
|
if durStr == "" {
|
|
return map[string]interface{}{"error": "duration is required"}, nil
|
|
}
|
|
if message == "" {
|
|
return map[string]interface{}{"error": "message is required"}, nil
|
|
}
|
|
|
|
dur, err := time.ParseDuration(durStr)
|
|
if err != nil {
|
|
return map[string]interface{}{"error": fmt.Sprintf("invalid duration %q: %v", durStr, err)}, nil
|
|
}
|
|
|
|
p.mu.Lock()
|
|
p.wg.Add(1)
|
|
p.mu.Unlock()
|
|
|
|
go func() {
|
|
defer p.wg.Done()
|
|
select {
|
|
case <-time.After(dur):
|
|
log.Printf("[timer] firing: %s (%s later)", message, dur)
|
|
s.InjectInterruptText("timer", "timer", fmt.Sprintf("timer: %s", message))
|
|
case <-p.stopCh:
|
|
log.Printf("[timer] cancelled: %s", message)
|
|
}
|
|
}()
|
|
|
|
doneAt := time.Now().Add(dur)
|
|
return map[string]interface{}{
|
|
"status": "timer_set",
|
|
"duration": durStr,
|
|
"message": message,
|
|
"done_at": doneAt.Format(time.RFC3339),
|
|
}, nil
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *Plugin) Stop() error {
|
|
close(p.stopCh)
|
|
p.wg.Wait()
|
|
return nil
|
|
}
|