Merge pull request #13 from toshi0383/ts-launchview-icon

Customizable LaunchView's Icon

Author: noppefoxwolf

Sources/DebugMenu/Entity/Options.swift

@@ -6,10 +6,22 @@
 //
 
 import Foundation
+import UIKit
 
 public enum Options {
     case showsWidgetOnLaunch
     case showsRecentItems
+    case launchIcon(UIImage)
 
     public static var `default`: [Options] = [.showsRecentItems]
+
+    var isShowsWidgetOnLaunch: Bool {
+        if case .showsWidgetOnLaunch = self { return true }
+        return false
+    }
+
+    var isShowsRecentItems: Bool {
+        if case .showsRecentItems = self { return true }
+        return false
+    }
 }

Sources/DebugMenu/View/FloatingViewController.swift

@@ -10,7 +10,7 @@ import UIKit
 
 internal class FloatingViewController: UIViewController {
     class View: UIView, TouchThrowing {}
-    private let launchView: LaunchView = .init()
+    private let launchView: LaunchView
     private let widgetView: WidgetView
     private let debuggerItems: [DebugItem]
     private var cancellables: Set<AnyCancellable> = []
@@ -24,6 +24,14 @@ internal class FloatingViewController: UIViewController {
         self.debuggerItems = debuggerItems
         self.widgetView = .init(dashboardItems: dashboardItems)
         self.options = options
+
+        launchView = .init(image: options.compactMap { option -> UIImage? in
+            if case .launchIcon(let image) = option {
+                return image
+            }
+            return nil
+        }.first)
+
         super.init(nibName: nil, bundle: nil)
     }
 
@@ -77,7 +85,7 @@ internal class FloatingViewController: UIViewController {
             gesture.moveInitialPosition(.topLeading)
         }
 
-        if options.contains(.showsWidgetOnLaunch) {
+        if options.contains(where: { $0.isShowsWidgetOnLaunch }) {
             widgetView.show()
         } else {
             widgetView.hide()

Sources/DebugMenu/View/InAppDebuggerViewController.swift

@@ -259,7 +259,7 @@ extension InAppDebuggerViewController {
             snapshot.appendItems(filteredItems, toSection: .items)
         } else {
             let recentItems = RecentItemStore(items: debuggerItems).get()
-            if !recentItems.isEmpty && options.contains(.showsRecentItems) {
+            if !recentItems.isEmpty && options.contains(where: { $0.isShowsRecentItems }) {
                 snapshot.appendSections([Section.recent])
                 snapshot.appendItems(recentItems, toSection: .recent)
             }

Sources/DebugMenu/View/LaunchView.swift

@@ -10,12 +10,11 @@ import UIKit
 class LaunchView: UIVisualEffectView {
     private let button: UIButton = .init(frame: .null)
 
-    init() {
+    init(image: UIImage? = nil) {
         super.init(effect: UIBlurEffect(style: .systemMaterialDark))
         frame = CGRect(x: 0, y: 0, width: 44, height: 44)
 
-        let image = UIImage(systemName: "ant.fill")
-        button.setImage(image, for: .normal)
+        button.setImage(image ?? UIImage(systemName: "ant.fill"), for: .normal)
         button.tintColor = UIColor.white
         button.translatesAutoresizingMaskIntoConstraints = false
         contentView.addSubview(button)