Change UIAction state without dismissing menu in iOS 16

ImageSwap.swift

//Bit of a crude example but it all works

var menu: UIMenu!
var selected = false

let toggleSelectionHandler: (UIAction) -> Void = { action in
	selected.toggle()
	// Actions are immutable once they're a child of a UIMenu, so we must create a copy before we modify anything
	let replacementAction = action.copy() as! UIAction
	if selected {
		replacementAction.state = .on
	} else {
		replacementAction.state = .off
	}
	button.menu = menu.replacingChildren([replacementAction])
}

let toggleSelection = UIAction(title: "Toggle", image: UIImage(systemName: "circle"), attributes: .keepsMenuPresented, handler: toggleSelectionHandler)

menu = UIMenu(title: "Test", children: [toggleSelection])

//I have a button I'm assigning this to, but you an use the menu however you like
button.menu = menu
button.showsMenuAsPrimaryAction = true