private static string DefaultScalarFieldMapping(GraphQlField field)
{
var propertyName = NamingHelper.CapitalizeFirst(field.Name);
if (propertyName == "From" || propertyName == "ValidFrom" || propertyName == "CreatedAt" ||
propertyName == "To" || propertyName == "ValidTo" || propertyName == "ModifiedAt" || propertyName.EndsWith("Timestamp"))
{
return("DateTimeOffset?");
}
return("string");
}
private static void GenerateEnum(GraphQlType type, StringBuilder builder)
{
builder.Append("public enum ");
builder.AppendLine(type.Name);
builder.AppendLine("{");
var enumValues = type.EnumValues.ToList();
for (var i = 0; i < enumValues.Count; i++)
{
var enumValue = enumValues[i];
builder.Append($" [EnumMember(Value=\"{enumValue.Name}\")] {NamingHelper.CapitalizeFirst(enumValue.Name)}");
if (i < enumValues.Count - 1)
{
builder.Append(",");
}
builder.AppendLine();
}
builder.AppendLine("}");
}
private static void GenerateDataClass(GraphQlType type, StringBuilder builder)
{
var className = $"{type.Name}{GraphQlGeneratorConfiguration.ClassPostfix}";
ValidateClassName(className);
builder.Append("public class ");
builder.AppendLine(className);
builder.AppendLine("{");
foreach (var field in type.Fields)
{
var propertyName = NamingHelper.CapitalizeFirst(field.Name);
string propertyType;
var fieldType = UnwrapNonNull(field.Type);
switch (fieldType.Kind)
{
case GraphQlTypeKindObject:
propertyType = $"{fieldType.Name}{GraphQlGeneratorConfiguration.ClassPostfix}";
break;
case GraphQlTypeKindEnum:
propertyType = $"{fieldType.Name}?";
break;
case GraphQlTypeKindList:
var itemType = IsObjectScalar(fieldType.OfType) ? "object" : $"{UnwrapNonNull(fieldType.OfType).Name}{GraphQlGeneratorConfiguration.ClassPostfix}";
var suggestedNetType = ScalarToNetType(UnwrapNonNull(fieldType.OfType).Name).TrimEnd('?');
if (!String.Equals(suggestedNetType, "object"))
{
itemType = suggestedNetType;
}
propertyType = $"ICollection<{itemType}>";
break;
case GraphQlTypeKindScalar:
switch (fieldType.Name)
{
case "Int":
propertyType = "int?";
break;
case "String":
propertyType = "string";
if (propertyName == "From" || propertyName == "ValidFrom" || propertyName == "CreatedAt" ||
propertyName == "To" || propertyName == "ValidTo" || propertyName == "ModifiedAt" || propertyName.EndsWith("Timestamp"))
{
propertyType = "DateTimeOffset?";
}
break;
case "Float":
propertyType = "decimal?";
break;
case "Boolean":
propertyType = "bool?";
break;
case "ID":
propertyType = "Guid?";
break;
default:
propertyType = "object";
break;
}
break;
default:
propertyType = "string";
break;
}
builder.AppendLine($" public {propertyType} {propertyName} {{ get; set; }}");
}
builder.Append("}");
}
private static void GenerateTypeQueryBuilder(GraphQlType type, StringBuilder builder)
{
var className = $"{type.Name}QueryBuilder{GraphQlGeneratorConfiguration.ClassPostfix}";
ValidateClassName(className);
builder.AppendLine($"public class {className} : GraphQlQueryBuilder<{className}>");
builder.AppendLine("{");
var fields = type.Fields.ToArray();
builder.AppendLine(" protected override IList<FieldMetadata> AllFields { get; } =");
builder.AppendLine(" new []");
builder.AppendLine(" {");
for (var i = 0; i < fields.Length; i++)
{
var comma = i == fields.Length - 1 ? null : ",";
var field = fields[i];
var fieldType = UnwrapNonNull(field.Type);
var isList = fieldType.Kind == GraphQlTypeKindList;
var isComplex = isList || IsObjectScalar(fieldType) || fieldType.Kind == GraphQlTypeKindObject;
builder.Append($" new FieldMetadata {{ Name = \"{field.Name}\"");
if (isComplex)
{
builder.Append(", IsComplex = true");
fieldType = isList ? UnwrapNonNull(fieldType.OfType) : fieldType;
if (fieldType.Kind != GraphQlTypeKindScalar)
{
builder.Append($", QueryBuilderType = typeof({fieldType.Name}QueryBuilder{GraphQlGeneratorConfiguration.ClassPostfix})");
}
}
builder.AppendLine($" }}{comma}");
}
builder.AppendLine(" };");
builder.AppendLine();
for (var i = 0; i < fields.Length; i++)
{
var field = fields[i];
var fieldType = UnwrapNonNull(field.Type);
if (fieldType.Kind == GraphQlTypeKindList)
{
fieldType = fieldType.OfType;
}
fieldType = UnwrapNonNull(fieldType);
var args = field.Args?.Where(a => UnwrapNonNull(a.Type).Kind == GraphQlTypeKindScalar || UnwrapNonNull(a.Type).Kind == GraphQlTypeKindEnum).ToArray() ?? new GraphQlArgument[0];
var methodParameters = String.Join(", ", args.Select(a => $"{(UnwrapNonNull(a.Type).Kind == GraphQlTypeKindEnum ? $"{UnwrapNonNull(a.Type).Name}?" : ScalarToNetType(UnwrapNonNull(a.Type).Name))} {a.Name} = null"));
var requiresFullBody = GraphQlGeneratorConfiguration.CSharpVersion == CSharpVersion.Compatible || args.Any();
var returnPrefix = requiresFullBody ? " return " : String.Empty;
if (fieldType.Kind == GraphQlTypeKindScalar || fieldType.Kind == GraphQlTypeKindEnum || fieldType.Kind == GraphQlTypeKindScalar)
{
builder.Append($" public {className} With{NamingHelper.CapitalizeFirst(field.Name)}({methodParameters})");
if (requiresFullBody)
{
builder.AppendLine();
builder.AppendLine(" {");
AppendArgumentDictionary(builder, args);
}
else
{
builder.Append(" => ");
}
builder.Append($"{returnPrefix}WithScalarField(\"{field.Name}\"");
if (args.Length > 0)
{
builder.Append(", args");
}
builder.AppendLine(");");
if (requiresFullBody)
{
builder.AppendLine(" }");
}
}
else
{
if (String.IsNullOrEmpty(fieldType.Name))
{
throw new InvalidOperationException($"Field '{field.Name}' type name not resolved. ");
}
var builderParameterName = NamingHelper.LowerFirst(fieldType.Name);
builder.Append($" public {className} With{NamingHelper.CapitalizeFirst(field.Name)}({fieldType.Name}QueryBuilder{GraphQlGeneratorConfiguration.ClassPostfix} {builderParameterName}QueryBuilder");
if (args.Length > 0)
{
builder.Append(", ");
builder.Append(methodParameters);
}
builder.Append(")");
if (requiresFullBody)
{
builder.AppendLine();
builder.AppendLine(" {");
}
else
{
builder.Append(" => ");
}
AppendArgumentDictionary(builder, args);
builder.Append($"{returnPrefix}WithObjectField(\"{field.Name}\", {builderParameterName}QueryBuilder");
if (args.Length > 0)
{
builder.Append(", args");
}
builder.AppendLine(");");
if (requiresFullBody)
{
builder.AppendLine(" }");
}
}
if (i < fields.Length - 1)
{
builder.AppendLine();
}
}
builder.AppendLine("}");
}
public static void GenerateDataClass(GraphQlType type, StringBuilder builder)
{
var className = $"{type.Name}{GraphQlGeneratorConfiguration.ClassPostfix}";
ValidateClassName(className);
builder.Append("public class ");
builder.AppendLine(className);
builder.AppendLine("{");
foreach (var field in type.Fields)
{
var propertyName = NamingHelper.CapitalizeFirst(field.Name);
string propertyType;
var fieldType = UnwrapNonNull(field.Type);
switch (fieldType.Kind)
{
case GraphQlTypeKindObject:
propertyType = $"{fieldType.Name}{GraphQlGeneratorConfiguration.ClassPostfix}";
break;
case GraphQlTypeKindEnum:
propertyType = $"{fieldType.Name}?";
break;
case GraphQlTypeKindList:
var itemType = IsObjectScalar(fieldType.OfType) ? "object" : $"{UnwrapNonNull(fieldType.OfType).Name}{GraphQlGeneratorConfiguration.ClassPostfix}";
var suggestedNetType = ScalarToNetType(UnwrapNonNull(fieldType.OfType).Name).TrimEnd('?');
if (!String.Equals(suggestedNetType, "object"))
{
itemType = suggestedNetType;
}
propertyType = $"ICollection<{itemType}>";
break;
case GraphQlTypeKindScalar:
switch (fieldType.Name)
{
case "Int":
propertyType = "int?";
break;
case "String":
propertyType = GraphQlGeneratorConfiguration.CustomScalarFieldMapping(field);
break;
case "Float":
propertyType = "decimal?";
break;
case "Boolean":
propertyType = "bool?";
break;
case "ID":
propertyType = "Guid?";
break;
default:
propertyType = "object";
break;
}
break;
default:
propertyType = "string";
break;
}
if (GraphQlGeneratorConfiguration.GenerateComments && !String.IsNullOrWhiteSpace(field.Description))
{
builder.AppendLine(" /// <summary>");
builder.AppendLine($" /// {field.Description}");
builder.AppendLine(" /// </summary>");
}
builder.AppendLine($" public {propertyType} {propertyName} {{ get; set; }}");
}
builder.Append("}");
}