-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProgram.cs
More file actions
69 lines (59 loc) · 2.03 KB
/
Program.cs
File metadata and controls
69 lines (59 loc) · 2.03 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
59
60
61
62
63
64
65
66
67
68
69
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using ModelContextProtocol;
using SplashShell.Services;
using SplashShell.Tools;
namespace SplashShell;
public class Program
{
public static async Task Main(string[] args)
{
// --console mode: run as ConPTY console worker process
if (args.Contains("--console"))
{
var exitCode = await ConsoleWorker.RunConsoleMode(args);
Environment.Exit(exitCode);
return;
}
// --test mode: run tests
if (args.Contains("--test"))
{
if (args.Contains("--conpty"))
{
Tests.ConPtyMinimalTest.Run();
return;
}
Tests.OscParserTests.Run();
Tests.CommandTrackerTests.Run();
Tests.PwshColorizerTests.Run();
Tests.ConsoleManagerTests.Run();
Tests.ConsoleWorkerTests.RunUnitTests();
if (args.Contains("--e2e"))
await Tests.ConsoleWorkerTests.Run();
return;
}
// Default: MCP server mode
var builder = Host.CreateApplicationBuilder(args);
// Suppress framework logging — only warnings and errors go to stderr.
// stdout is reserved for MCP JSON-RPC protocol.
builder.Logging.ClearProviders();
builder.Logging.AddConsole(options =>
{
options.LogToStandardErrorThreshold = LogLevel.Trace;
});
builder.Logging.SetMinimumLevel(LogLevel.Warning);
builder.Services
.AddSingleton<ConsoleManager>()
.AddSingleton<ProcessLauncher>();
builder.Services
.AddMcpServer()
.WithStdioServerTransport()
.WithTools<ShellTools>()
.WithTools<FileTools>();
var host = builder.Build();
var consoleManager = host.Services.GetRequiredService<ConsoleManager>();
consoleManager.Initialize();
await host.RunAsync();
}
}