SparqlResultSetTsvCodec.cs

using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using Waher.Content.Semantic.Model;
using Waher.Content.Text;
using Waher.Runtime.Inventory;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Objects.Matrices;
 
namespace Waher.Content.Semantic
{
    public class SparqlResultSetTsvCodec : IContentEncoder
    {
        public SparqlResultSetTsvCodec()
        {
        }
 
        public string[] ContentTypes => new string[0];
 
        public string[] FileExtensions => new string[0];
 
        public bool Encodes(object Object, out Grade Grade, params string[] AcceptedContentTypes)
        {
            if (Object is SparqlResultSet &&
                InternetContent.IsAccepted(TsvCodec.TsvContentTypes, AcceptedContentTypes))
            {
                Grade = Grade.Excellent;
                return true;
            }
            else if (Object is ObjectMatrix M && M.HasColumnNames &&
                InternetContent.IsAccepted(TsvCodec.TsvContentTypes, AcceptedContentTypes))
            {
                Grade = Grade.Ok;
                return true;
            }
            else if (Object is bool &&
                InternetContent.IsAccepted(TsvCodec.TsvContentTypes, AcceptedContentTypes))
            {
                Grade = Grade.Barely;
                return true;
            }
            else
            {
                Grade = Grade.NotAtAll;
                return false;
            }
        }
 
        public Task<KeyValuePair<byte[], string>> EncodeAsync(object Object, Encoding Encoding, params string[] AcceptedContentTypes)
        {
            string Text;
 
            if (Encoding is null)
                Encoding = Encoding.UTF8;
 
            if (Object is SparqlResultSet Result)
            {
                if (Result.BooleanResult.HasValue)
                {
                    string[][] Records = new string[1][];
                    Records[0] = new string[] { CommonTypes.Encode(Result.BooleanResult.Value) };
 
                    Text = TSV.Encode(Records);
                }
                else
                {
                    IMatrix M = Result.ToMatrix();
 
                    if (M is ObjectMatrix OM && !(OM.ColumnNames is null))
                    {
                        int i, c = OM.ColumnNames.Length;
 
                        for (i = 0; i < c; i++)
                            OM.ColumnNames[i] = "?" + OM.ColumnNames[i];
                    }
 
                    Text = TSV.Encode(M);
                }
            }
            else if (Object is ObjectMatrix M)
                Text = TSV.Encode(M, ElementToString, false);
            else if (Object is bool b)
            {
                string[][] Records = new string[1][];
                Records[0] = new string[] { CommonTypes.Encode(b) };
 
                Text = TSV.Encode(Records);
            }
            else
                throw new ArgumentException("Unable to encode object.", nameof(Object));
 
            byte[] Bin = Encoding.GetBytes(Text);
            string ContentType = TsvCodec.TsvContentTypes[0] + "; charset=" + Encoding.WebName;
 
            return Task.FromResult(new KeyValuePair<byte[], string>(Bin, ContentType));
        }
 
        private static string ElementToString(IElement E)
        {
            object Obj = E.AssociatedObjectValue;
 
            if (Obj is SemanticTriple Triple)
            {
                StringBuilder sb = new StringBuilder();
 
                sb.Append("<<");
                sb.Append(ElementToString(Triple.Subject));
                sb.Append(' ');
                sb.Append(ElementToString(Triple.Predicate));
                sb.Append(' ');
                sb.Append(ElementToString(Triple.Object));
                sb.Append(">>");
 
                return sb.ToString();
            }
            else if (Obj is ISemanticElement)
                return Obj.ToString();
            else
                return JSON.Encode(Obj?.ToString() ?? string.Empty);
        }
 
        public bool TryGetContentType(string FileExtension, out string ContentType)
        {
            ContentType = null;
            return false;
        }
 
        public bool TryGetFileExtension(string ContentType, out string FileExtension)
        {
            FileExtension = null;
            return false;
        }
    }
}