PlantUmlDocumentToImageConverter.cs

using System;
using System.IO;
using System.Threading.Tasks;
using Waher.Content;
using Waher.Content.Images;
using Waher.Runtime.Inventory;
using Waher.Script;
 
namespace Waher.Content.Markdown.PlantUml
{
    public class PlantUmlDocumentToImageConverter : IContentConverter
    {
        public PlantUmlDocumentToImageConverter()
        {
        }
 
        public string[] FromContentTypes => new string[] { PlantUmlCodec.DefaultContentType };
 
        public virtual string[] ToContentTypes
        {
            get
            {
                return new string[]
                {
                    ImageCodec.ContentTypePng,
                    ImageCodec.ContentTypeSvg
                };
            }
        }
 
        public virtual Grade ConversionGrade => Grade.Excellent;
 
        public async Task<bool> ConvertAsync(ConversionState State)
        {
            string GraphDescription;
 
            using (StreamReader rd = new StreamReader(State.From))
            {
                GraphDescription = rd.ReadToEnd();
            }
 
            string s = State.ToContentType;
            int i;
 
            i = s.IndexOf(';');
            if (i > 0)
                s = s.Substring(0, i);
 
            bool Png = string.Compare(s, ImageCodec.ContentTypePng, true) == 0;
            bool Svg = string.Compare(s, ImageCodec.ContentTypeSvg, true) == 0;
 
            if (!(State.PossibleContentTypes is null))
            {
                foreach (string ContentType in State.PossibleContentTypes)
                {
                    s = ContentType;
                    i = s.IndexOf(';');
                    if (i > 0)
                        s = s.Substring(0, i);
 
                    Png |= string.Compare(s, ImageCodec.ContentTypePng, true) == 0;
                    Svg |= string.Compare(s, ImageCodec.ContentTypeSvg, true) == 0;
                }
            }
 
            GraphInfo Graph;
 
            if (Svg)
            {
                Graph = await PlantUml.GetGraphInfo("uml", GraphDescription, ResultType.Svg, true);
                State.ToContentType = ImageCodec.ContentTypeSvg;
            }
            else if (Png)
            {
                Graph = await PlantUml.GetGraphInfo("uml", GraphDescription, ResultType.Png, true);
                State.ToContentType = ImageCodec.ContentTypePng;
            }
            else
                throw new Exception("Unable to convert document from " + State.FromContentType + " to " + State.ToContentType);
 
            byte[] Data = await Resources.ReadAllBytesAsync(Graph.ImageFileName);
 
            await State.To.WriteAsync(Data, 0, Data.Length);
 
            return false;
        }
 
    }
}