Exemple #1
0
        private static void BuildParameter(ParameterBuilder builder, System.Reflection.ParameterInfo paramInfo, int position, int count, CommandService service, IServiceProvider services)
        {
            var attributes = paramInfo.GetCustomAttributes();
            var paramType  = paramInfo.ParameterType;

            builder.Name = paramInfo.Name;

            builder.IsOptional   = paramInfo.IsOptional;
            builder.DefaultValue = paramInfo.HasDefaultValue ? paramInfo.DefaultValue : null;

            foreach (var attribute in attributes)
            {
                switch (attribute)
                {
                case SummaryAttribute summary:
                    builder.Summary = summary.Text;
                    break;

                case OverrideTypeReaderAttribute typeReader:
                    builder.TypeReader = GetTypeReader(service, paramType, typeReader.TypeReader, services);
                    break;

                case ParamArrayAttribute _:
                    builder.IsMultiple = true;
                    paramType          = paramType.GetElementType();
                    break;

                case ParameterPreconditionAttribute precon:
                    builder.AddPrecondition(precon);
                    break;

                case NameAttribute name:
                    builder.Name = name.Text;
                    break;

                case RemainderAttribute _:
                    if (position != count - 1)
                    {
                        throw new InvalidOperationException($"Remainder parameters must be the last parameter in a command. Parameter: {paramInfo.Name} in {paramInfo.Member.DeclaringType.Name}.{paramInfo.Member.Name}");
                    }

                    builder.IsRemainder = true;
                    break;

                default:
                    builder.AddAttributes(attribute);
                    break;
                }
            }

            builder.ParameterType = paramType;

            if (builder.TypeReader == null)
            {
                builder.TypeReader = service.GetDefaultTypeReader(paramType)
                                     ?? service.GetTypeReaders(paramType)?.FirstOrDefault().Value;
            }
        }
Exemple #2
0
        internal static TypeReader GetTypeReader(CommandService service, Type paramType, Type typeReaderType, IServiceProvider services)
        {
            var        readers = service.GetTypeReaders(paramType);
            TypeReader reader  = null;

            if (readers != null)
            {
                if (readers.TryGetValue(typeReaderType, out reader))
                {
                    return(reader);
                }
            }

            //We dont have a cached type reader, create one
            reader = ReflectionUtils.CreateObject <TypeReader>(typeReaderType.GetTypeInfo(), service, services);
            service.AddTypeReader(paramType, reader, false);

            return(reader);
        }
Exemple #3
0
        public override async Task <TypeReaderResult> ReadAsync(ITurnContext context, string input, IServiceProvider services)
        {
            var result = new T();
            var state = ReadState.LookingForParameter;
            int beginRead = 0, currentRead = 0;

            while (state != ReadState.End)
            {
                try
                {
                    var prop    = Read(out var arg);
                    var propVal = await ReadArgumentAsync(prop, arg).ConfigureAwait(false);

                    if (propVal != null)
                    {
                        prop.SetMethod.Invoke(result, new[] { propVal });
                    }
                    else
                    {
                        return(TypeReaderResult.FromError(CommandError.ParseFailed, $"Could not parse the argument for the parameter '{prop.Name}' as type '{prop.PropertyType}'."));
                    }
                }
                catch (Exception ex)
                {
                    //TODO: use the Exception overload after a rebase on latest
                    return(TypeReaderResult.FromError(CommandError.Exception, ex.Message));
                }
            }

            return(TypeReaderResult.FromSuccess(result));

            PropertyInfo Read(out string arg)
            {
                string currentParam = null;
                char   match        = '\0';

                for (; currentRead < input.Length; currentRead++)
                {
                    var currentChar = input[currentRead];
                    switch (state)
                    {
                    case ReadState.LookingForParameter:
                        if (Char.IsWhiteSpace(currentChar))
                        {
                            continue;
                        }
                        else
                        {
                            beginRead = currentRead;
                            state     = ReadState.InParameter;
                        }
                        break;

                    case ReadState.InParameter:
                        if (currentChar != ':')
                        {
                            continue;
                        }
                        else
                        {
                            currentParam = input.Substring(beginRead, currentRead - beginRead);
                            state        = ReadState.LookingForArgument;
                        }
                        break;

                    case ReadState.LookingForArgument:
                        if (Char.IsWhiteSpace(currentChar))
                        {
                            continue;
                        }
                        else
                        {
                            beginRead = currentRead;
                            state     = (QuotationAliasUtils.GetDefaultAliasMap.TryGetValue(currentChar, out match))
                                    ? ReadState.InQuotedArgument
                                    : ReadState.InArgument;
                        }
                        break;

                    case ReadState.InArgument:
                        if (!Char.IsWhiteSpace(currentChar))
                        {
                            continue;
                        }
                        else
                        {
                            return(GetPropAndValue(out arg));
                        }

                    case ReadState.InQuotedArgument:
                        if (currentChar != match)
                        {
                            continue;
                        }
                        else
                        {
                            return(GetPropAndValue(out arg));
                        }
                    }
                }

                if (currentParam == null)
                {
                    throw new InvalidOperationException("No parameter name was read.");
                }

                return(GetPropAndValue(out arg));

                PropertyInfo GetPropAndValue(out string argv)
                {
                    bool quoted = state == ReadState.InQuotedArgument;

                    state = (currentRead == (quoted ? input.Length - 1 : input.Length))
                        ? ReadState.End
                        : ReadState.LookingForParameter;

                    if (quoted)
                    {
                        argv = input.Substring(beginRead + 1, currentRead - beginRead - 1).Trim();
                        currentRead++;
                    }
                    else
                    {
                        argv = input.Substring(beginRead, currentRead - beginRead);
                    }

                    return(_tProps[currentParam]);
                }
            }

            async Task <object> ReadArgumentAsync(PropertyInfo prop, string arg)
            {
                var  elemType     = prop.PropertyType;
                bool isCollection = false;

                if (elemType.GetTypeInfo().IsGenericType&& elemType.GetGenericTypeDefinition() == typeof(IEnumerable <>))
                {
                    elemType     = prop.PropertyType.GenericTypeArguments[0];
                    isCollection = true;
                }

                var overridden = prop.GetCustomAttribute <OverrideTypeReaderAttribute>();
                var reader     = (overridden != null)
                    ? ModuleClassBuilder.GetTypeReader(_commands, elemType, overridden.TypeReader, services)
                    : (_commands.GetDefaultTypeReader(elemType)
                       ?? _commands.GetTypeReaders(elemType).FirstOrDefault().Value);

                if (reader != null)
                {
                    if (isCollection)
                    {
                        var method = _readMultipleMethod.MakeGenericMethod(elemType);
                        var task   = (Task <IEnumerable>)method.Invoke(null, new object[] { reader, context, arg.Split(','), services });
                        return(await task.ConfigureAwait(false));
                    }
                    else
                    {
                        return(await ReadSingle(reader, context, arg, services).ConfigureAwait(false));
                    }
                }
                return(null);
            }
        }