如何仅在从弹窗调用二级详情ViewController时添加按钮？

嘿，作为iOS新手能考虑到这种场景细节已经超棒啦！我给你梳理个一步步能落地的方案，完全贴合你的需求：

核心思路 #

我们需要给Detail View Controller加一个「来源标记」，判断它是从「弹窗触发的员工列表」跳转来的，还是从「正常导航菜单」进入的，再根据这个标记动态创建按钮。

步骤1：给Detail VC添加来源标记属性 #

首先在你的DetailViewController里新增一个布尔属性，用来区分进入路径：

class DetailViewController: UIViewController {
    // 你的现有属性，比如选中的员工模型
    var selectedEmployee: Employee!
    
    // 新增：标记是否从弹窗路径进入，默认false（正常导航进入）
    var isFromPopupSelection = false
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // 其他初始化逻辑...
    }
}

步骤2：给员工列表VC添加弹窗标记，并传递给Detail VC #

因为员工列表有两种展示方式：「弹窗」和「正常导航菜单」，所以我们需要给员工列表VC也加个标记，再把这个标记传给Detail VC：

2.1 在员工列表VC里添加弹窗标记

class EmployeeListViewController: UITableViewController {
    // 你的现有属性，比如员工数组
    var employees: [Employee] = []
    
    // 新增：标记当前列表是否以弹窗形式展示
    var isPresentedAsPopup = false
    
    // 选中行的逻辑
    override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        let employee = employees[indexPath.row]
        performSegue(withIdentifier: "ShowEmployeeDetail", sender: employee)
    }
    
    // 跳转前传递数据和标记
    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        guard segue.identifier == "ShowEmployeeDetail", 
              let detailVC = segue.destination as? DetailViewController,
              let selectedEmployee = sender as? Employee else {
            return
        }
        
        detailVC.selectedEmployee = selectedEmployee
        // 把弹窗标记传递给Detail VC
        detailVC.isFromPopupSelection = isPresentedAsPopup
    }
}

2.2 在第一个视图弹窗展示员工列表时，设置标记

当你从第一个视图的「OK按钮」点击弹出员工列表时，记得给列表VC的isPresentedAsPopup设为true：

// 第一个视图里的OK按钮点击事件
@objc func okButtonTapped() {
    guard let employeeListVC = storyboard?.instantiateViewController(withIdentifier: "EmployeeListVC") as? EmployeeListViewController else {
        return
    }
    // 标记这是弹窗形式的列表
    employeeListVC.isPresentedAsPopup = true
    present(employeeListVC, animated: true)
}

步骤3：在Detail VC里动态创建选择按钮 #

最后在Detail VC的viewDidLoad（或viewWillAppear）里，根据isFromPopupSelection的值来决定是否添加按钮：

override func viewDidLoad() {
    super.viewDidLoad()
    // 现有详情页展示逻辑...
    
    // 如果是从弹窗路径进入，才添加选择按钮
    if isFromPopupSelection {
        // 创建按钮
        let selectButton = UIButton(type: .system)
        selectButton.setTitle("选择该员工", for: .normal)
        selectButton.titleLabel?.font = UIFont.systemFont(ofSize: 16, weight: .medium)
        selectButton.addTarget(self, action: #selector(selectEmployeeAction), for: .touchUpInside)
        
        // 把按钮加到导航栏右侧（或者你想放在视图底部也可以，看需求）
        navigationItem.rightBarButtonItem = UIBarButtonItem(customView: selectButton)
    }
}

// 按钮点击事件：处理选择员工后的逻辑（比如回传数据给第一个视图）
@objc private func selectEmployeeAction() {
    // 示例：如果是弹窗层级，直接关闭详情页和列表页，把员工数据传给第一个视图
    if let firstVC = presentingViewController?.presentingViewController as? FirstViewController {
        firstVC.handleSelectedEmployee(selectedEmployee)
        dismiss(animated: true)
    }
}

额外小提示：数据回传的优雅方式 #

如果觉得用presentingViewController层层找太麻烦，可以给Detail VC加个闭包回调，比如：

// 在Detail VC里新增闭包属性
var onEmployeeSelected: ((Employee) -> Void)?

// 按钮点击时调用闭包
@objc private func selectEmployeeAction() {
    onEmployeeSelected?(selectedEmployee)
    dismiss(animated: true)
}

// 然后在员工列表VC的prepare方法里设置闭包
override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
    // ...其他代码
    detailVC.onEmployeeSelected = { [weak self] employee in
        self?.dismiss(animated: true) {
            if let firstVC = self?.presentingViewController as? FirstViewController {
                firstVC.handleSelectedEmployee(employee)
            }
        }
    }
}

这样代码逻辑更清晰，也避免了强引用问题～

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