public bool GenerateCode()
{
// Create compilationUnit
var compilationUnit = SyntaxFactory.CompilationUnit();
NamespaceDeclarationSyntax @namespace;
if (ApiSchema.IsSchemaEnumOrPropertyEnum())
{
IsEnum = true;
// Create a namespace
@namespace = SyntaxProjectFactory.CreateNamespace(
ApiProjectOptions,
NameConstants.Contracts);
var apiEnumSchema = ApiSchema.GetEnumSchema();
// Create an enum
var enumDeclaration = SyntaxEnumFactory.Create(apiEnumSchema.Item1.EnsureFirstCharacterToUpper(), apiEnumSchema.Item2);
if (enumDeclaration.HasAttributeOfAttributeType(typeof(FlagsAttribute)))
{
// Add using statement to compilationUnit
compilationUnit = compilationUnit.AddUsingStatements(new[] { "System" });
}
if (enumDeclaration.HasAttributeOfAttributeType(typeof(SuppressMessageAttribute)))
{
// Add using statement to compilationUnit
compilationUnit = compilationUnit.AddUsingStatements(new[] { "System.Diagnostics.CodeAnalysis" });
}
// Add the enum to the namespace.
@namespace = @namespace.AddMembers(enumDeclaration);
}
else
{
// Create a namespace
@namespace = IsSharedContract
? SyntaxProjectFactory.CreateNamespace(ApiProjectOptions, NameConstants.Contracts)
: SyntaxProjectFactory.CreateNamespace(ApiProjectOptions, NameConstants.Contracts, FocusOnSegmentName);
// Create class
var classDeclaration = SyntaxClassDeclarationFactory.Create(ApiSchemaKey.EnsureFirstCharacterToUpper())
.AddGeneratedCodeAttribute(ApiProjectOptions.ToolName, ApiProjectOptions.ToolVersion.ToString())
.WithLeadingTrivia(SyntaxDocumentationFactory.Create(ApiSchema));
// Create class-properties and add to class
if (ApiSchema.Properties != null)
{
if (ApiSchema.Type == OpenApiDataTypeConstants.Array)
{
var(key, _) = ApiProjectOptions.Document.Components.Schemas.FirstOrDefault(x => x.Key.Equals(ApiSchema.Title, StringComparison.OrdinalIgnoreCase));
if (string.IsNullOrEmpty(ApiSchema.Title))
{
ApiSchema.Title = ApiSchemaKey;
key = ApiSchemaKey;
}
if (string.IsNullOrEmpty(ApiSchema.Items.Title))
{
ApiSchema.Items.Title = ApiSchemaKey;
}
var title = key != null
? $"{ApiSchema.Title.EnsureFirstCharacterToUpperAndSingular()}List"
: ApiSchema.Title.EnsureFirstCharacterToUpper();
var propertyDeclaration = SyntaxPropertyDeclarationFactory.CreateListAuto(ApiSchema.Items.Title, title)
.WithLeadingTrivia(SyntaxDocumentationFactory.CreateSummary($"A list of {ApiSchema.Items.Title}."));
classDeclaration = classDeclaration.AddMembers(propertyDeclaration);
}
else
{
foreach (var property in ApiSchema.Properties)
{
var propertyDeclaration = SyntaxPropertyDeclarationFactory.CreateAuto(
property,
ApiSchema.Required,
ApiProjectOptions.ApiOptions.Generator.UseNullableReferenceTypes)
.WithLeadingTrivia(SyntaxDocumentationFactory.Create(property.Value));
classDeclaration = classDeclaration.AddMembers(propertyDeclaration);
}
}
var methodDeclaration = SyntaxMethodDeclarationFactory.CreateToStringMethod(ApiSchema.Properties);
if (methodDeclaration != null)
{
methodDeclaration = methodDeclaration.WithLeadingTrivia(SyntaxDocumentationFactory.CreateForOverrideToString());
classDeclaration = classDeclaration.AddMembers(methodDeclaration);
}
}
// Add using statement to compilationUnit
compilationUnit = compilationUnit.AddUsingStatements(ProjectContractDataFactory.CreateUsingList(ApiSchema));
// Add the class to the namespace.
@namespace = @namespace.AddMembers(classDeclaration);
}
// Add namespace to compilationUnit
compilationUnit = compilationUnit.AddMembers(@namespace);
// Set code property
Code = compilationUnit;
return(true);
}
private NamespaceDeclarationSyntax GenerateCodeForOtherThanEnum(
ref CompilationUnitSyntax?compilationUnit)
{
// Create a namespace
var @namespace = IsSharedContract
? SyntaxProjectFactory.CreateNamespace(ApiProjectOptions, NameConstants.Contracts)
: SyntaxProjectFactory.CreateNamespace(ApiProjectOptions, NameConstants.Contracts, FocusOnSegmentName);
// Create class
var classDeclaration = SyntaxClassDeclarationFactory.Create(ApiSchemaKey.EnsureFirstCharacterToUpper())
.AddGeneratedCodeAttribute(ApiProjectOptions.ToolName, ApiProjectOptions.ToolVersion.ToString())
.WithLeadingTrivia(SyntaxDocumentationFactory.Create(ApiSchema));
var hasAnyPropertiesAsArrayWithFormatTypeBinary = ApiSchema.HasAnyPropertiesAsArrayWithFormatTypeBinary();
// Create class-properties and add to class
if (ApiSchema.Properties is not null)
{
if (ApiSchema.IsTypeArray() ||
hasAnyPropertiesAsArrayWithFormatTypeBinary)
{
var(key, _) = ApiProjectOptions.Document.Components.Schemas.FirstOrDefault(x =>
x.Key.Equals(ApiSchema.Title, StringComparison.OrdinalIgnoreCase));
if (string.IsNullOrEmpty(ApiSchema.Title))
{
ApiSchema.Title = ApiSchemaKey;
key = ApiSchemaKey;
}
if (ApiSchema.Items is not null &&
string.IsNullOrEmpty(ApiSchema.Items.Title))
{
ApiSchema.Items.Title = ApiSchemaKey;
}
var title = key is not null
? $"{ApiSchema.Title.EnsureFirstCharacterToUpperAndSingular()}List"
: ApiSchema.Title.EnsureFirstCharacterToUpper();
var propertyDeclaration = hasAnyPropertiesAsArrayWithFormatTypeBinary
? SyntaxPropertyDeclarationFactory.CreateListAuto("IFormFile", ApiSchema.ExtractPropertyNameWhenHasAnyPropertiesOfArrayWithFormatTypeBinary())
.WithLeadingTrivia(
SyntaxDocumentationFactory.CreateSummary("A list of File(s)."))
: SyntaxPropertyDeclarationFactory.CreateListAuto(ApiSchema.Items !.Title, title)
.WithLeadingTrivia(
SyntaxDocumentationFactory.CreateSummary($"A list of {ApiSchema.Items.Title}."));
classDeclaration = classDeclaration.AddMembers(propertyDeclaration);
}
else
{
foreach (var property in ApiSchema.Properties)
{
var propertyDeclaration = SyntaxPropertyDeclarationFactory.CreateAuto(
property,
ApiSchema.Required,
ApiProjectOptions.UseNullableReferenceTypes)
.WithLeadingTrivia(SyntaxDocumentationFactory.Create(property.Value));
classDeclaration = classDeclaration.AddMembers(propertyDeclaration);
}
}
var methodDeclaration = SyntaxMethodDeclarationFactory.CreateToStringMethod(ApiSchema.Properties);
if (methodDeclaration is not null)
{
methodDeclaration = methodDeclaration.WithLeadingTrivia(SyntaxDocumentationFactory.CreateForOverrideToString());
classDeclaration = classDeclaration.AddMembers(methodDeclaration);
}
}
// Add using statement to compilationUnit
compilationUnit = compilationUnit !.AddUsingStatements(ProjectApiFactory.CreateUsingListForContractModel(ApiSchema));
// Add the class to the namespace.
@namespace = @namespace.AddMembers(classDeclaration);
return(@namespace);
}