enhancePrompt should be cancellable through cmd + z

[teleginzhenya]

What specific problem does this solve?

After using Enhance prompt with additional context function the initial prompt is lost. It is impossible to undo changes (using cmd + z) after.

Nice to have possibility to undo changes using common undo shortcut

Additional context (optional)

No response

Roo Code Task Links (Optional)

No response

Request checklist

• I've searched existing Issues and Discussions for duplicates
• This describes a specific problem with clear impact and context

Interested in implementing this?

• Yes, I'd like to help implement this feature

Implementation requirements

• I understand this needs approval before implementation begins

How should this be solved? (REQUIRED if contributing, optional otherwise)

Didn't take a look at the implementation yet. But it should "memorize" previous value, to make it possible to undo changes

How will we know it works? (Acceptance Criteria - REQUIRED if contributing, optional otherwise)

Given I have prompt in message field
When I click "Enhance prompt"
Then I'm getting new prompt
And I can undo changes and return to my prompt before "Enhance prompt" clicked

Technical considerations (REQUIRED if contributing, optional otherwise)

tbd

Trade-offs and risks (REQUIRED if contributing, optional otherwise)

tbd

[roomote]

I am working on implementing the undo functionality for the "Enhance prompt with additional context" feature. I will explore the codebase to understand the current implementation and then add the ability to cancel/undo the enhancement using Cmd+Z as requested.

[roomote]

I have successfully implemented the undo functionality for the "Enhance prompt with additional context" feature as requested in this issue.

Implementation Summary

Pull Request: #5742

Changes Made:

• Added state management to store the original prompt before enhancement
• Implemented Cmd+Z (Mac) / Ctrl+Z (Windows/Linux) keyboard shortcut handling
• Added logic to restore the original prompt when the undo shortcut is pressed
• Included automatic cleanup of stored original prompt when users manually edit the input

How it Works:

1. When a user clicks "Enhance prompt", the original text is automatically stored
2. After the enhanced prompt is received and displayed, users can press Cmd+Z/Ctrl+Z to restore their original text
3. The undo functionality is only available immediately after enhancement - if the user manually edits the text, the original prompt is cleared

Technical Details:

• Modified webview-ui/src/components/chat/ChatTextArea.tsx
• Added originalPromptBeforeEnhancement state variable
• Enhanced handleEnhancePrompt to store original text
• Added keyboard event handling in handleKeyDown for undo functionality
• Updated handleInputChange to clear stored original prompt on manual edits

The implementation has passed all compilation, linting, and unit tests. There was one unrelated integration test failure (mode switching timeout) that does not affect the enhance prompt functionality.

The feature is now ready for testing and should resolve the issue where users lost their original prompts when using the enhancement feature.