Fix specification of asset path

zhangao0086 · Jan 19, 2021 · 8e867f5 · 8e867f5
1 parent f7a5635
commit 8e867f5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/Sources/DKImagePickerController/Resource/DKImagePickerControllerResource.swift b/Sources/DKImagePickerController/Resource/DKImagePickerControllerResource.swift
@@ -99,8 +99,12 @@ public class DKImagePickerControllerResource: NSObject {
 private extension Bundle {
 
     class func imagePickerControllerBundle() -> Bundle {
-        let assetPath = Bundle(for: DKImagePickerControllerResource.self).resourcePath!
-        return Bundle(path: (assetPath as NSString).appendingPathComponent("DKImagePickerController.bundle"))!
+        #if SWIFT_PACKAGE
+            return Bundle.module
+        #else
+            let assetPath = Bundle(for: DKImagePickerControllerResource.self).resourcePath!
+            return Bundle(path: (assetPath as NSString).appendingPathComponent("DKImagePickerController.bundle"))!
+        #endif
     }
 
 }