EnumExtender: Support sealed classes.

Chocohead · Mar 28, 2023 · 11edbe4 · 11edbe4
1 parent 6cc4a93
commit 11edbe4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/com/chocohead/mm/EnumExtender.java b/src/com/chocohead/mm/EnumExtender.java
@@ -176,6 +176,10 @@ static Consumer<ClassNode> makeEnumExtender(EnumAdder builder) {
 			InsnList arrayFilling = new InsnList();
 
 			for (EnumAddition addition : builder.getAdditions()) {
+				String additionType = addition.isEnumSubclass() ? anonymousClassFactory.get() : node.name;
+				if (addition.isEnumSubclass() && node.permittedSubclasses != null) {
+					node.permittedSubclasses.add(additionType);
+				}
 				node.visitField(Opcodes.ACC_PUBLIC + Opcodes.ACC_FINAL + Opcodes.ACC_STATIC + Opcodes.ACC_ENUM, addition.name, 'L' + node.name + ';', null, null);
 
 				LabelNode stuffStart;
@@ -193,7 +197,6 @@ static Consumer<ClassNode> makeEnumExtender(EnumAdder builder) {
 					fieldSetting.add(stuffStart);
 				} else stuffStart = null;
 
-				String additionType = addition.isEnumSubclass() ? anonymousClassFactory.get() : node.name;
 				fieldSetting.add(new TypeInsnNode(Opcodes.NEW, additionType));
 				fieldSetting.add(new InsnNode(Opcodes.DUP));