Files
csharp/src/LibKubernetesGenerator/ParamHelper.cs

80 lines
2.6 KiB
C#
Raw Normal View History

using NJsonSchema;
using NSwag;
using Scriban.Runtime;
using System;
using System.Linq;
using System.Collections.Generic;
namespace LibKubernetesGenerator
{
internal class ParamHelper : IScriptObjectHelper
{
private readonly GeneralNameHelper generalNameHelper;
private readonly TypeHelper typeHelper;
public ParamHelper(GeneralNameHelper generalNameHelper, TypeHelper typeHelper)
{
this.generalNameHelper = generalNameHelper;
this.typeHelper = typeHelper;
}
public void RegisterHelper(ScriptObject scriptObject)
{
scriptObject.Import(nameof(GetModelCtorParam), new Func<JsonSchema, string>(GetModelCtorParam));
scriptObject.Import(nameof(IfParamContains), IfParamContains);
scriptObject.Import(nameof(FilterParameters), FilterParameters);
scriptObject.Import(nameof(GetParameterValueForWatch), new Func<OpenApiParameter, bool, string, string>(GetParameterValueForWatch));
}
public static bool IfParamContains(OpenApiOperation operation, string name)
{
var found = false;
foreach (var param in operation.Parameters)
{
if (param.Name == name)
{
found = true;
break;
}
}
return found;
}
public static IEnumerable<OpenApiParameter> FilterParameters(OpenApiOperation operation, string excludeParam)
{
return operation.Parameters.Where(p => p.Name != excludeParam);
}
public string GetParameterValueForWatch(OpenApiParameter parameter, bool watch, string init = "false")
{
if (parameter.Name == "watch")
{
return watch ? "true" : "false";
}
else
{
return generalNameHelper.GetDotNetNameOpenApiParameter(parameter, init);
}
}
public string GetModelCtorParam(JsonSchema schema)
{
return string.Join(", ", schema.Properties.Values
.OrderBy(p => !p.IsRequired)
.Select(p =>
{
var sp =
$"{typeHelper.GetDotNetType(p)} {generalNameHelper.GetDotNetName(p.Name, "fieldctor")}";
if (!p.IsRequired)
{
sp = $"{sp} = null";
}
return sp;
}));
}
}
}