feat(settings): basic LLMClients table and edit dialog

Author: Blarc

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/AppSettingsConfigurable.kt

@@ -3,6 +3,7 @@ package com.github.blarc.ai.commits.intellij.plugin.settings
 import com.aallam.openai.api.exception.OpenAIAPIException
 import com.github.blarc.ai.commits.intellij.plugin.*
 import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle.message
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientTable
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompts.Prompt
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompts.PromptTable
 import com.intellij.icons.AllIcons
@@ -29,7 +30,9 @@ class AppSettingsConfigurable : BoundConfigurable(message("settings.general.grou
     private val proxyTextField = JBTextField()
     private val socketTimeoutTextField = JBTextField()
     private var modelComboBox = ComboBox<String>()
+    private val llmClientTable = LLMClientTable()
     private val promptTable = PromptTable()
+    private lateinit var llmClientToolbarDecorator: ToolbarDecorator
     private lateinit var toolbarDecorator: ToolbarDecorator
     private lateinit var promptComboBox: Cell<ComboBox<Prompt>>
 
@@ -135,8 +138,21 @@ class AppSettingsConfigurable : BoundConfigurable(message("settings.general.grou
                 cell(verifyLabel)
                     .align(AlignX.RIGHT)
             }
+
+            row {
+                llmClientToolbarDecorator = ToolbarDecorator.createDecorator(llmClientTable.table)
+                    .setAddAction {
+                        llmClientTable.addLlmClient()
+                    }
+                    .disableUpDownActions()
+
+                cell(llmClientToolbarDecorator.createPanel())
+                    .align(Align.FILL)
+            }
         }
 
+
+
         group(JBLabel("Prompt")) {
             row {
                 label(message("settings.locale")).widthGroup("labelPrompt")

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/AppSettingsListCellRenderer.kt

@@ -1,6 +1,7 @@
 package com.github.blarc.ai.commits.intellij.plugin.settings
 
 import com.aallam.openai.api.model.ModelId
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClient
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompts.Prompt
 import java.awt.Component
 import java.util.*
@@ -16,14 +17,22 @@ class AppSettingsListCellRenderer : DefaultListCellRenderer() {
         cellHasFocus: Boolean
     ): Component {
         val component = super.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus)
-        if (value is Locale) {
-            text = value.displayLanguage
-        }
-        if (value is Prompt) {
-            text = value.name
-        }
-        if (value is ModelId) {
-            text = value.id
+        when (value) {
+            is Locale -> {
+                text = value.displayLanguage
+            }
+
+            is Prompt -> {
+                text = value.name
+            }
+
+            is ModelId -> {
+                text = value.id
+            }
+
+            is LLMClient -> {
+                text = value.displayName
+            }
         }
         return component
     }

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/LLMClient.kt

@@ -39,6 +39,8 @@ abstract class LLMClient(
         newToken: String
     )
 
+    abstract fun panel(): LLMClientPanel
+
     private fun saveToken(token: String) {
         try {
             PasswordSafe.instance.setPassword(getCredentialAttributes(displayName), token)

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/LLMClientPanel.kt

@@ -0,0 +1,9 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients
+
+import javax.swing.JComponent
+
+interface LLMClientPanel {
+
+    fun create(): JComponent
+
+}

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/LLMClientTable.kt

@@ -0,0 +1,92 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients
+
+import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle
+import com.github.blarc.ai.commits.intellij.plugin.createColumn
+import com.github.blarc.ai.commits.intellij.plugin.settings.AppSettings2
+import com.intellij.openapi.ui.ComboBox
+import com.intellij.openapi.ui.DialogWrapper
+import com.intellij.ui.JBCardLayout
+import com.intellij.ui.components.Panel
+import com.intellij.ui.dsl.builder.Align
+import com.intellij.ui.dsl.builder.panel
+import com.intellij.ui.table.TableView
+import com.intellij.util.ui.ListTableModel
+import javax.swing.DefaultComboBoxModel
+import javax.swing.ListSelectionModel.SINGLE_SELECTION
+
+class LLMClientTable {
+    private var llmClients = AppSettings2.instance.llmClients
+    private val tableModel = createTableModel()
+
+    val table = TableView(tableModel).apply {
+        setShowColumns(true)
+        setSelectionMode(SINGLE_SELECTION)
+    }
+
+    private fun createTableModel(): ListTableModel<LLMClient> = ListTableModel(
+        arrayOf(
+            createColumn<LLMClient>(AICommitsBundle.message("settings.prompt.name")) { llmClient -> llmClient.displayName },
+        ),
+        llmClients.toList()
+    )
+
+    fun addLlmClient(): LLMClient? {
+        val dialog = LLMClientDialog()
+
+        if (dialog.showAndGet()) {
+            return null
+        }
+        return null
+    }
+
+    private class LLMClientDialog(val newLlmClient: LLMClient? = null) : DialogWrapper(true) {
+        private val cardLayout = JBCardLayout()
+        private val cardPanel = Panel(null, cardLayout)
+        private val llmClients: List<LLMClient> = getLlmClients(newLlmClient)
+        private val llmClientsComboBox: ComboBox<LLMClient> = createLlmClientsComboBox(newLlmClient)
+
+        init {
+            title = newLlmClient?.let { "Edit LLM Client" } ?: "Add LLM Client"
+            llmClients.forEach {
+                cardPanel.add(it.displayName, it.panel().create())
+            }
+            cardLayout.show(cardPanel, llmClients[0].displayName)
+            init()
+        }
+
+        override fun createCenterPanel() = panel {
+            row("Type") {
+                cell(llmClientsComboBox)
+                    .align(Align.FILL)
+                    .applyToComponent {
+                        addItemListener { cardLayout.show(cardPanel, (it.item as LLMClient).displayName) }
+                    }
+            }
+            row {
+                cell(cardPanel)
+            }
+        }
+
+        private fun getLlmClients(newLLMClient: LLMClient?) : List<LLMClient> {
+            return if (newLLMClient == null) {
+                // TODO: Find a better way to create the list of all possible LLM Clients that implement LLMClient abstract class
+                listOf(
+                    OpenAIClient(),
+                    TestAIClient()
+                )
+            } else {
+                listOf(newLLMClient)
+            }
+        }
+
+        private fun createLlmClientsComboBox(newLLMClient: LLMClient?) : ComboBox<LLMClient> {
+            val comboBoxModel = DefaultComboBoxModel(getLlmClients(newLLMClient).toTypedArray())
+            with(ComboBox(comboBoxModel)){
+                // Type of LLM Client can not be changed when editing a LLM Client
+                if (newLLMClient != null) isEnabled = false
+                return this
+            }
+        }
+    }
+
+}

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/OpenAIClient.kt

@@ -70,6 +70,10 @@ class OpenAIClient(displayName: String = "OpenAI") : LLMClient(
         openAI.models()
     }
 
+    override fun panel(): LLMClientPanel {
+        return OpenAIClientPanel(this)
+    }
+
     private fun openAIConfig() = OpenAIConfig(
         token,
         host = host.takeIf { it.isNotBlank() }?.let { OpenAIHost(it) } ?: OpenAIHost.OpenAI,

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/OpenAIClientPanel.kt

@@ -0,0 +1,23 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients
+
+import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle
+import com.intellij.openapi.ui.ComboBox
+import com.intellij.ui.dsl.builder.bindItem
+import com.intellij.ui.dsl.builder.panel
+import com.intellij.ui.dsl.builder.toNullableProperty
+
+class OpenAIClientPanel(private val client: OpenAIClient) : LLMClientPanel {
+
+    private val hostComboBox = ComboBox(client.hosts.toTypedArray())
+
+    override fun create() = panel {
+        row {
+            label(AICommitsBundle.message("settings.openAIHost"))
+                .widthGroup("label")
+
+            cell(hostComboBox)
+                .bindItem(client::host.toNullableProperty())
+                .widthGroup("input")
+        }
+    }
+}

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/TestAIClient.kt

@@ -20,4 +20,7 @@ class TestAIClient(displayName: String = "TestAI") : LLMClient(
     override suspend fun verifyConfiguration(newHost: String, newProxy: String?, newTimeout: String, newToken: String) {
     }
 
+    override fun panel(): LLMClientPanel {
+        return TestAIClientPanel(this)
+    }
 }

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/TestAIClientPanel.kt

@@ -0,0 +1,22 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients
+
+import com.intellij.openapi.ui.ComboBox
+import com.intellij.ui.dsl.builder.bindItem
+import com.intellij.ui.dsl.builder.panel
+import com.intellij.ui.dsl.builder.toNullableProperty
+
+class TestAIClientPanel(private val client: TestAIClient): LLMClientPanel {
+
+    private val hostComboBox = ComboBox(client.hosts.toTypedArray())
+
+    override fun create() = panel {
+        row {
+            label("Test AI")
+                .widthGroup("label")
+
+            cell(hostComboBox)
+                .bindItem(client::host.toNullableProperty())
+                .widthGroup("input")
+        }
+    }
+}