mirror of https://github.com/Xpl0itR/protodec.git
Even more improvements
including but not limited to: - detect protobuf well-known types - detect NonUserCodeAttribute - detect ObsoleteAttribute to mark protobufs/fields as deprecated - parse generated client/server types for gRPC services - fix "optional" parsing - fix obfuscated name translation - fix foreign nested protobufs - support protobuf editions and corresponding options - rewrite protobuf models to better reflect the specification, including decoupling toplevels and files - rewrite a bunch of things to make more sense
This commit is contained in:
parent
68ee943b9d
commit
d43deef033
|
@ -10,13 +10,17 @@ Arguments:
|
||||||
target_assembly_path Either the path to the target assembly or a directory of assemblies, all of which be parsed.
|
target_assembly_path Either the path to the target assembly or a directory of assemblies, all of which be parsed.
|
||||||
out_path An existing directory to output into individual files, otherwise output to a single file.
|
out_path An existing directory to output into individual files, otherwise output to a single file.
|
||||||
Options:
|
Options:
|
||||||
|
--parse_service_servers Parses gRPC service definitions from server classes.
|
||||||
|
--parse_service_clients Parses gRPC service definitions from client classes.
|
||||||
--skip_enums Skip parsing enums and replace references to them with int32.
|
--skip_enums Skip parsing enums and replace references to them with int32.
|
||||||
--skip_properties_without_protoc_attribute Skip properties that aren't decorated with `GeneratedCode("protoc")` when parsing
|
--include_properties_without_non_user_code_attribute Includes properties that aren't decorated with `DebuggerNonUserCode` when parsing.
|
||||||
|
--include_service_methods_without_generated_code_attribute Includes methods that aren't decorated with `GeneratedCode("grpc_csharp_plugin")` when parsing gRPC services.
|
||||||
```
|
```
|
||||||
|
|
||||||
Limitations
|
Limitations
|
||||||
-----------
|
-----------
|
||||||
- Integers are assumed to be (u)int32/64 as CIL doesn't differentiate between them and sint32/64 and (s)fixed32/64.
|
- Integers are assumed to be (u)int32/64 as CIL doesn't differentiate between them and sint32/64 and (s)fixed32/64.
|
||||||
|
- Package names are not preserved in protobuf compilation so naturally we cannot recover them during decompilation, which may result in naming conflicts.
|
||||||
- When decompiling from [Il2CppDumper](https://github.com/Perfare/Il2CppDumper) DummyDLLs
|
- When decompiling from [Il2CppDumper](https://github.com/Perfare/Il2CppDumper) DummyDLLs
|
||||||
- The `Name` parameter of `OriginalNameAttribute` is not dumped. In this case, the CIL enum field names are used after conforming them to protobuf conventions
|
- The `Name` parameter of `OriginalNameAttribute` is not dumped. In this case, the CIL enum field names are used after conforming them to protobuf conventions
|
||||||
|
|
||||||
|
|
|
@ -7,6 +7,11 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "protodec", "src\protodec\pr
|
||||||
EndProject
|
EndProject
|
||||||
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "LibProtodec", "src\LibProtodec\LibProtodec.csproj", "{5F6DAD82-D9AD-4CE5-86E6-D20C9F059A4D}"
|
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "LibProtodec", "src\LibProtodec\LibProtodec.csproj", "{5F6DAD82-D9AD-4CE5-86E6-D20C9F059A4D}"
|
||||||
EndProject
|
EndProject
|
||||||
|
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "Solution Items", "Solution Items", "{22F217C3-0FC2-4D06-B5F3-AA1E3AFC402E}"
|
||||||
|
ProjectSection(SolutionItems) = preProject
|
||||||
|
README.md = README.md
|
||||||
|
EndProjectSection
|
||||||
|
EndProject
|
||||||
Global
|
Global
|
||||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||||
Debug|Any CPU = Debug|Any CPU
|
Debug|Any CPU = Debug|Any CPU
|
||||||
|
|
|
@ -35,16 +35,36 @@ public sealed class AssemblyInspector : IDisposable
|
||||||
|
|
||||||
public IEnumerable<Type> GetProtobufMessageTypes()
|
public IEnumerable<Type> GetProtobufMessageTypes()
|
||||||
{
|
{
|
||||||
Type? googleProtobufIMessage = LoadedTypes.SingleOrDefault(static type => type?.FullName == "Google.Protobuf.IMessage", null)
|
Type? iMessage = LoadedTypes.SingleOrDefault(static type => type?.FullName == "Google.Protobuf.IMessage", null)
|
||||||
?? AssemblyContext.LoadFromAssemblyName("Google.Protobuf")
|
?? AssemblyContext.LoadFromAssemblyName("Google.Protobuf")
|
||||||
.GetType("Google.Protobuf.IMessage");
|
.GetType("Google.Protobuf.IMessage");
|
||||||
return from type
|
|
||||||
in LoadedTypes
|
return LoadedTypes.Where(
|
||||||
where !type.IsNested
|
type => type is { IsNested: false, IsSealed: true }
|
||||||
&& type.IsSealed
|
|
||||||
&& type.Namespace?.StartsWith("Google.Protobuf", StringComparison.Ordinal) != true
|
&& type.Namespace?.StartsWith("Google.Protobuf", StringComparison.Ordinal) != true
|
||||||
&& type.IsAssignableTo(googleProtobufIMessage)
|
&& type.IsAssignableTo(iMessage));
|
||||||
select type;
|
}
|
||||||
|
|
||||||
|
public IEnumerable<Type> GetProtobufServiceClientTypes()
|
||||||
|
{
|
||||||
|
Type? clientBase = LoadedTypes.SingleOrDefault(static type => type?.FullName == "Grpc.Core.ClientBase", null)
|
||||||
|
?? AssemblyContext.LoadFromAssemblyName("Grpc.Core.Api")
|
||||||
|
.GetType("Grpc.Core.ClientBase");
|
||||||
|
|
||||||
|
return LoadedTypes.Where(
|
||||||
|
type => type is { IsNested: true, IsAbstract: false }
|
||||||
|
&& type.IsAssignableTo(clientBase));
|
||||||
|
}
|
||||||
|
|
||||||
|
public IEnumerable<Type> GetProtobufServiceServerTypes()
|
||||||
|
{
|
||||||
|
Type? bindServiceMethodAttribute = LoadedTypes.SingleOrDefault(static type => type?.FullName == "Grpc.Core.BindServiceMethodAttribute", null)
|
||||||
|
?? AssemblyContext.LoadFromAssemblyName("Grpc.Core.Api")
|
||||||
|
.GetType("Grpc.Core.BindServiceMethodAttribute");
|
||||||
|
|
||||||
|
return LoadedTypes.Where(
|
||||||
|
type => type is { IsNested: true, IsAbstract: true, DeclaringType: { IsNested: false, IsSealed: true, IsAbstract: true } }
|
||||||
|
&& type.GetCustomAttributesData().Any(attribute => attribute.AttributeType == bindServiceMethodAttribute));
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Dispose() =>
|
public void Dispose() =>
|
||||||
|
|
|
@ -1,46 +0,0 @@
|
||||||
// Copyright © 2023-2024 Xpl0itR
|
|
||||||
//
|
|
||||||
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
||||||
|
|
||||||
using System.CodeDom.Compiler;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using SystemEx.Collections;
|
|
||||||
|
|
||||||
namespace LibProtodec;
|
|
||||||
|
|
||||||
public sealed class Enum : Protobuf
|
|
||||||
{
|
|
||||||
public readonly List<KeyValuePair<int, string>> Fields = [];
|
|
||||||
|
|
||||||
public override void WriteFileTo(IndentedTextWriter writer)
|
|
||||||
{
|
|
||||||
this.WritePreambleTo(writer);
|
|
||||||
WriteTo(writer);
|
|
||||||
}
|
|
||||||
|
|
||||||
public override void WriteTo(IndentedTextWriter writer)
|
|
||||||
{
|
|
||||||
writer.Write("enum ");
|
|
||||||
writer.Write(this.Name);
|
|
||||||
writer.WriteLine(" {");
|
|
||||||
writer.Indent++;
|
|
||||||
|
|
||||||
if (Fields.ContainsDuplicateKey())
|
|
||||||
{
|
|
||||||
writer.WriteLine("option allow_alias = true;");
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach ((int id, string name) in Fields)
|
|
||||||
{
|
|
||||||
writer.Write(name);
|
|
||||||
writer.Write(" = ");
|
|
||||||
writer.Write(id);
|
|
||||||
writer.WriteLine(';');
|
|
||||||
}
|
|
||||||
|
|
||||||
writer.Indent--;
|
|
||||||
writer.Write('}');
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,26 +0,0 @@
|
||||||
// Copyright © 2024 Xpl0itR
|
|
||||||
//
|
|
||||||
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
||||||
|
|
||||||
namespace LibProtodec;
|
|
||||||
|
|
||||||
public static class FieldTypeName
|
|
||||||
{
|
|
||||||
public const string Double = "double";
|
|
||||||
public const string Float = "float";
|
|
||||||
public const string Int32 = "int32";
|
|
||||||
public const string Int64 = "int64";
|
|
||||||
public const string UInt32 = "uint32";
|
|
||||||
public const string UInt64 = "uint64";
|
|
||||||
public const string SInt32 = "sint32";
|
|
||||||
public const string SInt64 = "sint64";
|
|
||||||
public const string Fixed32 = "fixed32";
|
|
||||||
public const string Fixed64 = "fixed64";
|
|
||||||
public const string SFixed32 = "sfixed32";
|
|
||||||
public const string SFixed64 = "sfixed64";
|
|
||||||
public const string Bool = "bool";
|
|
||||||
public const string String = "string";
|
|
||||||
public const string Bytes = "bytes";
|
|
||||||
}
|
|
|
@ -1,98 +0,0 @@
|
||||||
// Copyright © 2023-2024 Xpl0itR
|
|
||||||
//
|
|
||||||
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
||||||
|
|
||||||
using System.CodeDom.Compiler;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.IO;
|
|
||||||
using System.Linq;
|
|
||||||
|
|
||||||
namespace LibProtodec;
|
|
||||||
|
|
||||||
public sealed class Message : Protobuf
|
|
||||||
{
|
|
||||||
public readonly HashSet<string> Imports = [];
|
|
||||||
public readonly Dictionary<string, int[]> OneOfs = [];
|
|
||||||
public readonly Dictionary<int, (bool IsOptional, string Type, string Name)> Fields = [];
|
|
||||||
public readonly Dictionary<string, Protobuf> Nested = [];
|
|
||||||
|
|
||||||
public override void WriteFileTo(IndentedTextWriter writer)
|
|
||||||
{
|
|
||||||
this.WritePreambleTo(writer);
|
|
||||||
|
|
||||||
if (Imports.Count > 0)
|
|
||||||
{
|
|
||||||
foreach (string import in Imports)
|
|
||||||
{
|
|
||||||
writer.Write("import \"");
|
|
||||||
writer.Write(import);
|
|
||||||
writer.WriteLine(".proto\";");
|
|
||||||
}
|
|
||||||
|
|
||||||
writer.WriteLine();
|
|
||||||
}
|
|
||||||
|
|
||||||
WriteTo(writer);
|
|
||||||
}
|
|
||||||
|
|
||||||
public override void WriteTo(IndentedTextWriter writer)
|
|
||||||
{
|
|
||||||
writer.Write("message ");
|
|
||||||
writer.Write(this.Name);
|
|
||||||
writer.WriteLine(" {");
|
|
||||||
writer.Indent++;
|
|
||||||
|
|
||||||
int[] oneOfs = OneOfs.SelectMany(static oneOf => oneOf.Value).ToArray();
|
|
||||||
|
|
||||||
foreach ((int fieldId, (bool, string, string) field) in Fields)
|
|
||||||
{
|
|
||||||
if (oneOfs.Contains(fieldId))
|
|
||||||
continue;
|
|
||||||
|
|
||||||
WriteField(writer, fieldId, field);
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach ((string name, int[] fieldIds) in OneOfs)
|
|
||||||
{
|
|
||||||
// ReSharper disable once StringLiteralTypo
|
|
||||||
writer.Write("oneof ");
|
|
||||||
writer.Write(name);
|
|
||||||
writer.WriteLine(" {");
|
|
||||||
writer.Indent++;
|
|
||||||
|
|
||||||
foreach (int fieldId in fieldIds)
|
|
||||||
{
|
|
||||||
WriteField(writer, fieldId, Fields[fieldId]);
|
|
||||||
}
|
|
||||||
|
|
||||||
writer.Indent--;
|
|
||||||
writer.WriteLine('}');
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach (Protobuf nested in Nested.Values)
|
|
||||||
{
|
|
||||||
nested.WriteTo(writer);
|
|
||||||
writer.WriteLine();
|
|
||||||
}
|
|
||||||
|
|
||||||
writer.Indent--;
|
|
||||||
writer.Write('}');
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void WriteField(TextWriter writer, int fieldId, (bool IsOptional, string Type, string Name) field)
|
|
||||||
{
|
|
||||||
if (field.IsOptional)
|
|
||||||
{
|
|
||||||
writer.Write("optional ");
|
|
||||||
}
|
|
||||||
|
|
||||||
writer.Write(field.Type);
|
|
||||||
writer.Write(' ');
|
|
||||||
writer.Write(field.Name);
|
|
||||||
writer.Write(" = ");
|
|
||||||
writer.Write(fieldId);
|
|
||||||
writer.WriteLine(';');
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -0,0 +1,29 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Fields;
|
||||||
|
|
||||||
|
public sealed class EnumField
|
||||||
|
{
|
||||||
|
public required string Name { get; init; }
|
||||||
|
public required int Id { get; init; }
|
||||||
|
|
||||||
|
public bool IsObsolete { get; init; }
|
||||||
|
|
||||||
|
public void WriteTo(System.IO.TextWriter writer)
|
||||||
|
{
|
||||||
|
writer.Write(Name);
|
||||||
|
writer.Write(" = ");
|
||||||
|
writer.Write(Id);
|
||||||
|
|
||||||
|
if (IsObsolete)
|
||||||
|
{
|
||||||
|
writer.Write(" [deprecated = true]");
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.WriteLine(';');
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,42 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
using System.IO;
|
||||||
|
using LibProtodec.Models.TopLevels;
|
||||||
|
using LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Fields;
|
||||||
|
|
||||||
|
public sealed class MessageField
|
||||||
|
{
|
||||||
|
public required IType Type { get; init; }
|
||||||
|
public required string Name { get; init; }
|
||||||
|
public required int Id { get; init; }
|
||||||
|
|
||||||
|
public bool IsObsolete { get; init; }
|
||||||
|
public bool HasHasProp { get; init; }
|
||||||
|
|
||||||
|
public void WriteTo(TextWriter writer, TopLevel topLevel, bool isOneOf)
|
||||||
|
{
|
||||||
|
if (HasHasProp && !isOneOf && Type is not Repeated)
|
||||||
|
{
|
||||||
|
writer.Write("optional ");
|
||||||
|
}
|
||||||
|
|
||||||
|
Protobuf.WriteTypeNameTo(writer, Type, topLevel);
|
||||||
|
writer.Write(' ');
|
||||||
|
writer.Write(Name);
|
||||||
|
writer.Write(" = ");
|
||||||
|
writer.Write(Id);
|
||||||
|
|
||||||
|
if (IsObsolete)
|
||||||
|
{
|
||||||
|
writer.Write(" [deprecated = true]");
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.WriteLine(';');
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,60 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
using System.CodeDom.Compiler;
|
||||||
|
using LibProtodec.Models.TopLevels;
|
||||||
|
using LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Fields;
|
||||||
|
|
||||||
|
public sealed class ServiceMethod
|
||||||
|
{
|
||||||
|
public required string Name { get; init; }
|
||||||
|
public required IType RequestType { get; init; }
|
||||||
|
public required IType ResponseType { get; init; }
|
||||||
|
|
||||||
|
public bool IsRequestStreamed { get; init; }
|
||||||
|
public bool IsResponseStreamed { get; init; }
|
||||||
|
public bool IsObsolete { get; init; }
|
||||||
|
|
||||||
|
public void WriteTo(IndentedTextWriter writer, TopLevel topLevel)
|
||||||
|
{
|
||||||
|
writer.Write("rpc ");
|
||||||
|
writer.Write(Name);
|
||||||
|
writer.Write(" (");
|
||||||
|
|
||||||
|
if (IsRequestStreamed)
|
||||||
|
{
|
||||||
|
writer.Write("stream ");
|
||||||
|
}
|
||||||
|
|
||||||
|
Protobuf.WriteTypeNameTo(writer, RequestType, topLevel);
|
||||||
|
writer.Write(") returns (");
|
||||||
|
|
||||||
|
if (IsResponseStreamed)
|
||||||
|
{
|
||||||
|
writer.Write("stream ");
|
||||||
|
}
|
||||||
|
|
||||||
|
Protobuf.WriteTypeNameTo(writer, ResponseType, topLevel);
|
||||||
|
writer.Write(')');
|
||||||
|
|
||||||
|
if (IsObsolete)
|
||||||
|
{
|
||||||
|
writer.WriteLine(" {");
|
||||||
|
writer.Indent++;
|
||||||
|
|
||||||
|
Protobuf.WriteOptionTo(writer, "deprecated", "true");
|
||||||
|
|
||||||
|
writer.Indent--;
|
||||||
|
writer.WriteLine('}');
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
writer.WriteLine(';');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,105 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
using System.CodeDom.Compiler;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.IO;
|
||||||
|
using System.Linq;
|
||||||
|
using LibProtodec.Models.TopLevels;
|
||||||
|
using LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models;
|
||||||
|
|
||||||
|
public sealed class Protobuf
|
||||||
|
{
|
||||||
|
private HashSet<string>? _imports;
|
||||||
|
public HashSet<string> Imports =>
|
||||||
|
_imports ??= [];
|
||||||
|
|
||||||
|
public readonly List<TopLevel> TopLevels = [];
|
||||||
|
|
||||||
|
public string? Edition { get; set; }
|
||||||
|
public string? AssemblyName { get; init; }
|
||||||
|
public string? Namespace { get; init; }
|
||||||
|
|
||||||
|
public string FileName =>
|
||||||
|
$"{TopLevels.FirstOrDefault()?.Name}.proto";
|
||||||
|
|
||||||
|
public void WriteTo(IndentedTextWriter writer)
|
||||||
|
{
|
||||||
|
writer.WriteLine("// Decompiled with protodec");
|
||||||
|
|
||||||
|
if (AssemblyName is not null)
|
||||||
|
{
|
||||||
|
writer.Write("// Assembly: ");
|
||||||
|
writer.WriteLine(AssemblyName);
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.WriteLine();
|
||||||
|
writer.WriteLine(
|
||||||
|
Edition is null
|
||||||
|
? """syntax = "proto3";"""
|
||||||
|
: $"""edition = "{Edition}";""");
|
||||||
|
|
||||||
|
if (_imports is not null)
|
||||||
|
{
|
||||||
|
writer.WriteLine();
|
||||||
|
|
||||||
|
foreach (string import in _imports)
|
||||||
|
{
|
||||||
|
writer.Write("import \"");
|
||||||
|
writer.Write(import);
|
||||||
|
writer.WriteLine("\";");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Namespace is not null)
|
||||||
|
{
|
||||||
|
writer.WriteLine();
|
||||||
|
WriteOptionTo(writer, "csharp_namespace", Namespace, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (TopLevel topLevel in TopLevels)
|
||||||
|
{
|
||||||
|
writer.WriteLine();
|
||||||
|
topLevel.WriteTo(writer);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void WriteOptionTo(TextWriter writer, string name, string value, bool quoteValue = false)
|
||||||
|
{
|
||||||
|
writer.Write("option ");
|
||||||
|
writer.Write(name);
|
||||||
|
writer.Write(" = ");
|
||||||
|
|
||||||
|
if (quoteValue)
|
||||||
|
{
|
||||||
|
writer.Write('\"');
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.Write(value);
|
||||||
|
|
||||||
|
if (quoteValue)
|
||||||
|
{
|
||||||
|
writer.Write('\"');
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.WriteLine(';');
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void WriteTypeNameTo(TextWriter writer, IType type, TopLevel topLevel)
|
||||||
|
{
|
||||||
|
if (type is TopLevel { Parent: not null } typeTopLevel && typeTopLevel.Parent != topLevel)
|
||||||
|
{
|
||||||
|
writer.Write(
|
||||||
|
typeTopLevel.QualifyName(topLevel));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
writer.Write(type.Name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,67 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
global using Enum = LibProtodec.Models.TopLevels.Enum;
|
||||||
|
using System.CodeDom.Compiler;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using LibProtodec.Models.Fields;
|
||||||
|
using LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.TopLevels;
|
||||||
|
|
||||||
|
public sealed class Enum : TopLevel, INestableType
|
||||||
|
{
|
||||||
|
public readonly List<EnumField> Fields = [];
|
||||||
|
|
||||||
|
public override void WriteTo(IndentedTextWriter writer)
|
||||||
|
{
|
||||||
|
writer.Write("enum ");
|
||||||
|
writer.Write(this.Name);
|
||||||
|
writer.WriteLine(" {");
|
||||||
|
writer.Indent++;
|
||||||
|
|
||||||
|
if (ContainsDuplicateField)
|
||||||
|
{
|
||||||
|
Protobuf.WriteOptionTo(writer, "allow_alias", "true");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this.IsObsolete)
|
||||||
|
{
|
||||||
|
Protobuf.WriteOptionTo(writer, "deprecated", "true");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (IsClosed)
|
||||||
|
{
|
||||||
|
Protobuf.WriteOptionTo(writer, "features.enum_type", "CLOSED");
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (EnumField field in Fields)
|
||||||
|
{
|
||||||
|
field.WriteTo(writer);
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.Indent--;
|
||||||
|
writer.Write('}');
|
||||||
|
}
|
||||||
|
|
||||||
|
public bool IsClosed { get; set; }
|
||||||
|
|
||||||
|
private bool ContainsDuplicateField
|
||||||
|
{
|
||||||
|
get
|
||||||
|
{
|
||||||
|
if (Fields.Count < 2)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
HashSet<int> set = [];
|
||||||
|
foreach (EnumField field in Fields)
|
||||||
|
if (!set.Add(field.Id))
|
||||||
|
return true;
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,69 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
using System.CodeDom.Compiler;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using LibProtodec.Models.Fields;
|
||||||
|
using LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.TopLevels;
|
||||||
|
|
||||||
|
public sealed class Message : TopLevel, INestableType
|
||||||
|
{
|
||||||
|
public readonly Dictionary<string, int[]> OneOfs = [];
|
||||||
|
public readonly Dictionary<int, MessageField> Fields = [];
|
||||||
|
public readonly Dictionary<string, INestableType> Nested = [];
|
||||||
|
|
||||||
|
public override void WriteTo(IndentedTextWriter writer)
|
||||||
|
{
|
||||||
|
writer.Write("message ");
|
||||||
|
writer.Write(this.Name);
|
||||||
|
writer.WriteLine(" {");
|
||||||
|
writer.Indent++;
|
||||||
|
|
||||||
|
if (this.IsObsolete)
|
||||||
|
{
|
||||||
|
Protobuf.WriteOptionTo(writer, "deprecated", "true");
|
||||||
|
}
|
||||||
|
|
||||||
|
int[] oneOfs = OneOfs.SelectMany(static oneOf => oneOf.Value).ToArray();
|
||||||
|
|
||||||
|
foreach (MessageField field in Fields.Values)
|
||||||
|
{
|
||||||
|
if (oneOfs.Contains(field.Id))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
field.WriteTo(writer, this, isOneOf: false);
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach ((string name, int[] fieldIds) in OneOfs)
|
||||||
|
{
|
||||||
|
// ReSharper disable once StringLiteralTypo
|
||||||
|
writer.Write("oneof ");
|
||||||
|
writer.Write(name);
|
||||||
|
writer.WriteLine(" {");
|
||||||
|
writer.Indent++;
|
||||||
|
|
||||||
|
foreach (int fieldId in fieldIds)
|
||||||
|
{
|
||||||
|
Fields[fieldId].WriteTo(writer, this, isOneOf: true);
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.Indent--;
|
||||||
|
writer.WriteLine('}');
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (INestableType nested in Nested.Values)
|
||||||
|
{
|
||||||
|
nested.WriteTo(writer);
|
||||||
|
writer.WriteLine();
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.Indent--;
|
||||||
|
writer.Write('}');
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,37 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
using System.CodeDom.Compiler;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using LibProtodec.Models.Fields;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.TopLevels;
|
||||||
|
|
||||||
|
public sealed class Service : TopLevel
|
||||||
|
{
|
||||||
|
public readonly List<ServiceMethod> Methods = [];
|
||||||
|
|
||||||
|
public override void WriteTo(IndentedTextWriter writer)
|
||||||
|
{
|
||||||
|
writer.Write("service ");
|
||||||
|
writer.Write(this.Name);
|
||||||
|
writer.WriteLine(" {");
|
||||||
|
writer.Indent++;
|
||||||
|
|
||||||
|
if (this.IsObsolete)
|
||||||
|
{
|
||||||
|
Protobuf.WriteOptionTo(writer, "deprecated", "true");
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (ServiceMethod method in Methods)
|
||||||
|
{
|
||||||
|
method.WriteTo(writer, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.Indent--;
|
||||||
|
writer.Write('}');
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,35 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
using System.CodeDom.Compiler;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.TopLevels;
|
||||||
|
|
||||||
|
public abstract class TopLevel
|
||||||
|
{
|
||||||
|
public required string Name { get; init; }
|
||||||
|
public bool IsObsolete { get; init; }
|
||||||
|
public Protobuf? Protobuf { get; set; }
|
||||||
|
public TopLevel? Parent { get; set; }
|
||||||
|
|
||||||
|
public string QualifyName(TopLevel topLevel)
|
||||||
|
{
|
||||||
|
List<string> names = [Name];
|
||||||
|
|
||||||
|
TopLevel? parent = Parent;
|
||||||
|
while (parent is not null && parent != topLevel)
|
||||||
|
{
|
||||||
|
names.Add(parent.Name);
|
||||||
|
parent = parent.Parent;
|
||||||
|
}
|
||||||
|
|
||||||
|
names.Reverse();
|
||||||
|
return string.Join('.', names);
|
||||||
|
}
|
||||||
|
|
||||||
|
public abstract void WriteTo(IndentedTextWriter writer);
|
||||||
|
}
|
|
@ -0,0 +1,14 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
public interface INestableType : IType
|
||||||
|
{
|
||||||
|
Protobuf? Protobuf { get; }
|
||||||
|
|
||||||
|
void WriteTo(System.CodeDom.Compiler.IndentedTextWriter writer);
|
||||||
|
}
|
|
@ -0,0 +1,18 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
public interface IType
|
||||||
|
{
|
||||||
|
string Name { get; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public sealed class External(string typeName) : IType
|
||||||
|
{
|
||||||
|
public string Name =>
|
||||||
|
typeName;
|
||||||
|
}
|
|
@ -0,0 +1,13 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
public sealed class Map(IType typeKey, IType typeVal) : IType
|
||||||
|
{
|
||||||
|
public string Name =>
|
||||||
|
$"map<{typeKey.Name}, {typeVal.Name}>";
|
||||||
|
}
|
|
@ -0,0 +1,13 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
public sealed class Repeated(IType type) : IType
|
||||||
|
{
|
||||||
|
public string Name =>
|
||||||
|
$"repeated {type.Name}";
|
||||||
|
}
|
|
@ -0,0 +1,27 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
// ReSharper disable StringLiteralTypo
|
||||||
|
public static class Scalar
|
||||||
|
{
|
||||||
|
public static readonly IType Bool = new External("bool");
|
||||||
|
public static readonly IType Bytes = new External("bytes");
|
||||||
|
public static readonly IType Double = new External("double");
|
||||||
|
public static readonly IType Fixed32 = new External("fixed32");
|
||||||
|
public static readonly IType Fixed64 = new External("fixed64");
|
||||||
|
public static readonly IType Float = new External("float");
|
||||||
|
public static readonly IType Int32 = new External("int32");
|
||||||
|
public static readonly IType Int64 = new External("int64");
|
||||||
|
public static readonly IType SFixed32 = new External("sfixed32");
|
||||||
|
public static readonly IType SFixed64 = new External("sfixed64");
|
||||||
|
public static readonly IType SInt32 = new External("sint32");
|
||||||
|
public static readonly IType SInt64 = new External("sint64");
|
||||||
|
public static readonly IType String = new External("string");
|
||||||
|
public static readonly IType UInt32 = new External("uint32");
|
||||||
|
public static readonly IType UInt64 = new External("uint64");
|
||||||
|
}
|
|
@ -0,0 +1,39 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec.Models.Types;
|
||||||
|
|
||||||
|
public static class WellKnown
|
||||||
|
{
|
||||||
|
public static readonly IType Any = new External("google.protobuf.Any");
|
||||||
|
public static readonly IType Api = new External("google.protobuf.Api");
|
||||||
|
public static readonly IType BoolValue = new External("google.protobuf.BoolValue");
|
||||||
|
public static readonly IType BytesValue = new External("google.protobuf.BytesValue");
|
||||||
|
public static readonly IType DoubleValue = new External("google.protobuf.DoubleValue");
|
||||||
|
public static readonly IType Duration = new External("google.protobuf.Duration");
|
||||||
|
public static readonly IType Empty = new External("google.protobuf.Empty");
|
||||||
|
public static readonly IType Enum = new External("google.protobuf.Enum");
|
||||||
|
public static readonly IType EnumValue = new External("google.protobuf.EnumValue");
|
||||||
|
public static readonly IType Field = new External("google.protobuf.Field");
|
||||||
|
public static readonly IType FieldMask = new External("google.protobuf.FieldMask");
|
||||||
|
public static readonly IType FloatValue = new External("google.protobuf.FloatValue");
|
||||||
|
public static readonly IType Int32Value = new External("google.protobuf.Int32Value");
|
||||||
|
public static readonly IType Int64Value = new External("google.protobuf.Int64Value");
|
||||||
|
public static readonly IType ListValue = new External("google.protobuf.ListValue");
|
||||||
|
public static readonly IType Method = new External("google.protobuf.Method");
|
||||||
|
public static readonly IType Mixin = new External("google.protobuf.Mixin");
|
||||||
|
public static readonly IType NullValue = new External("google.protobuf.NullValue");
|
||||||
|
public static readonly IType Option = new External("google.protobuf.Option");
|
||||||
|
public static readonly IType SourceContext = new External("google.protobuf.SourceContext");
|
||||||
|
public static readonly IType StringValue = new External("google.protobuf.StringValue");
|
||||||
|
public static readonly IType Struct = new External("google.protobuf.Struct");
|
||||||
|
public static readonly IType Syntax = new External("google.protobuf.Syntax");
|
||||||
|
public static readonly IType Timestamp = new External("google.protobuf.Timestamp");
|
||||||
|
public static readonly IType Type = new External("google.protobuf.Type");
|
||||||
|
public static readonly IType UInt32Value = new External("google.protobuf.UInt32Value");
|
||||||
|
public static readonly IType UInt64Value = new External("google.protobuf.UInt64Value");
|
||||||
|
public static readonly IType Value = new External("google.protobuf.Value");
|
||||||
|
}
|
|
@ -0,0 +1,16 @@
|
||||||
|
// Copyright © 2024 Xpl0itR
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
namespace LibProtodec;
|
||||||
|
|
||||||
|
[System.Flags]
|
||||||
|
public enum ParserOptions
|
||||||
|
{
|
||||||
|
None = 0,
|
||||||
|
SkipEnums = 1,
|
||||||
|
IncludePropertiesWithoutNonUserCodeAttribute = 2,
|
||||||
|
IncludeServiceMethodsWithoutGeneratedCodeAttribute = 4,
|
||||||
|
}
|
|
@ -1,47 +0,0 @@
|
||||||
// Copyright © 2023-2024 Xpl0itR
|
|
||||||
//
|
|
||||||
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
||||||
|
|
||||||
using System.CodeDom.Compiler;
|
|
||||||
using System.IO;
|
|
||||||
|
|
||||||
namespace LibProtodec;
|
|
||||||
|
|
||||||
public abstract class Protobuf
|
|
||||||
{
|
|
||||||
public required string Name { get; init; }
|
|
||||||
|
|
||||||
public string? AssemblyName { get; init; }
|
|
||||||
public string? Namespace { get; init; }
|
|
||||||
|
|
||||||
public abstract void WriteFileTo(IndentedTextWriter writer);
|
|
||||||
|
|
||||||
public abstract void WriteTo(IndentedTextWriter writer);
|
|
||||||
|
|
||||||
protected void WritePreambleTo(TextWriter writer) =>
|
|
||||||
WritePreambleTo(writer, AssemblyName, Namespace);
|
|
||||||
|
|
||||||
// ReSharper disable once MethodOverloadWithOptionalParameter
|
|
||||||
public static void WritePreambleTo(TextWriter writer, string? assemblyName = null, string? @namespace = null)
|
|
||||||
{
|
|
||||||
writer.WriteLine("// Decompiled with protodec");
|
|
||||||
|
|
||||||
if (assemblyName is not null)
|
|
||||||
{
|
|
||||||
writer.Write("// Assembly: ");
|
|
||||||
writer.WriteLine(assemblyName);
|
|
||||||
}
|
|
||||||
|
|
||||||
writer.WriteLine();
|
|
||||||
writer.WriteLine("""syntax = "proto3";""");
|
|
||||||
writer.WriteLine();
|
|
||||||
|
|
||||||
if (@namespace is not null)
|
|
||||||
{
|
|
||||||
writer.WriteLine($"""option csharp_namespace = "{@namespace}";""");
|
|
||||||
writer.WriteLine();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,4 +1,4 @@
|
||||||
// Copyright © 2023-2024 Xpl0itR
|
// Copyright © 2023-2024 Xpl0itR
|
||||||
//
|
//
|
||||||
// This Source Code Form is subject to the terms of the Mozilla Public
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
@ -7,89 +7,78 @@
|
||||||
using System;
|
using System;
|
||||||
using System.CodeDom.Compiler;
|
using System.CodeDom.Compiler;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
|
using System.Diagnostics;
|
||||||
using System.Diagnostics.CodeAnalysis;
|
using System.Diagnostics.CodeAnalysis;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using System.Reflection;
|
using System.Reflection;
|
||||||
using SystemEx;
|
using SystemEx;
|
||||||
using SystemEx.Collections;
|
|
||||||
using CommunityToolkit.Diagnostics;
|
using CommunityToolkit.Diagnostics;
|
||||||
|
using LibProtodec.Models;
|
||||||
|
using LibProtodec.Models.Fields;
|
||||||
|
using LibProtodec.Models.TopLevels;
|
||||||
|
using LibProtodec.Models.Types;
|
||||||
|
|
||||||
namespace LibProtodec;
|
namespace LibProtodec;
|
||||||
|
|
||||||
public delegate bool LookupFunc(string key, [MaybeNullWhen(false)] out string value);
|
public delegate bool TypeLookupFunc(Type type, [NotNullWhen(true)] out IType? fieldType, out string? import);
|
||||||
|
public delegate bool NameLookupFunc(string name, [MaybeNullWhen(false)] out string translatedName);
|
||||||
|
|
||||||
public sealed class ProtodecContext
|
public sealed class ProtodecContext
|
||||||
{
|
{
|
||||||
private readonly Dictionary<string, Protobuf> _protobufs = [];
|
private const BindingFlags PublicStatic = BindingFlags.Public | BindingFlags.Static;
|
||||||
private readonly HashSet<string> _currentDescent = [];
|
private const BindingFlags PublicInstanceDeclared = BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly;
|
||||||
|
|
||||||
public LookupFunc? CustomTypeLookup { get; init; }
|
private readonly Dictionary<string, TopLevel> _parsed = [];
|
||||||
|
|
||||||
public LookupFunc? CustomNameLookup { get; init; }
|
public readonly List<Protobuf> Protobufs = [];
|
||||||
|
|
||||||
public IReadOnlyDictionary<string, Protobuf> Protobufs =>
|
public NameLookupFunc? NameLookup { get; set; }
|
||||||
_protobufs;
|
|
||||||
|
public TypeLookupFunc TypeLookup { get; set; } =
|
||||||
|
LookupScalarAndWellKnownTypes;
|
||||||
|
|
||||||
public void WriteAllTo(IndentedTextWriter writer)
|
public void WriteAllTo(IndentedTextWriter writer)
|
||||||
{
|
{
|
||||||
Protobuf.WritePreambleTo(writer);
|
writer.WriteLine("// Decompiled with protodec");
|
||||||
|
writer.WriteLine();
|
||||||
|
writer.WriteLine("""syntax = "proto3";""");
|
||||||
|
writer.WriteLine();
|
||||||
|
|
||||||
foreach (Protobuf proto in _protobufs.Values)
|
foreach (TopLevel topLevel in Protobufs.SelectMany(static proto => proto.TopLevels))
|
||||||
{
|
{
|
||||||
proto.WriteTo(writer);
|
topLevel.WriteTo(writer);
|
||||||
writer.WriteLine();
|
writer.WriteLine();
|
||||||
writer.WriteLine();
|
writer.WriteLine();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void ParseMessage(Type type, bool skipEnums = false, bool skipPropertiesWithoutProtocAttribute = false)
|
public Message ParseMessage(Type messageClass, ParserOptions options = ParserOptions.None)
|
||||||
{
|
{
|
||||||
Guard.IsTrue(type.IsClass);
|
Guard.IsTrue(messageClass is { IsClass: true, IsSealed: true });
|
||||||
|
|
||||||
ParseMessageInternal(type, skipEnums, skipPropertiesWithoutProtocAttribute, null);
|
if (_parsed.TryGetValue(messageClass.FullName ?? messageClass.Name, out TopLevel? parsedMessage))
|
||||||
_currentDescent.Clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
public void ParseEnum(Type type)
|
|
||||||
{
|
{
|
||||||
Guard.IsTrue(type.IsEnum);
|
return (Message)parsedMessage;
|
||||||
|
|
||||||
ParseEnumInternal(type, null);
|
|
||||||
_currentDescent.Clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
private bool IsParsed(Type type, Message? parentMessage, out Dictionary<string, Protobuf> protobufs)
|
|
||||||
{
|
|
||||||
protobufs = parentMessage is not null && type.IsNested
|
|
||||||
? parentMessage.Nested
|
|
||||||
: _protobufs;
|
|
||||||
|
|
||||||
return protobufs.ContainsKey(type.Name)
|
|
||||||
|| !_currentDescent.Add(type.Name);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void ParseMessageInternal(Type messageClass, bool skipEnums, bool skipPropertiesWithoutProtocAttribute, Message? parentMessage)
|
|
||||||
{
|
|
||||||
if (IsParsed(messageClass, parentMessage, out Dictionary<string, Protobuf> protobufs))
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Message message = new()
|
Message message = new()
|
||||||
{
|
{
|
||||||
Name = TranslateProtobufName(messageClass.Name),
|
Name = TranslateTypeName(messageClass),
|
||||||
AssemblyName = messageClass.Assembly.FullName,
|
IsObsolete = HasObsoleteAttribute(messageClass.GetCustomAttributesData())
|
||||||
Namespace = messageClass.Namespace
|
|
||||||
};
|
};
|
||||||
|
_parsed.Add(messageClass.FullName ?? messageClass.Name, message);
|
||||||
|
|
||||||
FieldInfo[] idFields = messageClass.GetFields(BindingFlags.Public | BindingFlags.Static);
|
Protobuf protobuf = GetProtobuf(messageClass, message, options);
|
||||||
PropertyInfo[] properties = messageClass.GetProperties(BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly);
|
|
||||||
|
FieldInfo[] idFields = messageClass.GetFields(PublicStatic);
|
||||||
|
PropertyInfo[] properties = messageClass.GetProperties(PublicInstanceDeclared);
|
||||||
|
|
||||||
for (int pi = 0, fi = 0; pi < properties.Length; pi++, fi++)
|
for (int pi = 0, fi = 0; pi < properties.Length; pi++, fi++)
|
||||||
{
|
{
|
||||||
PropertyInfo property = properties[pi];
|
PropertyInfo property = properties[pi];
|
||||||
|
IList<CustomAttributeData> attributes = property.GetCustomAttributesData();
|
||||||
|
|
||||||
if ((skipPropertiesWithoutProtocAttribute && !HasProtocAttribute(property))
|
if (((options & ParserOptions.IncludePropertiesWithoutNonUserCodeAttribute) == 0 && !HasNonUserCodeAttribute(attributes))
|
||||||
|| property.GetMethod?.IsVirtual != false)
|
|| property.GetMethod?.IsVirtual != false)
|
||||||
{
|
{
|
||||||
fi--;
|
fi--;
|
||||||
|
@ -99,11 +88,10 @@ public sealed class ProtodecContext
|
||||||
Type propertyType = property.PropertyType;
|
Type propertyType = property.PropertyType;
|
||||||
|
|
||||||
// only OneOf enums are defined nested directly in the message class
|
// only OneOf enums are defined nested directly in the message class
|
||||||
if (propertyType.IsEnum
|
if (propertyType.IsEnum && propertyType.DeclaringType?.Name == messageClass.Name)
|
||||||
&& propertyType.DeclaringType?.Name == messageClass.Name)
|
|
||||||
{
|
{
|
||||||
string oneOfName = TranslateOneOfName(property.Name);
|
string oneOfName = TranslateOneOfPropName(property.Name);
|
||||||
int[] oneOfProtoFieldIds = propertyType.GetFields(BindingFlags.Public | BindingFlags.Static)
|
int[] oneOfProtoFieldIds = propertyType.GetFields(PublicStatic)
|
||||||
.Select(static field => (int)field.GetRawConstantValue()!)
|
.Select(static field => (int)field.GetRawConstantValue()!)
|
||||||
.Where(static id => id > 0)
|
.Where(static id => id > 0)
|
||||||
.ToArray();
|
.ToArray();
|
||||||
|
@ -118,126 +106,320 @@ public sealed class ProtodecContext
|
||||||
Guard.IsTrue(idField.IsLiteral);
|
Guard.IsTrue(idField.IsLiteral);
|
||||||
Guard.IsEqualTo(idField.FieldType.Name, nameof(Int32));
|
Guard.IsEqualTo(idField.FieldType.Name, nameof(Int32));
|
||||||
|
|
||||||
int msgFieldId = (int)idField.GetRawConstantValue()!;
|
bool msgFieldHasHasProp = false; // some field properties are immediately followed by an additional "Has" get-only boolean property
|
||||||
bool msgFieldIsOptional = false;
|
|
||||||
string msgFieldType = ParseFieldType(propertyType, skipEnums, skipPropertiesWithoutProtocAttribute, message);
|
|
||||||
string msgFieldName = TranslateMessageFieldName(property.Name);
|
|
||||||
|
|
||||||
// optional protobuf fields will generate an additional "Has" get-only boolean property immediately after the real property
|
|
||||||
if (properties.Length > pi + 1 && properties[pi + 1].PropertyType.Name == nameof(Boolean) && !properties[pi + 1].CanWrite)
|
if (properties.Length > pi + 1 && properties[pi + 1].PropertyType.Name == nameof(Boolean) && !properties[pi + 1].CanWrite)
|
||||||
{
|
{
|
||||||
msgFieldIsOptional = true;
|
msgFieldHasHasProp = true;
|
||||||
pi++;
|
pi++;
|
||||||
}
|
}
|
||||||
|
|
||||||
message.Fields.Add(msgFieldId, (msgFieldIsOptional, msgFieldType, msgFieldName));
|
MessageField field = new()
|
||||||
|
{
|
||||||
|
Type = ParseFieldType(propertyType, options, protobuf),
|
||||||
|
Name = TranslateMessageFieldName(property.Name),
|
||||||
|
Id = (int)idField.GetRawConstantValue()!,
|
||||||
|
IsObsolete = HasObsoleteAttribute(attributes),
|
||||||
|
HasHasProp = msgFieldHasHasProp
|
||||||
|
};
|
||||||
|
|
||||||
|
message.Fields.Add(field.Id, field);
|
||||||
}
|
}
|
||||||
|
|
||||||
protobufs.Add(message.Name, message);
|
return message;
|
||||||
}
|
}
|
||||||
|
|
||||||
private void ParseEnumInternal(Type enumEnum, Message? parentMessage)
|
public Enum ParseEnum(Type enumEnum, ParserOptions options = ParserOptions.None)
|
||||||
{
|
{
|
||||||
if (IsParsed(enumEnum, parentMessage, out Dictionary<string, Protobuf> protobufs))
|
Guard.IsTrue(enumEnum.IsEnum);
|
||||||
|
|
||||||
|
if (_parsed.TryGetValue(enumEnum.FullName ?? enumEnum.Name, out TopLevel? parsedEnum))
|
||||||
{
|
{
|
||||||
return;
|
return (Enum)parsedEnum;
|
||||||
}
|
}
|
||||||
|
|
||||||
Enum @enum = new()
|
Enum @enum = new()
|
||||||
{
|
{
|
||||||
Name = TranslateProtobufName(enumEnum.Name),
|
Name = TranslateTypeName(enumEnum),
|
||||||
AssemblyName = enumEnum.Assembly.FullName,
|
IsObsolete = HasObsoleteAttribute(enumEnum.GetCustomAttributesData())
|
||||||
Namespace = enumEnum.Namespace
|
|
||||||
};
|
};
|
||||||
|
_parsed.Add(enumEnum.FullName ?? enumEnum.Name, @enum);
|
||||||
|
|
||||||
foreach (FieldInfo field in enumEnum.GetFields(BindingFlags.Public | BindingFlags.Static))
|
Protobuf protobuf = GetProtobuf(enumEnum, @enum, options);
|
||||||
|
|
||||||
|
foreach (FieldInfo field in enumEnum.GetFields(PublicStatic))
|
||||||
{
|
{
|
||||||
int enumFieldId = (int)field.GetRawConstantValue()!;
|
@enum.Fields.Add(
|
||||||
string enumFieldName = TranslateEnumFieldName(field, @enum.Name);
|
new EnumField
|
||||||
|
{
|
||||||
@enum.Fields.Add(enumFieldId, enumFieldName);
|
Id = (int)field.GetRawConstantValue()!,
|
||||||
|
Name = TranslateEnumFieldName(field, @enum.Name),
|
||||||
|
IsObsolete = HasObsoleteAttribute(field.GetCustomAttributesData())
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
protobufs.Add(@enum.Name, @enum);
|
if (@enum.Fields.All(static field => field.Id != 0))
|
||||||
|
{
|
||||||
|
protobuf.Edition = "2023";
|
||||||
|
@enum.IsClosed = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
private string ParseFieldType(Type type, bool skipEnums, bool skipPropertiesWithoutProtocAttribute, Message message)
|
return @enum;
|
||||||
{
|
|
||||||
switch (type.Name)
|
|
||||||
{
|
|
||||||
case "ByteString":
|
|
||||||
return FieldTypeName.Bytes;
|
|
||||||
case nameof(String):
|
|
||||||
return FieldTypeName.String;
|
|
||||||
case nameof(Boolean):
|
|
||||||
return FieldTypeName.Bool;
|
|
||||||
case nameof(Double):
|
|
||||||
return FieldTypeName.Double;
|
|
||||||
case nameof(UInt32):
|
|
||||||
return FieldTypeName.UInt32;
|
|
||||||
case nameof(UInt64):
|
|
||||||
return FieldTypeName.UInt64;
|
|
||||||
case nameof(Int32):
|
|
||||||
return FieldTypeName.Int32;
|
|
||||||
case nameof(Int64):
|
|
||||||
return FieldTypeName.Int64;
|
|
||||||
case nameof(Single):
|
|
||||||
return FieldTypeName.Float;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public Service ParseService(Type serviceClass, ParserOptions options = ParserOptions.None)
|
||||||
|
{
|
||||||
|
Guard.IsTrue(serviceClass.IsClass);
|
||||||
|
|
||||||
|
bool? isClientClass = null;
|
||||||
|
if (serviceClass.IsAbstract)
|
||||||
|
{
|
||||||
|
if (serviceClass is { IsSealed: true, IsNested: false })
|
||||||
|
{
|
||||||
|
Type[] nested = serviceClass.GetNestedTypes();
|
||||||
|
serviceClass = nested.SingleOrDefault(static nested => nested is { IsAbstract: true, IsSealed: false })
|
||||||
|
?? nested.Single(static nested => nested is { IsClass: true, IsAbstract: false });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (serviceClass is { IsNested: true, IsAbstract: true, IsSealed: false })
|
||||||
|
{
|
||||||
|
isClientClass = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (serviceClass is { IsAbstract: false, IsNested: true, DeclaringType: not null })
|
||||||
|
{
|
||||||
|
isClientClass = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
Guard.IsNotNull(isClientClass);
|
||||||
|
|
||||||
|
if (_parsed.TryGetValue(serviceClass.DeclaringType!.FullName ?? serviceClass.DeclaringType!.Name, out TopLevel? parsedService))
|
||||||
|
{
|
||||||
|
return (Service)parsedService;
|
||||||
|
}
|
||||||
|
|
||||||
|
Service service = new()
|
||||||
|
{
|
||||||
|
Name = TranslateTypeName(serviceClass.DeclaringType),
|
||||||
|
IsObsolete = HasObsoleteAttribute(serviceClass.GetCustomAttributesData())
|
||||||
|
};
|
||||||
|
_parsed.Add(serviceClass.DeclaringType!.FullName ?? serviceClass.DeclaringType.Name, service);
|
||||||
|
|
||||||
|
Protobuf protobuf = NewProtobuf(serviceClass, service);
|
||||||
|
|
||||||
|
foreach (MethodInfo method in serviceClass.GetMethods(PublicInstanceDeclared))
|
||||||
|
{
|
||||||
|
IList<CustomAttributeData> attributes = method.GetCustomAttributesData();
|
||||||
|
if ((options & ParserOptions.IncludeServiceMethodsWithoutGeneratedCodeAttribute) == 0
|
||||||
|
&& !HasGeneratedCodeAttribute(attributes, "grpc_csharp_plugin"))
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
Type requestType, responseType, returnType = method.ReturnType;
|
||||||
|
bool streamReq, streamRes;
|
||||||
|
|
||||||
|
if (isClientClass.Value)
|
||||||
|
{
|
||||||
|
string returnTypeName = TranslateTypeName(returnType);
|
||||||
|
if (returnTypeName == "AsyncUnaryCall`1")
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
ParameterInfo[] parameters = method.GetParameters();
|
||||||
|
if (parameters.Length > 2)
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
Type firstParamType = parameters[0].ParameterType;
|
||||||
|
switch (returnType.GenericTypeArguments.Length)
|
||||||
|
{
|
||||||
|
case 2:
|
||||||
|
requestType = returnType.GenericTypeArguments[0];
|
||||||
|
responseType = returnType.GenericTypeArguments[1];
|
||||||
|
streamReq = true;
|
||||||
|
streamRes = returnTypeName == "AsyncDuplexStreamingCall`2";
|
||||||
|
break;
|
||||||
|
case 1:
|
||||||
|
requestType = firstParamType;
|
||||||
|
responseType = returnType.GenericTypeArguments[0];
|
||||||
|
streamReq = false;
|
||||||
|
streamRes = true;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
requestType = firstParamType;
|
||||||
|
responseType = returnType;
|
||||||
|
streamReq = false;
|
||||||
|
streamRes = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
ParameterInfo[] parameters = method.GetParameters();
|
||||||
|
Type firstParamType = parameters[0].ParameterType;
|
||||||
|
|
||||||
|
if (firstParamType.GenericTypeArguments.Length == 1)
|
||||||
|
{
|
||||||
|
streamReq = true;
|
||||||
|
requestType = firstParamType.GenericTypeArguments[0];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
streamReq = false;
|
||||||
|
requestType = firstParamType;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (returnType.GenericTypeArguments.Length == 1)
|
||||||
|
{
|
||||||
|
streamRes = false;
|
||||||
|
responseType = returnType.GenericTypeArguments[0];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
streamRes = true;
|
||||||
|
responseType = parameters[1].ParameterType.GenericTypeArguments[0];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
service.Methods.Add(
|
||||||
|
new ServiceMethod
|
||||||
|
{
|
||||||
|
Name = TranslateMethodName(method.Name),
|
||||||
|
IsObsolete = HasObsoleteAttribute(attributes),
|
||||||
|
RequestType = ParseFieldType(requestType, options, protobuf),
|
||||||
|
ResponseType = ParseFieldType(responseType, options, protobuf),
|
||||||
|
IsRequestStreamed = streamReq,
|
||||||
|
IsResponseStreamed = streamRes
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return service;
|
||||||
|
}
|
||||||
|
|
||||||
|
private IType ParseFieldType(Type type, ParserOptions options, Protobuf referencingProtobuf)
|
||||||
|
{
|
||||||
switch (type.GenericTypeArguments.Length)
|
switch (type.GenericTypeArguments.Length)
|
||||||
{
|
{
|
||||||
case 1:
|
case 1:
|
||||||
string t = ParseFieldType(type.GenericTypeArguments[0], skipEnums, skipPropertiesWithoutProtocAttribute, message);
|
return new Repeated(
|
||||||
return "repeated " + t;
|
ParseFieldType(type.GenericTypeArguments[0], options, referencingProtobuf));
|
||||||
case 2:
|
case 2:
|
||||||
string t1 = ParseFieldType(type.GenericTypeArguments[0], skipEnums, skipPropertiesWithoutProtocAttribute, message);
|
return new Map(
|
||||||
string t2 = ParseFieldType(type.GenericTypeArguments[1], skipEnums, skipPropertiesWithoutProtocAttribute, message);
|
ParseFieldType(type.GenericTypeArguments[0], options, referencingProtobuf),
|
||||||
return $"map<{t1}, {t2}>";
|
ParseFieldType(type.GenericTypeArguments[1], options, referencingProtobuf));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (CustomTypeLookup?.Invoke(type.Name, out string? fieldType) == true)
|
if (TypeLookup(type, out IType? fieldType, out string? import))
|
||||||
{
|
{
|
||||||
|
if (import is not null)
|
||||||
|
{
|
||||||
|
referencingProtobuf.Imports.Add(import);
|
||||||
|
}
|
||||||
|
|
||||||
return fieldType;
|
return fieldType;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (type.IsEnum)
|
if (type.IsEnum)
|
||||||
{
|
{
|
||||||
if (skipEnums)
|
if ((options & ParserOptions.SkipEnums) > 0)
|
||||||
{
|
{
|
||||||
return FieldTypeName.Int32;
|
return Scalar.Int32;
|
||||||
}
|
}
|
||||||
|
|
||||||
ParseEnumInternal(type, message);
|
fieldType = ParseEnum(type, options);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
ParseMessageInternal(type, skipEnums, skipPropertiesWithoutProtocAttribute, message);
|
fieldType = ParseMessage(type, options);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!type.IsNested)
|
Protobuf protobuf = ((INestableType)fieldType).Protobuf!;
|
||||||
|
if (referencingProtobuf != protobuf)
|
||||||
{
|
{
|
||||||
message.Imports.Add(type.Name);
|
referencingProtobuf.Imports.Add(protobuf.FileName);
|
||||||
}
|
}
|
||||||
|
|
||||||
return type.Name;
|
return fieldType;
|
||||||
}
|
}
|
||||||
|
|
||||||
private string TranslateProtobufName(string name) =>
|
private Protobuf NewProtobuf(Type topLevelType, TopLevel topLevel)
|
||||||
CustomNameLookup?.Invoke(name, out string? translatedName) == true
|
{
|
||||||
? translatedName
|
Protobuf protobuf = new()
|
||||||
: name;
|
{
|
||||||
|
AssemblyName = topLevelType.Assembly.FullName,
|
||||||
|
Namespace = topLevelType.Namespace
|
||||||
|
};
|
||||||
|
|
||||||
private string TranslateOneOfName(string oneOfEnumName) =>
|
topLevel.Protobuf = protobuf;
|
||||||
TranslateName(oneOfEnumName, out string translatedName)
|
protobuf.TopLevels.Add(topLevel);
|
||||||
? translatedName.TrimEnd("Case")
|
Protobufs.Add(protobuf);
|
||||||
: oneOfEnumName.TrimEnd("Case")
|
|
||||||
.ToSnakeCaseLower();
|
|
||||||
|
|
||||||
private string TranslateMessageFieldName(string fieldName) =>
|
return protobuf;
|
||||||
TranslateName(fieldName, out string translatedName)
|
}
|
||||||
|
|
||||||
|
private Protobuf GetProtobuf<T>(Type topLevelType, T topLevel, ParserOptions options)
|
||||||
|
where T : TopLevel, INestableType
|
||||||
|
{
|
||||||
|
Protobuf protobuf;
|
||||||
|
if (topLevelType.IsNested)
|
||||||
|
{
|
||||||
|
Type parent = topLevelType.DeclaringType!.DeclaringType!;
|
||||||
|
if (!_parsed.TryGetValue(parent.FullName ?? parent.Name, out TopLevel? parentTopLevel))
|
||||||
|
{
|
||||||
|
parentTopLevel = ParseMessage(parent, options);
|
||||||
|
}
|
||||||
|
|
||||||
|
protobuf = parentTopLevel.Protobuf!;
|
||||||
|
topLevel.Protobuf = protobuf;
|
||||||
|
topLevel.Parent = parentTopLevel;
|
||||||
|
|
||||||
|
((Message)parentTopLevel).Nested.Add(topLevelType.Name, topLevel);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
protobuf = NewProtobuf(topLevelType, topLevel);
|
||||||
|
}
|
||||||
|
|
||||||
|
return protobuf;
|
||||||
|
}
|
||||||
|
|
||||||
|
private string TranslateMethodName(string methodName) =>
|
||||||
|
NameLookup?.Invoke(methodName, out string? translatedName) == true
|
||||||
? translatedName
|
? translatedName
|
||||||
: fieldName.ToSnakeCaseLower();
|
: methodName;
|
||||||
|
|
||||||
|
private string TranslateOneOfPropName(string oneOfPropName)
|
||||||
|
{
|
||||||
|
if (NameLookup?.Invoke(oneOfPropName, out string? translatedName) != true)
|
||||||
|
{
|
||||||
|
if (IsBeebyted(oneOfPropName))
|
||||||
|
{
|
||||||
|
return oneOfPropName;
|
||||||
|
}
|
||||||
|
|
||||||
|
translatedName = oneOfPropName;
|
||||||
|
}
|
||||||
|
|
||||||
|
return translatedName!.TrimEnd("Case").ToSnakeCaseLower();
|
||||||
|
}
|
||||||
|
|
||||||
|
private string TranslateMessageFieldName(string fieldName)
|
||||||
|
{
|
||||||
|
if (NameLookup?.Invoke(fieldName, out string? translatedName) != true)
|
||||||
|
{
|
||||||
|
if (IsBeebyted(fieldName))
|
||||||
|
{
|
||||||
|
return fieldName;
|
||||||
|
}
|
||||||
|
|
||||||
|
translatedName = fieldName;
|
||||||
|
}
|
||||||
|
|
||||||
|
return translatedName!.ToSnakeCaseLower();
|
||||||
|
}
|
||||||
|
|
||||||
private string TranslateEnumFieldName(FieldInfo field, string enumName)
|
private string TranslateEnumFieldName(FieldInfo field, string enumName)
|
||||||
{
|
{
|
||||||
|
@ -250,9 +432,14 @@ public sealed class ProtodecContext
|
||||||
return originalName;
|
return originalName;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (TranslateName(field.Name, out string translatedName))
|
if (NameLookup?.Invoke(field.Name, out string? fieldName) != true)
|
||||||
{
|
{
|
||||||
return translatedName;
|
fieldName = field.Name;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!IsBeebyted(fieldName!))
|
||||||
|
{
|
||||||
|
fieldName = fieldName!.ToSnakeCaseUpper();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!IsBeebyted(enumName))
|
if (!IsBeebyted(enumName))
|
||||||
|
@ -260,26 +447,208 @@ public sealed class ProtodecContext
|
||||||
enumName = enumName.ToSnakeCaseUpper();
|
enumName = enumName.ToSnakeCaseUpper();
|
||||||
}
|
}
|
||||||
|
|
||||||
return enumName + '_' + field.Name.ToSnakeCaseUpper();
|
return enumName + '_' + fieldName;
|
||||||
}
|
}
|
||||||
|
|
||||||
private bool TranslateName(string name, out string translatedName)
|
private string TranslateTypeName(Type type)
|
||||||
{
|
{
|
||||||
if (CustomNameLookup?.Invoke(name, out translatedName!) == true)
|
if (NameLookup is null)
|
||||||
|
return type.Name;
|
||||||
|
|
||||||
|
string? fullName = type.FullName;
|
||||||
|
Guard.IsNotNull(fullName);
|
||||||
|
|
||||||
|
int genericArgs = fullName.IndexOf('[');
|
||||||
|
if (genericArgs != -1)
|
||||||
|
fullName = fullName[..genericArgs];
|
||||||
|
|
||||||
|
if (!NameLookup(fullName, out string? translatedName))
|
||||||
{
|
{
|
||||||
|
return type.Name;
|
||||||
|
}
|
||||||
|
|
||||||
|
int lastSlash = translatedName.LastIndexOf('/');
|
||||||
|
if (lastSlash != -1)
|
||||||
|
translatedName = translatedName[lastSlash..];
|
||||||
|
|
||||||
|
int lastDot = translatedName.LastIndexOf('.');
|
||||||
|
if (lastDot != -1)
|
||||||
|
translatedName = translatedName[lastDot..];
|
||||||
|
|
||||||
|
return translatedName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static bool LookupScalarAndWellKnownTypes(Type type, [NotNullWhen(true)] out IType? fieldType, out string? import)
|
||||||
|
{
|
||||||
|
switch (type.FullName)
|
||||||
|
{
|
||||||
|
case "System.String":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.String;
|
||||||
|
return true;
|
||||||
|
case "System.Boolean":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.Bool;
|
||||||
|
return true;
|
||||||
|
case "System.Double":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.Double;
|
||||||
|
return true;
|
||||||
|
case "System.UInt32":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.UInt32;
|
||||||
|
return true;
|
||||||
|
case "System.UInt64":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.UInt64;
|
||||||
|
return true;
|
||||||
|
case "System.Int32":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.Int32;
|
||||||
|
return true;
|
||||||
|
case "System.Int64":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.Int64;
|
||||||
|
return true;
|
||||||
|
case "System.Single":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.Float;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.ByteString":
|
||||||
|
import = null;
|
||||||
|
fieldType = Scalar.Bytes;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Any":
|
||||||
|
import = "google/protobuf/any.proto";
|
||||||
|
fieldType = WellKnown.Any;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Api":
|
||||||
|
import = "google/protobuf/api.proto";
|
||||||
|
fieldType = WellKnown.Api;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.BoolValue":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.BoolValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.BytesValue":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.BytesValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.DoubleValue":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.DoubleValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Duration":
|
||||||
|
import = "google/protobuf/duration.proto";
|
||||||
|
fieldType = WellKnown.Duration;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Empty":
|
||||||
|
import = "google/protobuf/empty.proto";
|
||||||
|
fieldType = WellKnown.Empty;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Enum":
|
||||||
|
import = "google/protobuf/type.proto";
|
||||||
|
fieldType = WellKnown.Enum;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.EnumValue":
|
||||||
|
import = "google/protobuf/type.proto";
|
||||||
|
fieldType = WellKnown.EnumValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Field":
|
||||||
|
import = "google/protobuf/type.proto";
|
||||||
|
fieldType = WellKnown.Field;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.FieldMask":
|
||||||
|
import = "google/protobuf/field_mask.proto";
|
||||||
|
fieldType = WellKnown.FieldMask;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.FloatValue":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.FloatValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Int32Value":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.Int32Value;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Int64Value":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.Int64Value;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.ListValue":
|
||||||
|
import = "google/protobuf/struct.proto";
|
||||||
|
fieldType = WellKnown.ListValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Method":
|
||||||
|
import = "google/protobuf/api.proto";
|
||||||
|
fieldType = WellKnown.Method;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Mixin":
|
||||||
|
import = "google/protobuf/api.proto";
|
||||||
|
fieldType = WellKnown.Mixin;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.NullValue":
|
||||||
|
import = "google/protobuf/struct.proto";
|
||||||
|
fieldType = WellKnown.NullValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Option":
|
||||||
|
import = "google/protobuf/type.proto";
|
||||||
|
fieldType = WellKnown.Option;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.SourceContext":
|
||||||
|
import = "google/protobuf/source_context.proto";
|
||||||
|
fieldType = WellKnown.SourceContext;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.StringValue":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.StringValue;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Struct":
|
||||||
|
import = "google/protobuf/struct.proto";
|
||||||
|
fieldType = WellKnown.Struct;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Syntax":
|
||||||
|
import = "google/protobuf/type.proto";
|
||||||
|
fieldType = WellKnown.Syntax;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Timestamp":
|
||||||
|
import = "google/protobuf/timestamp.proto";
|
||||||
|
fieldType = WellKnown.Timestamp;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Type":
|
||||||
|
import = "google/protobuf/type.proto";
|
||||||
|
fieldType = WellKnown.Type;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.UInt32Value":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.UInt32Value;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.UInt64Value":
|
||||||
|
import = "google/protobuf/wrappers.proto";
|
||||||
|
fieldType = WellKnown.UInt64Value;
|
||||||
|
return true;
|
||||||
|
case "Google.Protobuf.WellKnownTypes.Value":
|
||||||
|
import = "google/protobuf/struct.proto";
|
||||||
|
fieldType = WellKnown.Value;
|
||||||
return true;
|
return true;
|
||||||
}
|
|
||||||
|
|
||||||
translatedName = name;
|
default:
|
||||||
return IsBeebyted(name);
|
import = null;
|
||||||
|
fieldType = null;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReSharper disable once IdentifierTypo
|
// ReSharper disable once IdentifierTypo
|
||||||
private static bool IsBeebyted(string name) =>
|
private static bool IsBeebyted(string name) =>
|
||||||
name.Length == 11 && name.CountUpper() == 11;
|
name.Length == 11 && name.CountUpper() == 11;
|
||||||
|
|
||||||
private static bool HasProtocAttribute(MemberInfo member) =>
|
private static bool HasGeneratedCodeAttribute(IEnumerable<CustomAttributeData> attributes, string tool) =>
|
||||||
member.GetCustomAttributesData()
|
attributes.Any(attr => attr.AttributeType.Name == nameof(GeneratedCodeAttribute)
|
||||||
.Any(static attr => attr.AttributeType.Name == nameof(GeneratedCodeAttribute)
|
&& attr.ConstructorArguments[0].Value as string == tool);
|
||||||
&& attr.ConstructorArguments[0].Value as string == "protoc");
|
|
||||||
|
private static bool HasNonUserCodeAttribute(IEnumerable<CustomAttributeData> attributes) =>
|
||||||
|
attributes.Any(static attr => attr.AttributeType.Name == nameof(DebuggerNonUserCodeAttribute));
|
||||||
|
|
||||||
|
private static bool HasObsoleteAttribute(IEnumerable<CustomAttributeData> attributes) =>
|
||||||
|
attributes.Any(static attr => attr.AttributeType.Name == nameof(ObsoleteAttribute));
|
||||||
}
|
}
|
|
@ -1,8 +1,10 @@
|
||||||
using System;
|
using System;
|
||||||
using System.CodeDom.Compiler;
|
using System.CodeDom.Compiler;
|
||||||
|
using System.Collections.Generic;
|
||||||
using System.IO;
|
using System.IO;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using LibProtodec;
|
using LibProtodec;
|
||||||
|
using LibProtodec.Models;
|
||||||
|
|
||||||
const string indent = " ";
|
const string indent = " ";
|
||||||
const string help = """
|
const string help = """
|
||||||
|
@ -11,8 +13,11 @@ const string help = """
|
||||||
target_assembly_path Either the path to the target assembly or a directory of assemblies, all of which be parsed.
|
target_assembly_path Either the path to the target assembly or a directory of assemblies, all of which be parsed.
|
||||||
out_path An existing directory to output into individual files, otherwise output to a single file.
|
out_path An existing directory to output into individual files, otherwise output to a single file.
|
||||||
Options:
|
Options:
|
||||||
|
--parse_service_servers Parses gRPC service definitions from server classes.
|
||||||
|
--parse_service_clients Parses gRPC service definitions from client classes.
|
||||||
--skip_enums Skip parsing enums and replace references to them with int32.
|
--skip_enums Skip parsing enums and replace references to them with int32.
|
||||||
--skip_properties_without_protoc_attribute Skip properties that aren't decorated with `GeneratedCode("protoc")` when parsing
|
--include_properties_without_non_user_code_attribute Includes properties that aren't decorated with `DebuggerNonUserCode` when parsing.
|
||||||
|
--include_service_methods_without_generated_code_attribute Includes methods that aren't decorated with `GeneratedCode("grpc_csharp_plugin")` when parsing gRPC services.
|
||||||
""";
|
""";
|
||||||
|
|
||||||
if (args.Length < 2)
|
if (args.Length < 2)
|
||||||
|
@ -21,29 +26,63 @@ if (args.Length < 2)
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
string assemblyPath = args[0];
|
string assembly = args[0];
|
||||||
string outPath = Path.GetFullPath(args[1]);
|
string outPath = Path.GetFullPath(args[1]);
|
||||||
bool skipEnums = args.Contains("--skip_enums");
|
ParserOptions options = ParserOptions.None;
|
||||||
bool skipPropertiesWithoutProtocAttribute = args.Contains("--skip_properties_without_protoc_attribute");
|
|
||||||
|
|
||||||
using AssemblyInspector inspector = new(assemblyPath);
|
if (args.Contains("--skip_enums"))
|
||||||
|
options |= ParserOptions.SkipEnums;
|
||||||
|
|
||||||
|
if (args.Contains("--include_properties_without_non_user_code_attribute"))
|
||||||
|
options |= ParserOptions.IncludePropertiesWithoutNonUserCodeAttribute;
|
||||||
|
|
||||||
|
if (args.Contains("--include_service_methods_without_generated_code_attribute"))
|
||||||
|
options |= ParserOptions.IncludeServiceMethodsWithoutGeneratedCodeAttribute;
|
||||||
|
|
||||||
|
using AssemblyInspector inspector = new(assembly);
|
||||||
ProtodecContext ctx = new();
|
ProtodecContext ctx = new();
|
||||||
|
|
||||||
foreach (Type message in inspector.GetProtobufMessageTypes())
|
foreach (Type message in inspector.GetProtobufMessageTypes())
|
||||||
{
|
{
|
||||||
ctx.ParseMessage(message, skipEnums, skipPropertiesWithoutProtocAttribute);
|
ctx.ParseMessage(message, options);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (args.Contains("--parse_service_servers"))
|
||||||
|
{
|
||||||
|
foreach (Type service in inspector.GetProtobufServiceServerTypes())
|
||||||
|
{
|
||||||
|
ctx.ParseService(service, options);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (args.Contains("--parse_service_clients"))
|
||||||
|
{
|
||||||
|
foreach (Type service in inspector.GetProtobufServiceClientTypes())
|
||||||
|
{
|
||||||
|
ctx.ParseService(service, options);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (Directory.Exists(outPath))
|
if (Directory.Exists(outPath))
|
||||||
{
|
{
|
||||||
foreach (Protobuf proto in ctx.Protobufs.Values)
|
HashSet<string> writtenFiles = [];
|
||||||
{
|
|
||||||
string protoPath = Path.Join(outPath, proto.Name + ".proto");
|
|
||||||
|
|
||||||
using StreamWriter streamWriter = new(protoPath);
|
foreach (Protobuf protobuf in ctx.Protobufs)
|
||||||
|
{
|
||||||
|
// This workaround stops files from being overwritten in the case of a naming conflict,
|
||||||
|
// however the actual conflict will still have to be resolved manually
|
||||||
|
string fileName = protobuf.FileName;
|
||||||
|
while (!writtenFiles.Add(fileName))
|
||||||
|
{
|
||||||
|
fileName = '_' + fileName;
|
||||||
|
}
|
||||||
|
|
||||||
|
string protobufPath = Path.Join(outPath, fileName);
|
||||||
|
|
||||||
|
using StreamWriter streamWriter = new(protobufPath);
|
||||||
using IndentedTextWriter indentWriter = new(streamWriter, indent);
|
using IndentedTextWriter indentWriter = new(streamWriter, indent);
|
||||||
|
|
||||||
proto.WriteFileTo(indentWriter);
|
protobuf.WriteTo(indentWriter);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
|
|
Loading…
Reference in New Issue