YogaKit/Source/UIView+Yoga.h

@@ -8,10 +8,10 @@
 #import <TargetConditionals.h>
 
 #if TARGET_OS_OSX
-#import <AppKit/AppKit.h>
+@import AppKit;
 #define UIView NSView
 #else
-#import <UIKit/UIKit.h>
+@import UIKit;
 #endif
 
 #import "YogaKit.h"

YogaKit/Source/UIView+Yoga.m

@@ -5,9 +5,9 @@
  * LICENSE file in the root directory of this source tree.
  */
 
-#import <objc/runtime.h>
 #import "UIView+Yoga.h"
 #import "YGLayout+Private.h"
+@import ObjectiveC;
 
 static const void* kYGYogaAssociatedKey = &kYGYogaAssociatedKey;
 

YogaKit/Source/YogaKit.h

@@ -8,9 +8,9 @@
 #import <TargetConditionals.h>
 
 #if TARGET_OS_OSX
-#import <AppKit/AppKit.h>
+@import AppKit;
 #else
-#import <UIKit/UIKit.h>
+@import UIKit;
 #endif
 
 #import "../../yoga/YGEnums.h"

YogaKit/Source/modulemap/YogaKit-umbrella.h

@@ -1,2 +0,0 @@
-#import "../YogaKit.h"
-#import "../UIView+Yoga.h"

YogaKit/Source/modulemap/module.modulemap

@@ -1,4 +1,5 @@
 module YogaKit {
-    umbrella header "YogaKit-umbrella.h"
+    header "../YogaKit.h"
+    header "../UIView+Yoga.h"
     export *
 }

yoga/modulemap/Yoga-umbrella.h

@@ -1,8 +0,0 @@
-#ifdef __OBJC__
-#import <Foundation/Foundation.h>
-#endif
-
-#import "../YGEnums.h"
-#import "../YGMacros.h"
-#import "../YGValue.h"
-#import "../Yoga.h"

yoga/modulemap/module.modulemap

@@ -1,4 +1,7 @@
 module Yoga [extern_c] {
-    umbrella header "Yoga-umbrella.h"
+    header "../Yoga.h"
+    header "../YGValue.h"
+    header "../YGMacros.h"
+    header "../YGEnums.h"
     export *
 }