Rewrite

cap-js · Jan 30, 2024 · a19322c · a19322c
1 parent 435f4dd
commit a19322c
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/lib/visitor.js b/lib/visitor.js
@@ -449,16 +449,14 @@ class Visitor {
             case 'aspect':
                 this.#printAspect(name, entity)
                 break
-            case 'type':
+            case 'type': {
                 // types like inline definitions can be used very similarly to entities.
                 // They can be extended, contain inline enums, etc., so we treat them as entities.
                 // Enums seem to be the only actual "type".
-                if (entity.enum) {
-                    this.#printType(name, entity)
-                } else {
-                    this.#printEntity(name, entity)    
-                }
+                const handler = entity.enum ? this.#printType : this.#printEntity
+                handler.bind(this)(name, entity)
                 break
+            }
             case 'event':
                 this.#printEvent(name, entity)
                 break