File size: 4,605 Bytes
064bfd6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 | import { useState } from 'react'
import {
type AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
logEvent,
} from '../../services/analytics/index.js'
import { sanitizeToolNameForAnalytics } from '../../services/analytics/metadata.js'
import { useSetAppState } from '../../state/AppState.js'
import type { ToolUseConfirm } from './PermissionRequest.js'
import { logUnaryPermissionEvent } from './utils.js'
/**
* Shared feedback-mode state + handlers for shell permission dialogs (Bash,
* PowerShell). Encapsulates the yes/no input-mode toggle, feedback text state,
* focus tracking, and reject handling.
*/
export function useShellPermissionFeedback({
toolUseConfirm,
onDone,
onReject,
explainerVisible,
}: {
toolUseConfirm: ToolUseConfirm
onDone: () => void
onReject: () => void
explainerVisible: boolean
}): {
yesInputMode: boolean
noInputMode: boolean
yesFeedbackModeEntered: boolean
noFeedbackModeEntered: boolean
acceptFeedback: string
rejectFeedback: string
setAcceptFeedback: (v: string) => void
setRejectFeedback: (v: string) => void
focusedOption: string
handleInputModeToggle: (option: string) => void
handleReject: (feedback?: string) => void
handleFocus: (value: string) => void
} {
const setAppState = useSetAppState()
const [rejectFeedback, setRejectFeedback] = useState('')
const [acceptFeedback, setAcceptFeedback] = useState('')
const [yesInputMode, setYesInputMode] = useState(false)
const [noInputMode, setNoInputMode] = useState(false)
const [focusedOption, setFocusedOption] = useState('yes')
// Track whether user ever entered feedback mode (persists after collapse)
const [yesFeedbackModeEntered, setYesFeedbackModeEntered] = useState(false)
const [noFeedbackModeEntered, setNoFeedbackModeEntered] = useState(false)
// Handle Tab key toggling input mode for Yes/No options
function handleInputModeToggle(option: string) {
// Notify that user is interacting with the dialog
toolUseConfirm.onUserInteraction()
const analyticsProps = {
toolName: sanitizeToolNameForAnalytics(
toolUseConfirm.tool.name,
) as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
isMcp: toolUseConfirm.tool.isMcp ?? false,
}
if (option === 'yes') {
if (yesInputMode) {
setYesInputMode(false)
logEvent('tengu_accept_feedback_mode_collapsed', analyticsProps)
} else {
setYesInputMode(true)
setYesFeedbackModeEntered(true)
logEvent('tengu_accept_feedback_mode_entered', analyticsProps)
}
} else if (option === 'no') {
if (noInputMode) {
setNoInputMode(false)
logEvent('tengu_reject_feedback_mode_collapsed', analyticsProps)
} else {
setNoInputMode(true)
setNoFeedbackModeEntered(true)
logEvent('tengu_reject_feedback_mode_entered', analyticsProps)
}
}
}
function handleReject(feedback?: string) {
const trimmedFeedback = feedback?.trim()
const hasFeedback = !!trimmedFeedback
// Log escape if no feedback was provided (user pressed ESC)
if (!hasFeedback) {
logEvent('tengu_permission_request_escape', {
explainer_visible: explainerVisible,
})
// Increment escape count for attribution tracking
setAppState(prev => ({
...prev,
attribution: {
...prev.attribution,
escapeCount: prev.attribution.escapeCount + 1,
},
}))
}
logUnaryPermissionEvent(
'tool_use_single',
toolUseConfirm,
'reject',
hasFeedback,
)
if (trimmedFeedback) {
toolUseConfirm.onReject(trimmedFeedback)
} else {
toolUseConfirm.onReject()
}
onReject()
onDone()
}
function handleFocus(value: string) {
// Notify that user is interacting with the dialog (only if focus changed)
// This prevents triggering on the initial mount/render
if (value !== focusedOption) {
toolUseConfirm.onUserInteraction()
}
// Reset input mode when navigating away, but only if no text typed
if (value !== 'yes' && yesInputMode && !acceptFeedback.trim()) {
setYesInputMode(false)
}
if (value !== 'no' && noInputMode && !rejectFeedback.trim()) {
setNoInputMode(false)
}
setFocusedOption(value)
}
return {
yesInputMode,
noInputMode,
yesFeedbackModeEntered,
noFeedbackModeEntered,
acceptFeedback,
rejectFeedback,
setAcceptFeedback,
setRejectFeedback,
focusedOption,
handleInputModeToggle,
handleReject,
handleFocus,
}
}
|