-
Notifications
You must be signed in to change notification settings - Fork 49
/
HotReloadClientEmit.cs
140 lines (105 loc) · 4.91 KB
/
HotReloadClientEmit.cs
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
using Microsoft.Build.Locator;
using Microsoft.CodeAnalysis.MSBuild;
using Microsoft.CodeAnalysis;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CSharp;
using System.IO;
namespace MauiReactor.HotReloadConsole
{
internal class HotReloadClientEmit : HotReloadClient
{
private Compilation? _projectCompilation;
private record ParsedFileInfo(string FilePath, SyntaxTree SyntaxTree, DateTime LastModified);
private readonly Dictionary<string, ParsedFileInfo> _parsedFiles = new();
static HotReloadClientEmit()
{
MSBuildLocator.RegisterDefaults();
}
public HotReloadClientEmit(Options options) : base(options)
{
}
protected override Task<bool> CompileProject(Stream stream, Stream pdbStream, CancellationToken cancellationToken)
{
if (_projectCompilation == null)
{
throw new InvalidOperationException();
}
var result = _projectCompilation.Emit(stream, pdbStream, cancellationToken: cancellationToken);
if (!result.Success)
{
Console.WriteLine(string.Join(Environment.NewLine, result.Diagnostics.Select(_ => _.ToString())));
}
return Task.FromResult(result.Success);
}
public override async Task Startup(CancellationToken cancellationToken)
{
await SetupProjectCompilation(cancellationToken);
}
private async Task SetupProjectCompilation(CancellationToken cancellationToken)
{
Console.Write($"Setting up build pipeline for {_projFileName} project...");
var properties = new Dictionary<string, string>
{
{ "Configuration", "Debug" },
{ "TargetFramework", _options.Framework ?? throw new InvalidOperationException() }
};
_workspace ??= MSBuildWorkspace.Create(properties);
_workspace.CloseSolution();
_project = await _workspace.OpenProjectAsync(Path.Combine(_workingDirectory, $"{_projFileName}.csproj"), cancellationToken: cancellationToken);
foreach (var projectReference in _project.AllProjectReferences)
{
}
_projectCompilation = await _project.GetCompilationAsync(cancellationToken);
if (_projectCompilation == null)
{
throw new InvalidOperationException();
}
_parsedFiles.Clear();
foreach (var syntaxTree in _projectCompilation.SyntaxTrees)
{
_parsedFiles[syntaxTree.FilePath] = new ParsedFileInfo(syntaxTree.FilePath, syntaxTree, File.GetLastWriteTime(syntaxTree.FilePath));
}
Console.WriteLine("done.");
}
protected override async Task<bool> HandleFileChangeNotifications(IEnumerable<FileChangeNotification> notifications, CancellationToken cancellationToken)
{
if (_projectCompilation == null)
{
throw new InvalidOperationException();
}
if (notifications.Any(_ =>
_.FileChangeKind == FileChangeKind.Created ||
_.FileChangeKind == FileChangeKind.Deleted ||
_.FileChangeKind == FileChangeKind.Renamed && !_parsedFiles.ContainsKey(_.FilePath)))
{
await SetupProjectCompilation(cancellationToken);
return true;
}
bool requiredHotReload = false;
foreach (var notification in notifications)
{
if (!_parsedFiles.TryGetValue(notification.FilePath, out var parsedFileInfo))
{
//this is a notification for a file not included in the project according to
//the current solution configuration
continue;
};
var currentFileLastWriteTime = File.GetLastWriteTime(notification.FilePath);
if (parsedFileInfo.LastModified != currentFileLastWriteTime)
{
var newSyntaxTree = SyntaxFactory.ParseSyntaxTree(
await ReadAllTextFileAsync(notification.FilePath, cancellationToken), cancellationToken: cancellationToken);
Console.WriteLine($"Replacing syntax tree for: {Path.GetRelativePath(_workingDirectory, notification.FilePath)}");
_projectCompilation = _projectCompilation.ReplaceSyntaxTree(parsedFileInfo.SyntaxTree, newSyntaxTree);
_parsedFiles[notification.FilePath] = new ParsedFileInfo(notification.FilePath, newSyntaxTree, currentFileLastWriteTime);
requiredHotReload = true;
}
}
return requiredHotReload;
}
}
}