mirror of
https://github.com/aaru-dps/Aaru.git
synced 2025-12-16 19:24:25 +00:00
412 lines
18 KiB
C#
412 lines
18 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using Microsoft.CodeAnalysis;
|
|
using Microsoft.CodeAnalysis.CSharp.Syntax;
|
|
|
|
namespace Aaru.Generators;
|
|
|
|
[Generator]
|
|
public class PluginRegisterGenerator : ISourceGenerator
|
|
{
|
|
#region ISourceGenerator Members
|
|
|
|
/// <inheritdoc />
|
|
public void Initialize(GeneratorInitializationContext context) =>
|
|
|
|
// Nothing to do
|
|
context.RegisterForSyntaxNotifications(() => new PluginFinder());
|
|
|
|
/// <inheritdoc />
|
|
public void Execute(GeneratorExecutionContext context)
|
|
{
|
|
/*
|
|
#if DEBUG
|
|
if(!Debugger.IsAttached)
|
|
{
|
|
Debugger.Launch();
|
|
}
|
|
#endif
|
|
*/
|
|
|
|
ClassDeclarationSyntax pluginRegister = ((PluginFinder)context.SyntaxReceiver)?.Register;
|
|
|
|
if(pluginRegister == null) return;
|
|
|
|
var @namespace =
|
|
(pluginRegister.Ancestors().FirstOrDefault(x => x is FileScopedNamespaceDeclarationSyntax) as
|
|
FileScopedNamespaceDeclarationSyntax)?.Name.ToString();
|
|
|
|
@namespace ??=
|
|
(pluginRegister.Ancestors().FirstOrDefault(x => x is NamespaceDeclarationSyntax) as
|
|
NamespaceDeclarationSyntax)?.ToString();
|
|
|
|
string className = pluginRegister.Identifier.Text;
|
|
|
|
List<string> archives = ((PluginFinder)context.SyntaxReceiver)?.Archives;
|
|
List<string> checksums = ((PluginFinder)context.SyntaxReceiver)?.Checksums;
|
|
List<string> fileSystems = ((PluginFinder)context.SyntaxReceiver)?.FileSystems;
|
|
List<string> filters = ((PluginFinder)context.SyntaxReceiver)?.Filters;
|
|
List<string> floppyImagePlugins = ((PluginFinder)context.SyntaxReceiver)?.FloppyImagePlugins;
|
|
List<string> partitionPlugins = ((PluginFinder)context.SyntaxReceiver)?.PartitionPlugins;
|
|
List<string> mediaImagePlugins = ((PluginFinder)context.SyntaxReceiver)?.MediaImagePlugins;
|
|
List<string> readOnlyFileSystems = ((PluginFinder)context.SyntaxReceiver)?.ReadOnlyFileSystems;
|
|
List<string> writableFloppyImagePlugins = ((PluginFinder)context.SyntaxReceiver)?.WritableFloppyImagePlugins;
|
|
List<string> writableImagePlugins = ((PluginFinder)context.SyntaxReceiver)?.WritableImagePlugins;
|
|
List<string> byteAddressableImagePlugins = ((PluginFinder)context.SyntaxReceiver)?.ByteAddressableImagePlugins;
|
|
|
|
StringBuilder sb = new();
|
|
|
|
sb.AppendLine("""
|
|
// /***************************************************************************
|
|
// Aaru Data Preservation Suite
|
|
// ----------------------------------------------------------------------------
|
|
//
|
|
// Filename : Register.g.cs
|
|
// Author(s) : Natalia Portillo <claunia@claunia.com>
|
|
//
|
|
// --[ Description ] ----------------------------------------------------------
|
|
//
|
|
// Registers all plugins in this assembly.
|
|
//
|
|
// --[ License ] --------------------------------------------------------------
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the
|
|
// "Software"), to deal in the Software without restriction, including
|
|
// without limitation the rights to use, copy, modify, merge, publish,
|
|
// distribute, sublicense, and/or sell copies of the Software, and to
|
|
// permit persons to whom the Software is furnished to do so, subject to
|
|
// the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included
|
|
// in all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
// IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
// TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
// SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
//
|
|
// ----------------------------------------------------------------------------
|
|
// Copyright © 2011-2024 Natalia Portillo
|
|
// ****************************************************************************/
|
|
""");
|
|
|
|
sb.AppendLine();
|
|
sb.AppendLine("using System;");
|
|
sb.AppendLine("using System.Collections.Generic;");
|
|
sb.AppendLine("using Aaru.CommonTypes.Interfaces;");
|
|
sb.AppendLine("using Microsoft.Extensions.DependencyInjection;");
|
|
sb.AppendLine();
|
|
sb.AppendLine($"namespace {@namespace};");
|
|
sb.AppendLine();
|
|
sb.AppendLine($"public sealed partial class {className} : IPluginRegister");
|
|
sb.AppendLine("{");
|
|
|
|
if(archives?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterArchivePlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in archives.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IArchive, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterArchivePlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(checksums?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterChecksumPlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in checksums.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IChecksum, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterChecksumPlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(fileSystems?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterFilesystemPlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in fileSystems.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IFilesystem, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterFilesystemPlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(filters?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterFilterPlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in filters.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IFilter, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterFilterPlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(floppyImagePlugins?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterFloppyImagePlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in floppyImagePlugins.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IFloppyImage, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterFloppyImagePlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(mediaImagePlugins?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterMediaImagePlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in mediaImagePlugins.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IMediaImage, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterMediaImagePlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(partitionPlugins?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterPartitionPlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in partitionPlugins.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IPartition, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterPartitionPlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(readOnlyFileSystems?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterReadOnlyFilesystemPlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in readOnlyFileSystems.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IReadOnlyFilesystem, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterReadOnlyFilesystemPlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(writableFloppyImagePlugins?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterWritableFloppyImagePlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in writableFloppyImagePlugins.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IWritableFloppyImage, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterWritableFloppyImagePlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(writableImagePlugins?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterWritableImagePlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in writableImagePlugins.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IBaseWritableImage, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterWritableImagePlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine();
|
|
|
|
if(byteAddressableImagePlugins?.Count > 0)
|
|
{
|
|
sb.AppendLine(" public void RegisterByteAddressablePlugins(IServiceCollection services)");
|
|
sb.AppendLine(" {");
|
|
|
|
foreach(string plugin in byteAddressableImagePlugins.Distinct())
|
|
sb.AppendLine($" services.AddTransient<IByteAddressableImage, {plugin}>();");
|
|
|
|
sb.AppendLine(" }");
|
|
}
|
|
else
|
|
sb.AppendLine(" public void RegisterByteAddressablePlugins(IServiceCollection services) {}");
|
|
|
|
sb.AppendLine("}");
|
|
|
|
context.AddSource("Register.g.cs", sb.ToString());
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Nested type: PluginFinder
|
|
|
|
sealed class PluginFinder : ISyntaxReceiver
|
|
{
|
|
public List<string> Archives { get; } = new();
|
|
public List<string> Checksums { get; } = new();
|
|
public List<string> FileSystems { get; } = new();
|
|
public List<string> Filters { get; } = new();
|
|
public List<string> FloppyImagePlugins { get; } = new();
|
|
public List<string> MediaImagePlugins { get; } = new();
|
|
public List<string> PartitionPlugins { get; } = new();
|
|
public List<string> ReadOnlyFileSystems { get; } = new();
|
|
public List<string> WritableFloppyImagePlugins { get; } = new();
|
|
public List<string> WritableImagePlugins { get; } = new();
|
|
public List<string> ByteAddressableImagePlugins { get; } = new();
|
|
public ClassDeclarationSyntax Register { get; private set; }
|
|
|
|
#region ISyntaxReceiver Members
|
|
|
|
public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
|
|
{
|
|
if(syntaxNode is not ClassDeclarationSyntax plugin) return;
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IPluginRegister") ==
|
|
true)
|
|
Register = plugin;
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IArchive") ==
|
|
true)
|
|
{
|
|
if(!Archives.Contains(plugin.Identifier.Text)) Archives.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IChecksum") ==
|
|
true)
|
|
{
|
|
if(!Checksums.Contains(plugin.Identifier.Text)) Checksums.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IFilesystem") ==
|
|
true)
|
|
{
|
|
if(!FileSystems.Contains(plugin.Identifier.Text)) FileSystems.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IFilter") ==
|
|
true)
|
|
{
|
|
if(!Filters.Contains(plugin.Identifier.Text)) Filters.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IFloppyImage") ==
|
|
true)
|
|
{
|
|
if(!FloppyImagePlugins.Contains(plugin.Identifier.Text)) FloppyImagePlugins.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText is "IMediaImage"
|
|
or "IOpticalMediaImage"
|
|
or "IFloppyImage"
|
|
or "ITapeImage") ==
|
|
true)
|
|
{
|
|
if(!MediaImagePlugins.Contains(plugin.Identifier.Text)) MediaImagePlugins.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IPartition") ==
|
|
true)
|
|
{
|
|
if(!PartitionPlugins.Contains(plugin.Identifier.Text)) PartitionPlugins.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IReadOnlyFilesystem") ==
|
|
true)
|
|
{
|
|
if(!ReadOnlyFileSystems.Contains(plugin.Identifier.Text))
|
|
ReadOnlyFileSystems.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IWritableFloppyImage") ==
|
|
true)
|
|
{
|
|
if(!WritableFloppyImagePlugins.Contains(plugin.Identifier.Text))
|
|
WritableFloppyImagePlugins.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText is "IWritableImage"
|
|
or "IWritableOpticalImage"
|
|
or "IWritableTapeImage"
|
|
or "IByteAddressableImage") ==
|
|
true)
|
|
{
|
|
if(!WritableImagePlugins.Contains(plugin.Identifier.Text))
|
|
WritableImagePlugins.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
if(plugin.BaseList?.Types.Any(t => ((t as SimpleBaseTypeSyntax)?.Type as IdentifierNameSyntax)?.Identifier
|
|
.ValueText ==
|
|
"IByteAddressableImage") ==
|
|
true)
|
|
{
|
|
if(!ByteAddressableImagePlugins.Contains(plugin.Identifier.Text))
|
|
ByteAddressableImagePlugins.Add(plugin.Identifier.Text);
|
|
}
|
|
|
|
MediaImagePlugins.AddRange(WritableImagePlugins.Where(t => !ByteAddressableImagePlugins.Contains(t)));
|
|
FileSystems.AddRange(ReadOnlyFileSystems);
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
|
|
#endregion
|
|
} |