* First commit

pkg/heartbeat/service.go

@@ -0,0 +1,134 @@
+package heartbeat
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"sync"
+	"time"
+)
+
+type HeartbeatService struct {
+	workspace   string
+	onHeartbeat func(string) (string, error)
+	interval    time.Duration
+	enabled     bool
+	mu          sync.RWMutex
+	stopChan    chan struct{}
+}
+
+func NewHeartbeatService(workspace string, onHeartbeat func(string) (string, error), intervalS int, enabled bool) *HeartbeatService {
+	return &HeartbeatService{
+		workspace:   workspace,
+		onHeartbeat: onHeartbeat,
+		interval:    time.Duration(intervalS) * time.Second,
+		enabled:     enabled,
+		stopChan:    make(chan struct{}),
+	}
+}
+
+func (hs *HeartbeatService) Start() error {
+	hs.mu.Lock()
+	defer hs.mu.Unlock()
+
+	if hs.running() {
+		return nil
+	}
+
+	if !hs.enabled {
+		return fmt.Errorf("heartbeat service is disabled")
+	}
+
+	go hs.runLoop()
+
+	return nil
+}
+
+func (hs *HeartbeatService) Stop() {
+	hs.mu.Lock()
+	defer hs.mu.Unlock()
+
+	if !hs.running() {
+		return
+	}
+
+	close(hs.stopChan)
+}
+
+func (hs *HeartbeatService) running() bool {
+	select {
+	case <-hs.stopChan:
+		return false
+	default:
+		return true
+	}
+}
+
+func (hs *HeartbeatService) runLoop() {
+	ticker := time.NewTicker(hs.interval)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-hs.stopChan:
+			return
+		case <-ticker.C:
+			hs.checkHeartbeat()
+		}
+	}
+}
+
+func (hs *HeartbeatService) checkHeartbeat() {
+	hs.mu.RLock()
+	if !hs.enabled || !hs.running() {
+		hs.mu.RUnlock()
+		return
+	}
+	hs.mu.RUnlock()
+
+	prompt := hs.buildPrompt()
+
+	if hs.onHeartbeat != nil {
+		_, err := hs.onHeartbeat(prompt)
+		if err != nil {
+			hs.log(fmt.Sprintf("Heartbeat error: %v", err))
+		}
+	}
+}
+
+func (hs *HeartbeatService) buildPrompt() string {
+	notesDir := filepath.Join(hs.workspace, "memory")
+	notesFile := filepath.Join(notesDir, "HEARTBEAT.md")
+
+	var notes string
+	if data, err := os.ReadFile(notesFile); err == nil {
+		notes = string(data)
+	}
+
+	now := time.Now().Format("2006-01-02 15:04")
+
+	prompt := fmt.Sprintf(`# Heartbeat Check
+
+Current time: %s
+
+Check if there are any tasks I should be aware of or actions I should take.
+Review the memory file for any important updates or changes.
+Be proactive in identifying potential issues or improvements.
+
+%s
+`, now, notes)
+
+	return prompt
+}
+
+func (hs *HeartbeatService) log(message string) {
+	logFile := filepath.Join(hs.workspace, "memory", "heartbeat.log")
+	f, err := os.OpenFile(logFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
+	if err != nil {
+		return
+	}
+	defer f.Close()
+
+	timestamp := time.Now().Format("2006-01-02 15:04:05")
+	f.WriteString(fmt.Sprintf("[%s] %s\n", timestamp, message))
+}