public static NestedText GenSerializer(Arr <Arg> args, Option <int> typeNumber, string funcName) { Text GenSerializer(TgType type) => type.Match( primitive: x => $"Write{x.Type}", typeRef: x => "WriteSerializable", vector: x => Concat( $"WriteVector<{TgTypeConverter.ConvertType(x.Type)}>(", GenSerializer(x.Type), ")" ) ); Option <Text> GenNonFlagArgSerializer(Arg arg) => arg.Kind.Match( _: () => throw new Exception("WTF"), required: _ => GenSerializer(arg.Type).Apply(Some), optional: x => arg.Type == TgType.OfPrimitive(PrimitiveType.True) ? None : Concat( $"WriteOption<{TgTypeConverter.ConvertType(arg.Type)}>(", GenSerializer(arg.Type), ")" ).Apply(Some) ).Map(s => Concat($"Write({arg.Name}, bw, ", s, ");") ); Text GenMaskSerializer(IEnumerable <(string, int)> maskArgs) => maskArgs.ToArr() .Apply(Optional).Filter(xs => xs.Count > 0) .Map(xs => xs .Map(x => $"MaskBit({x.Item2}, {x.Item1})").Map(String).Reduce((x, y) => Concat(x, " | ", y)) ) .IfNone("0") .Apply(mask => Concat("Write(", mask, ", bw, WriteInt);")); Option <Text> GenArgSerializer(Arg arg) => arg.Kind.Match( _: () => GenNonFlagArgSerializer(arg), flags: _ => args.Choose(x => x.Kind.Match( _: () => None, optional: optional => ($"{x.Name}", optional.Flag.Bit).Apply(Some) )).Apply(GenMaskSerializer) ); var body = args.Choose(GenArgSerializer).Map(Line).Scope().Apply(s => typeNumber .Map(Helpers.TypeNumber).Map(x => Line($"WriteUint(bw, {x});")).Map(typeNumSer => Scope(typeNumSer, s)) .IfNone(s) ); var def = Scope( Line($"void {funcName}(BinaryWriter bw)"), Line("{"), Indent(1, body), Line("}") ); return(def); }
public static NestedText GenTagYamlifier(Arr <Arg> args, string typeName) { Text GenYamlifier(bool tagOnly, TgType type) => type.Match( primitive: x => $"Yamlifier.Write{x.Type}", typeRef: x => $"{TgTypeConverter.ConvertType(type, cmpWrapper: false)}.Yamlify(tagOnly: {(tagOnly ? "true" : "false")})", vector: x => Concat( $"Yamlifier.StringifyVector<{TgTypeConverter.ConvertType(x.Type, cmpWrapper: false)}>(", GenYamlifier(tagOnly, x.Type), ")" ) );
static NestedText GenFunc(Signature func, string funcName) { var argsWithoutFlags = func.Args .Filter(x => x.Kind.Match(_: () => true, flags: _ => false)) .ToArr(); var funcArgs = argsWithoutFlags .Map(x => ( name: x.Name, lowerName: Helpers.LowerFirst(x.Name), type: TgTypeConverter.ConvertArgType(x), isRef: TgTypeConverter.IsRefArgType(x) )).ToArray(); // usually it is a higher-order function, i am too lazy to modify the scheme just for this case var isWrapper = func.Args.Exists(x => x.Type == TgType.OfTypeRef("X") || x.Type == TgType.OfTypeRef("!X")); var resType = isWrapper ? "TFuncRes" : TgTypeConverter.ConvertType(func.ResultType); var classAccess = isWrapper ? "" : "public "; var classTemplates = isWrapper ? "<TFunc, TFuncRes>" : ""; var classAnnotations = isWrapper ? $": ITgFunc<{resType}> where TFunc : class, ITgFunc<{resType}>" : $": ITgFunc<{resType}>, IEquatable<{funcName}>, IComparable<{funcName}>, IComparable"; var resDes = isWrapper ? "Query.DeserializeResult(br);" // it is 'Query' all the time, i am too lazy : Concat("Read(br, ", SerializerGen.GenTypeDeserializer(func.ResultType), ");"); var resultDeserializer = Scope( Line($"{resType} ITgFunc<{resType}>.DeserializeResult(BinaryReader br) =>"), Indent(1, Line(resDes)) ); return(Scope( Line($"{classAccess}sealed class {funcName}{classTemplates} {classAnnotations}"), Line("{"), IndentedScope(1, funcArgs.Map(arg => Line($"public {arg.type} {arg.name} {{ get; }}")).Scope(), Scope( Line(""), Scope( Line($"public {funcName}("), IndentedScope(1, $",{Environment.NewLine}", funcArgs.Map(arg => Line($"{arg.type} {arg.lowerName}")) ), Line(") {"), IndentedScope(1, funcArgs.Map(arg => Line( $"{arg.name} = {arg.lowerName}" + $"{(arg.isRef ? $" ?? throw new ArgumentNullException(nameof({arg.lowerName}))" : "")};" )) ), Line("}") ), Line(""), Line(""), isWrapper ? Scope(new NestedText[0]) : Scope( WithGen.GenWith(argsWithoutFlags, funcName), Line(""), RelationsGen.GenRelations(funcName, argsWithoutFlags), Line("") ), SerializerGen.GenSerializer(func.Args, typeNumber: func.TypeNumber, "ITgSerializable.Serialize"), Line(""), resultDeserializer ) ), Line("}") )); }