Files
Enterprise-Knowledge-Base/cli/internal/agent/mode.go
T
nullkey a39805aa89 feat(cli): scaffold weknora CLI module foundation (PR-1)
Empty cli/ Go module with Factory + Options skeleton that all v0.x commands
plug into.

Foundation packages (10 internal/):
- cmdutil: Factory(3 closure: Config/Client/Prompter) + Options + Errors
  with namespaced ErrorCode + typed predicates (IsAuthError / IsNotFound /
  IsTransient / IsAuthExpired) + Exporter + AddJSONFlags + ExitCode/PrintError
- iostreams: package singleton with TTY detection; ColorEnabled honors
  NO_COLOR / TERM=dumb on demand
- agent: mode + detect + annotations (Stripe pkg/useragent pattern;
  CLAUDECODE / CURSOR_AGENT / CODEX_* / AIDER_* / CONTINUE_* /
  OPENCODE_* / GEMINICODER env detection)
- build: ldflags-injected version/commit/date
- safepaths: separator-aware path-traversal protection
- config: yaml.v3 schema + atomic save 0600 (XDG-style)
- format: success/failure envelope (code/hint/request_id/risk/retryable/
  console_url) with typed RiskLevel
- prompt: Prompter interface + AgentPrompter (rejects in agent mode)
- secrets: file 0600 store with weknora:<context>:<key> namespace
- testutil: XDGTempDir(t) test helper

Smoke command: weknora version (-/--json). Build/test/vet matrix in
.github/workflows/cli.yml across linux x macos x windows x Go 1.24.

Factory.Client returns CodeLocalUnimplemented until PR-3 wires the SDK
adapter; PR-1 commands (version) don't need it.
2026-05-07 23:19:46 +08:00

57 lines
1.6 KiB
Go

package agent
import (
"os"
"github.com/spf13/cobra"
)
// agentSugarFlags lists the flags ApplyAgentSugar sets to their agent-friendly
// defaults when triggered. Each entry is no-op'd at runtime if the leaf hasn't
// registered the flag (e.g. version doesn't have --no-progress).
var agentSugarFlags = []struct{ name, value string }{
{"json", "true"},
{"no-interactive", "true"},
{"no-progress", "true"},
}
// ApplyAgentSugar evaluates --agent / WEKNORA_AGENT / env-based AI agent
// detection, then sets json / no-interactive / no-progress to their
// agent-friendly defaults unless the user already set them.
func ApplyAgentSugar(cmd *cobra.Command) {
if !ShouldUseAgentMode(cmd) {
return
}
for _, kv := range agentSugarFlags {
if f := cmd.Flags().Lookup(kv.name); f != nil && !cmd.Flags().Changed(kv.name) {
_ = cmd.Flags().Set(kv.name, kv.value)
}
}
}
// ShouldUseAgentMode returns true if any of the three triggers are active:
// (1) --agent flag, (2) WEKNORA_AGENT=1 env, (3) a known agent env var is set.
func ShouldUseAgentMode(cmd *cobra.Command) bool {
if cmd != nil {
if v, err := cmd.Flags().GetBool("agent"); err == nil && v {
return true
}
}
if truthy(os.Getenv("WEKNORA_AGENT")) {
return true
}
return DetectAIAgent() != ""
}
// truthy treats any non-empty value other than "0" / "false" / "FALSE" as
// truthy. This is wider than strconv.ParseBool by design — users set
// WEKNORA_AGENT=1, =true, or =yes interchangeably across CI shells.
func truthy(v string) bool {
switch v {
case "", "0", "false", "FALSE":
return false
default:
return true
}
}