serein-flow/Serein.Script/SereinScriptToCsharpScript.cs

using Serein.Library.Utils;
using Serein.Script.Node;
using Serein.Script.Node.FlowControl;
using System.Text;

namespace Serein.Script
{
    /// <summary>
    /// 将 Serein 脚本转换为 C# 脚本的类
    /// </summary>
    public class SereinScriptToCsharpScript
    {
        public const string ClassName = nameof(SereinScriptToCsharpScript);

        public SereinScriptMethodInfo sereinScriptMethodInfo;

        /// <summary>
        /// 符号表
        /// </summary>
        private readonly Dictionary<ASTNode, Type> _symbolInfos;

        /// <summary>
        /// 记录方法节点是否需要异步调用
        /// </summary>
        private readonly Dictionary<ASTNode, bool> _asyncMethods;

        /// <summary>
        /// 临时变量表
        /// </summary>
        private readonly Dictionary<string, Type> _local = [];

        private bool _isTaskMain = false;

        public SereinScriptToCsharpScript(SereinScriptTypeAnalysis analysis)
        {
            _symbolInfos = analysis.NodeSymbolInfos;
            _asyncMethods = analysis.AsyncMethods;
            _isTaskMain = _asyncMethods.Any(kvp => kvp.Value);
        }

        private readonly StringBuilder _codeBuilder = new StringBuilder();
        private int _indentLevel = 0;

        private void AppendLine(string code)
        {
            _codeBuilder.AppendLine(new string(' ', _indentLevel * 4) + code);
        }

        private void Append(string code)
        {
            _codeBuilder.Append(code);
        }

        private void CompleteCurrentStatement()
        {
            _codeBuilder.Append($";{Environment.NewLine}");

        }
        private string Tab =>  new string (' ', _indentLevel* 4);

        private void Indent() => _indentLevel++;
        private void Unindent() => _indentLevel--;

        private List<Action<StringBuilder>> _classDefinitions = new List<Action<StringBuilder>>();

        public SereinScriptMethodInfo CompileToCSharp(string mehtodName, ProgramNode programNode, Dictionary<string, Type>? param)
        {
            _codeBuilder.Clear();
            sereinScriptMethodInfo = new SereinScriptMethodInfo()
            {
                ClassName = ClassName,
                ParamInfos = new List<SereinScriptMethodInfo.SereinScriptParamInfo>(),
                MethodName = mehtodName,
            };
            var sb = _codeBuilder;
            var methodResultType = _symbolInfos[programNode];
            if (methodResultType == typeof(void))
            {
                methodResultType = typeof(object);
            }
            var taskFullName = typeof(Task).FullName;
            string? returnContent;
            if (_isTaskMain)
            {
                returnContent = $"global::{taskFullName}<global::{methodResultType.GetFriendlyName()}>";
                sereinScriptMethodInfo.IsAsync = true;
            }
            else
            {
                returnContent = $"global::{methodResultType.GetFriendlyName()}";
                sereinScriptMethodInfo.IsAsync = false;
            }

            AppendLine($"public partial class {ClassName}");
            AppendLine("{");
            Indent();
            if (param is null || param.Count == 0)
            {
                // 生成方法签名
                AppendLine($"public static {returnContent} {mehtodName}()");
            }
            else
            {
                // 生成方法签名
                AppendLine($"public static {returnContent} {mehtodName}({GetMethodParamster(param)})");
            }
            AppendLine("{");
            Indent();
            // 生成变量节点
            var idfNodesTemp = _symbolInfos.Keys.Where(key => key is IdentifierNode)
                                            .OfType<IdentifierNode>().ToList();
            var idfNodes = (param is null) switch
            {
                true => idfNodesTemp.DistinctBy(n => n.Name).ToList(),
                false => idfNodesTemp.DistinctBy(n => n.Name).DistinctByCondition(n => !param.ContainsKey(n.Name)).ToList(),
            };
            foreach (var idf in idfNodes)
            {
                var varName = idf.Name;
                var varType = _symbolInfos[idf];
                AppendLine($"global::{varType.GetFriendlyName()} {varName} = default; // 变量");
            }
            AppendLine("");

            // 递归遍历节点生成代码
            foreach (var stmt in programNode.Statements)
            {
                ConvertCode(stmt);
                if (stmt is not (IfNode or WhileNode))
                {
                    CompleteCurrentStatement();
                }
            }
            if (_symbolInfos[programNode] == typeof(void))
            {
                AppendLine("");
                AppendLine("return null;");
            }

            Unindent();
            AppendLine("}");
            Unindent();
            AppendLine("}");

            foreach (var cd in _classDefinitions)
            {
                cd.Invoke(sb);
            }
            sereinScriptMethodInfo.CsharpCode = sb.ToString();
            sereinScriptMethodInfo.ReturnType = methodResultType;
            return sereinScriptMethodInfo;
        }

        private string GetMethodParamster(Dictionary<string, Type> param)
        {
            var values = param.Select(kvp =>
            {
                var paramName = kvp.Key;
                var type = kvp.Value;
                _local[paramName] = type;
                sereinScriptMethodInfo.ParamInfos.Add(new SereinScriptMethodInfo.SereinScriptParamInfo
                {
                    ParameterType = type,
                    ParamName = paramName,
                });
                return $"global::{type.GetFriendlyName()} {paramName}";
            });
            return string.Join(',', values);
        }
        
        private void ConvertCode(ASTNode node)
        {
            switch (node)
            {
                case ProgramNode programNode: // 程序开始节点
                    break;
                case ReturnNode returnNode: // 程序退出节点
                    void ConvertCodeOfReturnNode(ReturnNode returnNode)
                    {
                        Append(Tab);
                        if (returnNode.Value is not null)
                        {
                            Append($"return ");
                            ConvertCode(returnNode.Value);
                        }
                        else
                        {
                            AppendLine("return defult");
                        }
                    }
                    ConvertCodeOfReturnNode(returnNode);
                    break;
                #region 字面量解析
                case NullNode nullNode: // null
                    Append("null");
                    break;
                case CharNode charNode: // char字面量
                    Append($"'{charNode.Value}'");
                    break;
                case RawStringNode rawStringNode: // 原始字符串
                    AppendLine("");
                    Append("\"\"\"");
                    Append(rawStringNode.Value);
                    Append("\"\"\"");
                    break;
                case StringNode stringNode: // 字符串字面量
                    void ConvertCodeOfStringNode(StringNode stringNode)
                    {
                        static string EscapeForCSharpString(string input)
                        {
                            return input
                                .Replace("\\", "\\\\")
                                .Replace("\"", "\\\"")
                                .Replace("\0", "\\0")
                                .Replace("\a", "\\a")
                                .Replace("\b", "\\b")
                                .Replace("\f", "\\f")
                                .Replace("\n", "\\n")
                                .Replace("\r", "\\r")
                                .Replace("\t", "\\t")
                                .Replace("\v", "\\v");
                        }
                        var value = stringNode.Value;
                        var sp = value.Split(Environment.NewLine);
                        if(sp.Length == 1)
                        {
                            var content = EscapeForCSharpString(value);
                            Append($"\"{content}\"");
                        }
                        else
                        {
                            Append($"\"");
                            for (int index = 0; index < sp.Length; index++)
                            {
                                string? s = sp[index];
                                var content = EscapeForCSharpString(s);
                                if(index == 0)
                                {
                                    Append(content);
                                }
                                else 
                                {
                                    if (OperatingSystem.IsWindows())
                                    {
                                        Append($"\\r\\n{content}");
                                    }
                                    else if (OperatingSystem.IsLinux())
                                    {
                                        Append($"\\n{content}");
                                    }
                                }
                               
                               
                            }
                            Append($"\"");
                        }
                        

                    }
                    ConvertCodeOfStringNode(stringNode);
                    break;
                case BooleanNode booleanNode: // 布尔值字面量
                    Append($"{(booleanNode.Value ? "true" : "false")}");
                    break;
                case NumberIntNode numberIntNode: // int整型数值字面量
                    Append($"{numberIntNode.Value}");
                    break;
                case NumberLongNode numberLongNode: // long整型数值字面量
                    Append($"{numberLongNode.Value}");
                    break;
                case NumberFloatNode numberFloatNode: // float浮点数值字面量
                    Append($"{numberFloatNode.Value}f");
                    break;
                case NumberDoubleNode numberDoubleNode: // double浮点数值字面量
                    Append($"{numberDoubleNode.Value}d");
                    break;
                #endregion
                case IdentifierNode identifierNode: // 变量定义
                    void ConvertCodeOfIdentifierNode(IdentifierNode identifierNode)
                    {
                        var varName = identifierNode.Name;
                        Append(varName);
                    }
                    ConvertCodeOfIdentifierNode(identifierNode);
                    break;
                case IfNode ifNode: // if语句结构
                    void ConvertCodeOfIfNode(IfNode ifNOde)
                    {
                        
                        Append($"{Tab}if(");
                        ConvertCode(ifNOde.Condition); // 解析条件
                        Append($" == true){Environment.NewLine}");
                        AppendLine("{");
                        Indent();
                        foreach (var item in ifNOde.TrueBranch)
                        {
                            ConvertCode(item);
                            CompleteCurrentStatement();
                        }
                        Unindent();
                        AppendLine("}");
                        AppendLine("else");
                        AppendLine("{");
                        Indent();
                        foreach (var item in ifNOde.FalseBranch)
                        {
                            ConvertCode(item);
                            CompleteCurrentStatement();
                        }
                        Unindent();
                        AppendLine("}");
                    }
                    ConvertCodeOfIfNode(ifNode);
                    break;
                case WhileNode whileNode: // while语句结构
                    void ConvertCodeOfWhileNode(WhileNode whileNode)
                    {
                        AppendLine("");
                        Append($"while(");
                        ConvertCode(whileNode.Condition); // 解析条件
                        Append($" == {true})");
                        AppendLine("{");
                        Indent();
                        foreach (var item in whileNode.Body)
                        {
                            ConvertCode(item);
                            CompleteCurrentStatement();
                        }
                        Unindent();
                        AppendLine("}");
                    }
                    ConvertCodeOfWhileNode(whileNode);
                    break;
                case AssignmentNode assignmentNode: // 变量赋值语句
                    void ConvertCodeOfAssignmentNode(AssignmentNode assignmentNode)
                    {
                        
                        Append(Tab);
                        ConvertCode(assignmentNode.Target);
                        Append(" = ");
                        if (assignmentNode.Value is null)
                        {
                            Append("null");
                        }
                        else
                        {
                            ConvertCode(assignmentNode.Value);
                        }
                    }
                    ConvertCodeOfAssignmentNode(assignmentNode);
                    break;
                case BinaryOperationNode binaryOperationNode: // 二元运算操作
                    void ConvertCodeOfBinaryOperationNode(BinaryOperationNode binaryOperationNode)
                    {
                        Append("(");
                        ConvertCode(binaryOperationNode.Left); // 左操作数
                        Append(binaryOperationNode.Operator); // 操作符
                        ConvertCode(binaryOperationNode.Right); // 左操作数
                        Append(")");
                    }
                    ConvertCodeOfBinaryOperationNode(binaryOperationNode);
                    break;
                case CollectionAssignmentNode collectionAssignmentNode:
                    void ConvertCodeOfCollectionAssignmentNode(CollectionAssignmentNode collectionAssignmentNode)
                    {
                        ConvertCode(collectionAssignmentNode.Collection);
                        Append(" = ");
                        ConvertCode(collectionAssignmentNode.Value);
                        AppendLine(string.Empty);
                    }
                    ConvertCodeOfCollectionAssignmentNode(collectionAssignmentNode);
                    break;
                case CollectionIndexNode collectionIndexNode: // 集合类型操作
                    void ConvertCodeOfCollectionIndexNode(CollectionIndexNode collectionIndexNode)
                    {
                        ConvertCode(collectionIndexNode.Collection);
                        Append("[");
                        ConvertCode(collectionIndexNode.Index);
                        Append("]");
                    }
                    ConvertCodeOfCollectionIndexNode(collectionIndexNode);
                    break;
                case ArrayDefintionNode arrayDefintionNode:
                    void ConvertCodeOfArrayDefintionNode(ArrayDefintionNode arrayDefintionNode)
                    {
                        var arrType = this._symbolInfos[arrayDefintionNode];
                        var tab = new string(' ', (_indentLevel + 1) * 4);
                        Append($"new global::{arrType.GetFriendlyName()}{{{Environment.NewLine}");
                        for (int index = 0; index < arrayDefintionNode.Elements.Count; index++)
                        {
                            ASTNode? e = arrayDefintionNode.Elements[index];
                            Append(tab + "    ");
                            ConvertCode(e);
                            if (index < arrayDefintionNode.Elements.Count - 1)
                                Append($", {Environment.NewLine}");

                        }
                        Append($"{Environment.NewLine}{tab}}}");
                    }
                    ConvertCodeOfArrayDefintionNode(arrayDefintionNode);
                    break;
                case ClassTypeDefinitionNode classTypeDefinitionNode: // 类型定义
                    void ConvertCodeOfClassTypeDefinitionNode(ClassTypeDefinitionNode classTypeDefinitionNode)
                    {
                        _classDefinitions.Add((sb) =>
                        {
                            var type = _symbolInfos[classTypeDefinitionNode.ClassType];
                            AppendLine($"public class {type.FullName}");
                            AppendLine("{");
                            Indent();
                            foreach (var property in classTypeDefinitionNode.Propertys)
                            {
                                var propertyName = property.Key;
                                var propertyType = _symbolInfos[property.Value];
                                AppendLine($"public global::{propertyType.GetFriendlyName()} {propertyName} {{ get; set; }}");
                            }
                            Unindent();
                            AppendLine("}");
                        });
                    }
                    ConvertCodeOfClassTypeDefinitionNode(classTypeDefinitionNode);
                    break;
                case TypeNode typeNode: // 类型
                    break;
                case ObjectInstantiationNode objectInstantiationNode: // 类型实例化
                    void ConvertCodeOfObjectInstantiationNode(ObjectInstantiationNode objectInstantiationNode)
                    {
                        var type = _symbolInfos[objectInstantiationNode.Type];
                        Append($"new global::{type}");
                        if (objectInstantiationNode.Arguments.Count > 0)
                        {
                            Append("(");
                            for (int i = 0; i < objectInstantiationNode.Arguments.Count; i++)
                            {
                                ConvertCode(objectInstantiationNode.Arguments[i]);
                                if (i < objectInstantiationNode.Arguments.Count - 1)
                                {
                                    Append(", ");
                                }
                            }
                            Append(")");
                        }
                        else
                        {
                            Append("()");
                        }
                        if (objectInstantiationNode.CtorAssignments.Count > 0)
                        {
                            AppendLine("{");
                            Indent();
                            foreach (var assignment in objectInstantiationNode.CtorAssignments)
                            {
                                ConvertCode(assignment);
                            }
                            Unindent();
                            AppendLine("}");
                        }
                    }
                    ConvertCodeOfObjectInstantiationNode(objectInstantiationNode);
                    break;
                case CtorAssignmentNode ctorAssignmentNode: // 构造器赋值
                    void ConvertCodeOfCtorAssignmentNode(CtorAssignmentNode ctorAssignmentNode)
                    {
                        var propertyName = ctorAssignmentNode.MemberName;
                        var value = ctorAssignmentNode.Value;
                        Append($"{propertyName} = ");
                        ConvertCode(value);
                        AppendLine(",");
                    }
                    ConvertCodeOfCtorAssignmentNode(ctorAssignmentNode);
                    break;
                case MemberAccessNode memberAccessNode: // 对象成员访问
                    void ConvertCodeOfMemberAccessNode(MemberAccessNode memberAccessNode)
                    {
                        ConvertCode(memberAccessNode.Object);
                        Append($".{memberAccessNode.MemberName}");
                    }
                    ConvertCodeOfMemberAccessNode(memberAccessNode);
                    break;
                case MemberAssignmentNode memberAssignmentNode: // 对象成员赋值
                    void ConvertCodeOfMemberAssignmentNode(MemberAssignmentNode memberAssignmentNode)
                    {
                        ConvertCode(memberAssignmentNode.Object);
                        Append($".{memberAssignmentNode.MemberName} = ");
                        ConvertCode(memberAssignmentNode.Value);
                    }
                    ConvertCodeOfMemberAssignmentNode(memberAssignmentNode);
                    break;
                case MemberFunctionCallNode memberFunctionCallNode: // 对象方法调用
                    void ConvertCodeOfMemberFunctionCallNode(MemberFunctionCallNode memberFunctionCallNode)
                    {
                        var isAsync = _asyncMethods.TryGetValue(memberFunctionCallNode, out var isAsyncValue) && isAsyncValue;
                        if (isAsync)
                        {
                            Append($"(await ");
                        }

                        ConvertCode(memberFunctionCallNode.Object);
                        Append($".{memberFunctionCallNode.FunctionName}(");
                        for (int i = 0; i < memberFunctionCallNode.Arguments.Count; i++)
                        {
                            ASTNode? argNode = memberFunctionCallNode.Arguments[i];
                            ConvertCode(argNode);
                            if (i < memberFunctionCallNode.Arguments.Count - 1)
                            {
                                Append(", ");
                            }
                        }
                        Append($")");
                        if (isAsync)
                        {
                            Append($")");
                        }


                    }
                    ConvertCodeOfMemberFunctionCallNode(memberFunctionCallNode);
                    break;
                case FunctionCallNode functionCallNode: // 外部挂载的函数调用
                    void ConvertCodeOfFunctionCallNode(FunctionCallNode functionCallNode)
                    {
                        var isAsync = _asyncMethods.TryGetValue(functionCallNode, out var isAsyncValue) && isAsyncValue;
                        if (isAsync)
                        {
                            Append($"(await ");
                        }
                        var funcName = functionCallNode.FunctionName switch
                        {
                            "int" => "@int",
                            "bool" => "@bool",
                            "double" => "@double",
                            "long" => "@long",
                            "decimal" => "@decimal",
                            "float" => "@float",
                            "byte" => "@byte",

                            _ => functionCallNode.FunctionName,
                        };
                        Append($"global::Serein.Library.ScriptBaseFunc.{funcName}(");
                        for (int i = 0; i < functionCallNode.Arguments.Count; i++)
                        {
                            ASTNode? argNode = functionCallNode.Arguments[i];
                            ConvertCode(argNode);
                            if (i < functionCallNode.Arguments.Count - 1)
                            {
                                Append(", ");
                            }
                        }
                        Append($")");
                        if (isAsync)
                        {
                            Append($")");
                        }
                    }
                    ConvertCodeOfFunctionCallNode(functionCallNode);
                    break;
                default: // 未定义的节点类型
                    break;
            }
        }

        
    }


}