6.7. Getting the expense back to ViewController

We now need to get the expense back when we press buttonAdd. We need to tweak the data model Expense to accommodate the image, right?

Update Expense data model:

Let's open the Expense.swift file and update the file as the following:

//
//  Expense.swift
//  App6
//
//  Created by Sakib Miazi on 5/18/23.
//

import Foundation
import UIKit

struct Expense{
    var title: String?
    var amount: Double?
    var type: String?
    var image: UIImage?
    
    init(title: String, amount: Double, type: String, image: UIImage) {
        self.title = title
        self.amount = amount
        self.type = type
        self.image = image
    }
    
}

Update AddExpenseViewController's onAddButtonTapped() method:

We write:

//MARK: action for tapping buttonAdd..
@objc func onAddButtonTapped(){
    var title:String = ""
    if let titleText = addExpenseScreen.textFieldTitle.text{
        if !titleText.isEmpty{
            title = titleText
        }else{
            //do your thing to alert user...
            return
        }
    }
    
    var amount = 0.0
    if let amountText = addExpenseScreen.textFieldAmount.text{
        if !amountText.isEmpty{
            if let uwAmount = Double(amountText){
                amount = uwAmount
            }else{
                //alert the user that it's not a valid input...
                return
            }
        
            
        }else{
            //do your thing to alert the user...
            return
        }
        
    }
    
    let newExpense = Expense(
                        title: title, 
                        amount: amount, 
                        type: selectedType, 
                        image: pickedImage ?? (UIImage(systemName: "photo"))!
                    )
    delegate.delegateOnAddExpense(expense: newExpense)
    navigationController?.popViewController(animated: true)
}

Here, we are creating a new expense with the title, amount, type, and image. The pickedImage might be nil since it is Optional. So if it becomes nil (at any point), we send a default system image named "photo" to be safe.

Update the TableView to display the image in the cell

Open ViewController.swift. We need to update where we are adopting the TableView protocools.

//MARK: adopting the procols for TableView...
extension ViewController: UITableViewDelegate, UITableViewDataSource{
    //MARK: returns the number of rows in the current section...
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return expenses.count
    }
    
    //MARK: populate a cell for the currecnt row...
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "expenses", for: indexPath) as! TableViewExpenseCell
        cell.labelTitle.text = expenses[indexPath.row].title
        if let uwAmount = expenses[indexPath.row].amount{
            cell.labelAmount.text = "Cost: $\(uwAmount)"
        }
        if let uwType = expenses[indexPath.row].type{
            cell.labelType.text = "Type: \(uwType)"
        }
        
        //MARK: setting the image of the receipt...
        if let uwImage = expenses[indexPath.row].image{
            cell.imageReceipt.image = uwImage
        }
        
        return cell
    }
    
    //MARK: deal with user interaction with a cell...
    func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        print(self.expenses[indexPath.row])
    }


}

We unwrap the image, and then set it to cell.imageReceipt.image.

Finally, let's run the app!