@@ -8,6 +8,7 @@ import com.google.gson.Gson
88import com.intellij.ide.BrowserUtil
99import com.intellij.ide.util.RunOnceUtil
1010import com.intellij.openapi.application.runInEdt
11+ import com.intellij.openapi.fileEditor.FileEditorManager
1112import com.intellij.openapi.options.ShowSettingsUtil
1213import com.intellij.openapi.project.Project
1314import com.intellij.ui.jcef.JBCefJSQuery.Response
@@ -22,8 +23,7 @@ import kotlinx.coroutines.flow.merge
2223import kotlinx.coroutines.flow.onEach
2324import kotlinx.coroutines.launch
2425import org.cef.browser.CefBrowser
25- import org.eclipse.lsp4j.Position
26- import org.eclipse.lsp4j.Range
26+ import org.eclipse.lsp4j.TextDocumentIdentifier
2727import software.aws.toolkits.core.utils.error
2828import software.aws.toolkits.core.utils.getLogger
2929import software.aws.toolkits.core.utils.warn
@@ -34,7 +34,6 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.AmazonQLspService
3434import software.aws.toolkits.jetbrains.services.amazonq.lsp.encryption.JwtEncryptionManager
3535import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.AwsServerCapabilitiesProvider
3636import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatCommunicationManager
37- import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.getTextDocumentIdentifier
3837import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.AUTH_FOLLOW_UP_CLICKED
3938import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.AuthFollowUpClickNotification
4039import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ButtonClickNotification
@@ -71,7 +70,6 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CopyC
7170import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CopyCodeToClipboardParams
7271import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CreatePromptNotification
7372import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CreatePromptParams
74- import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CursorState
7573import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.EncryptedChatParams
7674import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.EncryptedQuickActionChatParams
7775import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.FeedbackNotification
@@ -107,6 +105,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabBa
107105import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabBarActionRequest
108106import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventParams
109107import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventRequest
108+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.util.LspEditorUtil
109+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.util.LspEditorUtil.toUriString
110110import software.aws.toolkits.jetbrains.services.amazonq.util.command
111111import software.aws.toolkits.jetbrains.services.amazonq.util.tabType
112112import software.aws.toolkits.jetbrains.services.amazonq.webview.theme.AmazonQTheme
@@ -236,19 +236,9 @@ class BrowserConnector(
236236 requestFromUi.params.prompt.escapedPrompt,
237237 node.command
238238 )
239- val textDocumentIdentifier = getTextDocumentIdentifier(project)
240- val cursorState = CursorState (
241- Range (
242- Position (
243- 0 ,
244- 0
245- ),
246- Position (
247- 1 ,
248- 1
249- )
250- )
251- )
239+ val editor = FileEditorManager .getInstance(project).selectedTextEditor
240+ val textDocumentIdentifier = editor?.let { TextDocumentIdentifier (toUriString(it.virtualFile)) }
241+ val cursorState = editor?.let { LspEditorUtil .getCursorState(it) }
252242
253243 val chatParams = ChatParams (
254244 requestFromUi.params.tabId,
0 commit comments