BuilderStepMethod.cs

using M31.FluentApi.Generator.CodeBuilding;
using M31.FluentApi.Generator.CodeGeneration.CodeBoardActors.Commons;
using M31.FluentApi.Generator.CodeGeneration.CodeBoardElements;
using M31.FluentApi.Generator.SourceGenerators.Generics;

namespace M31.FluentApi.Generator.CodeGeneration.CodeBoardActors.BuilderMethodsGeneration;

internal abstract class BuilderStepMethod : BuilderMethod
{
    protected BuilderStepMethod(BuilderMethod builderMethod)
        : base(builderMethod)
    {
    }

    internal abstract Method BuildMethodCode(BuilderAndTargetInfo info, ReservedVariableNames reservedVariableNames);

    protected Method CreateMethod(string defaultReturnType, params string[] modifiers)
    {
        MethodSignature methodSignature = CreateMethodSignature(defaultReturnType, null, modifiers);
        Method method = new Method(methodSignature);
        foreach (string comment in Comments.GetLines())
        {
            method.AddCommentLine(comment);
        }

        return method;
    }

    protected Method CreateInterfaceMethod(
        string interfaceName,
        string defaultReturnType,
        params string[] modifiers)
    {
        MethodSignature methodSignature = CreateMethodSignature(defaultReturnType, interfaceName, modifiers);
        Method method = new Method(methodSignature);
        foreach (string comment in Comments.GetLines())
        {
            method.AddCommentLine(comment);
        }

        return method;
    }

    private MethodSignature CreateMethodSignature(
        string defaultReturnType,
        string? explicitInterfacePrefix,
        params string[] modifiers)
    {
        string returnType = ReturnTypeToRespect ?? defaultReturnType;

        MethodSignature signature = MethodSignature.Create(returnType, MethodName, explicitInterfacePrefix, false);
        signature.AddModifiers(modifiers);
        CodeBuildingHelpers.AddGenericParameters(signature, GenericInfo);

        foreach (Parameter parameter in Parameters)
        {
            signature.AddParameter(parameter);
        }

        return signature;
    }

    protected void CreateBody(Method method, string instancePrefix, ReservedVariableNames reservedVariableNames)
    {
        List<string> bodyCode = BuildBodyCode(instancePrefix, reservedVariableNames, ReturnTypeToRespect);
        foreach (string bodyLine in bodyCode)
        {
            method.AppendBodyLine(bodyLine);
        }
    }

    protected void CreateReturnStatement(Method method, string defaultReturnStatement)
    {
        if (ReturnTypeToRespect == null)
        {
            method.AppendBodyLine(defaultReturnStatement);
        }
    }
}