-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCommandLineWorker.cs
More file actions
58 lines (45 loc) · 1.77 KB
/
CommandLineWorker.cs
File metadata and controls
58 lines (45 loc) · 1.77 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
using CliFx;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
namespace Modulight.Modules.CommandLine
{
class CommandLineWorker : BackgroundService
{
public CommandLineWorker(IServiceProvider services, IHostApplicationLifetime lifetime, ICommandLineModuleCollection collection)
{
Services = services;
Lifetime = lifetime;
Collection = collection;
}
IServiceProvider Services { get; }
IHostApplicationLifetime Lifetime { get; }
ICommandLineModuleCollection Collection { get; }
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{
await using var scope = Services.CreateAsyncScope();
var app = new CliApplicationBuilder();
{
HashSet<Type> addedCommands = new();
foreach (var module in Collection.LoadedModules)
{
stoppingToken.ThrowIfCancellationRequested();
var cmanifest = Collection.GetManifest(module.GetType());
foreach (var cmd in cmanifest.Commands)
{
stoppingToken.ThrowIfCancellationRequested();
if (addedCommands.Contains(cmd))
continue;
app.AddCommand(cmd);
addedCommands.Add(cmd);
}
}
}
var exitCode = await app
.UseTypeActivator(scope.ServiceProvider.GetRequiredService)
.Build()
.RunAsync().ConfigureAwait(false);
Environment.ExitCode = exitCode;
Lifetime.StopApplication();
}
}
}