- add: script abilities

This commit is contained in:
2025-07-13 10:08:07 +03:00
parent a475148543
commit 138b769f81
10 changed files with 287 additions and 64 deletions

View File

@@ -75,6 +75,11 @@ public sealed class FileApi : IReadWriteFileApi
return File.Exists(fullPath);
}
public IEnumerable<string> EnumerateDirectories(string path)
{
return Directory.GetDirectories(Path.Join(RootPath, path)).Select(p=>p.Replace(RootPath,"").Substring(1));
}
private IEnumerable<string> GetAllFiles(){
if(!Directory.Exists(RootPath)) return [];

View File

@@ -12,15 +12,23 @@ namespace Nebula.Shared.Services;
[ServiceRegister]
public class AssemblyService
{
private readonly List<Assembly> _assemblies = new();
private readonly Dictionary<string, Assembly> _assemblyCache = new();
private readonly ILogger _logger;
private readonly HashSet<string> _resolvingAssemblies = new();
private List<AssemblyApi> _mountedApis = [];
public Action<Assembly>? OnAssemblyLoaded;
public IReadOnlyList<Assembly> Assemblies => _assemblyCache.Values.ToList().AsReadOnly();
public AssemblyService(DebugService debugService)
{
_logger = debugService.GetLogger(this);
AssemblyLoadContext.Default.ResolvingUnmanagedDll += LoadContextOnResolvingUnmanaged;
AssemblyLoadContext.Default.Resolving += (context, name) => OnAssemblyResolving(context, name);
ZstdImportResolver.ResolveLibrary += (name, assembly1, path) =>
{
if (name.Equals("SharpZstd.Native"))
@@ -36,21 +44,41 @@ public class AssemblyService
};
}
public IReadOnlyList<Assembly> Assemblies => _assemblies;
private Assembly? OnAssemblyResolving(AssemblyLoadContext context, AssemblyName name)
{
if (_resolvingAssemblies.Contains(name.FullName))
{
_logger.Debug($"Already resolving {name.Name}, skipping.");
return null; // Prevent recursive resolution
}
Assembly? assembly;
if (_assemblyCache.TryGetValue(name.Name ?? "", out assembly))
{
return assembly;
}
foreach (var api in _mountedApis)
{
if((assembly = OnAssemblyResolving(context, name, api)) != null)
return assembly;
}
return null;
}
public AssemblyApi Mount(IFileApi fileApi)
{
var asmApi = new AssemblyApi(fileApi);
AssemblyLoadContext.Default.Resolving += (context, name) => OnAssemblyResolving(context, name, asmApi);
AssemblyLoadContext.Default.ResolvingUnmanagedDll += LoadContextOnResolvingUnmanaged;
_mountedApis.Add(asmApi);
return asmApi;
}
public bool TryGetLoader(Assembly clientAssembly, [NotNullWhen(true)] out ILoaderEntryPoint? loader)
{
loader = null;
// Find ILoaderEntryPoint with the LoaderEntryPointAttribute
var attrib = clientAssembly.GetCustomAttribute<LoaderEntryPointAttribute>();
if (attrib == null)
{
@@ -79,9 +107,11 @@ public class AssemblyService
assembly = AssemblyLoadContext.Default.LoadFromStream(asm, pdb);
_logger.Log("LOADED ASSEMBLY " + name);
if (!_assemblies.Contains(assembly)) _assemblies.Add(assembly);
if (_assemblyCache.TryAdd(name, assembly))
{
OnAssemblyLoaded?.Invoke(assembly);
}
asm.Dispose();
pdb?.Dispose();
@@ -103,21 +133,18 @@ public class AssemblyService
private Assembly? OnAssemblyResolving(AssemblyLoadContext context, AssemblyName name, AssemblyApi assemblyApi)
{
if (_resolvingAssemblies.Contains(name.FullName))
lock (_resolvingAssemblies)
{
_logger.Debug($"Already resolving {name.Name}, skipping.");
return null; // Prevent recursive resolution
}
try
{
_resolvingAssemblies.Add(name.FullName);
_logger.Debug($"Resolving assembly from FileAPI: {name.Name}");
return TryOpenAssembly(name.Name!, assemblyApi, out var assembly) ? assembly : null;
}
finally
{
_resolvingAssemblies.Remove(name.FullName);
try
{
_resolvingAssemblies.Add(name.FullName);
_logger.Debug($"Resolving assembly from FileAPI: {name.Name}");
return TryOpenAssembly(name.Name!, assemblyApi, out var assembly) ? assembly : null;
}
finally
{
_resolvingAssemblies.Remove(name.FullName);
}
}
}

View File

@@ -23,7 +23,7 @@ public class FileService
Directory.CreateDirectory(RootPath);
}
public IReadWriteFileApi CreateFileApi(string path)
public FileApi CreateFileApi(string path)
{
_logger.Debug($"Creating file api for {path}");
return new FileApi(Path.Join(RootPath, path));