src/OneScript.Core/Compilation/Binding/SymbolScope.cs - develop - smwwiki/OneScript_fork

src/OneScript.Core/Compilation/Binding/SymbolScope.cs
137 строк4 KB
Andrei Ovsiankin
Два метода хелпера
02 июн 2024, 21:43
02 июн 2024, 21:437680922
/*----------------------------------------------------------
This Source Code Form is subject to the terms of the
Mozilla Public License, v.2.0. If a copy of the MPL
was not distributed with this file, You can obtain one
at http://mozilla.org/MPL/2.0/.
----------------------------------------------------------*/

using System.Collections.Generic;
using System.Reflection;
using OneScript.Contexts;
using OneScript.Runtime.Binding;
using OneScript.Values;

namespace OneScript.Compilation.Binding
{
    public class SymbolScope
    {
        public SymbolsCollection<IMethodSymbol> Methods { get; } =
            new SymbolsCollection<IMethodSymbol>();

        public SymbolsCollection<IVariableSymbol> Variables { get; } =
            new SymbolsCollection<IVariableSymbol>();


        public int DefineVariable(IVariableSymbol symbol)
        {
            return Variables.Add(symbol);
        }

        public int DefineMethod(IMethodSymbol symbol)
        {
            return Methods.Add(symbol);
        }

        #region Static part
        
        public static SymbolScope FromObject(BslObjectValue target)
        {
            var scope = new SymbolScope();

            var type = target.GetType();
            foreach (var info in type.GetMethods())
            {
                var attr = info.GetCustomAttribute<ContextMethodAttribute>();
                if(attr == null)
                    continue;
                
                var symbol = new BslMethodSymbol
                {
                    Method = new ContextMethodInfo(info),
                };

                scope.Methods.Add(symbol);
            }
            
            foreach (var info in type.GetProperties())
            {
                var attr = info.GetCustomAttribute<ContextMethodAttribute>();
                if(attr == null)
                    continue;
                
                var symbol = new BslPropertySymbol
                {
                    Property = new ContextPropertyInfo(info),
                };

                scope.Variables.Add(symbol);
            }

            return scope;
        }

        public static SymbolScope FromContext(IRuntimeContextInstance target)
        {
            var scope = new SymbolScope();
            for (int i = 0; i < target.GetPropCount(); i++)
            {
                var targetProp = target.GetPropertyInfo(i);
                scope.Variables.Add(new BslPropertySymbol { Property = targetProp });
            }
            
            for (int i = 0; i < target.GetMethodsCount(); i++)
            {
                var targetMeth = target.GetMethodInfo(i);
                scope.Methods.Add(new BslMethodSymbol { Method = targetMeth });
            }

            return scope;
        }

        public static SymbolScope FromContext(IContext target)
        {
            var scope = new SymbolScope();
            
            foreach (var info in target.GetMethods())
            {
                var symbol = new BslMethodSymbol
                {
                    Method = info
                };

                scope.Methods.Add(symbol);
            }
            
            foreach (var info in target.GetProperties())
            {
                var symbol = new BslPropertySymbol
                {
                    Property = info
                };

                scope.Variables.Add(symbol);
            }

            return scope;
        }

        public static SymbolScope FromMethodsAndProperties(IEnumerable<BslMethodInfo> methods,
            IEnumerable<BslPropertyInfo> properties)
        {
            var scope = new SymbolScope();
            foreach (var methodInfo in methods)
            {
                scope.Methods.Add(methodInfo.ToSymbol());
            }

            foreach (var propertyInfo in properties)
            {
                scope.Variables.Add(propertyInfo.ToSymbol());
            }

            return scope;
        }
        
        #endregion
    }
}