diff --git a/src/main/java/io/github/lonamiwebs/overgram/generator/Generator.java b/src/main/java/io/github/lonamiwebs/overgram/generator/Generator.java index c5b0304..c4aa8c8 100644 --- a/src/main/java/io/github/lonamiwebs/overgram/generator/Generator.java +++ b/src/main/java/io/github/lonamiwebs/overgram/generator/Generator.java @@ -18,7 +18,7 @@ public class Generator { writeAbstract(writer, types); } try (final Writer writer = new BufferedWriter(new FileWriter(typesFile))) { - writeCode(writer, "Types", "TLObject", types); + writeCode(writer, "Types", null, types); } try (final Writer writer = new BufferedWriter(new FileWriter(functionsFile))) { writeCode(writer, "Functions", "TLRequest", functions); @@ -68,7 +68,12 @@ public class Generator { writer.write("public static class "); writer.write(TLObject.toCamelCase(object.name)); writer.write(" extends "); - writer.write(extendsName); + if (extendsName == null) { + writer.write("Abstract."); + writer.write(TLObject.toCamelCase(object.type)); + } else { + writer.write(extendsName); + } writer.write(" {\n"); for (int i = 0; i < object.args.size(); ++i) {