src/ScriptEngine/Hosting/ServiceRegistrationExtensions.cs - develop - EvilBeaver/OneScript

src/ScriptEngine/Hosting/ServiceRegistrationExtensions.cs
70 строк3 KB
EvilBeaver
Вынесен IAttachableContext
14 сен 2022, 20:55
14 сен 2022, 20:55893b373
/*----------------------------------------------------------
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;
using System.Reflection;
using OneScript.Contexts;
using OneScript.DependencyInjection;
using OneScript.Language.SyntaxAnalysis;
using ScriptEngine.Machine;
using ScriptEngine.Machine.Contexts;

namespace ScriptEngine.Hosting
{
    public static class ServiceRegistrationExtensions
    {
        public static IEngineBuilder SetupConfiguration(this IEngineBuilder b, Action<ConfigurationProviders> setup)
        {
            setup(b.ConfigurationProviders);
            b.Services.RegisterSingleton(b.ConfigurationProviders);
            return b;
        }
        
        public static ExecutionContext AddAssembly(this ExecutionContext env, Assembly asm, Predicate<Type> filter = null)
        {
            var discoverer = env.Services.Resolve<ContextDiscoverer>();
            discoverer.DiscoverClasses(asm, filter);
            discoverer.DiscoverGlobalContexts(env.GlobalNamespace, asm, filter);
            return env;
        }
        
        public static ExecutionContext AddGlobalContext(this ExecutionContext env, IAttachableContext context)
        {
            env.GlobalNamespace.InjectObject(context);
            env.GlobalInstances.RegisterInstance(context);
            return env;
        }

        public static IServiceDefinitions UseImports(this IServiceDefinitions services)
        {
            services.RegisterEnumerable<IDirectiveHandler, ImportDirectivesHandler>();
            services.RegisterSingleton<IDependencyResolver, NullDependencyResolver>();
            return services;
        }
        
        public static IServiceDefinitions UseImports<T>(this IServiceDefinitions services)
            where T : class, IDependencyResolver
        {
            services.RegisterEnumerable<IDirectiveHandler, ImportDirectivesHandler>();
            services.RegisterSingleton<IDependencyResolver, T>();
            return services;
        }
        
        public static IServiceDefinitions UseImports(this IServiceDefinitions services, Func<IServiceContainer, IDependencyResolver> factory)
        {
            services.RegisterEnumerable<IDirectiveHandler, ImportDirectivesHandler>();
            services.RegisterSingleton<IDependencyResolver>(factory);
            return services;
        }
        
        public static IServiceDefinitions AddDirectiveHandler<T>(this IServiceDefinitions services) where T : class, IDirectiveHandler
        {
            services.RegisterEnumerable<IDirectiveHandler, T>();
            return services;
        }
    }
}