clients: extend completion request context for intelij idea

clients/intellij/build.gradle.kts

@@ -18,7 +18,7 @@ intellij {
   version.set("2022.2.5")
   type.set("IC") // Target IDE Platform
 
-  plugins.set(listOf(/* Plugin Dependencies */))
+  plugins.set(listOf("git4idea"))
 }
 
 tasks {
@@ -67,3 +67,4 @@ tasks {
     channels.set(listOf("alpha"))
   }
 }
+

clients/intellij/src/main/kotlin/com/tabbyml/intellijtabby/agent/Agent.kt

@@ -222,9 +222,22 @@ class Agent : ProcessAdapter() {
     val filepath: String,
     val language: String,
     val text: String,
+    val indentation: String?,
     val position: Int,
     val manually: Boolean?,
-  )
+    val workspace: String?,
+    val git: GitContext?
+  ) {
+    data class GitContext(
+      val root: String,
+      val remotes: List<GitRemote>
+    ) {
+      data class GitRemote(
+        val name: String,
+        val url: String,
+      )
+    }
+  }
 
   data class CompletionResponse(
     val id: String,

clients/intellij/src/main/kotlin/com/tabbyml/intellijtabby/agent/AgentService.kt

@@ -7,7 +7,6 @@ import com.intellij.notification.Notification
 import com.intellij.notification.NotificationType
 import com.intellij.notification.Notifications
 import com.intellij.openapi.Disposable
-import com.intellij.openapi.actionSystem.ActionManager
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.application.ApplicationInfo
@@ -23,8 +22,11 @@ import com.intellij.openapi.keymap.Keymap
 import com.intellij.openapi.keymap.KeymapManagerListener
 import com.intellij.openapi.options.ShowSettingsUtil
 import com.intellij.openapi.progress.ProgressIndicator
+import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiDocumentManager
 import com.intellij.psi.PsiFile
+import com.intellij.psi.codeStyle.CodeStyleSettingsManager
+import com.tabbyml.intellijtabby.git.GitContextProvider
 import com.tabbyml.intellijtabby.settings.ApplicationConfigurable
 import com.tabbyml.intellijtabby.settings.ApplicationSettingsState
 import com.tabbyml.intellijtabby.settings.KeymapSettings
@@ -36,6 +38,7 @@ import kotlinx.coroutines.flow.*
 class AgentService : Disposable {
   private val logger = Logger.getInstance(AgentService::class.java)
   private var agent: Agent = Agent()
+  private var gitContextProvider: GitContextProvider = service<GitContextProvider>()
   val scope: CoroutineScope = CoroutineScope(Dispatchers.IO)
 
   private var initFailedNotification: Notification? = null
@@ -267,23 +270,42 @@ class AgentService : Disposable {
 
   suspend fun provideCompletion(editor: Editor, offset: Int, manually: Boolean = false): Agent.CompletionResponse? {
     waitForInitialized()
-    return ReadAction.compute<PsiFile, Throwable> {
-      editor.project?.let { project ->
+    val project = editor.project ?: return null
+
+    return ReadAction.compute<PsiFile?, Throwable> {
         PsiDocumentManager.getInstance(project).getPsiFile(editor.document)
-      }
-    }?.let { file ->
+      }?.let { file ->
       agent.provideCompletions(
         Agent.CompletionRequest(
           file.virtualFile.path,
           file.getLanguageId(),
           editor.document.text,
+          getIndention(project, file),
           offset,
           manually,
+          project.basePath,
+          getRemotesForProjectGit(project, file)
         )
       )
     }
   }
 
+  private fun getIndention(project: Project, file: PsiFile): String? {
+    val codeStyleSettingsManager = CodeStyleSettingsManager.getInstance(project)
+    val projectSettings = codeStyleSettingsManager.mainProjectCodeStyle ?: return null
+    val indentionOptions = projectSettings.getCommonSettings(file.language).indentOptions ?: return null
+
+    if (indentionOptions.USE_TAB_CHARACTER) {
+      return "\t"
+    }
+
+    return " ".repeat(indentionOptions.INDENT_SIZE)
+  }
+
+  private fun getRemotesForProjectGit(project: Project, file: PsiFile): Agent.CompletionRequest.GitContext? {
+    return gitContextProvider.getGitContextForFile(project, file.virtualFile)
+  }
+
   suspend fun postEvent(event: Agent.LogEventRequest) {
     waitForInitialized()
     agent.postEvent(event)

clients/intellij/src/main/kotlin/com/tabbyml/intellijtabby/git/DefaultGitContextProvider.kt

@@ -0,0 +1,11 @@
+package com.tabbyml.intellijtabby.git
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import com.tabbyml.intellijtabby.agent.Agent
+
+class DefaultGitContextProvider : GitContextProvider {
+    override fun getGitContextForFile(project: Project, file: VirtualFile): Agent.CompletionRequest.GitContext? {
+        return null
+    }
+}

clients/intellij/src/main/kotlin/com/tabbyml/intellijtabby/git/Git4IdeaGitContextProvider.kt

@@ -0,0 +1,27 @@
+package com.tabbyml.intellijtabby.git
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import com.tabbyml.intellijtabby.agent.Agent
+import git4idea.repo.GitRepositoryManager
+
+class Git4IdeaGitContextProvider : GitContextProvider {
+    override fun getGitContextForFile(project: Project, file: VirtualFile): Agent.CompletionRequest.GitContext? {
+        val manger = GitRepositoryManager.getInstance(project)
+        val repository = manger.getRepositoryForFile(file)
+        if (repository === null) return null
+
+        val remotes = ArrayList<Agent.CompletionRequest.GitContext.GitRemote>()
+
+        for (remote in repository.remotes) {
+            val url = remote.firstUrl
+            if (url !== null) {
+                remotes.add(Agent.CompletionRequest.GitContext.GitRemote(remote.name, url))
+            }
+        }
+
+        if (remotes.isEmpty()) return null
+
+        return Agent.CompletionRequest.GitContext(repository.root.path, remotes)
+    }
+}

clients/intellij/src/main/kotlin/com/tabbyml/intellijtabby/git/GitContextProvider.kt

@@ -0,0 +1,9 @@
+package com.tabbyml.intellijtabby.git
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import com.tabbyml.intellijtabby.agent.Agent
+
+interface GitContextProvider {
+    fun getGitContextForFile(project: Project, file: VirtualFile): Agent.CompletionRequest.GitContext?
+}

clients/intellij/src/main/resources/META-INF/git4idea-features.xml

@@ -0,0 +1,7 @@
+<idea-plugin>
+    <extensions defaultExtensionNs="com.intellij">
+        <applicationService serviceInterface="com.tabbyml.intellijtabby.git.GitContextProvider"
+                            serviceImplementation="com.tabbyml.intellijtabby.git.Git4IdeaGitContextProvider"
+                            overrides="true" />
+    </extensions>
+</idea-plugin>

clients/intellij/src/main/resources/META-INF/plugin.xml

@@ -28,6 +28,8 @@
     <!-- Product and plugin compatibility requirements.
          Read more: https://plugins.jetbrains.com/docs/intellij/plugin-compatibility.html -->
     <depends>com.intellij.modules.platform</depends>
+    <depends optional="true" config-file="git4idea-features.xml">Git4Idea</depends>
+
 
     <!-- Extension points defined by the plugin.
          Read more: https://plugins.jetbrains.com/docs/intellij/plugin-extension-points.html -->
@@ -50,6 +52,8 @@
                            displayType="STICKY_BALLOON"
                            bundle="strings"
                            key="tabby.warning"/>
+        <applicationService serviceInterface="com.tabbyml.intellijtabby.git.GitContextProvider"
+                            serviceImplementation="com.tabbyml.intellijtabby.git.DefaultGitContextProvider" />
     </extensions>
 
     <actions>