Now, we will take advantage of the capability that Swift offers us to extend an existing class to add specific members. In this case, we will add an instance method to the previously defined AngryCat class. The following lines add the doSomethingWith method to the existing AngryCat class. The code file for the sample is included in the swift_3_oop_chapter_05_11 folder:
public extension AngryCat {
public func doSomethingWith(cat: AngryCat) {
if let angryCatAlien = cat as? AngryCatAlien {
angryCatAlien.appear()
} else if let angryCatKnight = cat as? AngryCatKnight {
angryCatKnight.unsheathSword()
} else if let angryCatWizard = cat as? AngryCatWizard {
print("My spell power is \(angryCatWizard.spellPower)")
} else {
print("This AngryCat doesn't have cool skills.")
}
}
...