-
Notifications
You must be signed in to change notification settings - Fork 29
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
stub a UseHost extension method to configure app with an IHost and IHostBuilder Inject CommandContext as a service
- Loading branch information
1 parent
87fbb45
commit 632ff60
Showing
3 changed files
with
153 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,27 @@ | ||
| <Project Sdk="Microsoft.NET.Sdk"> | ||
|
|
||
| <PropertyGroup> | ||
| <AssemblyTitle>CommandDotNet.Hosting</AssemblyTitle> | ||
| <Description>Integrates Microsoft.Extensions.Hosting with CommandDotNet</Description> | ||
| </PropertyGroup> | ||
| <ItemGroup> | ||
| <Compile Remove="output\**" /> | ||
| </ItemGroup> | ||
| <ItemGroup> | ||
| <EmbeddedResource Remove="output\**" /> | ||
| </ItemGroup> | ||
| <ItemGroup> | ||
| <None Remove="output\**" /> | ||
| </ItemGroup> | ||
| <ItemGroup> | ||
| <PackageReference Include="Microsoft.Extensions.Hosting" Version="3.1.4" /> | ||
| <PackageReference Include="Nullable" Version="1.2.1"> | ||
| <PrivateAssets>all</PrivateAssets> | ||
| <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets> | ||
| </PackageReference> | ||
| </ItemGroup> | ||
| <ItemGroup> | ||
| <ProjectReference Include="..\CommandDotNet\CommandDotNet.csproj" /> | ||
| </ItemGroup> | ||
|
|
||
| </Project> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,116 @@ | ||
| using System; | ||
| using System.IO; | ||
| using System.Threading.Tasks; | ||
| using CommandDotNet.Execution; | ||
| using Microsoft.Extensions.Configuration; | ||
| using Microsoft.Extensions.DependencyInjection; | ||
| using Microsoft.Extensions.Hosting; | ||
|
|
||
| namespace CommandDotNet.Hosting | ||
| { | ||
| public static class HostingExtensions | ||
| { | ||
| public static AppRunner UseHost(this AppRunner runner, | ||
| string[]? args = null, | ||
| Action<IHostBuilder>? configureHost = null, | ||
| Func<IHostBuilder> hostBuilderFactory = null, | ||
| bool supportReplSessions = false) | ||
| { | ||
| var hostBuilder = hostBuilderFactory?.Invoke() ?? Host.CreateDefaultBuilder(args); | ||
|
|
||
| hostBuilder.ConfigureHostConfiguration(hostConfig => | ||
| { | ||
| hostConfig.SetBasePath(Directory.GetCurrentDirectory()); | ||
| }); | ||
|
|
||
| configureHost?.Invoke(hostBuilder); | ||
| return runner.UseHost(hostBuilder, supportReplSessions); | ||
| } | ||
|
|
||
| public static AppRunner UseHost(this AppRunner runner, IHostBuilder hostBuilder, bool supportReplSessions = false) | ||
| { | ||
| if (hostBuilder == null) | ||
| { | ||
| throw new ArgumentNullException(nameof(hostBuilder)); | ||
| } | ||
|
|
||
| return runner.Configure(cfg => | ||
| { | ||
| hostBuilder.ConfigureServices(services => | ||
| { | ||
| if (supportReplSessions) | ||
| { | ||
| services.AddScoped<ScopedServices>(); | ||
| services.AddScoped(p => p.GetRequiredService<ScopedServices>().CommandContext!); | ||
| services.AddScoped(p => p.GetRequiredService<ScopedServices>().CommandContext!.Console); | ||
| } | ||
| else | ||
| { | ||
| services.AddSingleton<ScopedServices>(); | ||
| services.AddSingleton(p => p.GetRequiredService<ScopedServices>().CommandContext!); | ||
| services.AddSingleton(p => p.GetRequiredService<ScopedServices>().CommandContext!.Console); | ||
| } | ||
| }); | ||
|
|
||
| var host = hostBuilder.Build(); | ||
|
|
||
| cfg.UseMiddleware(RunHost, MiddlewareSteps.CancellationHandler + 1000); | ||
| cfg.UseMiddleware(InjectCommandContextForDI, MiddlewareSteps.CancellationHandler + 1001); | ||
| cfg.Services.Add(new Config(host, supportReplSessions)); | ||
| }); | ||
| } | ||
|
|
||
| private static async Task<int> RunHost(CommandContext context, ExecutionDelegate next) | ||
| { | ||
| var config = context.AppConfig.Services.GetOrThrow<Config>(); | ||
|
|
||
| if (config.IsStarted) | ||
| { | ||
| // this call is within a REPL session. | ||
| // do not recreate the host | ||
| return await next(context); | ||
| } | ||
|
|
||
| config.IsStarted = true; | ||
| await config.Host.StartAsync(); | ||
| var result = await next(context); | ||
| await config.Host.StopAsync(); | ||
|
|
||
| return result; | ||
| } | ||
|
|
||
| private class Config | ||
| { | ||
| public bool IsStarted; | ||
| public IHost Host { get; } | ||
| public bool SupportReplSessions { get; } | ||
|
|
||
| public Config(IHost host, bool supportReplSessions) | ||
| { | ||
| Host = host ?? throw new ArgumentNullException(nameof(host)); | ||
| SupportReplSessions = supportReplSessions; | ||
| } | ||
| } | ||
|
|
||
| private class ScopedServices | ||
| { | ||
| public CommandContext? CommandContext; | ||
| } | ||
|
|
||
| private static Task<int> InjectCommandContextForDI(CommandContext context, ExecutionDelegate next) | ||
| { | ||
| var config = context.AppConfig.Services.GetOrThrow<Config>(); | ||
| var serviceProvider = config.Host.Services; | ||
|
|
||
| if (config.SupportReplSessions) | ||
| { | ||
| using var scope = serviceProvider.CreateScope(); | ||
| scope.ServiceProvider.GetRequiredService<ScopedServices>().CommandContext = context; | ||
| return next(context); | ||
| } | ||
|
|
||
| serviceProvider.GetRequiredService<ScopedServices>().CommandContext = context; | ||
| return next(context); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters