Method Cleanup

cli/agent/session_workspace.go:203–236

Cleanup removes the session workspace unless CHATCLI_AGENT_KEEP_TMPDIR=true. Safe to call multiple times.

()

// Cleanup removes the session workspace unless CHATCLI_AGENT_KEEP_TMPDIR=true.
// Safe to call multiple times.
func (ws *SessionWorkspace) Cleanup() {
	if ws == nil {
		return
	}

	sessionWorkspaceMu.Lock()
	defer sessionWorkspaceMu.Unlock()

	// Unregister first so a late tool call can't land a file into a
	// directory we're about to delete.
	UnregisterAuxReadPath(ws.Root)
	engine.UnregisterAuxPath(ws.Root)
	SetBudgetResultDir("") // fall back to default
	fireResultDirSetters("")

	if ws.keep {
		if ws.logger != nil {
			ws.logger.Info("Keeping session workspace (CHATCLI_AGENT_KEEP_TMPDIR=true)",
				zap.String("root", ws.Root))
		}
	} else {
		if err := os.RemoveAll(ws.Root); err != nil && ws.logger != nil {
			ws.logger.Warn("Failed to remove session workspace",
				zap.String("root", ws.Root),
				zap.Error(err))
		}
	}

	_ = os.Unsetenv("CHATCLI_AGENT_TMPDIR")

	if sessionWorkspace == ws {
		sessionWorkspace = nil
	}
}

// AuxAllowedPaths returns paths the engine and read validator should treat as
// inside the boundary, in addition to the workspace root. Includes the