feat: [New billing] Implement model picker category badge.

Author: ethanyhou

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/protocol/CopilotModel.java

@@ -22,6 +22,7 @@ public class CopilotModel {
   private CopilotModelBilling billing;
   private String degradationReason;
   private String providerName;
+  private String modelPickerCategory;
 
   /**
    * Policy for the model.
@@ -169,6 +170,14 @@ public void setProviderName(String providerName) {
     this.providerName = providerName;
   }
 
+  public String getModelPickerCategory() {
+    return modelPickerCategory;
+  }
+
+  public void setModelPickerCategory(String modelPickerCategory) {
+    this.modelPickerCategory = modelPickerCategory;
+  }
+
   @Override
   public boolean equals(Object obj) {
     if (this == obj) {
@@ -185,14 +194,15 @@ public boolean equals(Object obj) {
         && Objects.equals(degradationReason, other.degradationReason) && Objects.equals(id, other.id)
         && isChatDefault == other.isChatDefault && isChatFallback == other.isChatFallback
         && Objects.equals(modelFamily, other.modelFamily) && Objects.equals(modelName, other.modelName)
+        && Objects.equals(modelPickerCategory, other.modelPickerCategory)
         && Objects.equals(modelPolicy, other.modelPolicy) && preview == other.preview
         && Objects.equals(providerName, other.providerName) && Objects.equals(scopes, other.scopes);
   }
 
   @Override
   public int hashCode() {
     return Objects.hash(billing, capabilities, degradationReason, id, isChatDefault, isChatFallback, modelFamily,
-        modelName, modelPolicy, preview, providerName, scopes);
+        modelName, modelPickerCategory, modelPolicy, preview, providerName, scopes);
   }
 
   @Override
@@ -210,6 +220,7 @@ public String toString() {
     builder.append("billing", billing);
     builder.append("degradationReason", degradationReason);
     builder.append("providerName", providerName);
+    builder.append("modelPickerCategory", modelPickerCategory);
     return builder.toString();
   }
 

com.microsoft.copilot.eclipse.ui/src/com/microsoft/copilot/eclipse/ui/swt/CategoryBadge.java

@@ -0,0 +1,72 @@
+package com.microsoft.copilot.eclipse.ui.swt;
+
+import org.apache.commons.lang3.StringUtils;
+import org.eclipse.swt.SWT;
+import org.eclipse.swt.graphics.Color;
+import org.eclipse.swt.graphics.GC;
+import org.eclipse.swt.graphics.Point;
+import org.eclipse.swt.layout.GridData;
+import org.eclipse.swt.widgets.Composite;
+import org.eclipse.swt.widgets.Display;
+
+/**
+ * A custom-painted badge widget that displays a model picker category label with a rounded border in the category's
+ * theme color.
+ */
+public class CategoryBadge extends Composite {
+
+  private static final int HORIZONTAL_PADDING = 6;
+  private static final int VERTICAL_PADDING = 1;
+
+  /**
+   * Creates a category badge for the given category string. Returns {@code null} if the category has no known color
+   * mapping.
+   *
+   * @param parent the parent composite
+   * @param category the raw category value from the API (e.g. "powerful")
+   * @return the badge composite, or {@code null} if the category is unrecognised
+   */
+  public static CategoryBadge create(Composite parent, String category) {
+    Color badgeColor = getCategoryColor(parent.getDisplay(), category);
+    if (badgeColor == null) {
+      return null;
+    }
+    return new CategoryBadge(parent, category, badgeColor);
+  }
+
+  private CategoryBadge(Composite parent, String category, Color badgeColor) {
+    super(parent, SWT.NONE);
+    setLayoutData(new GridData(SWT.LEFT, SWT.NONE, false, false));
+
+    String displayText = StringUtils.capitalize(category);
+
+    GC gc = new GC(this);
+    Point textExtent = gc.textExtent(displayText);
+    gc.dispose();
+
+    int badgeWidth = textExtent.x + 2 * HORIZONTAL_PADDING;
+    int badgeHeight = textExtent.y + 2 * VERTICAL_PADDING;
+    GridData gd = (GridData) getLayoutData();
+    gd.widthHint = badgeWidth;
+    gd.heightHint = badgeHeight;
+
+    addPaintListener(e -> {
+      e.gc.setAntialias(SWT.ON);
+      e.gc.setForeground(badgeColor);
+      e.gc.drawRoundRectangle(0, 0, badgeWidth - 1, badgeHeight - 1,
+          badgeHeight, badgeHeight);
+      e.gc.drawText(displayText, HORIZONTAL_PADDING, VERTICAL_PADDING, true);
+    });
+  }
+
+  private static Color getCategoryColor(Display display, String category) {
+    if ("Powerful".equalsIgnoreCase(category)) {
+      return CssConstants.getCategoryPowerfulColor(display);
+    } else if ("Versatile".equalsIgnoreCase(category)) {
+      return CssConstants.getCategoryVersatileColor(display);
+    } else if ("Lightweight".equalsIgnoreCase(category)) {
+      return CssConstants.getCategoryLightweightColor(display);
+    }
+    return null;
+  }
+}

com.microsoft.copilot.eclipse.ui/src/com/microsoft/copilot/eclipse/ui/swt/CssConstants.java

@@ -80,6 +80,36 @@ public static Color getPopupBgColor(Display display) {
     return new Color(display, 255, 255, 255);
   }
 
+  /**
+   * Returns the color for the "Powerful" model picker category badge.
+   */
+  public static Color getCategoryPowerfulColor(Display display) {
+    if (UiUtils.isDarkTheme()) {
+      return new Color(display, 116, 167, 220); // #74A7DC
+    }
+    return new Color(display, 15, 108, 189); // #0F6CBD
+  }
+
+  /**
+   * Returns the color for the "Versatile" model picker category badge.
+   */
+  public static Color getCategoryVersatileColor(Display display) {
+    if (UiUtils.isDarkTheme()) {
+      return new Color(display, 222, 225, 229); // #DEE1E5
+    }
+    return new Color(display, 97, 97, 97); // #616161
+  }
+
+  /**
+   * Returns the color for the "Lightweight" model picker category badge.
+   */
+  public static Color getCategoryLightweightColor(Display display) {
+    if (UiUtils.isDarkTheme()) {
+      return new Color(display, 110, 185, 110); // #6EB96E
+    }
+    return new Color(display, 16, 124, 16); // #107C10
+  }
+
   /**
    * Get the focused background color for dropdown popup items based on the current theme.
    */

com.microsoft.copilot.eclipse.ui/src/com/microsoft/copilot/eclipse/ui/swt/ModelHoverContentProvider.java

@@ -44,6 +44,10 @@ public ModelHoverContentProvider(CopilotModel model) {
   public void configureHover(Composite parent, DropdownItem item) {
     renderHeader(parent, item);
 
+    if (StringUtils.isNotBlank(model.getModelPickerCategory())) {
+      CategoryBadge.create(parent, model.getModelPickerCategory());
+    }
+
     // Degradation warning
     if (StringUtils.isNotBlank(model.getDegradationReason())) {
       addWarningRow(parent, model.getDegradationReason());