LambdaMethodModel.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Amazon.Lambda.Annotations.SourceGenerator.Models.Attributes;

namespace Amazon.Lambda.Annotations.SourceGenerator.Models
{
    /// <summary>
    /// Represents original method model used for generating the code from the template.
    /// </summary>
    public class LambdaMethodModel
    {
        private AttributeModel<LambdaFunctionAttribute> _lambdaFunctionAttribute;


        /// <summary>
        /// Returns true if original method returns void
        /// </summary>
        public bool ReturnsVoid { get; set; }

        /// <summary>
        /// Returns true if original method returns <see cref="System.Threading.Tasks.Task"/>
        /// </summary>
        public bool ReturnsVoidTask { get; set; }

        /// <summary>
        /// Returns true if original method returns <see cref="System.Threading.Tasks.Task<T>"/>
        /// </summary>
        public bool ReturnsGenericTask { get; set; }

        /// <summary>
        /// Returns true if either the ReturnsVoidTask or ReturnsGenericTask are true
        /// </summary>
        public bool ReturnsVoidOrGenericTask => ReturnsVoidTask || ReturnsGenericTask;

        /// <summary>
        /// Gets or sets the return type of the method.
        /// </summary>
        public TypeModel ReturnType { get; set; }

        /// <summary>
        /// Returns true if the Lambda function returns either IHttpResult or Task<IHttpResult>
        /// </summary>
        public bool ReturnsIHttpResults
        {
            get
            {
                if(ReturnsVoid)
                {
                    return false;
                }

                if(ReturnType.FullName == TypeFullNames.IHttpResult)
                {
                    return true;
                }
                if(ReturnsGenericTask && ReturnType.TypeArguments.Count == 1 && ReturnType.TypeArguments[0].FullName == TypeFullNames.IHttpResult)
                {
                    return true;
                }


                return false;
            }
        }

        /// <summary>
        /// Returns true if the Lambda function returns either IAuthorizerResult or Task&lt;IAuthorizerResult&gt;
        /// </summary>
        public bool ReturnsIAuthorizerResult
        {
            get
            {
                if (ReturnsVoid)
                {
                    return false;
                }

                if (ReturnType.FullName == TypeFullNames.IAuthorizerResult)
                {
                    return true;
                }
                if (ReturnsGenericTask && ReturnType.TypeArguments.Count == 1 && ReturnType.TypeArguments[0].FullName == TypeFullNames.IAuthorizerResult)
                {
                    return true;
                }

                return false;
            }
        }

        /// <summary>
        /// Returns true if the Lambda function returns either ApplicationLoadBalancerResponse or Task&lt;ApplicationLoadBalancerResponse&gt;
        /// </summary>
        public bool ReturnsApplicationLoadBalancerResponse
        {
            get
            {
                if (ReturnsVoid)
                {
                    return false;
                }

                if (ReturnType.FullName == TypeFullNames.ApplicationLoadBalancerResponse)
                {
                    return true;
                }
                if (ReturnsGenericTask && ReturnType.TypeArguments.Count == 1 && ReturnType.TypeArguments[0].FullName == TypeFullNames.ApplicationLoadBalancerResponse)
                {
                    return true;
                }

                return false;
            }
        }

        /// <summary>
        /// Returns true if the Lambda function returns either void, Task, SQSBatchResponse or Task<SQSBatchResponse>
        /// </summary>
        public bool ReturnsVoidTaskOrSqsBatchResponse
        {
            get
            {
                if (ReturnsVoid || ReturnsVoidTask || ReturnType.FullName == TypeFullNames.SQSBatchResponse)
                {
                    return true;
                }
                return ReturnsGenericTask && ReturnType.TypeArguments.Count == 1 && ReturnType.TypeArguments[0].FullName == TypeFullNames.SQSBatchResponse;
            }
        }

        /// <summary>
        /// Gets or sets the parameters of original method. If this method has no parameters, returns
        /// an empty list.
        /// </summary>
        public IList<ParameterModel> Parameters { get; set; }

        /// <summary>
        /// Gets or sets name of the original method.
        /// </summary>
        public string Name { get; set; }

        /// <summary>
        /// Gets or sets name of the original method.
        /// </summary>
        public string ExecutableAssemblyHandlerName => $"{this.Name.ToLower()}_handler";

        /// <summary>
        /// Returns true if original method uses dependency injection.
        /// </summary>
        public bool UsingDependencyInjection { get; set; }

        public bool UsingHostBuilderForDependencyInjection { get; set; }

        /// <summary>
        /// Gets or sets the namespace for the nearest enclosing namespace. Returns null if the
        /// symbol isn't contained in a namespace.
        /// </summary>
        public string ContainingNamespace { get; set; }

        /// <summary>
        /// Gets or sets the name of the assemblying containing the Lambda function.
        /// </summary>
        public string ContainingAssembly { get; set; }

        /// <summary>
        /// Gets or sets type of Lambda event
        /// </summary>
        public HashSet<EventType> Events { get; set; }

        /// <summary>
        /// Gets or sets the <see cref="TypeModel"/> for the containing type. Returns null if the
        /// symbol is not contained within a type.
        /// </summary>
        public TypeModel ContainingType { get; set; }

        /// <summary>
        /// Gets or sets the attributes of original method. There always exist <see cref="Annotations.LambdaFunctionAttribute"/> in the list.
        /// </summary>
        public IList<AttributeModel> Attributes { get; set; }

        /// <summary>
        /// Gets <see cref="Annotations.LambdaFunctionAttribute"/> attribute.
        /// </summary>
        public AttributeModel<LambdaFunctionAttribute> LambdaFunctionAttribute
        {
            get
            {
                if (_lambdaFunctionAttribute == null)
                {
                    var model = Attributes.First(att => att.Type.FullName == TypeFullNames.LambdaFunctionAttribute);
                    if (model is AttributeModel<LambdaFunctionAttribute> lambdaFunctionAttributeModel)
                    {
                        _lambdaFunctionAttribute = lambdaFunctionAttributeModel;
                    }
                    else
                    {
                        throw new InvalidOperationException($"Lambda method must has a {TypeFullNames.LambdaFunctionAttribute} attribute");
                    }
                }

                return _lambdaFunctionAttribute;
            }
        }
    }
}