Enhance MediaInfoManager and Reporter for improved media playback monitoring

- Integrated Combine framework into MediaInfoManager for reactive media state changes.
- Added notification handling for playback state changes, allowing the Reporter to respond to media updates.
- Updated the Reporter class to utilize the new media monitoring capabilities, sending media info alongside window focus data.
- Refactored the prepareSend method to accommodate optional media information, enhancing reporting flexibility.

These changes improve the application's ability to track and report media playback status in real-time, enhancing user experience and functionality.

Signed-off-by: Innei <tukon479@gmail.com>

Author: Innei

ProcessReporter/Core/MediaInfoManager/MediaInfoManager.swift

@@ -3,6 +3,7 @@
 // Created by Innei on 2025/4/11.
 
 import AppKit
+import Combine
 import Foundation
 
 // Recreating the MRContent classes in Swift
@@ -33,6 +34,78 @@ typealias MRMediaRemoteGetNowPlayingApplicationPIDFunction = @convention(c) (
 ) -> Void
 
 public class MediaInfoManager: NSObject {
+  // Notification name for media playing state changes
+  private static let playingStateChangedNotificationName =
+    "kMRMediaRemoteNowPlayingApplicationIsPlayingDidChangeNotification"
+  private static let applicationChangedNotificationName =
+    "kMRMediaRemoteNowPlayingApplicationDidChangeNotification"
+  private static let infoChangedNotificationName =
+    "kMRMediaRemoteNowPlayingInfoDidChangeNotification"
+
+  // Callback for when playback state changes
+  public typealias PlaybackStateChangedCallback = (MediaInfo) -> Void
+
+  // Store the callback
+  private static var playbackStateChangedCallback: PlaybackStateChangedCallback?
+
+  // Observer for the notification
+  private static var observer: Any?
+
+  // Static initializer to set up the notification center
+  private static let setupOnce: Void = {
+    loadMediaRemoteFramework()
+  }()
+  private static var cancellables = Set<AnyCancellable>()
+  private static var mediaInfo: MediaInfo?
+
+  // Load MediaRemote framework and register for notifications
+  private static func loadMediaRemoteFramework() {
+    let url = URL(fileURLWithPath: "/System/Library/PrivateFrameworks/MediaRemote.framework")
+    guard CFBundleCreate(kCFAllocatorDefault, url as CFURL) != nil else {
+      print("Failed to load MediaRemote framework")
+      return
+    }
+
+    for name in [
+      playingStateChangedNotificationName, applicationChangedNotificationName,
+      infoChangedNotificationName,
+    ] {
+
+      NotificationCenter.default.publisher(for: Notification.Name(name)).sink { _ in
+        if let callback = MediaInfoManager.playbackStateChangedCallback {
+          DispatchQueue.main.async {
+            guard let mediaInfo = MediaInfoManager.getMediaInfo() else {
+              return
+            }
+            callback(mediaInfo)
+          }
+        }
+      }.store(in: &cancellables)
+    }
+  }
+
+  // Setup the notification observer
+  public static func startMonitoringPlaybackChanges(
+    callback: @escaping PlaybackStateChangedCallback
+  ) {
+    // Make sure framework is loaded
+    _ = setupOnce
+
+    // Remove existing observer if there is one
+    if let observer = observer {
+      NotificationCenter.default.removeObserver(observer)
+    }
+
+    // Store the callback
+    playbackStateChangedCallback = callback
+  }
+
+  // Stop monitoring playback changes
+  public static func stopMonitoringPlaybackChanges() {
+    cancellables.removeAll()
+    playbackStateChangedCallback = nil
+  }
+
   public static func getMediaInfo() -> MediaInfo? {
     if let nowPlayingInfo = getNowPlayingInfo() {
       let name = nowPlayingInfo["name"] as? String
@@ -53,7 +126,8 @@ public class MediaInfoManager: NSObject {
         name: name, artist: artist, album: album, image: artworkData, duration: duration,
         elapsedTime: elapsedTime, processID: processID, processName: processName,
         executablePath: executablePath, playing: playing,
-        applicationIdentifier: bundleID)
+        applicationIdentifier: bundleID
+      )
     }
     return nil
   }
@@ -73,17 +147,22 @@ public class MediaInfoManager: NSObject {
       // Get function pointers from the framework
       let getMRMediaRemoteGetNowPlayingInfo = unsafeBitCast(
         CFBundleGetFunctionPointerForName(bundle, "MRMediaRemoteGetNowPlayingInfo" as CFString),
-        to: MRMediaRemoteGetNowPlayingInfoFunction.self)
+        to: MRMediaRemoteGetNowPlayingInfoFunction.self
+      )
 
       let getMRMediaRemoteGetNowPlayingApplicationIsPlaying = unsafeBitCast(
         CFBundleGetFunctionPointerForName(
-          bundle, "MRMediaRemoteGetNowPlayingApplicationIsPlaying" as CFString),
-        to: MRMediaRemoteGetNowPlayingApplicationIsPlayingFunction.self)
+          bundle, "MRMediaRemoteGetNowPlayingApplicationIsPlaying" as CFString
+        ),
+        to: MRMediaRemoteGetNowPlayingApplicationIsPlayingFunction.self
+      )
 
       let getMRMediaRemoteGetNowPlayingApplicationPID = unsafeBitCast(
         CFBundleGetFunctionPointerForName(
-          bundle, "MRMediaRemoteGetNowPlayingApplicationPID" as CFString),
-        to: MRMediaRemoteGetNowPlayingApplicationPIDFunction.self)
+          bundle, "MRMediaRemoteGetNowPlayingApplicationPID" as CFString
+        ),
+        to: MRMediaRemoteGetNowPlayingApplicationPIDFunction.self
+      )
 
       // Get playing status
       var isPlaying = false

ProcessReporter/Core/Reporter/Reporter.swift

@@ -174,18 +174,51 @@ class Reporter {
 	private func monitor() {
 		ApplicationMonitor.shared.startMouseMonitoring()
 		ApplicationMonitor.shared.startWindowFocusMonitoring()
-		ApplicationMonitor.shared.onWindowFocusChanged = { [unowned self] info in
-			if PreferencesDataModel.shared.focusReport.value {
+		ApplicationMonitor.shared.onWindowFocusChanged = { [weak self] info in
+			guard let self = self else { return }
+			if PreferencesDataModel.shared.focusReport.value
+				&& PreferencesDataModel.shared.enabledTypes.value.types.contains(.process)
+			{
 				self.prepareSend(windowInfo: info)
 			}
 		}
 
+		MediaInfoManager.startMonitoringPlaybackChanges { [weak self] mediaInfo in
+			guard let self = self else { return }
+			if PreferencesDataModel.shared.enabledTypes.value.types.contains(.media) {
+				self.prepareSend(
+					windowInfo: ApplicationMonitor.shared.getFocusedWindowInfo(),
+					mediaInfo: mediaInfo
+				)
+			}
+		}
 		statusItemManager.toggleStatusItemIcon(.syncing)
 	}
 
 	private var reporterInitializedTime: Date
 
-	private func prepareSend(windowInfo: FocusedWindowInfo) {
+	private func prepareSend(
+		windowInfo optionalWindowInfo: FocusedWindowInfo?,
+		mediaInfo optionalMediaInfo: MediaInfo? = nil
+	) {
+
+		var windowInfo: FocusedWindowInfo!
+		if let optionalWindowInfo = optionalWindowInfo {
+			windowInfo = optionalWindowInfo
+		} else {
+			windowInfo = ApplicationMonitor.shared.getFocusedWindowInfo()
+			if windowInfo == nil {
+				return
+			}
+		}
+
+		var mediaInfo: MediaInfo?
+		if let optionalMediaInfo = optionalMediaInfo {
+			mediaInfo = optionalMediaInfo
+		} else {
+			mediaInfo = MediaInfoManager.getMediaInfo()
+		}
+
 		let appName = windowInfo.appName
 		let now = Date()
 		// Ignore the first 2 seconds after initialization to wait for the setting synchronization to complete
@@ -205,8 +238,6 @@ class Reporter {
 			statusItemManager.toggleStatusItemIcon(.syncing)
 		}
 
-		let mediaInfo = MediaInfoManager.getMediaInfo()
-
 		var dataModel = ReportModel(
 			windowInfo: nil,
 			integrations: [],
@@ -218,8 +249,8 @@ class Reporter {
 			// Filter media name
 
 			if !cachedFilteredMediaAppNames.contains(mediaInfo.processName),
-			   !shouldIgnoreArtistNull
-			   || (mediaInfo.artist != nil && !mediaInfo.artist!.isEmpty)
+				!shouldIgnoreArtistNull
+					|| (mediaInfo.artist != nil && !mediaInfo.artist!.isEmpty)
 			{
 				dataModel.setMediaInfo(mediaInfo)
 			}
@@ -254,8 +285,8 @@ class Reporter {
 
 		let interval = PreferencesDataModel.shared.sendInterval.value
 		timer = Timer.scheduledTimer(
-			withTimeInterval: TimeInterval(interval.rawValue), repeats: true)
-		{ [weak self] _ in
+			withTimeInterval: TimeInterval(interval.rawValue), repeats: true
+		) { [weak self] _ in
 			Task { @MainActor in
 				guard let self = self else { return }
 				if let info = ApplicationMonitor.shared.getFocusedWindowInfo() {
@@ -284,7 +315,7 @@ class Reporter {
 		let extensions: [ReporterExtension] = [
 			MixSpaceReporterExtension(),
 			S3ReporterExtension(),
-			SlackReporterExtension()
+			SlackReporterExtension(),
 		]
 
 		for ext in extensions {
@@ -363,7 +394,8 @@ extension Reporter {
 		let d3 = Observable.combineLatest(
 			preferences.mixSpaceIntegration,
 			preferences.s3Integration,
-			preferences.slackIntegration).subscribe { [weak self] _ in
+			preferences.slackIntegration
+		).subscribe { [weak self] _ in
 			guard let self = self else { return }
 			Task {
 				await self.updateExtensions()

ProcessReporter/Core/Utilities/ApplicationMonitor.swift

@@ -43,7 +43,7 @@ class ApplicationMonitor {
                     NSWorkspace.shared.open(
                         URL(
                             string:
-                            "x-apple.systempreferences:com.apple.preference.security?Privacy_Accessibility"
+                                "x-apple.systempreferences:com.apple.preference.security?Privacy_Accessibility"
                         )!)
                 }
             }
@@ -58,14 +58,16 @@ class ApplicationMonitor {
         let appElement = AXUIElementCreateApplication(pid)
 
         var mainWindow: CFTypeRef?
-        let mainWindowError = AXUIElementCopyAttributeValue(appElement, kAXMainWindowAttribute as CFString, &mainWindow)
+        let mainWindowError = AXUIElementCopyAttributeValue(
+            appElement, kAXMainWindowAttribute as CFString, &mainWindow)
         let window = mainWindow
         guard mainWindowError == .success else {
             return nil
         }
 
         var title: CFTypeRef?
-        let titleError = AXUIElementCopyAttributeValue(window as! AXUIElement, kAXTitleAttribute as CFString, &title)
+        let titleError = AXUIElementCopyAttributeValue(
+            window as! AXUIElement, kAXTitleAttribute as CFString, &title)
         guard titleError == .success, let titleString = title as? String else {
             return nil
         }
@@ -75,15 +77,16 @@ class ApplicationMonitor {
 
     func getFocusedWindowInfo() -> FocusedWindowInfo? {
         guard isAccessibilityEnabled() else {
-            checkAndRequestAccessibilityPermissions()
+            ToastManager.shared.error(
+                "Accessibility permissions are required to monitor window changes.")
             return nil
         }
 
         guard let app = NSWorkspace.shared.frontmostApplication else {
             return nil
         }
         let applicationIdentifier = app.bundleIdentifier ?? ""
-        
+
         if IgnoreSystemApplication.contains(applicationIdentifier) {
             return nil
         }
@@ -143,7 +146,7 @@ class ApplicationMonitor {
             queue: .main
         ) { [weak self] _ in
             guard let self = self,
-                  let windowInfo = self.getFocusedWindowInfo()
+                let windowInfo = self.getFocusedWindowInfo()
             else {
                 return
             }