private static string MakeFileName(FileType type, Kind kind)
{
switch (type)
{
case FileType.ModelClass:
return(NameHelper.MakeModelClassName(kind) + ".cs");
case FileType.GeneratorClass:
return(NameHelper.MakeGeneratorClassName(kind) + ".cs");
case FileType.ScaffolderClass:
return(NameHelper.MakeScaffolderClassName(kind) + ".cs");
case FileType.GeneratorInterface:
return(NameHelper.MakeGeneratorInterfaceName(kind) + ".cs");
case FileType.ScaffolderInterface:
return(NameHelper.MakeScaffolderInterfaceName(kind) + ".cs");
default:
return(string.Empty + ".cs");
}
}
private static string GenerateServicesClass(CodeGenerationOptions options)
{
Clear();
string ns = options.RootNamespace;
WriteUsings(options, Kind.None);
WriteLine($"namespace {ns}");
using (OpenBlock())
{
WriteLine("public static class OpenApiServices");
using (OpenBlock())
{
WriteLine("private static IServiceProvider _serviceProvider;");
WriteLine();
WriteLine("public static TService GetService<TService>()");
using (OpenBlock())
{
WriteLine("return ServiceProvider.GetService<TService>();");
}
WriteLine();
WriteLine("private static IServiceProvider ServiceProvider");
using (OpenBlock())
{
WriteLine("get");
using (OpenBlock())
{
WriteLine("if (_serviceProvider == null)");
using (OpenBlock())
{
WriteLine("_serviceProvider = new ServiceCollection()");
PushIndent();
WriteLine($".AddSingleton<GeneratorDependencies>()");
WriteLine($".AddSingleton<ScaffolderDependencies>()");
WriteLine($".AddSingleton<IFileWriter, FileWriter>()");
WriteLine($".AddSingleton<IPathHelper, PathHelper>()");
WriteLine($".AddSingleton<ITextProvider, TextProvider>()");
WriteLine($".AddSingleton<INamespaceHelper, NamespaceHelper>()");
WriteLine($".AddSingleton<IScaffoldingManager, ScaffoldingManager>()");
foreach (var kind in options.Kinds)
{
string generatorInterface = NameHelper.MakeGeneratorInterfaceName(kind);
string generatorClass = NameHelper.MakeGeneratorClassName(kind);
WriteLine($".AddSingleton<{generatorInterface}, {generatorClass}>()");
}
foreach (var kind in options.Kinds)
{
string scaffolderInterface = NameHelper.MakeScaffolderInterfaceName(kind);
string scaffolderClass = NameHelper.MakeScaffolderClassName(kind);
WriteLine($".AddSingleton<{scaffolderInterface}, {scaffolderClass}>()");
}
WriteLine(".BuildServiceProvider();");
PopIndent();
}
WriteLine("return _serviceProvider;");
}
}
}
}
return(GetText());
}
private static string GenerateScaffolderClass(CodeGenerationOptions options, Kind kind)
{
Clear();
string ns = NamespaceHelper.MakeNamespace(options.RootNamespace, kind);
WriteUsings(options, kind);
WriteLine($"namespace {ns}");
using (OpenBlock())
{
string modelName = NameHelper.MakeModelClassName(kind);
string typeName = NameHelper.MakeScaffolderClassName(kind);
string baseTypeName = NameHelper.MakeScaffolderInterfaceName(kind);
string generatorInterfaceName = NameHelper.MakeGeneratorInterfaceName(kind);
WriteLine($"public class {typeName} : AbstractScaffolder, {baseTypeName}");
using (OpenBlock())
{
WriteLine($"public {typeName}(ScaffolderDependencies dependencies, {generatorInterfaceName} generator) : base(dependencies)");
using (OpenBlock())
{
WriteLine("Generator = generator;");
}
WriteLine();
WriteLine($"protected {generatorInterfaceName} Generator {{ get; }} ");
WriteLine();
WriteLine($"public void Save({modelName} model)");
using (OpenBlock())
{
if (kind == Kind.Context || kind == Kind.Supervisor)
{
WriteLine($"Dependencies.FileWriter.WriteFile(model.File);");
WriteLine($"Dependencies.FileWriter.WriteFile(model.Interface);");
}
else
{
WriteLine($"Dependencies.FileWriter.WriteFiles(model.Files);");
}
}
WriteLine();
WriteLine($"public {modelName} ScaffoldModel(OpenApiOptions options)");
using (OpenBlock())
{
WriteLine($"var model = new {modelName}();");
if (kind == Kind.Context)
{
WriteLine("var classFile = new ScaffoldedFile();");
WriteLine("var interfaceFile = new ScaffoldedFile();");
WriteLine("classFile.Code = Generator.WriteClassCode(options.Document, options.RootNamespace);");
WriteLine($"classFile.Path = Dependencies.PathHelper.Context(options.OutputDir, options.ContextClassName);");
WriteLine("interfaceFile.Code = Generator.WriteInterfaceCode(options.Document, options.RootNamespace);");
WriteLine($"interfaceFile.Path = Dependencies.PathHelper.Context(options.OutputDir, options.ContextInterfaceName);");
WriteLine("model.File = classFile;");
WriteLine("model.Interface = interfaceFile;");
}
else if (kind == Kind.Supervisor)
{
WriteLine("var classFile = new ScaffoldedFile();");
WriteLine("var interfaceFile = new ScaffoldedFile();");
WriteLine("classFile.Code = Generator.WriteClassCode(options.Document, options.RootNamespace);");
WriteLine($"classFile.Path = Dependencies.PathHelper.Supervisor(options.OutputDir, options.SupervisorClassName);");
WriteLine("interfaceFile.Code = Generator.WriteInterfaceCode(options.Document, options.RootNamespace);");
WriteLine($"interfaceFile.Path = Dependencies.PathHelper.Supervisor(options.OutputDir, options.SupervisorInterfaceName);");
WriteLine("model.File = classFile;");
WriteLine("model.Interface = interfaceFile;");
}
else
{
WriteLine("foreach(var kvp in options.Document.Components.Schemas)");
using (OpenBlock())
{
WriteLine("var name = kvp.Key;");
WriteLine("var schema = kvp.Value;");
WriteLine($"var code = Generator.WriteCode(schema, name, Dependencies.Namespace.{kind}(options.RootNamespace));");
WriteLine($"var path = Dependencies.PathHelper.{kind}(options.OutputDir, name);");
WriteLine($"var file = new ScaffoldedFile {{ Code = code, Path = path }};");
WriteLine($"model.Files.Add(file);");
}
}
WriteLine("return model;");
}
}
}
return(GetText());
}