feat: sync Go Auggie hook integration

This commit is contained in:
ZacharyZhang-NY committed 2026-07-19 08:55:00 -04:00
1 parent e2f319914b
commit e334e84333
8 files changed
+1430 -2

No files matched your search

+141
View File
@@ -0,0 +1,141 @@
package plugins
import "fmt"
// AugmentPlugin manages Auggie native global user hooks.
type AugmentPlugin struct{}
func (p *AugmentPlugin) Install(config InstallConfig) error {
if config.AgentID == "" {
return fmt.Errorf("agent ID is required")
}
document, err := readAugmentConfig()
if err != nil {
return err
}
if config.GuardScriptPath == "" {
return fmt.Errorf("guard script path is required")
}
guardExists, err := regularFileExists(
config.GuardScriptPath, "Elydora guard runtime",
)
if err != nil {
return err
}
if !guardExists {
return fmt.Errorf(
"Elydora guard runtime is missing: %s", config.GuardScriptPath,
)
}
hookPath, err := hookScriptPath(config.AgentID)
if err != nil {
return err
}
if config.HookScript != "" {
hookPath = config.HookScript
}
nodePath, err := resolveNodeRuntime()
if err != nil {
return err
}
if err := validateAugmentMatchers(document.hooks, nodePath); err != nil {
return err
}
runtimeRoot, err := augmentRuntimeRoot()
if err != nil {
return err
}
wrappers := resolveAugmentWrapperPaths(runtimeRoot, config.AgentID)
hooks, _ := removeManagedAugmentHooks(document.hooks, "", runtimeRoot)
hooks["PreToolUse"] = append(
hooks["PreToolUse"],
buildAugmentGroup(buildAugmentHandler(wrappers.guard)),
)
hooks["PostToolUse"] = append(
hooks["PostToolUse"],
buildAugmentGroup(buildAugmentHandler(wrappers.audit)),
)
next := cloneAugmentObject(document.root)
next["hooks"] = renderAugmentHooks(hooks)
runtimeConfig := config
runtimeConfig.AgentName = augmentAgentKey
if err := GenerateHookScript(hookPath, runtimeConfig); err != nil {
return fmt.Errorf("generate hook script: %w", err)
}
if err := writeHookFileAtomic(
wrappers.guard,
buildAugmentWrapper(nodePath, config.GuardScriptPath),
0700,
); err != nil {
return fmt.Errorf("write Auggie guard wrapper: %w", err)
}
if err := writeHookFileAtomic(
wrappers.audit,
buildAugmentWrapper(nodePath, hookPath),
0700,
); err != nil {
return fmt.Errorf("write Auggie audit wrapper: %w", err)
}
if err := writeHookJSONObjectAtomic(document.configPath, next); err != nil {
return fmt.Errorf("write Auggie settings: %w", err)
}
fmt.Println("Auggie: user-level PreToolUse and PostToolUse hooks installed.")
return nil
}
func (p *AugmentPlugin) Uninstall(agentID string) error {
document, err := readAugmentConfig()
if err != nil || !document.exists {
return err
}
runtimeRoot, err := augmentRuntimeRoot()
if err != nil {
return err
}
hooks, changed := removeManagedAugmentHooks(
document.hooks, agentID, runtimeRoot,
)
if !changed {
return nil
}
next := cloneAugmentObject(document.root)
if len(hooks) == 0 {
delete(next, "hooks")
} else {
next["hooks"] = renderAugmentHooks(hooks)
}
if len(next) == 0 {
return removeHookFile(document.configPath, "Auggie settings")
}
return writeHookJSONObjectAtomic(document.configPath, next)
}
func (p *AugmentPlugin) Status() (PluginStatus, error) {
document, err := readAugmentConfig()
status := PluginStatus{
AgentName: augmentAgentKey,
DisplayName: "Augment Code CLI",
ConfigPath: document.configPath,
}
if err != nil {
return status, err
}
runtimeRoot, err := augmentRuntimeRoot()
if err != nil {
return status, err
}
contracts := augmentRuntimeContracts(document.hooks, runtimeRoot)
if len(contracts) == 0 {
return status, nil
}
status.HookConfigured = true
status.HookScriptExists, err = augmentRuntimeFilesExist(
contracts, runtimeRoot,
)
if err != nil {
return status, err
}
status.Installed = status.HookScriptExists
return status, nil
}