2021-12-22 17:16:44 -08:00
|
|
|
using Microsoft.CodeAnalysis;
|
2021-10-14 06:55:19 -07:00
|
|
|
using NSwag;
|
|
|
|
|
|
2021-12-22 17:16:44 -08:00
|
|
|
namespace LibKubernetesGenerator
|
2021-10-14 06:55:19 -07:00
|
|
|
{
|
2021-10-24 08:14:22 -07:00
|
|
|
internal class ModelGenerator
|
2021-10-14 06:55:19 -07:00
|
|
|
{
|
|
|
|
|
private readonly ClassNameHelper classNameHelper;
|
2024-04-15 10:43:05 -07:00
|
|
|
private readonly ScriptObjectFactory scriptObjectFactory;
|
2021-10-14 06:55:19 -07:00
|
|
|
|
2024-04-15 10:43:05 -07:00
|
|
|
public ModelGenerator(ClassNameHelper classNameHelper, ScriptObjectFactory scriptObjectFactory)
|
2021-10-14 06:55:19 -07:00
|
|
|
{
|
|
|
|
|
this.classNameHelper = classNameHelper;
|
2024-04-15 10:43:05 -07:00
|
|
|
this.scriptObjectFactory = scriptObjectFactory;
|
2021-10-14 06:55:19 -07:00
|
|
|
}
|
|
|
|
|
|
2023-10-10 09:49:47 -07:00
|
|
|
public void Generate(OpenApiDocument swagger, IncrementalGeneratorPostInitializationContext context)
|
2021-10-14 06:55:19 -07:00
|
|
|
{
|
2024-04-15 10:43:05 -07:00
|
|
|
var sc = scriptObjectFactory.CreateScriptObject();
|
|
|
|
|
|
|
|
|
|
|
2021-12-22 17:16:44 -08:00
|
|
|
foreach (var kv in swagger.Definitions)
|
2021-10-14 06:55:19 -07:00
|
|
|
{
|
2021-12-22 17:16:44 -08:00
|
|
|
var def = kv.Value;
|
2021-10-14 06:55:19 -07:00
|
|
|
var clz = classNameHelper.GetClassNameForSchemaDefinition(def);
|
2024-04-15 10:43:05 -07:00
|
|
|
|
|
|
|
|
sc.SetValue("clz", clz, true);
|
|
|
|
|
sc.SetValue("def", def, true);
|
|
|
|
|
sc.SetValue("properties", def.Properties.Values, true);
|
|
|
|
|
|
|
|
|
|
context.RenderToContext("Model.cs.template", sc, $"Models_{clz}.g.cs");
|
2021-10-14 06:55:19 -07:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|