feat: add appIcon

jaywcjlove · Oct 1, 2024 · ea1c694 · ea1c694
1 parent e6a52af
commit ea1c694
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -18,4 +18,5 @@ DeviceInfo.systemVersion.patchVersion        // e.g "0"
 DeviceInfo.systemMajorVersion   // e.g "1"
 DeviceInfo.systemMinorVersion   // e.g "3"
 DeviceInfo.systemPatchVersion   // e.g "0"
-```
+DeviceInfo.appIcon              // -> UIImage | NSImage
+```
diff --git a/Sources/DeviceInfo/DeviceInfo.swift b/Sources/DeviceInfo/DeviceInfo.swift
@@ -52,6 +52,23 @@ public struct DeviceInfo {
     public static let isDarkMode: Bool = { NSApp?.effectiveAppearance.isDarkMode ?? false }()
     #elseif canImport(WatchKit)
     #endif
+
+    #if canImport(UIKit)
+    static let appIcon: UIImage? = getAppIcon()
+    static func getAppIcon() -> UIImage? {
+        if let iconsDictionary = Bundle.main.infoDictionary?["CFBundleIcons"] as? [String: Any],
+           let primaryIconsDictionary = iconsDictionary["CFBundlePrimaryIcon"] as? [String: Any],
+           let iconFiles = primaryIconsDictionary["CFBundleIconFiles"] as? [String],
+           let lastIcon = iconFiles.last {
+            return UIImage(named: lastIcon)
+        }
+        return nil
+    }
+    #elseif canImport(AppKit)
+    @MainActor
+    static let appIcon: NSImage = NSApp.applicationIconImage!
+    #elseif canImport(WatchKit)
+    #endif
 }