AI-Studio/app/MindWork AI Studio/Provider/ConfidenceLevelExtensions.cs

using AIStudio.Settings;
using AIStudio.Tools.PluginSystem;

namespace AIStudio.Provider;

public static class ConfidenceLevelExtensions
{
    private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(ConfidenceLevelExtensions).Namespace, nameof(ConfidenceLevelExtensions));
    
    extension(ConfidenceLevel level)
    {
        public string GetName() => level switch
        {
            ConfidenceLevel.NONE => TB("No provider selected"),
        
            ConfidenceLevel.UNTRUSTED => TB("Untrusted"),
            ConfidenceLevel.UNKNOWN => TB("Unknown"),
            ConfidenceLevel.VERY_LOW => TB("Very Low"),
            ConfidenceLevel.LOW => TB("Low"),
            ConfidenceLevel.MODERATE => TB("Moderate"),
            ConfidenceLevel.MEDIUM => TB("Medium"),
            ConfidenceLevel.HIGH => TB("High"),
        
            _ => TB("Unknown confidence level"),
        };

        public string GetColor(SettingsManager settingsManager) => (level, settingsManager.IsDarkMode) switch
        {
            (ConfidenceLevel.NONE, false) => "#cccccc",
            (ConfidenceLevel.NONE, true) => "#666666",

            (ConfidenceLevel.UNKNOWN, false) => "#777777",
            (ConfidenceLevel.UNKNOWN, true) => "#aaaaaa",
        
            (ConfidenceLevel.UNTRUSTED, false) => "#ff0000",
            (ConfidenceLevel.UNTRUSTED, true) => "#800000",
        
            (ConfidenceLevel.VERY_LOW, false) => "#ff6600",
            (ConfidenceLevel.VERY_LOW, true) => "#803300",
        
            (ConfidenceLevel.LOW, false) => "#ffcc00",
            (ConfidenceLevel.LOW, true) => "#806600",
        
            (ConfidenceLevel.MODERATE, false) => "#99cc00",
            (ConfidenceLevel.MODERATE, true) => "#4d6600",
        
            (ConfidenceLevel.MEDIUM, false) => "#86b300",
            (ConfidenceLevel.MEDIUM, true) => "#394d00",
        
            (ConfidenceLevel.HIGH, false) => "#009933",
            (ConfidenceLevel.HIGH, true) => "#004d1a",
        
            (_, false) => "#cc6600",
            (_, true) => "#663300",
        };

        public string SetColorStyle(SettingsManager settingsManager) => $"--confidence-color: {level.GetColor(settingsManager)};";
    }
}