GraphVizCodec.cs

using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using Waher.Runtime.Inventory;
using Waher.Runtime.IO;
 
namespace Waher.Content.Markdown.GraphViz
{
    public class GraphVizCodec : IContentDecoder, IContentEncoder
    {
        public GraphVizCodec()
        {
        }
 
        public const string DefaultContentType = "text/vnd.graphviz";
 
        public static readonly string[] GraphVizContentTypes = new string[]
        {
            DefaultContentType
        };
 
        public static readonly string[] GraphVizFileExtensions = new string[]
        {
            "gv", "dot", "neato", "fdp", "sfdp", "twopi", "circo"
        };
 
        public string[] ContentTypes => GraphVizContentTypes;
 
        public string[] FileExtensions => GraphVizFileExtensions;
 
        public bool Decodes(string ContentType, out Grade Grade)
        {
            if (Array.IndexOf(GraphVizContentTypes, ContentType) >= 0)
            {
                Grade = Grade.Barely;
                return true;
            }
            else
            {
                Grade = Grade.NotAtAll;
                return false;
            }
        }
 
        public Task<ContentResponse> DecodeAsync(string ContentType, byte[] Data, Encoding Encoding,
            KeyValuePair<string, string>[] Fields, Uri BaseUri, ICodecProgress Progress)
        {
            string GraphViz = Strings.GetString(Data, Encoding);
            return Task.FromResult(new ContentResponse(ContentType, new GraphVizDocument(GraphViz), Data));
        }
 
        public bool TryGetContentType(string FileExtension, out string ContentType)
        {
            switch (FileExtension.ToLower())
            {
                case "gv":
                case "dot":
                case "neato":
                case "fdp":
                case "sfdp":
                case "twopi":
                case "circo":
                    ContentType = GraphVizContentTypes[0];
                    return true;
 
                default:
                    ContentType = string.Empty;
                    return false;
            }
        }
 
        public bool TryGetFileExtension(string ContentType, out string FileExtension)
        {
            switch (ContentType.ToLower())
            {
                case DefaultContentType:
                    FileExtension = GraphVizFileExtensions[0];
                    return true;
 
                default:
                    FileExtension = string.Empty;
                    return false;
            }
        }
 
        public bool Encodes(object Object, out Grade Grade, params string[] AcceptedContentTypes)
        {
            if (Object is GraphVizDocument)
            {
                if (InternetContent.IsAccepted(GraphVizContentTypes, AcceptedContentTypes))
                {
                    Grade = Grade.Barely;
                    return true;
                }
            }
 
            Grade = Grade.NotAtAll;
            return false;
        }
 
        public Task<ContentResponse> EncodeAsync(object Object, Encoding Encoding, ICodecProgress Progress, params string[] AcceptedContentTypes)
        {
            if (!InternetContent.IsAccepted(GraphVizContentTypes, out string ContentType, AcceptedContentTypes))
                return Task.FromResult(new ContentResponse(new ArgumentException("Unable to encode object, or content type not accepted.", nameof(Object))));
 
            string s;
 
            if (Object is GraphVizDocument GraphVizDoc)
                s = GraphVizDoc.GraphDescription;
            else
                s = Object.ToString();
 
            if (Encoding is null)
            {
                ContentType += "; charset=utf-8";
                Encoding = Encoding.UTF8;
            }
            else
                ContentType += "; charset=" + Encoding.WebName;
 
            return Task.FromResult(new ContentResponse(ContentType, Object, Encoding.GetBytes(s)));
        }
    }
}