@@ -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
@@ -35,7 +35,6 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.encryption.JwtEncryp
3535import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.AwsServerCapabilitiesProvider
3636import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatCommunicationManager
3737import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.FlareUiMessage
38- import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.getTextDocumentIdentifier
3938import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.AUTH_FOLLOW_UP_CLICKED
4039import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.AuthFollowUpClickNotification
4140import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ButtonClickNotification
@@ -108,6 +107,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabBa
108107import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabBarActionRequest
109108import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventParams
110109import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventRequest
110+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.util.LspEditorUtil
111+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.util.LspEditorUtil.toUriString
111112import software.aws.toolkits.jetbrains.services.amazonq.util.command
112113import software.aws.toolkits.jetbrains.services.amazonq.util.tabType
113114import software.aws.toolkits.jetbrains.services.amazonq.webview.theme.AmazonQTheme
@@ -238,19 +239,9 @@ class BrowserConnector(
238239 requestFromUi.params.prompt.escapedPrompt,
239240 node.command
240241 )
241- val textDocumentIdentifier = getTextDocumentIdentifier(project)
242- val cursorState = CursorState (
243- Range (
244- Position (
245- 0 ,
246- 0
247- ),
248- Position (
249- 1 ,
250- 1
251- )
252- )
253- )
242+ val editor = FileEditorManager .getInstance(project).selectedTextEditor
243+ val textDocumentIdentifier = editor?.let { TextDocumentIdentifier (toUriString(it.virtualFile)) }
244+ val cursorState = editor?.let { LspEditorUtil .getCursorState(it) }
254245
255246 val chatParams = ChatParams (
256247 requestFromUi.params.tabId,
0 commit comments