iOS Swift中能否自定义UIImagePickerController实现相机相册切换

无法直接通过自定义UIImagePickerController实现相机界面内切换到相册的功能——系统的UIImagePickerController是封闭的控件，苹果不允许修改它的内部UI结构，强行添加子视图不仅可能违反App Store审核规则，还会因为不同iOS版本的界面布局差异导致兼容性问题。

推荐以下两种可行方案：

方案1：临时叠加自定义按钮（仅适合测试场景） #

在相机模式的UIImagePickerController上手动添加相册跳转按钮，点击后关闭当前相机控制器，再打开相册模式的选择器。

class TempImagePicker: UIImagePickerController {
    override func viewDidLoad() {
        super.viewDidLoad()
        guard sourceType == .camera else { return }
        
        // 添加相册切换按钮
        let albumBtn = UIButton(type: .system)
        albumBtn.setImage(UIImage(systemName: "photo.on.rectangle"), for: .normal)
        albumBtn.frame = CGRect(x: view.bounds.width - 60, y: view.bounds.height - 120, width: 40, height: 40)
        albumBtn.addTarget(self, action: #selector(switchToAlbum), for: .touchUpInside)
        view.addSubview(albumBtn)
    }
    
    @objc private func switchToAlbum() {
        dismiss(animated: true) {
            let albumPicker = UIImagePickerController()
            albumPicker.sourceType = .photoLibrary
            albumPicker.delegate = self.delegate
            self.presentingViewController?.present(albumPicker, animated: true)
        }
    }
}

⚠️ 注意：此方案存在审核风险，且按钮位置会随iOS版本变化偏移，不建议用于正式发布的应用。

方案2：基于PhotoKit+AVFoundation实现完全自定义选择器（推荐） #

自己搭建相机界面+相册浏览功能，完全控制UI逻辑，实现和WhatsApp、系统相机一致的无缝切换体验。

核心实现步骤 #

1. 用AVFoundation搭建相机预览层，实现拍照功能；
2. 用PhotoKit获取相册最新照片，在相机界面添加相册缩略图入口；
3. 点击缩略图跳转到自定义相册页面，完成选择后返回。

代码片段示例 #

import AVFoundation
import Photos

class CustomCameraVC: UIViewController {
    private let captureSession = AVCaptureSession()
    private let previewLayer = AVCaptureVideoPreviewLayer()
    private var latestAlbumAsset: PHAsset?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        setupCamera()
        setupAlbumThumbnail()
    }
    
    // 配置相机会话（需提前申请相机权限）
    private func setupCamera() {
        captureSession.sessionPreset = .photo
        // 省略相机输入输出配置代码
        previewLayer.session = captureSession
        previewLayer.frame = view.bounds
        view.layer.addSublayer(previewLayer)
        captureSession.startRunning()
    }
    
    // 添加左下角相册缩略图入口
    private func setupAlbumThumbnail() {
        // 获取最新的一张相册照片
        let fetchOpts = PHFetchOptions()
        fetchOpts.sortDescriptors = [NSSortDescriptor(key: "creationDate", ascending: false)]
        fetchOpts.fetchLimit = 1
        let fetchResult = PHAsset.fetchAssets(with: .image, options: fetchOpts)
        latestAlbumAsset = fetchResult.firstObject
        
        let thumbnailBtn = UIButton(frame: CGRect(x: 20, y: view.bounds.height - 120, width: 60, height: 60))
        thumbnailBtn.layer.cornerRadius = 8
        thumbnailBtn.clipsToBounds = true
        thumbnailBtn.addTarget(self, action: #selector(openAlbum), for: .touchUpInside)
        
        // 加载缩略图
        if let asset = latestAlbumAsset {
            PHImageManager.default().requestImage(for: asset, targetSize: CGSize(width: 120, height: 120), contentMode: .aspectFill, options: nil) { image, _ in
                thumbnailBtn.setImage(image, for: .normal)
            }
        }
        view.addSubview(thumbnailBtn)
    }
    
    @objc private func openAlbum() {
        // 跳转到自定义相册浏览页面
        let albumVC = CustomAlbumVC()
        albumVC.delegate = self
        present(albumVC, animated: true)
    }
}

// 相册页面代理协议
protocol CustomAlbumVCDelegate: AnyObject {
    func didSelectImage(_ image: UIImage)
}

这种方案完全符合苹果审核规则，兼容性强，能实现和目标应用一致的交互效果。

内容的提问来源于stack exchange，提问作者doc_id