Cherry pick himanshu

Author: nang-dev

src/api/providers/pearai/pearai.ts

@@ -24,6 +24,7 @@ export class PearAIHandler extends BaseProvider implements SingleCompletionHandl
 	constructor(options: ApiHandlerOptions) {
 		super()
 		if (!options.pearaiApiKey) {
+			vscode.commands.executeCommand("pearai-roo-cline.PearAIKeysNotFound", undefined)
 			vscode.window.showErrorMessage("PearAI API key not found.", "Login to PearAI").then(async (selection) => {
 				if (selection === "Login to PearAI") {
 					const extensionUrl = `${vscode.env.uriScheme}://pearai.pearai/auth`

src/extension.ts

@@ -165,6 +165,15 @@ export async function activate(context: vscode.ExtensionContext) {
 		}),
 	)
 
+	context.subscriptions.push(
+		vscode.commands.registerCommand("pearai-roo-cline.PearAIKeysNotFound", async () => {
+			const provider = await ClineProvider.getInstance()
+			if (provider) {
+				provider.postMessageToWebview({ type: "action", action: "PearAIKeysNotFound" })
+			}
+		}),
+	)
+
 	/*
 	We use the text document content provider API to show the left side for diff view by creating a virtual document for the original content. This makes it readonly so users know to edit the right side if they want to keep their changes.
 

src/shared/ExtensionMessage.ts

@@ -76,6 +76,7 @@ export interface ExtensionMessage {
 		| "promptsButtonClicked"
 		| "didBecomeVisible"
 		| "focusInput"
+		| "PearAIKeysNotFound"
 	invoke?: "newChat" | "sendMessage" | "primaryButtonClick" | "secondaryButtonClick" | "setChatBoxMessage"
 	state?: ExtensionState
 	images?: string[]

webview-ui/src/components/chat/ChatView.tsx

@@ -119,9 +119,11 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 	const apiMetrics = useMemo(() => getApiMetrics(modifiedMessages), [modifiedMessages])
 
 	const [inputValue, setInputValue] = useState("")
+	const [lastInputValue, setLastInputValue] = useState("")
 	const textAreaRef = useRef<HTMLTextAreaElement>(null)
 	const [textAreaDisabled, setTextAreaDisabled] = useState(false)
 	const [selectedImages, setSelectedImages] = useState<string[]>([])
+	const [lastInputImages, setLastInputImages] = useState<string[]>([])
 
 	// we need to hold on to the ask because useEffect > lastMessage will always let us know when an ask comes in and handle it, but by the time handleMessage is called, the last message might not be the ask anymore (it could be a say that followed)
 	const [clineAsk, setClineAsk] = useState<ClineAsk | undefined>(undefined)
@@ -362,6 +364,8 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 
 	const handleChatReset = useCallback(() => {
 		// Only reset message-specific state, preserving mode.
+		setLastInputValue(inputValue)
+		setLastInputImages(selectedImages)
 		setInputValue("")
 		setTextAreaDisabled(true)
 		setSelectedImages([])
@@ -371,7 +375,7 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 		// setPrimaryButtonText(undefined)
 		// setSecondaryButtonText(undefined)
 		disableAutoScrollRef.current = false
-	}, [])
+	}, [inputValue, selectedImages])
 
 	const handleSendMessage = useCallback(
 		(text: string, images: string[]) => {
@@ -546,6 +550,13 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 						case "focusInput":
 							textAreaRef.current?.focus()
 							break
+						case "PearAIKeysNotFound":
+							setInputValue(lastInputValue)
+							setTextAreaDisabled(false)
+							setSelectedImages(lastInputImages)
+							setEnableButtons(true)
+							disableAutoScrollRef.current = true
+							break
 					}
 					break
 				case "selectedImages":
@@ -586,6 +597,12 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 			handleSetChatBoxMessage,
 			handlePrimaryButtonClick,
 			handleSecondaryButtonClick,
+			lastInputValue,
+			lastInputImages,
+			setInputValue,
+			setTextAreaDisabled,
+			setSelectedImages,
+			setEnableButtons,
 		],
 	)