feat: add a utils method to get all enums in a type recursivly

src/index.ts

@@ -1,4 +1,6 @@
-export * from "./models"
 export { Parser } from "./parser"
 export * from "./rewrite"
 export * from "./utils/prettify"
+
+export * from "./models"
+export * from "./writer"

src/models/UnionType.ts

@@ -5,6 +5,7 @@ import { Type } from "./Type"
 export class UnionType extends Type {
 	public types: Array<Type>
 
+	// TODO these `types` should be a parameter instead of creating a dependency to the `typescript` lib here
 	constructor(node: UnionTypeNode) {
 		super()
 		this.types = node.types.map(TypeFactory.create)

src/writer/index.ts

@@ -0,0 +1 @@
+export * from "./utils/getAllEnumsInside"

src/writer/utils/getAllEnumsInside.ts

@@ -0,0 +1,30 @@
+import { ArrayType, EnumType, IntersectionType, TypeLiteral, UnionType } from "../../models"
+import { Type } from "../../models/Type"
+import { L } from "../../utils/logger"
+
+export function getAllEnumsInside(type: Type): Set<EnumType> {
+	const s = new Set<EnumType>()
+
+	function append(set: Set<EnumType>) {
+		set.forEach((e) => s.add(e))
+	}
+
+	L.d("<getAllEnumsInside>")
+	if (type instanceof UnionType) {
+		L.d("isUnionType")
+		type.types.map(getAllEnumsInside).forEach(append)
+	} else if (type instanceof IntersectionType) {
+		L.d("isIntersection")
+		type.types.map(getAllEnumsInside).forEach(append)
+	} else if (type instanceof ArrayType) {
+		L.d("isArray")
+		append(getAllEnumsInside(type.arrayType))
+	} else if (type instanceof TypeLiteral) {
+		L.d("isTypeLiteral")
+		Array.from(type.properties.values()).map(getAllEnumsInside).forEach(append)
+	} else if (type instanceof EnumType) {
+		s.add(type)
+	}
+
+	return s
+}