Hastlayer-SDK

using Hast.Common.Services;
using Hast.Layer;
using Hast.Transformer.Helpers;
using Hast.Transformer.Models;
using ICSharpCode.Decompiler.CSharp;
using ICSharpCode.Decompiler.CSharp.Syntax;
using ICSharpCode.Decompiler.IL;
using System.Collections.Generic;
using System.Linq;

namespace Hast.Transformer.Services;

/// <summary>
/// Converts a conditional expression, i.e. an expression with a ternary operator into an if-else statement.
/// </summary>
/// <example>
/// The following expression: numberOfStepsInIteration = testMode ? 1 : KpzKernels.GridWidth * KpzKernels.GridHeight;
///
/// ...will be converted into the below form: if (testMode) numberOfStepsInIteration = 1; else numberOfStepsInIteration
/// = KpzKernels.GridWidth * KpzKernels.GridHeight;.
/// </example>
public class ConditionalExpressionsToIfElsesConverter : IConverter
{
    private readonly IHashProvider _hashProvider;

    public IEnumerable<string> Dependencies { get; } = [nameof(InstanceMethodsToStaticConverter)];

    public ConditionalExpressionsToIfElsesConverter(IHashProvider hashProvider) => _hashProvider = hashProvider;

    public void Convert(
        SyntaxTree syntaxTree,
        IHardwareGenerationConfiguration configuration,
        IKnownTypeLookupTable knownTypeLookupTable) =>
        syntaxTree.AcceptVisitor(new ConditionalExpressionsConvertingVisitor(_hashProvider));

    private sealed class ConditionalExpressionsConvertingVisitor : DepthFirstAstVisitor
    {
        private readonly IHashProvider _hashProvider;
        public ConditionalExpressionsConvertingVisitor(IHashProvider hashProvider) => _hashProvider = hashProvider;

        public override void VisitConditionalExpression(ConditionalExpression conditionalExpression)
        {
            base.VisitConditionalExpression(conditionalExpression);

            // Simple "variable = condition ? value1 : value2" expressions are easily handled but ones not in this form
            // need some further work.
            if (conditionalExpression.Parent is not AssignmentExpression assignment ||
                assignment.Left is not IdentifierExpression ||
                assignment.Parent is not ExpressionStatement)
            {
                var variableName = _hashProvider.ComputeHash("conditional", conditionalExpression.GetFullName());

                var resolveResult = conditionalExpression.GetResolveResult();

                var variableType = resolveResult.Type;

                ILVariableResolveResult CreateILVariableResolveResult() =>
                    VariableHelper.CreateILVariableResolveResult(VariableKind.Local, variableType, variableName);

                // First creating a variable for the result.
                var variableDeclaration =
                    new VariableDeclarationStatement(TypeHelper.CreateAstType(variableType), variableName);
                variableDeclaration.Variables.Single().AddAnnotation(CreateILVariableResolveResult());
                AstInsertionHelper.InsertStatementBefore(
                    conditionalExpression.FindFirstParentStatement(),
                    variableDeclaration);

                // Then moving the conditional expression so its result is assigned to the variable.
                var newConditionalExpression = conditionalExpression.Clone<ConditionalExpression>();
                assignment = new AssignmentExpression(
                    new IdentifierExpression(variableName).WithAnnotation(CreateILVariableResolveResult()),
                    newConditionalExpression);

                assignment.AddAnnotation(resolveResult);

                AstInsertionHelper.InsertStatementAfter(variableDeclaration, new ExpressionStatement(assignment));

                // And finally swapping out the original expression with the variable reference.
                conditionalExpression.ReplaceWith(new IdentifierExpression(variableName).WithAnnotation(CreateILVariableResolveResult()));
                conditionalExpression = newConditionalExpression;
            }

            // Enclosing the assignments into BlockStatements because this is also what normal if-else statements are
            // decompiled into. This is also necessary to establish a variable scope.
            var trueAssignment = assignment.Clone<AssignmentExpression>();
            trueAssignment.Right = conditionalExpression.TrueExpression.Clone();
            var trueBlock = new BlockStatement();
            trueBlock.Statements.Add(new ExpressionStatement(trueAssignment));

            var falseAssignment = assignment.Clone<AssignmentExpression>();
            falseAssignment.Right = conditionalExpression.FalseExpression.Clone();
            var falseBlock = new BlockStatement();
            falseBlock.Statements.Add(new ExpressionStatement(falseAssignment));

            conditionalExpression.Parent.Parent
                .ReplaceWith(new IfElseStatement(
                    conditionalExpression.Condition.Clone(), trueBlock, falseBlock));
        }
    }
}