MPMediaPickerController wrapped for SwiftUI

MediaPicker.swift

//
//  MediaPicker.swift
//  Music Triage
//
//  Created by Kyle Hughes on 9/6/22.
//

import Foundation
import MediaPlayer
import SwiftUI

struct MediaPicker {
    @Binding private var selection: MPMediaItemCollection?
    
    @Environment(\.dismiss) private var dismiss
    
    // MARK: Internal Initialization
    
    init(selection: Binding<MPMediaItemCollection?>) {
        _selection = selection
    }
}

// MARK: - UIViewControllerRepresentable Extension

extension MediaPicker: UIViewControllerRepresentable {
    // MARK: Creating and Updating the View Controller
    
    func makeCoordinator() -> Coordinator {
        Coordinator(dismiss: dismiss, selection: $selection)
    }

    func makeUIViewController(context: UIViewControllerRepresentableContext<MediaPicker>) -> MPMediaPickerController {
        let picker = MPMediaPickerController(mediaTypes: [.music])
        picker.allowsPickingMultipleItems = true
        picker.delegate = context.coordinator
        picker.showsItemsWithProtectedAssets = true
        
        return picker
    }

    func updateUIViewController(
        _ uiViewController: MPMediaPickerController,
        context: UIViewControllerRepresentableContext<MediaPicker>
    ) {
        // NO-OP
    }
}

// MARK: - MediaPicker.Coordinator Definition

extension MediaPicker {
    final class Coordinator: NSObject {
        private let dismiss: DismissAction
        private let selection: Binding<MPMediaItemCollection?>
        
        // MARK: Internal Initialization
        
        init(dismiss: DismissAction, selection: Binding<MPMediaItemCollection?>) {
            self.dismiss = dismiss
            self.selection = selection
        }
    }
}

// MARK: - MPMediaPickerControllerDelegate Extension

extension MediaPicker.Coordinator: MPMediaPickerControllerDelegate {
    // MARK: Responding to User Actions
    
    func mediaPicker(
        _ mediaPicker: MPMediaPickerController,
        didPickMediaItems mediaItemCollection: MPMediaItemCollection
    ) {
        selection.wrappedValue = mediaItemCollection
        
        dismiss()
    }
    
    func mediaPickerDidCancel(_ mediaPicker: MPMediaPickerController) {
        dismiss()
    }
}