He agregado una barra de navegación en la parte superior de un controlador de vista. Estoy tratando de controlar si un botón es visible en función de una condición, pero tengo problemas para agregar el botón. Hasta ahora tengo

var addButton: UIBarButtonItem = UIBarButtonItem(title: "test", style: .done, target: self, action: #selector(addTapped))

override func viewDidLoad() {
    super.viewDidLoad()

    let boool = true
    if boool {
        self.navigationItem.rightBarButtonItem = self.addButton
    }
    else {
        self.navigationItem.rightBarButtonItem = nil
    }
}

func addTapped(sender: AnyObject) {
    print("hjxdbsdhjbv")
}

Creo que no funciona correctamente porque he agregado una barra de navegación en el VC, en lugar de usar un controlador de navegación y trabajar con la barra allí. Me preguntaba si había una manera de trabajar con esta barra de navegación.