Implement getting inline specs

fern-api · Jan 14, 2025 · 6062279 · 6062279
1 parent f3ef8c6
commit 6062279
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 12 deletions.
diff --git a/...rs/java/generator-utils/src/main/java/com/fern/java/generators/AbstractTypeGenerator.java b/...rs/java/generator-utils/src/main/java/com/fern/java/generators/AbstractTypeGenerator.java
@@ -9,7 +9,9 @@
 import com.squareup.javapoet.ClassName;
 import com.squareup.javapoet.JavaFile;
 import com.squareup.javapoet.TypeSpec;
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
 import java.util.Set;
 
 public abstract class AbstractTypeGenerator extends AbstractFileGenerator {
@@ -33,15 +35,30 @@ public TypeSpec getTypeSpec() {
     }
 
     private List<TypeSpec> getInlineTypeSpecs() {
-        //        if (generatorContext.getCustomConfig().enableInlineTypes()) {
-        //            return List.of();
-        //        }
-        //
-        //        List<TypeDeclaration> declarations = getInlineTypeDeclarations();
-        //        for (TypeDeclaration declaration : declarations) {
-        //
-        //        }
-        return List.of();
+        if (generatorContext.getCustomConfig().enableInlineTypes()) {
+            return List.of();
+        }
+
+        List<TypeDeclaration> declarations = getInlineTypeDeclarations();
+        List<TypeSpec> result = new ArrayList<>();
+        for (TypeDeclaration declaration : declarations) {
+            Optional<AbstractTypeGenerator> generator = declaration
+                    .getShape()
+                    .visit(new SingleTypeGenerator(
+                            generatorContext,
+                            declaration.getName(),
+                            className.nestedClass(declaration
+                                    .getName()
+                                    .getName()
+                                    .getPascalCase()
+                                    .getSafeName()),
+                            TypesGenerator.getGeneratedInterfaces(generatorContext),
+                            false,
+                            reservedTypeNames));
+            generator.map(AbstractTypeGenerator::getTypeSpec).ifPresent(result::add);
+        }
+
+        return result;
     }
 
     protected String resolveName(String rawName) {

diff --git a/generators/java/generator-utils/src/main/java/com/fern/java/generators/TypesGenerator.java b/generators/java/generator-utils/src/main/java/com/fern/java/generators/TypesGenerator.java
@@ -43,7 +43,7 @@ public TypesGenerator(AbstractGeneratorContext<?, ?> generatorContext) {
     }
 
     public Result generateFiles() {
-        Map<TypeId, GeneratedJavaInterface> generatedInterfaces = getGeneratedInterfaces();
+        Map<TypeId, GeneratedJavaInterface> generatedInterfaces = getGeneratedInterfaces(generatorContext);
         Map<TypeId, GeneratedJavaFile> generatedTypes = KeyedStream.stream(typeDeclarations)
                 .map(typeDeclaration -> {
                     ClassName className =
@@ -65,10 +65,12 @@ public Result generateFiles() {
         return new Result(generatedInterfaces, generatedTypes);
     }
 
-    private Map<TypeId, GeneratedJavaInterface> getGeneratedInterfaces() {
+    public static Map<TypeId, GeneratedJavaInterface> getGeneratedInterfaces(
+            AbstractGeneratorContext<?, ?> generatorContext) {
         Set<TypeId> interfaceCandidates = generatorContext.getInterfaceIds();
         return interfaceCandidates.stream().collect(Collectors.toMap(Function.identity(), typeId -> {
-            TypeDeclaration typeDeclaration = typeDeclarations.get(typeId);
+            TypeDeclaration typeDeclaration =
+                    generatorContext.getTypeDeclarations().get(typeId);
             ObjectTypeDeclaration objectTypeDeclaration = typeDeclaration
                     .getShape()
                     .getObject()