fix: check if superclass has local header (#444)

blacktop · Apr 18, 2024 · af4360b · af4360b
1 parent 3bfd710
commit af4360b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/commands/macho/objc.go b/internal/commands/macho/objc.go
@@ -890,8 +890,12 @@ func (o *ObjC) processForwardDeclarations(m *macho.File) (map[string]Imports, er
 	}
 
 	for _, class := range classes {
-		imp := Imports{}
-		imp.Imports = append(imp.Imports, class.SuperClass+".h")
+		var imp Imports
+		if slices.Contains(classNames, class.SuperClass) {
+			imp.Locals = append(imp.Locals, class.SuperClass+".h")
+		} else {
+			imp.Classes = append(imp.Classes, class.SuperClass)
+		}
 		for _, prot := range class.Protocols {
 			if slices.Contains(protoNames, prot.Name) {
 				imp.Locals = append(imp.Locals, prot.Name+"-Protocol.h")