test(jb): add autocomplete test

.github/workflows/pr_checks.yaml

@@ -96,7 +96,7 @@ jobs:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
 
   install-gui:
-    needs: [install-root, install-core]
+    needs: [ install-root, install-core ]
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -157,7 +157,7 @@ jobs:
           npx tsc --noEmit
 
   binary-checks:
-    needs: [install-root, install-core]
+    needs: [ install-root, install-core ]
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -193,7 +193,7 @@ jobs:
           npx tsc --noEmit
 
   install-vscode:
-    needs: [install-root, install-core]
+    needs: [ install-root, install-core ]
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -277,7 +277,7 @@ jobs:
 
   vscode-get-test-file-matrix:
     runs-on: ubuntu-latest
-    needs: [install-root, install-vscode]
+    needs: [ install-root, install-vscode ]
     outputs:
       test_file_matrix: ${{ steps.vscode-get-test-file-matrix.outputs.test_file_matrix }}
     steps:
@@ -306,7 +306,7 @@ jobs:
 
   vscode-package-extension:
     runs-on: ubuntu-latest
-    needs: [install-vscode, install-core]
+    needs: [ install-vscode, install-core ]
     steps:
       - uses: actions/checkout@v4
       - uses: actions/setup-node@v4
@@ -338,7 +338,7 @@ jobs:
 
   vscode-download-e2e-dependencies:
     runs-on: ubuntu-latest
-    needs: [install-vscode, install-core]
+    needs: [ install-vscode, install-core ]
     steps:
       - uses: actions/checkout@v4
       - uses: actions/setup-node@v4
@@ -445,7 +445,7 @@ jobs:
           path: extensions/vscode/e2e/storage/screenshots
 
   gui-tests:
-    needs: [install-gui, install-core]
+    needs: [ install-gui, install-core ]
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -476,7 +476,7 @@ jobs:
           npm test
 
   jetbrains-tests:
-    needs: [install-root, install-core]
+    needs: [ install-root, core-checks ]
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -573,3 +573,5 @@ jobs:
           name: jb-failure-report
           path: |
             ${{ github.workspace }}/extensions/intellij/build/reports
+
+

core/protocol/passThrough.ts

@@ -61,6 +61,8 @@ export const WEBVIEW_TO_CORE_PASS_THROUGH: (keyof ToCoreFromWebviewProtocol)[] =
   ];
 
 // Message types to pass through from core to webview
+// Note: If updating these values, make a corresponding update in
+// extensions/intellij/src/main/kotlin/com/github/continuedev/continueintellijextension/constants/MessageTypes.kt
 export const CORE_TO_WEBVIEW_PASS_THROUGH: (keyof ToWebviewFromCoreProtocol)[] =
   [
     "configUpdate",

extensions/intellij/.gitignore

@@ -3,4 +3,6 @@ src/main/resources/webview
 src/main/resources/bin
 src/main/resources/config_schema.json
 src/main/resources/continue_rc_schema.json
-video
+video
+src/test/kotlin/com/github/continuedev/continueintellijextension/e2e/test-continue/*
+!src/test/kotlin/com/github/continuedev/continueintellijextension/e2e/test-continue/config.json

extensions/intellij/.run/Run Test IDE.run.xml

@@ -0,0 +1,27 @@
+<component name="ProjectRunConfigurationManager">
+    <configuration default="false" name="Run Test IDE" type="GradleRunConfiguration" factoryName="Gradle">
+        <ExternalSystemSettings>
+            <option name="executionName"/>
+            <option name="externalProjectPath" value="$PROJECT_DIR$/extensions/intellij"/>
+            <option name="externalSystemIdString" value="GRADLE"/>
+            <option name="scriptParameters" value=""/>
+            <option name="taskDescriptions">
+                <list/>
+            </option>
+            <option name="taskNames">
+                <list>
+                    <option value="runIdeForUiTests"/>
+                </list>
+            </option>
+            <option name="vmOptions"/>
+        </ExternalSystemSettings>
+        <ExternalSystemDebugServerProcess>true</ExternalSystemDebugServerProcess>
+        <ExternalSystemReattachDebugProcess>true</ExternalSystemReattachDebugProcess>
+        <DebugAllEnabled>false</DebugAllEnabled>
+        <RunAsTest>false</RunAsTest>
+        <method v="2">
+            <option name="Gradle.BeforeRunTask" enabled="true" tasks="clean"
+                    externalProjectPath="$PROJECT_DIR$/extensions/intellij" vmOptions="" scriptParameters=""/>
+        </method>
+    </configuration>
+</component>

extensions/intellij/.run/Run Tests.run.xml

@@ -0,0 +1,24 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="Run Tests" type="GradleRunConfiguration" factoryName="Gradle" focusToolWindowBeforeRun="true">
+    <ExternalSystemSettings>
+      <option name="executionName" />
+      <option name="externalProjectPath" value="$PROJECT_DIR$/extensions/intellij" />
+      <option name="externalSystemIdString" value="GRADLE" />
+      <option name="scriptParameters" value="" />
+      <option name="taskDescriptions">
+        <list />
+      </option>
+      <option name="taskNames">
+        <list>
+          <option value="test" />
+        </list>
+      </option>
+      <option name="vmOptions" />
+    </ExternalSystemSettings>
+    <ExternalSystemDebugServerProcess>true</ExternalSystemDebugServerProcess>
+    <ExternalSystemReattachDebugProcess>true</ExternalSystemReattachDebugProcess>
+    <DebugAllEnabled>false</DebugAllEnabled>
+    <RunAsTest>false</RunAsTest>
+    <method v="2" />
+  </configuration>
+</component>

extensions/intellij/.run/Run e2e tests.run.xml

@@ -0,0 +1,7 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="Run e2e tests" type="CompoundRunConfigurationType">
+    <toRun name="Run Test IDE" type="GradleRunConfiguration" />
+    <toRun name="Run Tests" type="GradleRunConfiguration" />
+    <method v="2" />
+  </configuration>
+</component>

extensions/intellij/CONTRIBUTING.md

@@ -101,7 +101,9 @@ toolbar.
   _Run | Debugging Actions | Reload Changed Classes`_
     - This will often fail on new imports, schema changes etc. In that case, you need to stop and restart the extension
 - `gui`: Changes will be reloaded automatically
-- `core`: Run `npm run build` from the `binary` directory (requires restarting the `Start Core Dev Server` task)
+- `core`: Run `npm run build -- --os [darwin | linux | win32]` from the `binary` directory (requires
+  restarting the
+  `Start Core Dev Server` task)
 
 ### Setting breakpoints
 

extensions/intellij/build.gradle.kts

@@ -130,7 +130,10 @@ tasks {
     // Configure UI tests plugin
     // Read more: https://github.com/JetBrains/intellij-ui-test-robot
     runIdeForUiTests {
-        environment("CONTINUE_GLOBAL_DIR", "src/test/kotlin/com/github/continuedev/continueintellijextension/e2e/test-continue")
+        environment(
+            "CONTINUE_GLOBAL_DIR",
+            "${rootProject.projectDir}/src/test/kotlin/com/github/continuedev/continueintellijextension/e2e/test-continue"
+        )
         systemProperty("robot-server.port", "8082")
         systemProperty("ide.mac.message.dialogs.as.sheets", "false")
         systemProperty("jb.privacy.policy.text", "<!--999.999-->")
@@ -141,6 +144,7 @@ tasks {
         systemProperty("idea.trust.all.projects", "true")
         systemProperty("ide.show.tips.on.startup.default.value", "false")
         systemProperty("ide.browser.jcef.jsQueryPoolSize", "10000")
+        systemProperty("ide.browser.jcef.contextMenu.devTools.enabled", "true")
 
         // This is to ensure we load the GUI with OSR enabled. We have logic that
         // renders with OSR disabled below a particular IDE version.
@@ -180,6 +184,5 @@ tasks {
 
     test {
         useJUnitPlatform()
-        environment("CONTINUE_GLOBAL_DIR", "src/test/kotlin/com/github/continuedev/continueintellijextension/e2e/test-continue")
     }
 }

extensions/intellij/src/main/kotlin/com/github/continuedev/continueintellijextension/autocomplete/AutocompleteService.kt

@@ -294,12 +294,13 @@ class AutocompleteService(private val project: Project) {
     }
 
     private fun isInjectedFile(editor: Editor): Boolean {
-        val psiFile = runReadAction { PsiDocumentManager.getInstance(project).getPsiFile(editor.document) }
-        if (psiFile == null) {
-            return false
-        }
-        val response = runReadAction { psiFile.isInjectedText() }
-        return response
+        return ApplicationManager.getApplication().executeOnPooledThread<Boolean> {
+            ApplicationManager.getApplication().runReadAction<Boolean> {
+                val psiFile =
+                    PsiDocumentManager.getInstance(project).getPsiFile(editor.document) ?: return@runReadAction false
+                return@runReadAction psiFile.isInjectedText()
+            }
+        }.get()
     }
 
     fun hideCompletions(editor: Editor) {

extensions/intellij/src/main/kotlin/com/github/continuedev/continueintellijextension/constants/MessageTypes.kt

@@ -50,13 +50,78 @@ class MessageTypes {
             "showToast",
         )
 
+        // Note: If updating these values, make a corresponding update in
+        // core/protocol/passThrough.ts
         val PASS_THROUGH_TO_WEBVIEW = listOf(
             "configUpdate",
             "getDefaultModelTitle",
-            "indexProgress",
+            "indexProgress", // Codebase
+            "indexing/statusUpdate", // Docs, etc.
+            "addContextItem",
             "refreshSubmenuItems",
+            "isContinueInputFocused",
             "didChangeAvailableProfiles",
-            "addContextItem"
+            "setTTSActive",
+            "getWebviewHistoryLength",
+            "getCurrentSessionId",
+            "signInToControlPlane",
+            "openDialogMessage",
+            "docs/suggestions",
+        )
+
+        // Note: If updating these values, make a corresponding update in
+        // core/protocol/passThrough.ts
+        val PASS_THROUGH_TO_CORE = listOf(
+            "abort",
+            "history/list",
+            "history/delete",
+            "history/load",
+            "history/save",
+            "devdata/log",
+            "config/addModel",
+            "config/addContextProvider",
+            "config/newPromptFile",
+            "config/ideSettingsUpdate",
+            "config/getSerializedProfileInfo",
+            "config/deleteModel",
+            "config/listProfiles",
+            "config/openProfile",
+            "context/getContextItems",
+            "context/getSymbolsForFiles",
+            "context/loadSubmenuItems",
+            "context/addDocs",
+            "context/removeDocs",
+            "context/indexDocs",
+            "autocomplete/complete",
+            "autocomplete/cancel",
+            "autocomplete/accept",
+            "command/run",
+            "tts/kill",
+            "llm/complete",
+            "llm/streamComplete",
+            "llm/streamChat",
+            "llm/listModels",
+            "streamDiffLines",
+            "chatDescriber/describe",
+            "stats/getTokensPerDay",
+            "stats/getTokensPerModel",
+            // Codebase
+            "index/setPaused",
+            "index/forceReIndex",
+            "index/forceReIndexFiles",
+            "index/indexingProgressBarInitialized",
+            // Docs, etc.
+            "indexing/reindex",
+            "indexing/abort",
+            "indexing/setPaused",
+            "docs/getSuggestedDocs",
+            "docs/initStatuses",
+            //
+            "completeOnboarding",
+            "addAutocompleteModel",
+            "profiles/switch",
+            "didChangeSelectedProfile",
+            "tools/call",
         )
     }
 }