传递数据取决于tableView单元格中的按钮

Question

I have a TableView where I display all my data and each cell might have 1-2 buttons. I read many topics and understand how to add target for each button through my ViewController. Since these buttons will be forwarded to the same VC and display images, I have the following code. In my TableViewCell subclass I have 2 buttons

class CODetailsTicketCell: UITableViewCel {

      var onButtonTapped: (() -> Void)? = nil

      @IBAction func firstBtnTapped(_ sender: UIButton) {
          if let onButtonTapped = self.onButtonTapped {
              onButtonTapped()
          }
         print("First button was pressed")

     }

      @IBAction func secondBtnTapped(_ sender: UIButton) {
          if let onButtonTapped = self.onButtonTapped {
             onButtonTapped()
          }
          print("Second button was pressed")
     }
}

In my ViewController in cellForRowAt indexPath I have the following code

let message = messages[indexPath.row]

if let cell = tableView.dequeueReusableCell(withIdentifier: "COTicketsCell", for: indexPath) as? CODetailsTicketCell {
    cell.configureCell(openTickets: message)
    cell.onButtonTapped = {
         self.performSegue(withIdentifier: "toImageVC", sender: message)
     } 

     return cell

In order to pass the data through segue I use the following code in prepareForSegue

override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
    if segue.identifier == "toImageVC" {
        let navigationController = segue.destination as? UINavigationController
        if let targetController = navigationController?.topViewController as? ImageVC {
            if let data = sender as? OpenTicketsData {
                targetController.loadImageURL = URL(string: data.firstImageUrl)
            }

        }
    }
}

Everything is working FINE but I can't check for button tag in prepareForSegue. Basically, currently both buttons send the same data

targetController.loadImageURL = URL(string: data.firstImageUrl)

How can I pass data based on the button pressed? I tried to do something like this but seems it's wrong and not working.

let button = sender as? UIButton
if let data = sender as? OpenTicketsData {
   if button?.tag == 1 {
      targetController.loadImageURL = URL(string: data.firstImageUrl)
   } else if button?.tag == 2 {
      targetController.loadImageURL = URL(string: data.secondImageUrl)
   }
}

Answer 1

You can either separate it into 2 different events or

class CODetailsTicketCell: UITableViewCell {

      var onButtonTapped: ((_ sender: UIButton) -> Void)? = nil

      @IBAction func firstBtnTapped(_ sender: UIButton) {
          if let onButtonTapped = self.onButtonTapped {
              onButtonTapped?(sender)
          }
         print("First button was pressed")

     }

      @IBAction func secondBtnTapped(_ sender: UIButton) {
          if let onButtonTapped = self.onButtonTapped {
              onButtonTapped(sender)
          }
          print("Second button was pressed")
     }
}

In your assignment of the onButtonTapped , remember to add [weak self] if you ever use self to avoid the retain cycle.

cell.onButtonTapped = { [weak self] sender in
    if sender.tag == 1 {
        // Do something
    } else {
        // Do other thing
    }
}