feat(gui): add token usage progress bar to chat panel

Add visual progress bar showing token usage ratio against model's context window limit, with color-coded indicators (yellow/orange/red) for usage thresholds and detailed usage statistics display.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/TokenUsagePanel.kt

@@ -3,7 +3,10 @@ package cc.unitmesh.devti.gui.chat.ui
 import cc.unitmesh.devti.llm2.TokenUsageEvent
 import cc.unitmesh.devti.llm2.TokenUsageListener
 import cc.unitmesh.devti.llms.custom.Usage
+import cc.unitmesh.devti.settings.AutoDevSettingsState
+import cc.unitmesh.devti.settings.model.LLMModelManager
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.components.service
 import com.intellij.openapi.project.Project
 import com.intellij.ui.components.JBLabel
 import com.intellij.util.ui.JBUI
@@ -12,8 +15,11 @@ import com.intellij.util.ui.components.BorderLayoutPanel
 import java.awt.BorderLayout
 import java.awt.Color
 import java.awt.Font
+import java.awt.GridBagConstraints
+import java.awt.GridBagLayout
 import javax.swing.Box
 import javax.swing.JPanel
+import javax.swing.JProgressBar
 import javax.swing.SwingConstants
 
 /**
@@ -25,9 +31,12 @@ class TokenUsagePanel(private val project: Project) : BorderLayoutPanel() {
     private val completionTokensLabel = JBLabel("0", SwingConstants.RIGHT)
     private val totalTokensLabel = JBLabel("0", SwingConstants.RIGHT)
     private val modelLabel = JBLabel("", SwingConstants.LEFT)
+    private val progressBar = JProgressBar(0, 100)
+    private val usageRatioLabel = JBLabel("", SwingConstants.CENTER)
 
     private var currentUsage = Usage()
     private var currentModel: String? = null
+    private var maxContextWindowTokens: Long = 0
 
     init {
         setupUI()
@@ -38,13 +47,55 @@ class TokenUsagePanel(private val project: Project) : BorderLayoutPanel() {
         isOpaque = false
         border = JBUI.Borders.empty(4, 8)
 
+        // Setup progress bar
+        progressBar.apply {
+            isStringPainted = false
+            preferredSize = java.awt.Dimension(150, 16)
+            minimumSize = java.awt.Dimension(100, 16)
+            font = font.deriveFont(Font.PLAIN, 10f)
+            isOpaque = false
+        }
+        
+        // Setup usage ratio label
+        usageRatioLabel.apply {
+            font = font.deriveFont(Font.PLAIN, 10f)
+            foreground = UIUtil.getContextHelpForeground()
+        }
+        
+        // Create main layout
+        val mainPanel = JPanel(GridBagLayout())
+        mainPanel.isOpaque = false
+        
+        val gbc = GridBagConstraints()
+        
+        // Top row: Model info and progress bar
+        gbc.gridx = 0
+        gbc.gridy = 0
+        gbc.anchor = GridBagConstraints.WEST
+        gbc.fill = GridBagConstraints.NONE
         // Create left panel for model info
         val leftPanel = JPanel(BorderLayout())
         leftPanel.isOpaque = false
         modelLabel.font = modelLabel.font.deriveFont(Font.PLAIN, 11f)
         modelLabel.foreground = UIUtil.getContextHelpForeground()
         leftPanel.add(modelLabel, BorderLayout.WEST)
 
+        mainPanel.add(leftPanel, gbc)
+        
+        // Progress bar and ratio in the middle
+        gbc.gridx = 1
+        gbc.weightx = 1.0
+        gbc.fill = GridBagConstraints.HORIZONTAL
+        gbc.insets = JBUI.insets(0, 8, 0, 8)
+        
+        val progressPanel = JPanel(BorderLayout())
+        progressPanel.isOpaque = false
+        progressPanel.add(progressBar, BorderLayout.CENTER)
+        progressPanel.add(usageRatioLabel, BorderLayout.SOUTH)
+        
+        mainPanel.add(progressPanel, gbc)
+        
+        // Bottom row: Token stats
         // Create right panel for token stats
         val rightPanel = JPanel()
         rightPanel.isOpaque = false
@@ -77,9 +128,14 @@ class TokenUsagePanel(private val project: Project) : BorderLayoutPanel() {
         })
         rightPanel.add(totalTokensLabel)
 
+        gbc.gridx = 2
+        gbc.weightx = 0.0
+        gbc.fill = GridBagConstraints.NONE
+        gbc.insets = JBUI.insets(0, 0, 0, 0)
+        mainPanel.add(rightPanel, gbc)
+        
         // Add panels to main layout
-        addToLeft(leftPanel)
-        addToRight(rightPanel)
+        addToCenter(mainPanel)
 
         // Initially hidden
         isVisible = false
@@ -99,10 +155,17 @@ class TokenUsagePanel(private val project: Project) : BorderLayoutPanel() {
             currentUsage = event.usage
             currentModel = event.model
 
+            // Get max tokens for current model
+            updateMaxTokens()
+            
+            // Update token displays
             promptTokensLabel.text = formatTokenCount(event.usage.promptTokens ?: 0)
             completionTokensLabel.text = formatTokenCount(event.usage.completionTokens ?: 0)
             totalTokensLabel.text = formatTokenCount(event.usage.totalTokens ?: 0)
 
+            // Update progress bar
+            updateProgressBar(event.usage.totalTokens ?: 0)
+            
             if (!event.model.isNullOrBlank()) {
                 modelLabel.text = "Model: ${event.model}"
             }
@@ -114,6 +177,45 @@ class TokenUsagePanel(private val project: Project) : BorderLayoutPanel() {
         }
     }
 
+    private fun updateMaxTokens() {
+        try {
+            val settings = AutoDevSettingsState.getInstance()
+            val modelManager = LLMModelManager(project, settings) {}
+            val limits = modelManager.getUsedMaxToken()
+            maxContextWindowTokens = limits.maxContextWindowTokens?.toLong() ?: 0
+        } catch (e: Exception) {
+            // Fallback to default if unable to get limits
+            maxContextWindowTokens = 4096
+        }
+    }
+    
+    private fun updateProgressBar(totalTokens: Long) {
+        if (maxContextWindowTokens <= 0) {
+            progressBar.isVisible = false
+            usageRatioLabel.isVisible = false
+            return
+        }
+        
+        val usageRatio = (totalTokens.toDouble() / maxContextWindowTokens * 100).toInt()
+        progressBar.value = usageRatio.coerceIn(0, 100)
+        
+        // Update color based on usage ratio
+        progressBar.foreground = when {
+            usageRatio >= 90 -> Color.RED
+            usageRatio >= 75 -> Color.ORANGE
+            usageRatio >= 50 -> Color.YELLOW
+            else -> UIUtil.getPanelBackground().brighter()
+        }
+        
+        // Update ratio label
+        usageRatioLabel.text = "${formatTokenCount(totalTokens)}/${formatTokenCount(maxContextWindowTokens)} (${usageRatio}%)"
+        
+        progressBar.isVisible = true
+        usageRatioLabel.isVisible = true
+        
+        progressBar.toolTipText = "Token usage: $usageRatio% of context window"
+    }
+    
     private fun formatTokenCount(count: Long): String {
         return when {
             count >= 1_000_000 -> String.format("%.1fM", count / 1_000_000.0)
@@ -129,10 +231,15 @@ class TokenUsagePanel(private val project: Project) : BorderLayoutPanel() {
         ApplicationManager.getApplication().invokeLater {
             currentUsage = Usage()
             currentModel = null
+            maxContextWindowTokens = 0
             promptTokensLabel.text = "0"
             completionTokensLabel.text = "0"
             totalTokensLabel.text = "0"
             modelLabel.text = ""
+            progressBar.value = 0
+            progressBar.isVisible = false
+            usageRatioLabel.text = ""
+            usageRatioLabel.isVisible = false
             isVisible = false
             revalidate()
             repaint()