AI-Studio/app/MindWork AI Studio/Settings/DataModel/DataSourceSecurityExtensions.cs

using AIStudio.Tools.PluginSystem;

namespace AIStudio.Settings.DataModel;

public static class DataSourceSecurityExtensions
{
    private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(DataSourceSecurityExtensions).Namespace, nameof(DataSourceSecurityExtensions));
    
    extension(DataSourceSecurity security)
    {
        public string ToSelectionText() => security switch
        {
            DataSourceSecurity.NOT_SPECIFIED => TB("Please select a security policy"),
        
            DataSourceSecurity.ALLOW_ANY => TB("This data source can be used with any LLM provider. Your data may be sent to a cloud-based provider."),
            DataSourceSecurity.SELF_HOSTED => TB("This data source can only be used with a self-hosted LLM provider. Your data will not be sent to any cloud-based provider."),
        
            _ => TB("Unknown security policy")
        };

        public string ToInfoText() => security switch
        {
            DataSourceSecurity.NOT_SPECIFIED => TB("The security of the data source is not specified yet. You cannot use this data source until you specify a security policy."),
        
            DataSourceSecurity.ALLOW_ANY => TB("This data source can be used with any LLM provider. Your data may be sent to a cloud-based provider."),
            DataSourceSecurity.SELF_HOSTED => TB("This data source can only be used with a self-hosted LLM provider. Your data will not be sent to any cloud-based provider."),
        
            _ => TB("Unknown security policy")
        };

        public TextColor GetColor() => security switch
        {
            DataSourceSecurity.ALLOW_ANY => TextColor.WARN,
            DataSourceSecurity.SELF_HOSTED => TextColor.SUCCESS,
        
            _ => TextColor.ERROR
        };
    }
    
    /// <param name="held">The data security held so far.</param>
    extension(DataSourceSecurity held)
    {
        /// <summary>
        /// The data security which demands what the one held and the one arriving demand together.
        /// </summary>
        /// <remarks>
        /// Data which may only be used with self-hosted providers keeps them as the only choice, no matter
        /// what arrives later: the data was seen. Data which may be used with any provider marks the holder
        /// as one which holds data of a data source, while a restriction set earlier stays. NOT_SPECIFIED
        /// demands nothing and changes nothing. The order does not matter.
        /// </remarks>
        /// <param name="arriving">What the data arriving demands.</param>
        /// <returns>The tightened data security; never looser than either of the two.</returns>
        public DataSourceSecurity TightenedBy(DataSourceSecurity arriving) => (held, arriving) switch
        {
            (DataSourceSecurity.SELF_HOSTED, _) or (_, DataSourceSecurity.SELF_HOSTED) => DataSourceSecurity.SELF_HOSTED,
            (_, DataSourceSecurity.ALLOW_ANY) => DataSourceSecurity.ALLOW_ANY,
            _ => held,
        };
    }
}