Skip to content

Commit

Permalink
NTypewriter.SourceGenerator - #106 - bump Microsoft.CodeAnalysis.*.dl…
Browse files Browse the repository at this point in the history 
…l from 3.11 to 4.0.1 & more sophisticated assembly resolver
  • Loading branch information
@NeVeSpl
NeVeSpl committed Mar 8, 2024
1 parent 0309787 commit af6cc87
Show file tree
Hide file tree
Showing 4 changed files with 230 additions and 85 deletions.
183 changes: 143 additions & 40 deletions NTypewriter.SourceGenerator/AssemblyResolver.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,70 +8,173 @@ namespace NTypewriter.SourceGenerator
{
internal static class AssemblyResolver
{
public static Assembly Resolve(object sender, ResolveEventArgs args)
private static readonly string TempPath = Path.Combine(Path.GetTempPath(), "NTSG");


public static void Register()
{
Directory.CreateDirectory(TempPath);
Log(DateTime.Now.ToString());

AppDomain.CurrentDomain.AssemblyResolve += AssemblyResolver.Resolve;

PreLoadAssemblies();
}
private static void PreLoadAssemblies()
{
ResolveFromAssemblyName(new AssemblyName("Microsoft.CodeAnalysis, Version=3.11.0.0"));
ResolveFromAssemblyName(new AssemblyName("Microsoft.CodeAnalysis.CSharp, Version=3.11.0.0"));
ResolveFromAssemblyName(new AssemblyName("Microsoft.CodeAnalysis.CSharp.Workspaces, Version=3.11.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35"));
ResolveFromAssemblyName(new AssemblyName("Microsoft.CodeAnalysis.Workspaces, Version=3.11.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35"));
ResolveFromAssemblyName(new AssemblyName("Microsoft.CodeAnalysis.CSharp.Scripting, Version=3.11.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35"));
ResolveFromAssemblyName(new AssemblyName("Microsoft.CodeAnalysis.Scripting, Version=3.11.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35"));
ResolveFromAssemblyName(new AssemblyName("System.Text.Json, Version=6.0.0.0, Culture=neutral, PublicKeyToken=cc7b13ffcd2ddd51"));
ResolveFromAssemblyName(new AssemblyName("Microsoft.Bcl.AsyncInterfaces, Version=6.0.0.0, Culture=neutral, PublicKeyToken=cc7b13ffcd2ddd51"));
ResolveFromAssemblyName(new AssemblyName("Basic.Reference.Assemblies.NetStandard20, Version=1.0.0.0, Culture=neutral, PublicKeyToken=00aeae93c2ffe759"));
}


private static void Log(string message)
{
var enrichedMessage = $"AssemblyResolver: {message}";
Trace.WriteLine(enrichedMessage);

try
{
var path = Path.Combine(TempPath, $"AppDomain_{AppDomain.CurrentDomain.FriendlyName}_AssemblyResolver.log");
File.AppendAllText(path, enrichedMessage + "\r\n");
}
catch (Exception ex)
{
}
}
public static void DumpLodedAssemblies()
{
try
{
var lodedAssemblies = AppDomain.CurrentDomain.GetAssemblies();
var path = Path.Combine(TempPath, $"AppDomain_{AppDomain.CurrentDomain.FriendlyName}_LodedAssemblies.log");
var entries = lodedAssemblies.Select(x => x.GetLogEntry());

File.WriteAllLines(path, entries);
}
catch (Exception ex)
{
Log(ex.ToString());
}
}


private static Assembly Resolve(object sender, ResolveEventArgs args)
{
var asmName = new AssemblyName(args.Name);
Trace.WriteLine($"AssemblyResolver.Resolve : {asmName.FullName}");
if (asmName.Name.EndsWith(".resources")) return null;

return ResolveFromAssemblyName(asmName);
}
private static Assembly ResolveFromAssemblyName(AssemblyName asmName)
{
DumpLodedAssemblies();

Log($"requested {asmName.Name}, {asmName.Version}");

var lodedAssemblies = AppDomain.CurrentDomain.GetAssemblies();

var loadedAssembly = AppDomain.CurrentDomain.GetAssemblies().FirstOrDefault(a => a.GetName().FullName == asmName.FullName);
//

var loadedAssembly = lodedAssemblies.FirstOrDefault(a => a.GetName().FullName == asmName.FullName);
if (loadedAssembly != null)
{
Trace.WriteLine($"AssemblyResolver.Resolve : resolved from AppDomain.CurrentDomain");
Log($"resolved from AppDomain.CurrentDomain - {loadedAssembly.GetName().Version}");
return loadedAssembly;
}

var manifestResourceNames = Assembly.GetExecutingAssembly().GetManifestResourceNames();

var version = asmName.Version;
//

if (asmName.Name == "Scriban.Signed" && version == new Version(5, 0, 0, 0))
var loadedAssemblyRedirected = lodedAssemblies.Where(a => a.GetName().Name == asmName.Name).OrderByDescending(x => x.GetName().Version).FirstOrDefault();
if (loadedAssemblyRedirected != null)
{
version = new Version(5, 9, 0, 0);
Log($"resolved from AppDomain.CurrentDomain -> redirect - {loadedAssemblyRedirected.GetName().Version}");
return loadedAssemblyRedirected;
}
if (asmName.Name == "Basic.Reference.Assemblies.NetStandard20" && version == new Version(1, 0, 0, 0))

//

if (asmName.Name.StartsWith("Microsoft.CodeAnalysis"))
{
version = new Version(1, 4, 5, 0);
var microsoftCodeAnalysisAssembly = lodedAssemblies.FirstOrDefault(x => x.GetName().Name == "Microsoft.CodeAnalysis");
var microsoftCodeAnalysisPath = microsoftCodeAnalysisAssembly != null ? Path.GetDirectoryName(microsoftCodeAnalysisAssembly.Location) : null;

if (!string.IsNullOrEmpty(microsoftCodeAnalysisPath))
{
var roslynAsmPath = Path.Combine(microsoftCodeAnalysisPath, asmName.Name + ".dll");
if (File.Exists(roslynAsmPath))
{
var roslynAsm = LoadAssembly(roslynAsmPath);
if (roslynAsm != null)
{
Log($"resolved from Roslyn - {roslynAsm.GetName().Version}");
return roslynAsm;
}
}
}
}

//

var dllName = $"{asmName.Name}.v{version}.dll";
var manifestResourceNames = Assembly.GetExecutingAssembly().GetManifestResourceNames();

using (Stream resourceStream = Assembly.GetExecutingAssembly().GetManifestResourceStream(dllName))
var version = asmName.Version;
var dllNameWithVer = manifestResourceNames.FirstOrDefault(x => x.StartsWith(asmName.Name+".v"));

if (!string.IsNullOrEmpty(dllNameWithVer))
{
if (resourceStream == null)
{
Trace.WriteLine($"AssemblyResolver.Resolve : failure");
return null;
}
try
using (Stream resourceStream = Assembly.GetExecutingAssembly().GetManifestResourceStream(dllNameWithVer))
{

var tempPath = Path.Combine(Path.GetTempPath(), "NTSG");
var filePath = Path.Combine(tempPath, dllName);
Directory.CreateDirectory(tempPath);

if (!File.Exists(filePath))
{
using (FileStream fileStream = File.Create(filePath))
if (resourceStream != null)
{
try
{
resourceStream.CopyTo(fileStream);
var filePath = Path.Combine(TempPath, dllNameWithVer);

if (!File.Exists(filePath))
{
using (FileStream fileStream = File.Create(filePath))
{
resourceStream.CopyTo(fileStream);
}
}
var tempASM = LoadAssembly(filePath);
Log($"resolved from TEMP - {tempASM.GetName().Version}");
return tempASM;
}
catch
{

}
}

return Assembly.LoadFile(filePath);
using (MemoryStream memoryStream = new MemoryStream())
{
resourceStream.CopyTo(memoryStream);
var streamAsm = Assembly.Load(memoryStream.ToArray());
Log($"resolved from ResourceStream - {streamAsm.GetName().Version}");
return streamAsm;
}
}
}
catch
{
}

}
Log($"failure");
return null;
}

using (MemoryStream memoryStream = new MemoryStream())
{
Trace.WriteLine($"AssemblyResolver.Resolve : resolved from ResourceStream");
resourceStream.CopyTo(memoryStream);
return Assembly.Load(memoryStream.ToArray());
}
}

private static Assembly LoadAssembly(string filePath)
{
//var b = File.ReadAllBytes(filePath);
var assembly = Assembly.LoadFile(filePath);

return assembly;
}
}
}
15 changes: 15 additions & 0 deletions NTypewriter.SourceGenerator/Extensions/System.Reflection/AssemblyExtensions.cs
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
using System;
using System.Collections.Generic;
using System.Reflection;
using System.Text;

namespace System.Reflection
{
internal static class AssemblyExtensions
{
public static string GetLogEntry(this Assembly assembly)
{
return $"{assembly.GetName().Name,-48} {assembly.GetName().Version} {(assembly.IsDynamic ? null : assembly.Location)} {assembly.FullName}";
}
}
}

0 comments on commit af6cc87

Please sign in to comment.