bd sync: 2025-11-24 11:28:32
This commit is contained in:
154
cmd/bd/tips.go
154
cmd/bd/tips.go
@@ -5,6 +5,7 @@ import (
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strconv"
|
||||
"sync"
|
||||
@@ -152,3 +153,156 @@ func recordTipShown(store storage.Storage, tipID string) {
|
||||
value := time.Now().Format(time.RFC3339)
|
||||
_ = store.SetMetadata(context.Background(), key, value)
|
||||
}
|
||||
|
||||
// InjectTip adds a dynamic tip to the registry at runtime.
|
||||
// This enables tips to be programmatically added based on detected conditions.
|
||||
//
|
||||
// Parameters:
|
||||
// - id: Unique identifier for the tip (used for frequency tracking)
|
||||
// - message: The tip message to display to the user
|
||||
// - priority: Higher values = shown first when eligible (e.g., 100 for critical, 30 for suggestions)
|
||||
// - frequency: Minimum time between showings (e.g., 24*time.Hour for daily)
|
||||
// - probability: Chance of showing when eligible (0.0 to 1.0)
|
||||
// - condition: Function that returns true when tip should be eligible
|
||||
//
|
||||
// Example usage:
|
||||
//
|
||||
// // Critical security update - always show
|
||||
// InjectTip("security_update", "CRITICAL: Security update available!", 100, 0, 1.0, func() bool { return true })
|
||||
//
|
||||
// // New version available - frequent but not always
|
||||
// InjectTip("upgrade_available", "New version available", 90, 7*24*time.Hour, 0.8, func() bool { return true })
|
||||
//
|
||||
// // Feature suggestion - occasional
|
||||
// InjectTip("try_filters", "Try using filters", 50, 14*24*time.Hour, 0.4, func() bool { return true })
|
||||
func InjectTip(id, message string, priority int, frequency time.Duration, probability float64, condition func() bool) {
|
||||
tipsMutex.Lock()
|
||||
defer tipsMutex.Unlock()
|
||||
|
||||
// Check if tip with this ID already exists - update it if so
|
||||
for i, tip := range tips {
|
||||
if tip.ID == id {
|
||||
tips[i] = Tip{
|
||||
ID: id,
|
||||
Condition: condition,
|
||||
Message: message,
|
||||
Frequency: frequency,
|
||||
Priority: priority,
|
||||
Probability: probability,
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Add new tip
|
||||
tips = append(tips, Tip{
|
||||
ID: id,
|
||||
Condition: condition,
|
||||
Message: message,
|
||||
Frequency: frequency,
|
||||
Priority: priority,
|
||||
Probability: probability,
|
||||
})
|
||||
}
|
||||
|
||||
// RemoveTip removes a tip from the registry by ID.
|
||||
// This is useful for removing dynamically injected tips when they are no longer relevant.
|
||||
// It is safe to call with a non-existent ID (no-op).
|
||||
func RemoveTip(id string) {
|
||||
tipsMutex.Lock()
|
||||
defer tipsMutex.Unlock()
|
||||
|
||||
for i, tip := range tips {
|
||||
if tip.ID == id {
|
||||
tips = append(tips[:i], tips[i+1:]...)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// isClaudeDetected checks if the user is running within a Claude Code environment.
|
||||
// Detection methods:
|
||||
// - CLAUDE_CODE environment variable (set by Claude Code)
|
||||
// - ANTHROPIC_CLI environment variable
|
||||
// - Presence of ~/.claude directory (Claude Code config)
|
||||
func isClaudeDetected() bool {
|
||||
// Check environment variables set by Claude Code
|
||||
if os.Getenv("CLAUDE_CODE") != "" || os.Getenv("ANTHROPIC_CLI") != "" {
|
||||
return true
|
||||
}
|
||||
|
||||
// Check if ~/.claude directory exists (Claude Code stores config here)
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
if _, err := os.Stat(filepath.Join(home, ".claude")); err == nil {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// isClaudeSetupComplete checks if the beads Claude integration is properly configured.
|
||||
// Checks for either global or project-level installation of the beads hooks.
|
||||
func isClaudeSetupComplete() bool {
|
||||
// Check for global installation
|
||||
home, err := os.UserHomeDir()
|
||||
if err == nil {
|
||||
commandFile := filepath.Join(home, ".claude", "commands", "prime_beads.md")
|
||||
hooksDir := filepath.Join(home, ".claude", "hooks")
|
||||
|
||||
// Check for prime_beads command
|
||||
if _, err := os.Stat(commandFile); err == nil {
|
||||
// Check for sessionstart hook (could be a file or directory)
|
||||
hookPath := filepath.Join(hooksDir, "sessionstart")
|
||||
if _, err := os.Stat(hookPath); err == nil {
|
||||
return true // Global hooks installed
|
||||
}
|
||||
// Also check PreToolUse hook which is used by beads
|
||||
preToolUsePath := filepath.Join(hooksDir, "PreToolUse")
|
||||
if _, err := os.Stat(preToolUsePath); err == nil {
|
||||
return true // Global hooks installed
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Check for project-level installation
|
||||
commandFile := ".claude/commands/prime_beads.md"
|
||||
hooksDir := ".claude/hooks"
|
||||
|
||||
if _, err := os.Stat(commandFile); err == nil {
|
||||
hookPath := filepath.Join(hooksDir, "sessionstart")
|
||||
if _, err := os.Stat(hookPath); err == nil {
|
||||
return true // Project hooks installed
|
||||
}
|
||||
preToolUsePath := filepath.Join(hooksDir, "PreToolUse")
|
||||
if _, err := os.Stat(preToolUsePath); err == nil {
|
||||
return true // Project hooks installed
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// initDefaultTips registers the built-in tips.
|
||||
// Called during initialization to populate the tip registry.
|
||||
func initDefaultTips() {
|
||||
// Claude setup tip - suggest running bd setup claude when Claude is detected
|
||||
// but the integration is not configured
|
||||
InjectTip(
|
||||
"claude_setup",
|
||||
"Run 'bd setup claude' to enable automatic context recovery in Claude Code",
|
||||
100, // Highest priority - this is important for Claude users
|
||||
24*time.Hour, // Daily minimum gap
|
||||
0.6, // 60% chance when eligible (~4 times per week)
|
||||
func() bool {
|
||||
return isClaudeDetected() && !isClaudeSetupComplete()
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
// init initializes the tip system with default tips
|
||||
func init() {
|
||||
initDefaultTips()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user