You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.6 KiB
69 lines
2.6 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Discord.Commands
|
|
{
|
|
internal static class EnumTypeReader
|
|
{
|
|
public static TypeReader GetReader(Type type)
|
|
{
|
|
Type baseType = Enum.GetUnderlyingType(type);
|
|
var constructor = typeof(EnumTypeReader<>).MakeGenericType(baseType).GetTypeInfo().DeclaredConstructors.First();
|
|
return (TypeReader)constructor.Invoke(new object[] { type, PrimitiveParsers.Get(baseType) });
|
|
}
|
|
}
|
|
|
|
internal class EnumTypeReader<T> : TypeReader
|
|
{
|
|
private readonly IReadOnlyDictionary<string, object> _enumsByName;
|
|
private readonly IReadOnlyDictionary<T, object> _enumsByValue;
|
|
private readonly Type _enumType;
|
|
private readonly TryParseDelegate<T> _tryParse;
|
|
|
|
public EnumTypeReader(Type type, TryParseDelegate<T> parser)
|
|
{
|
|
_enumType = type;
|
|
_tryParse = parser;
|
|
|
|
var byNameBuilder = ImmutableDictionary.CreateBuilder<string, object>();
|
|
var byValueBuilder = ImmutableDictionary.CreateBuilder<T, object>();
|
|
|
|
foreach (var v in Enum.GetNames(_enumType))
|
|
{
|
|
var parsedValue = Enum.Parse(_enumType, v);
|
|
byNameBuilder.Add(v.ToLower(), parsedValue);
|
|
if (!byValueBuilder.ContainsKey((T)parsedValue))
|
|
byValueBuilder.Add((T)parsedValue, parsedValue);
|
|
}
|
|
|
|
_enumsByName = byNameBuilder.ToImmutable();
|
|
_enumsByValue = byValueBuilder.ToImmutable();
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override Task<TypeReaderResult> ReadAsync(ICommandContext context, string input, IServiceProvider services)
|
|
{
|
|
object enumValue;
|
|
|
|
if (_tryParse(input, out T baseValue))
|
|
{
|
|
if (_enumsByValue.TryGetValue(baseValue, out enumValue))
|
|
return Task.FromResult(TypeReaderResult.FromSuccess(enumValue));
|
|
else
|
|
return Task.FromResult(TypeReaderResult.FromError(CommandError.ParseFailed, $"Value is not a {_enumType.Name}."));
|
|
}
|
|
else
|
|
{
|
|
if (_enumsByName.TryGetValue(input.ToLower(), out enumValue))
|
|
return Task.FromResult(TypeReaderResult.FromSuccess(enumValue));
|
|
else
|
|
return Task.FromResult(TypeReaderResult.FromError(CommandError.ParseFailed, $"Value is not a {_enumType.Name}."));
|
|
}
|
|
}
|
|
}
|
|
}
|