MoussaHellal

let addItemNotfication = NSNotification.Name(rawValue: "addItem")

MoussaHellal

func createObservers() {
        NotificationCenter.default.addObserver(self, selector: #selector(TabBarController.setBadge(notification:)), name: addItemNotfication, object: nil)
}

@objc func setBadge(notification: NSNotification) {
        //get the existant badge value
        let badgeValue = Int(self.tabBar.items![0].badgeValue!)!
        // set the new badge value
        self.tabBar.items![0].badgeValue = "\(badgeValue + 1)"
}

MoussaHellal

override func viewDidLoad() {
        super.viewDidLoad()
        // add badge to my cart tab icon with value of zero
        self.tabBar.items![0].badgeValue = "0"
        createObservers()
}

MoussaHellal

import UIKit

class TabBarController: UITabBarController {
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // add badge to my cart tab icon with value of zero
        self.tabBar.items![0].badgeValue = "0"
        createObservers()

MoussaHellal

 NotificationCenter.default.post(name: addItemNotfication, object: nil,userInfo: nil)

MoussaHellal

deinit {
        NotificationCenter.default.removeObserver(self)
}

MoussaHellal

class MainViewController: UIViewController {
    @IBOutlet weak var changeButton: UIButton!
    
    override func viewDidLoad() {
        super.viewDidLoad()
    }
    
    @IBAction func changeButtonDidTapped(_ sender: Any) {
        let chooseVC = storyboard?.instantiateViewController(identifier: "chooseVC") as! ChooseViewController
        chooseVC.modalPresentationStyle = .fullScreen

MoussaHellal

class ChooseViewController: UIViewController {
    
    @IBOutlet weak var sunButton: SpaceCustomButton!
    @IBOutlet weak var mercuryButton: SpaceCustomButton!
    @IBOutlet weak var venusButton: SpaceCustomButton!
    @IBOutlet weak var earthButton: SpaceCustomButton!
    @IBOutlet weak var moonButton: SpaceCustomButton!
    @IBOutlet weak var marsButton: SpaceCustomButton!
    @IBOutlet weak var jupiterButton: SpaceCustomButton!
    @IBOutlet weak var saturnButton: SpaceCustomButton!

MoussaHellal

import UIKit

protocol ChoosePlanetMoonDelegate {
    func didChoosePlanetWith(image: UIImage)
}

MoussaHellal

class MainViewController: UIViewController, ChoosePlanetMoonDelegate {
    @IBOutlet weak var mainPlanetImage: UIImageView!
    @IBOutlet weak var changeButton: UIButton!

    override func viewDidLoad() {
        super.viewDidLoad()
    }
    
    @IBAction func changeButtonDidTapped(_ sender: Any) {
        let chooseVC = storyboard?.instantiateViewController(identifier: "chooseVC") as! ChooseViewController