rainote/PacketMediator
1
0
Fork 0
Code Issues 1 Pull requests 12 Projects Releases Packages 1 Wiki Activity Actions

rough outline

Browse source
This commit is contained in:
Timothy Schenk 2025-01-18 21:32:43 +01:00
parent 6386a9ec16
commit b6cf60e909
Signed by: rainote
SSH key fingerprint: SHA256:pnkNSDwpAnaip00xaZlVFHKKsS7T8UtOomMzvs0yITE
4 changed files with 189 additions and 136 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
PacketMediator.Generator/PacketMediator.Generator.Sample/Examples.cs
View file

@ -63,7 +63,8 @@ public class HandlerA : IPacketHandler<StructA, RandomSession> {
}
public class TestHandler : IPacketHandler<StructB, RandomSession> {
public Task HandleAsync(StructB packet, RandomSession session, CancellationToken cancellationToken) {
public async Task HandleAsync(StructB packet, RandomSession session, CancellationToken cancellationToken) {
await Task.Delay(2000, cancellationToken);
throw new NotImplementedException();
}
}

20
RaiNote.PacketMediator/IntermediatePacketHandlerData.cs
View file

@ -4,14 +4,26 @@ using Microsoft.CodeAnalysis;
namespace RaiNote.PacketMediator;
class IntermediatePacketHandlerData {
public IntermediatePacketHandlerData(INamedTypeSymbol? symbol, string packetHandlerIdentifier, IntermediatePacketStructHandlerData? packetStructHandlerData) {
Symbol = symbol;
internal class IntermediatePacketHandlerData {
public IntermediatePacketHandlerData(string packetHandlerIdentifier,
IntermediatePacketStructHandlerData? packetStructHandlerData) {
PacketHandlerIdentifier = packetHandlerIdentifier;
PacketStructHandlerData = packetStructHandlerData;
}
public INamedTypeSymbol? Symbol { get; set; }
public string PacketHandlerIdentifier { get; set; }
public IntermediatePacketStructHandlerData? PacketStructHandlerData { get; set; }
}
internal record IntermediatePacketStructData(
Location SymbolLocation,
string PacketStructFullIdentifier,
string EnumValue,
string EnumTypeFullIdentifier,
string EnumMemberIdentifier,
string EnumMaxValue,
bool ImplementsInterface);
internal record IntermediateHandlerAndStructTuple(
IntermediatePacketHandlerData HandlerData,
IntermediatePacketStructData StructData);

12
RaiNote.PacketMediator/IntermediatePacketStructHandlerData.cs
View file

@ -4,12 +4,12 @@ using Microsoft.CodeAnalysis;
namespace RaiNote.PacketMediator;
class IntermediatePacketStructHandlerData {
public IntermediatePacketStructHandlerData(ITypeSymbol packetStructSymbol, ITypeSymbol sessionSymbol) {
PacketStructSymbol = packetStructSymbol;
SessionSymbol = sessionSymbol;
internal class IntermediatePacketStructHandlerData {
public IntermediatePacketStructHandlerData(string packetStructFullIdentifier, string sessionFullIdentifier) {
PacketStructFullIdentifier = packetStructFullIdentifier;
SessionFullIdentifier = sessionFullIdentifier;
}
public ITypeSymbol PacketStructSymbol { get; set; }
public ITypeSymbol SessionSymbol { get; set; }
public string PacketStructFullIdentifier { get; set; }
public string SessionFullIdentifier { get; set; }
}

182
RaiNote.PacketMediator/PacketMediatorGenerator.cs
View file

@ -24,7 +24,9 @@ public class PacketMediatorGenerator : IIncrementalGenerator {
public void Initialize(IncrementalGeneratorInitializationContext context) {
context.RegisterPostInitializationOutput(ctx => {
ctx.AddSource("PacketMediatorStatic.g.cs", SourceText.From(@"
ctx.AddSource("PacketMediatorStatic.g.cs",
SourceText.From("""
using System;
using System.Diagnostics;
using System.Threading;
@ -65,8 +67,8 @@ public interface IPacketHandler<in TIncomingPacket, in TSession> : IPacketHandle
}
using var activity = new ActivitySource(nameof(PacketMediator)).StartActivity(nameof(HandleAsync));
activity?.AddTag(""Handler"", ToString());
activity?.AddTag(""Packet"", packet.ToString());
activity?.AddTag("Handler", ToString());
activity?.AddTag("Packet", packet.ToString());
await HandleAsync(tPacket, session, cancellationToken);
return true;
@ -79,7 +81,8 @@ public interface IPacketHandler<in TSession>
{
Task<bool> TryHandleAsync(IIncomingPacket packet, TSession session, CancellationToken cancellationToken);
}
", Encoding.UTF8));
""", Encoding.UTF8));
});
// Find all struct declarations
var structsWithAttributes = context.SyntaxProvider
@ -97,8 +100,13 @@ public interface IPacketHandler<in TSession>
};
var implementsInterface = symbol != null && symbol.AllInterfaces
.Any(i => requiredInterfaces.Contains(i.Name, StringComparer.Ordinal));
if (!implementsInterface)
return default;
if (!implementsInterface) {
// TODO: https://github.com/dotnet/roslyn/blob/main/docs/features/incremental-generators.cookbook.md#issue-diagnostics
// or Analyzer
var diagnostic = Diagnostic.Create(_rpmGen001Diagnostic, symbol?.Locations.First(),
symbol?.ToDisplayString());
}
// Check for the marker attribute
var attribute = symbol?.GetAttributes()
.FirstOrDefault(attr => {
@ -116,7 +124,7 @@ public interface IPacketHandler<in TSession>
return false;
});
if (attribute == null) {
return default;
return null;
}
var attributeConstructorArgument = attribute.ConstructorArguments[0];
@ -130,95 +138,94 @@ public interface IPacketHandler<in TSession>
var enumMaxValue = enumType?.GetMembers()
.OfType<IFieldSymbol>().Max(x => x.ConstantValue);
return (symbol, structName: structDeclaration.Identifier.Text, value: enumValue, enumType,
enumMember,
enumMaxValue, implementsInterface);
if (symbol == null || enumMember == null || enumMaxValue == null || enumType == null ||
enumValue == null)
return null;
var intermediatePacketStructData = new IntermediatePacketStructData(symbol.Locations.First(),
symbol.ToDisplayString(),
enumValue.ToString(),
enumType.ToDisplayString(),
enumMember.ToDisplayString(),
enumMaxValue.ToString(), implementsInterface);
return intermediatePacketStructData;
})
.Where(result => result != default);
.Where(result => result != null);
var packetHandlers = context.SyntaxProvider.CreateSyntaxProvider(
var packetHandlerValues = context.SyntaxProvider.CreateSyntaxProvider(
predicate: (node, _) => node is ClassDeclarationSyntax,
(syntaxContext, cancellationToken) => {
var classDeclaration = (ClassDeclarationSyntax)syntaxContext.Node;
var model = syntaxContext.SemanticModel;
var symbol =
ModelExtensions.GetDeclaredSymbol(model, classDeclaration, cancellationToken: cancellationToken) as
INamedTypeSymbol;
var packetStruct = (symbol?.Interfaces.Select(interfaceSyntax => {
if (!interfaceSyntax.Name.SequenceEqual("IPacketHandler")) {
return null;
}
if (interfaceSyntax.TypeArguments.Length != 2)
return null;
var genericStructArgument = interfaceSyntax.TypeArguments[0];
var genericSessionArgument = interfaceSyntax.TypeArguments[1];
return new IntermediatePacketStructHandlerData(genericStructArgument, genericSessionArgument);
}) ?? throw new InvalidOperationException("1")).FirstOrDefault(x => x != null);
if (packetStruct == null)
return null;
return new IntermediatePacketHandlerData(symbol, classDeclaration.Identifier.Text, packetStruct);
}).Where(result => result != null);
TransformPacketHandlers).Where(result => result != null);
var location =
context.SyntaxProvider.CreateSyntaxProvider(predicate: static (node, _) => InterceptorPredicate(node),
transform: static (context, ct) => InterceptorTransform(context, ct))
.Where(candidate => candidate is not null);
var combinedResults = structsWithAttributes.Collect().Combine(packetHandlers.Collect());
var combinedResults = structsWithAttributes.Collect().Combine(packetHandlerValues.Collect()).Select(
static (tuple, cancellationToken) => {
var (structDatas, handlerDatas) = tuple;
var matchingData = handlerDatas.Select(handlerData => {
if (handlerData == null)
return null;
var structs = structDatas.Where(sData => {
var equals =
sData != null && sData.PacketStructFullIdentifier.Equals(handlerData.PacketStructHandlerData
?.PacketStructFullIdentifier);
return equals;
}).FirstOrDefault();
if (structs == null)
return null;
var intermediateHandlerAndStructTuple = new IntermediateHandlerAndStructTuple(handlerData, structs);
return intermediateHandlerAndStructTuple;
});
var intermediateHandlerAndStructTuples = matchingData.Where(x => x != null);
return intermediateHandlerAndStructTuples;
});
// Collect and generate the dictionary
context.RegisterSourceOutput(combinedResults, (ctx, result) => {
var (packetStructs, packetHandlerDatas) = result;
var combinedInfo = result.ToList();
var combinedInfo = packetHandlerDatas.Select(handler => {
var respectiveStruct = packetStructs.FirstOrDefault(pStruct =>
pStruct.structName.SequenceEqual(handler?.PacketStructHandlerData?.PacketStructSymbol.Name ??
string.Empty));
return (handler, respectiveStruct);
});
var packetHandlerData = packetHandlerDatas.FirstOrDefault();
if (combinedInfo.Count <= 0)
return;
var packetHandlerData = combinedInfo.First()?.HandlerData;
var usedValues = new List<long>();
var highestValue = long.Parse(packetStructs.FirstOrDefault().enumMaxValue?.ToString());
var intermediatePacketStructData = combinedInfo.First()?.StructData;
if (intermediatePacketStructData?.EnumMaxValue == null)
return;
var highestValue = long.Parse(intermediatePacketStructData.EnumMaxValue);
var ms = new MemoryStream();
var sw = new StreamWriter(ms, Encoding.UTF8);
sw.AutoFlush = true;
var enumTypeString = packetStructs.FirstOrDefault().enumType?.ToDisplayString();
var sessionTypeString = packetHandlerData?.PacketStructHandlerData?.SessionSymbol.ToDisplayString();
var enumTypeString = intermediatePacketStructData?.EnumTypeFullIdentifier;
var sessionTypeString = packetHandlerData?.PacketStructHandlerData?.SessionFullIdentifier;
sw.WriteLine($$"""
using System;
using System.Threading;
using System.Threading.Tasks;
public static class PacketHandlerMediator
{
public async static Task Handle(IServiceProvider serviceProvider, byte[] data,{{enumTypeString}} opcode ,{{packetHandlerData?.PacketStructHandlerData?.SessionSymbol}} session, CancellationToken cancellationToken){
public async static Task Handle(IServiceProvider serviceProvider, byte[] data,{{enumTypeString}} opcode ,{{packetHandlerData?.PacketStructHandlerData?.SessionFullIdentifier}} session, CancellationToken cancellationToken){
switch(opcode)
{
""");
foreach (var ((handler, (symbol, structName, value, _, enumMember, _, implementsInterface)), i) in
combinedInfo.Select((value, i)=> (value, i))) {
if (!implementsInterface) {
var diagnostic = Diagnostic.Create(_rpmGen001Diagnostic, symbol?.Locations.FirstOrDefault(),
structName);
ctx.ReportDiagnostic(diagnostic);
continue;
}
var tempVal = long.Parse(value?.ToString()!,
var valueTuples = combinedInfo.Select((value, i) => (value, i));
foreach (var ((handlerData, packetStructData), i) in valueTuples) {
var tempVal = long.Parse(packetStructData.EnumValue,
new NumberFormatInfo());
usedValues.Add(tempVal);
sw.WriteLine($"""
case {enumMember}:
var packet = new {handler?.PacketStructHandlerData?.PacketStructSymbol.ToDisplayString()}();
case {packetStructData.EnumMemberIdentifier}:
var packet = new {handlerData.PacketHandlerIdentifier}();
packet.Deserialize(data);
_ = {handler?.Symbol?.ToDisplayString()}.HandleAsync(packet, session, cancellationToken);
_ = {handlerData.PacketHandlerIdentifier}.HandleAsync(packet, session, cancellationToken);
return;
""");
}
@ -247,7 +254,8 @@ public interface IPacketHandler<in TSession>
""");
sw.Flush();
ctx.AddSource("PacketHandlerMediator.g.cs", SourceText.From(sw.BaseStream, Encoding.UTF8, canBeEmbedded: true));
ctx.AddSource("PacketHandlerMediator.g.cs",
SourceText.From(sw.BaseStream, Encoding.UTF8, canBeEmbedded: true));
sw.Close();
ms.Close();
var stringWriter = new StringWriter();
@ -255,12 +263,14 @@ public interface IPacketHandler<in TSession>
idWriter.WriteLine("using Microsoft.Extensions.DependencyInjection;");
idWriter.WriteLine("public static class ServiceExtensions{");
idWriter.Indent++;
idWriter.WriteLine("public static void AddPacketHandlerServices(this IServiceCollection serviceCollection){");
idWriter.WriteLine(
"public static void AddPacketHandlerServices(this IServiceCollection serviceCollection){");
idWriter.Indent++;
idWriter.WriteLine("// PacketHandler Service Generation");
foreach (var handlerData in packetHandlerDatas) {
idWriter.WriteLine($"serviceCollection.AddScoped<{handlerData?.Symbol?.ToDisplayString()}>();");
foreach (var (handlerData, _) in combinedInfo) {
idWriter.WriteLine($"serviceCollection.AddScoped<{handlerData.PacketHandlerIdentifier}>();");
}
idWriter.Indent--;
idWriter.WriteLine("}");
idWriter.Indent--;
@ -269,17 +279,47 @@ public interface IPacketHandler<in TSession>
});
}
private static IntermediatePacketHandlerData? TransformPacketHandlers(GeneratorSyntaxContext syntaxContext,
CancellationToken cancellationToken) {
var classDeclaration = (ClassDeclarationSyntax)syntaxContext.Node;
var model = syntaxContext.SemanticModel;
var symbol =
ModelExtensions.GetDeclaredSymbol(model, classDeclaration, cancellationToken: cancellationToken) as
INamedTypeSymbol;
var packetStruct = (symbol.Interfaces.Select(interfaceSyntax => {
if (!interfaceSyntax.Name.Equals("IPacketHandler")) {
return null;
}
if (interfaceSyntax.TypeArguments.Length != 2) return null;
var genericStructArgument = interfaceSyntax.TypeArguments[0];
var genericSessionArgument = interfaceSyntax.TypeArguments[1];
return new IntermediatePacketStructHandlerData(genericStructArgument.ToDisplayString(),
genericSessionArgument.ToDisplayString());
}) ?? throw new InvalidOperationException("1")).FirstOrDefault(x => x != null);
if (packetStruct == null) return null;
var intermediatePacketHandlerData = new IntermediatePacketHandlerData(symbol.ToDisplayString(), packetStruct);
return intermediatePacketHandlerData;
}
// https://andrewlock.net/creating-a-source-generator-part-11-implementing-an-interceptor-with-a-source-generator/
private static CandidateInvocation? InterceptorTransform(GeneratorSyntaxContext context, CancellationToken cancellationToken) {
private static CandidateInvocation? InterceptorTransform(GeneratorSyntaxContext context,
CancellationToken cancellationToken) {
if (context.Node is InvocationExpressionSyntax {
Expression: MemberAccessExpressionSyntax { Name: { } nameSyntax }
} invocation &&
context.SemanticModel.GetOperation(context.Node, cancellationToken) is IInvocationOperation targetOperation &&
context.SemanticModel.GetOperation(context.Node,
cancellationToken) is IInvocationOperation targetOperation &&
targetOperation.TargetMethod is
{ Name: "AddPacketHandlerServices", ContainingNamespace: { Name: "RaiNote.PacketMediator" } }
) {
#pragma warning disable RSEXPERIMENTAL002 // / Experimental interceptable location API
if (context.SemanticModel.GetInterceptableLocation(invocation, cancellationToken: cancellationToken) is { } location) {
if (context.SemanticModel.GetInterceptableLocation(invocation, cancellationToken: cancellationToken) is
{ } location) {
// Return the location details and the full type details
return new CandidateInvocation(location);
}