Expandable cell won't expand - ios

Question

I have created two expandable cells but the second one doesn't expand when I tap on it
I googled a lot of ways how to do this but this one seemed the shortest and the simplest one.
private var dateCellExpanded: Bool = false
override func viewDidLoad() {
super.viewDidLoad()
// For removing the extra empty spaces of TableView below
self.tableView!.tableFooterView = UIView()
}
override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
if indexPath.row == 0 {
if dateCellExpanded {
dateCellExpanded = false
} else {
dateCellExpanded = true
}
tableView.beginUpdates()
tableView.endUpdates()
}
}
override func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
if indexPath.row == 0 {
if dateCellExpanded {
return 250
} else {
return 40
}
}
return 40
}
I expect to be able to add as many cells as I want and have them expand on tap, but only the first one does.
Here's what happens

Answer 1

It's only working for the first row because you're limiting it to the first row with if indexPath.row == 0.
You'll need an array of Bool rather than a single Bool as you need to track the state for all rows.
Just initialize the array to an array of false values with the count of your rows. Set NUMBER_OF_ROWS below accordingly.
Then, when selecting a cell, you flip (or toggle()) the boolean for that row and ask the table to update.
private var isCellExpanded: [Bool] = Array(repeating: false, count: NUMBER_OF_ROWS)
override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
isCellExpanded[indexPath.row].toggle()
tableView.beginUpdates()
tableView.endUpdates()
}
override func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
if isCellExpanded[indexPath.row] {
return 250
} else {
return 40
}
}
Note that this assumes that you're only using one section. Otherwise you'll have to adjust the data structure [Bool] to [[Bool]].