AI-Studio/app/MindWork AI Studio/Tools/Validation/FileExtensionValidation.cs

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

namespace AIStudio.Tools.Validation;

/// <summary>
/// Provides centralized validation for file extensions.
/// </summary>
public static class FileExtensionValidation
{
    private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(FileExtensionValidation).Namespace, nameof(FileExtensionValidation));

    /// <summary>
    /// Defines the use cases for file extension validation.
    /// </summary>
    public enum UseCase
    {
        /// <summary>
        /// No specific use case; general validation.
        /// </summary>
        NONE,
        
        /// <summary>
        /// Validating for directly loading content into the UI. In this state, there might be no provider selected yet.
        /// </summary>
        DIRECTLY_LOADING_CONTENT,
        
        /// <summary>
        /// Validating for attaching content to a message or prompt.
        /// </summary>
        ATTACHING_CONTENT,
    }

    /// <summary>
    /// Validates the file extension and sends appropriate MessageBus notifications when invalid.
    /// </summary>
    /// <param name="useCae">The validation use case.</param>
    /// <param name="filePath">The file path to validate.</param>
    /// <param name="validateMediaFileTypes">Whether to validate media file types against provider capabilities.</param>
    /// <param name="provider">The selected provider.</param>
    /// <returns>True if valid, false if invalid (error/warning already sent via MessageBus).</returns>
    public static async Task<bool> IsExtensionValidWithNotifyAsync(UseCase useCae, string filePath, bool validateMediaFileTypes = true, Settings.Provider? provider = null)
    {
        if (string.Equals(Path.GetExtension(filePath), ".doc", StringComparison.OrdinalIgnoreCase))
        {
            await MessageBus.INSTANCE.SendWarning(new(
                Icons.Material.Filled.Description,
                TB("This file format is not supported. Please convert the .doc file to .docx (e.g. with Microsoft Word).")));
            return false;
        }

        if (FileTypes.IsAllowedPath(filePath, FileTypes.EXECUTABLES))
        {
            await MessageBus.INSTANCE.SendError(new(
                Icons.Material.Filled.AppBlocking,
                TB("Executables are not allowed")));
            return false;
        }

        if (FileTypes.IsAllowedPath(filePath, FileTypes.IMAGE))
        {
            switch (useCae)
            {
                // In this use case, we cannot guarantee that a provider is selected yet:
                case UseCase.DIRECTLY_LOADING_CONTENT:
                    await MessageBus.INSTANCE.SendWarning(new(
                        Icons.Material.Filled.ImageNotSupported,
                        TB("Images are not supported at this place")));
                    return false;
                
                //
                // Whether AI Studio can prepare an image for a model does not depend on the
                // provider. This holds for templates as well, therefore: a HEIC image attached
                // there could never be sent.
                //
                case UseCase.ATTACHING_CONTENT when !FileTypes.IsAllowedPath(filePath, FileTypes.MODEL_IMAGE):
                    await MessageBus.INSTANCE.SendWarning(new(
                        Icons.Material.Filled.ImageNotSupported,
                        string.Format(TB("AI Studio cannot give {0} images to an AI model. Please convert the image to JPG or PNG."), Path.GetExtension(filePath).TrimStart('.').ToUpperInvariant())));
                    return false;

                // In this use case, we don't validate the provider capabilities:
                case UseCase.ATTACHING_CONTENT when !validateMediaFileTypes:
                    return true;
                
                // In this use case, we can check the provider capabilities:
                case UseCase.ATTACHING_CONTENT when provider?.SupportsImageInput() is true:
                    return true;
                
                // We know that images are not supported:
                case UseCase.ATTACHING_CONTENT:
                    await MessageBus.INSTANCE.SendWarning(new(
                        Icons.Material.Filled.ImageNotSupported,
                        TB("Images are not supported by the selected provider and model")));
                    return false;
                
                default:
                    await MessageBus.INSTANCE.SendWarning(new(
                        Icons.Material.Filled.ImageNotSupported,
                        TB("Images are not supported yet")));
                    return false;
            }
        }

        if (FileTypes.IsAllowedPath(filePath, FileTypes.VIDEO))
        {
            await MessageBus.INSTANCE.SendWarning(new(
                Icons.Material.Filled.FeaturedVideo,
                TB("Videos are not supported yet")));
            return false;
        }

        if (FileTypes.IsAllowedPath(filePath, FileTypes.AUDIO))
        {
            await MessageBus.INSTANCE.SendWarning(new(
                Icons.Material.Filled.AudioFile,
                TB("Audio files are not supported yet")));
            return false;
        }

        if (FileTypes.IsAllowedPath(filePath, FileTypes.DOCUMENT))
            return true;

        await MessageBus.INSTANCE.SendWarning(new(
            Icons.Material.Filled.InsertDriveFile,
            TB("Unsupported file type")));
        return false;
    }
}