如何在UITableView自定义单元格间实现键盘上一步/下一步/完成按钮功能

嘿各位，刚接触Swift不久，最近在做一个UITableView表单页面，每行都是带输入控件（UITextField、UITextView这类）的自定义单元格，需要实现键盘上一步/下一步/完成按钮来在各个输入框间切换焦点。翻了不少方案，但要么不够通用，要么没考虑到下一目标单元格可能在屏幕外的情况——点了下一步结果焦点切过去了，但单元格没滚动到视野里，用户体验特别差。

折腾一阵后，我自己整理了一套可行的方案，分享出来给大家参考，也欢迎各位大佬提改进建议！

核心思路 #

• 给每个输入控件绑定对应的单元格索引，快速定位上/下一个输入控件所在的单元格
• 统一管理所有输入控件的代理，集中处理键盘按钮的点击事件
• 切换焦点前先判断目标单元格是否在可视区域，不在的话先滚动到对应位置，再设置焦点

具体实现步骤 #

1. 给单元格和输入控件添加标识 #

自定义单元格时，给输入控件关联对应的IndexPath属性，方便后续定位：

class InputTableViewCell: UITableViewCell {
    @IBOutlet weak var inputTextField: UITextField!
    @IBOutlet weak var inputTextView: UITextView!
    var cellIndexPath: IndexPath?
}

2. 统一管理输入控件的映射 #

在ViewController里，维护一个输入控件与IndexPath的映射表，同时给每个输入控件设置代理：

var inputControls: [IndexPath: UITextInput] = [:]

override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    let cell = tableView.dequeueReusableCell(withIdentifier: "InputCell", for: indexPath) as! InputTableViewCell
    cell.cellIndexPath = indexPath
    
    // 根据单元格类型绑定对应的输入控件
    if let textField = cell.inputTextField {
        textField.delegate = self
        inputControls[indexPath] = textField
    } else if let textView = cell.inputTextView {
        textView.delegate = self
        inputControls[indexPath] = textView
    }
    return cell
}

3. 实现键盘切换逻辑 #

通过代理方法处理键盘按钮事件，先滚动到目标单元格再切换焦点：

extension YourViewController: UITextFieldDelegate, UITextViewDelegate {
    // UITextField的返回键处理
    func textFieldShouldReturn(_ textField: UITextField) -> Bool {
        return switchFocus(from: textField, isNext: true)
    }
    
    // UITextView的回车处理（可根据需求调整）
    func textView(_ textView: UITextView, shouldChangeTextIn range: NSRange, replacementText text: String) -> Bool {
        if text == "\n" {
            return switchFocus(from: textView, isNext: true)
        }
        return true
    }
    
    // 通用焦点切换方法，isNext区分上一步/下一步
    private func switchFocus(from input: UITextInput, isNext: Bool) -> Bool {
        guard let currentIndexPath = inputControls.first(where: { $0.value == input })?.key else { return false }
        
        // 计算目标索引
        let targetRow = isNext ? currentIndexPath.row + 1 : currentIndexPath.row - 1
        guard targetRow >= 0 && targetRow < tableView.numberOfRows(inSection: currentIndexPath.section) else {
            // 超出范围，收起键盘
            input.resignFirstResponder()
            return true
        }
        let targetIndexPath = IndexPath(row: targetRow, section: currentIndexPath.section)
        
        // 滚动到目标单元格
        tableView.scrollToRow(at: targetIndexPath, at: .middle, animated: true)
        
        // 等滚动完成后再设置焦点（避免因单元格复用导致的控件为空）
        DispatchQueue.main.asyncAfter(deadline: .now() + 0.2) {
            if let targetInput = self.inputControls[targetIndexPath] {
                targetInput.becomeFirstResponder()
            }
        }
        return true
    }
}

4. 处理单元格复用问题 #

因为UITableView复用机制，要在单元格显示/隐藏时更新映射表：

override func tableView(_ tableView: UITableView, willDisplay cell: UITableViewCell, forRowAt indexPath: IndexPath) {
    guard let inputCell = cell as? InputTableViewCell else { return }
    if let textField = inputCell.inputTextField {
        inputControls[indexPath] = textField
    } else if let textView = inputCell.inputTextView {
        inputControls[indexPath] = textView
    }
}

override func tableView(_ tableView: UITableView, didEndDisplaying cell: UITableViewCell, forRowAt indexPath: IndexPath) {
    inputControls.removeValue(forKey: indexPath)
}

目前方案的待优化点 #

• 滚动完成的延迟时间（0.2秒）是硬编码的，有没有更优雅的方式监听滚动结束事件？
• 目前只处理了同section内的切换，跨section的场景还没覆盖
• 对于UITextView，直接把回车当成下一步可能不符合部分需求，需要更灵活的控件类型区分逻辑

欢迎各位大佬提出改进思路，或者分享你们遇到过的类似场景的解决方案！

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