rabidus-test/Assets/Plugins/QFSW/Quantum Console/Source/Scripts/Parsing/QuantumParser.cs

384 lines
14 KiB
C#

using QFSW.QC.Utilities;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text.RegularExpressions;
using UnityEngine;
namespace QFSW.QC
{
/// <summary>
/// Handles parsing values to use as console inputs.
/// </summary>
public class QuantumParser
{
private readonly IQcParser[] _parsers;
private readonly IQcGrammarConstruct[] _grammarConstructs;
private readonly ConcurrentDictionary<Type, IQcParser> _parserLookup = new ConcurrentDictionary<Type, IQcParser>();
private readonly HashSet<Type> _unparseableLookup = new HashSet<Type>();
private readonly Func<string, Type, object> _recursiveParser;
/// <summary>
/// Creates a Parser Serializer with a custom set of parsers.
/// </summary>
/// <param name="parsers">The IQcParsers to use in this Quantum Parser.</param>
/// <param name="grammarConstructs">The IQcGrammarConstructs to use in this Quantum Parser</param>
public QuantumParser(IEnumerable<IQcParser> parsers, IEnumerable<IQcGrammarConstruct> grammarConstructs)
{
_recursiveParser = Parse;
_parsers = parsers.OrderByDescending(x => x.Priority)
.ToArray();
_grammarConstructs = grammarConstructs.OrderBy(x => x.Precedence)
.ToArray();
}
/// <summary>
/// Creates a Quantum Parser with the default injected parsers.
/// </summary>
public QuantumParser() : this(new InjectionLoader<IQcParser>().GetInjectedInstances(), new InjectionLoader<IQcGrammarConstruct>().GetInjectedInstances())
{
}
public IQcParser GetParser(Type type)
{
if (_parserLookup.ContainsKey(type))
{
return _parserLookup[type];
}
else if (!_unparseableLookup.Contains(type))
{
foreach (IQcParser parser in _parsers)
{
try
{
if (parser.CanParse(type))
{
return _parserLookup[type] = parser;
}
}
catch (Exception e)
{
Debug.LogError($"{parser.GetType().GetDisplayName()}.CanParse is malformed and throws");
Debug.LogException(e);
}
}
_unparseableLookup.Add(type);
}
return null;
}
public bool CanParse(Type type)
{
return GetParser(type) != null;
}
private IQcGrammarConstruct GetMatchingGrammar(string value, Type type)
{
foreach (IQcGrammarConstruct grammar in _grammarConstructs)
{
try
{
if (grammar.Match(value, type))
{
return grammar;
}
}
catch (Exception e)
{
Debug.LogError($"{grammar.GetType().GetDisplayName()}.Match is malformed and throws");
Debug.LogException(e);
}
}
return null;
}
/// <summary>
/// Parses a serialized string of data.
/// </summary>
/// <typeparam name="T">The type of the value to parse.</typeparam>
/// <param name="value">The string to parse.</param>
/// <returns>The parsed value.</returns>
public T Parse<T>(string value)
{
return (T)Parse(value, typeof(T));
}
/// <summary>
/// Parses a serialized string of data.
/// </summary>
/// <param name="value">The string to parse.</param>
/// <param name="type">The type of the value to parse.</param>
/// <returns>The parsed value.</returns>
public object Parse(string value, Type type)
{
value = value.ReduceScope('(', ')');
if (type.IsClass && value == "null")
{
return null;
}
IQcGrammarConstruct grammar = GetMatchingGrammar(value, type);
if (grammar != null)
{
try
{
return grammar.Parse(value, type, _recursiveParser);
}
catch (ParserException) { throw; }
catch (Exception e)
{
throw new Exception($"Parsing of {type.GetDisplayName()} via {grammar} failed:\n{e.Message}", e);
}
}
IQcParser parser = GetParser(type);
if (parser == null)
{
throw new ArgumentException($"Cannot parse object of type '{type.GetDisplayName()}'");
}
try
{
return parser.Parse(value, type, _recursiveParser);
}
catch (ParserException) { throw; }
catch (Exception e)
{
throw new Exception($"Parsing of {type.GetDisplayName()} via {parser} failed:\n{e.Message}", e);
}
}
#region Type Parser
private static readonly Dictionary<Type, string> _typeDisplayNames = new Dictionary<Type, string>
{
{ typeof(int), "int" }, { typeof(float), "float" }, { typeof(decimal), "decimal" },
{ typeof(double), "double" }, { typeof(string), "string" }, { typeof(bool), "bool" },
{ typeof(byte), "byte" }, { typeof(sbyte), "sbyte" }, { typeof(uint), "uint" },
{ typeof(short), "short" }, { typeof(ushort), "ushort" }, { typeof(long), "long" },
{ typeof(ulong), "ulong" }, { typeof(char), "char" }, { typeof(object), "object" }
};
private static readonly Dictionary<string, Type> _reverseTypeDisplayNames = _typeDisplayNames.Invert();
private static readonly Assembly[] _loadedAssemblies = AppDomain.CurrentDomain.GetAssemblies();
private static readonly string[] _defaultNamespaces = new string[] { "System", "System.Collections", "System.Collections.Generic", "UnityEngine", "UnityEngine.UI", "QFSW.QC" };
private static readonly List<string> _namespaceTable = new List<string>(_defaultNamespaces);
private static readonly Regex _arrayTypeRegex = new Regex(@"^.*\[,*\]$");
private static readonly Regex _genericTypeRegex = new Regex(@"^.+<.*>$");
private static readonly Regex _tupleTypeRegex = new Regex(@"^\(.*\)$");
/// <summary>
/// Resets the namespace table to its initial state.
/// </summary>
[Command("reset-namespaces", "Resets the namespace table to its initial state")]
public static void ResetNamespaceTable()
{
_namespaceTable.Clear();
_namespaceTable.AddRange(_defaultNamespaces);
}
/// <summary>
/// Adds a namespace to the table so that it can be used to type resolution.
/// </summary>
[Command("use-namespace", "Adds a namespace to the table so that it can be used to type resolution")]
public static void AddNamespace(string namespaceName)
{
if (!_namespaceTable.Contains(namespaceName))
{
_namespaceTable.Add(namespaceName);
}
}
/// <summary>
/// Removes a namespace to the table so that it is no longer used to type resolution.
/// </summary>
[Command("remove-namespace", "Removes a namespace from the table")]
public static void RemoveNamespace(string namespaceName)
{
if (_namespaceTable.Contains(namespaceName))
{
_namespaceTable.Remove(namespaceName);
}
else
{
throw new ArgumentException($"No namespace named {namespaceName} was present in the table");
}
}
[Command("all-namespaces", "Displays all of the namespaces currently in use by the namespace table")]
private static string ShowNamespaces()
{
_namespaceTable.Sort();
if (_namespaceTable.Count == 0) { return "Namespace table is empty"; }
else { return string.Join("\n", _namespaceTable); }
}
/// <summary>
/// Returns a copy of the namespace table.
/// </summary>
public static IEnumerable<string> GetAllNamespaces() { return _namespaceTable; }
/// <summary>
/// Parses and infers the type specified by the string.
/// </summary>
/// <returns>The parsed type.</returns>
/// <param name="typeName">The type to parse.</param>
public static Type ParseType(string typeName)
{
typeName = typeName.Trim();
if (_reverseTypeDisplayNames.ContainsKey(typeName))
{
return _reverseTypeDisplayNames[typeName];
}
if (_tupleTypeRegex.IsMatch(typeName))
{
return ParseTupleType(typeName);
}
if (_arrayTypeRegex.IsMatch(typeName))
{
return ParseArrayType(typeName);
}
if (_genericTypeRegex.IsMatch(typeName))
{
return ParseGenericType(typeName);
}
if (typeName.Contains('`'))
{
string genericName = typeName.Split('`')[0];
if (_reverseTypeDisplayNames.ContainsKey(genericName))
{
return _reverseTypeDisplayNames[genericName];
}
}
return ParseTypeBaseCase(typeName);
}
private static Type ParseArrayType(string typeName)
{
int arrayPos = typeName.LastIndexOf('[');
int arrayRank = typeName.CountFromIndex(',', arrayPos) + 1;
Type elementType = ParseType(typeName.Substring(0, arrayPos));
return arrayRank > 1
? elementType.MakeArrayType(arrayRank)
: elementType.MakeArrayType();
}
private static Type ParseGenericType(string typeName)
{
string[] parts = typeName.Split(new[] { '<' }, 2);
string[] genericArgNames = $"<{parts[1]}".ReduceScope('<', '>').SplitScoped(',');
string incompleteGenericName = $"{parts[0]}`{Math.Max(1, genericArgNames.Length)}";
Type incompleteGenericType = ParseType(incompleteGenericName);
if (genericArgNames.All(string.IsNullOrWhiteSpace))
{
return incompleteGenericType;
}
Type[] genericArgs = genericArgNames.Select(ParseType).ToArray();
return incompleteGenericType.MakeGenericType(genericArgs);
}
private static Type ParseTupleType(string typeName)
{
string inner = typeName.Substring(1, typeName.Length - 2);
Type[] parts = inner
.SplitScoped(',')
.Select(ParseType)
.ToArray();
return CreateTupleType(parts);
}
private static readonly Type[] _valueTupleTypes =
{
typeof(ValueTuple<>),
typeof(ValueTuple<,>),
typeof(ValueTuple<,,>),
typeof(ValueTuple<,,,>),
typeof(ValueTuple<,,,,>),
typeof(ValueTuple<,,,,,>),
typeof(ValueTuple<,,,,,,>),
typeof(ValueTuple<,,,,,,,>)
};
private static Type CreateTupleType(Type[] types)
{
const int maxFlatTupleSize = 8;
if (types.Length > maxFlatTupleSize - 1)
{
Type[] innerTypes = types.Skip(maxFlatTupleSize - 1).ToArray();
types = types
.Take(maxFlatTupleSize - 1)
.Concat(CreateTupleType(innerTypes).Yield())
.ToArray();
}
return _valueTupleTypes[types.Length - 1].MakeGenericType(types);
}
private static Type ParseTypeBaseCase(string typeName)
{
return GetTypeFromAssemblies(typeName, _loadedAssemblies, false, false)
?? GetTypeFromAssemblies(typeName, _namespaceTable, _loadedAssemblies, false, false)
?? GetTypeFromAssemblies(typeName, _loadedAssemblies, false, true)
?? GetTypeFromAssemblies(typeName, _namespaceTable, _loadedAssemblies, true, true);
}
private static Type GetTypeFromAssemblies(string typeName, IEnumerable<string> namespaces, IEnumerable<Assembly> assemblies, bool throwOnError, bool ignoreCase)
{
foreach (string namespaceName in namespaces)
{
Type type = GetTypeFromAssemblies($"{namespaceName}.{typeName}", assemblies, false, ignoreCase);
if (type != null) { return type; }
}
if (throwOnError)
{
throw new TypeLoadException($"No type of name '{typeName}' could be found in the specified assemblies and namespaces.");
}
return null;
}
private static Type GetTypeFromAssemblies(string typeName, IEnumerable<Assembly> assemblies, bool throwOnError, bool ignoreCase)
{
foreach (Assembly assembly in assemblies)
{
Type type = Type.GetType($"{typeName}, {assembly.FullName}", false, ignoreCase);
if (type != null) { return type; }
}
if (throwOnError)
{
throw new TypeLoadException($"No type of name '{typeName}' could be found in the specified assemblies.");
}
return null;
}
#endregion
}
}