From 1de6350c7dcf6c4e33e6d308394c0cfa9a305421 Mon Sep 17 00:00:00 2001 From: CodeLiturgy Date: Mon, 5 Sep 2022 02:01:17 +0100 Subject: [PATCH] Prepare getone template --- .../Sources/EfMethods/EfMethodsGenerator.cs | 26 ++++++++++--------- 1 file changed, 14 insertions(+), 12 deletions(-) diff --git a/src/BlueWest.MapTo/Sources/EfMethods/EfMethodsGenerator.cs b/src/BlueWest.MapTo/Sources/EfMethods/EfMethodsGenerator.cs index ebc005d..0ab9f46 100644 --- a/src/BlueWest.MapTo/Sources/EfMethods/EfMethodsGenerator.cs +++ b/src/BlueWest.MapTo/Sources/EfMethods/EfMethodsGenerator.cs @@ -50,20 +50,13 @@ namespace MapTo private static void AddGeneratedExtensions(GeneratorExecutionContext context, Compilation compilation, IEnumerable candidateMembers, SourceGenerationOptions options) { - + string addSourceTemplate = GetCsxMethodTemplate(context, "AddToEntityTemplate"); + string updateSourceTemplate = GetCsxMethodTemplate(context, "UpdateEntityTemplate"); + string getOneByTemplate = GetCsxMethodTemplate(context, "GetOneByTemplate"); + foreach (var candidateMember in candidateMembers) { - string addSourceTemplate = context.AdditionalFiles - .FirstOrDefault(x => x.Path.Contains("AddToEntityTemplate"))? - .GetText()? - .ToString() ?? string.Empty; - - string updateSourceTemplate = context.AdditionalFiles - .FirstOrDefault(x => x.Path.Contains("UpdateEntityTemplate"))? - .GetText()? - .ToString() ?? string.Empty; - - + var mappingContext = EfGeneratorContext.Create(compilation, options, candidateMember); mappingContext.Diagnostics.ForEach(context.ReportDiagnostic); @@ -82,5 +75,14 @@ namespace MapTo context.AddSource(hintName, source); } } + + private static string GetCsxMethodTemplate(GeneratorExecutionContext context, string templateName) + { + return context.AdditionalFiles + .FirstOrDefault(x => x.Path.Contains(templateName))? + .GetText()? + .ToString() ?? string.Empty; + } + } } \ No newline at end of file