FileFlowsPlugins/ImageNodes/Images/ImageNode.cs

using System.Text.RegularExpressions;
using ImageMagick;
using SixLabors.ImageSharp.Formats;
using SixLabors.ImageSharp.Formats.Bmp;
using SixLabors.ImageSharp.Formats.Gif;
using SixLabors.ImageSharp.Formats.Jpeg;
using SixLabors.ImageSharp.Formats.Pbm;
using SixLabors.ImageSharp.Formats.Png;
using SixLabors.ImageSharp.Formats.Tga;
using SixLabors.ImageSharp.Formats.Tiff;
using SixLabors.ImageSharp.Formats.Webp;

namespace FileFlows.ImageNodes.Images;

public abstract class ImageNode : ImageBaseNode
{
    [Select(nameof(FormatOptions), 1)]
    public string Format { get; set; }
    
    private static List<ListOption> _FormatOptions;
    public static List<ListOption> FormatOptions
    {
        get
        {
            if (_FormatOptions == null)
            {
                _FormatOptions = new List<ListOption>
                {
                    new () { Value = "", Label = "Same as source"},
                    new () { Value = IMAGE_FORMAT_BMP, Label = "Bitmap"},
                    new () { Value = IMAGE_FORMAT_GIF, Label = "GIF"},
                    new () { Value = IMAGE_FORMAT_JPEG, Label = "JPEG"},
                    new () { Value = IMAGE_FORMAT_PBM, Label = "PBM"},
                    new () { Value = IMAGE_FORMAT_PNG, Label = "PNG"},
                    new () { Value = IMAGE_FORMAT_TIFF, Label = "TIFF"},
                    new () { Value = IMAGE_FORMAT_TGA, Label = "TGA" },
                    new () { Value = IMAGE_FORMAT_WEBP, Label = "WebP"},
                };
            }
            return _FormatOptions;
        }
    }

    protected (IImageFormat? format, string file) GetFormat(NodeParameters args)
    {
        IImageFormat? format = null;
        
        var newFile = FileHelper.Combine(args.TempPath, Guid.NewGuid().ToString());
        switch (this.Format)
        {
            case IMAGE_FORMAT_BMP:
                newFile = newFile + ".bmp";
                format = BmpFormat.Instance;
                break; 
            case IMAGE_FORMAT_GIF:
                newFile = newFile + ".gif";
                format = GifFormat.Instance;
                break; 
            case IMAGE_FORMAT_JPEG:
                newFile = newFile + ".jpg";
                format = JpegFormat.Instance;
                break;
            case IMAGE_FORMAT_PBM:
                newFile = newFile + ".pbm";
                format = PbmFormat.Instance;
                break; 
            case IMAGE_FORMAT_PNG:
                newFile = newFile + ".png";
                format = PngFormat.Instance;
                break; 
            case IMAGE_FORMAT_TIFF:
                newFile = newFile + ".tiff";
                format = TiffFormat.Instance;
                break;
            case IMAGE_FORMAT_TGA:
                newFile = newFile + ".tga";
                format = TgaFormat.Instance;
                break; 
            case IMAGE_FORMAT_WEBP:
                newFile = newFile + ".webp";
                format = WebpFormat.Instance;
                break;
            case "HEIC":
                // cant save to this format, save to PNG
                newFile = newFile + ".png";
                format = PngFormat.Instance;
                break; 
            default:
                newFile = newFile + "." + args.WorkingFile.Substring(args.WorkingFile.LastIndexOf(".") + 1);
                newFile = Regex.Replace(newFile, @"\.heic$", ".png", RegexOptions.IgnoreCase);
                break;
        }

        return (format, newFile);
    }

    protected void SaveImage(NodeParameters args, Image img, string file, IImageFormat format, bool updateWorkingFile = true)
    {
        string local = args.FileService.FileIsLocal(file)
            ? file
            : FileHelper.Combine(args.TempPath, Guid.NewGuid() + FileHelper.GetExtension(file));
        
        using var outStream = new System.IO.FileStream(local, System.IO.FileMode.Create);
        img.Save(outStream, format);

        if (local != file && args.FileService.FileMove(local, file).Failed(out string error))
        {
            args.Logger?.ELog("Failed to move saved file: " + error);
            return;
        }

        if (updateWorkingFile)
        {
            args.SetWorkingFile(file);
            UpdateImageInfo(args, img.Height, img.Height, format.Name, Variables);
        }
    }

    protected void SaveImage(NodeParameters args, ImageMagick.MagickImage img, string file, bool updateWorkingFile = true)
    {
        string local = args.FileService.FileIsLocal(file)
            ? file
            : FileHelper.Combine(args.TempPath, Guid.NewGuid() + FileHelper.GetExtension(file));
        
        using var outStream = new System.IO.FileStream(local, System.IO.FileMode.Create);

        string origExtension = FileHelper.GetExtension(args.WorkingFile).ToLowerInvariant().TrimStart('.');
        string newExtension = FileHelper.GetExtension(file).ToLowerInvariant().TrimStart('.');
        if (origExtension != newExtension)
        {
            switch (newExtension)
            {
                case "jpg":
                case "jpeg":
                    img.Format = MagickFormat.Jpeg;
                    break;
                case "png":
                    img.Format = MagickFormat.Png;
                    break;
                case "gif":
                    img.Format = MagickFormat.Gif;
                    break;
                case "bmp":
                    img.Format = MagickFormat.Bmp;
                    break;
                case "tga":
                    img.Format = MagickFormat.Tga;
                    break;
                case "webp":
                    img.Format = MagickFormat.WebP;
                    break;
                case "webm":
                    img.Format = MagickFormat.WebM;
                    break;
                default:
                    if (Enum.TryParse(newExtension, true, out MagickFormat format))
                        img.Format = format;
                    break;
            }
        }

        img.Write(outStream);
        
        
        if (updateWorkingFile)
        {
            args.SetWorkingFile(file);
            using var image = Image.Load(local, out IImageFormat format);
            UpdateImageInfo(args, img.Width, img.Height, format.Name, Variables);
        }
        
        if (local != file && args.FileService.FileMove(local, file).Failed(out string error))
        {
            args.Logger?.ELog("Failed to move saved file: " + error);
        }
    }
}