Bug with multiple Storyboards

I have a project with multiple Storyboard files, and for some reason, when I run the script on my project folder I get this:

//MARK: - Storyboards
enum Storyboards: String {
    case Filter = "Filter"
    case Main = "Main"
    case Results = "Results"

    private var instance:UIStoryboard {
        return UIStoryboard(name: self.rawValue, bundle: nil)
    }

    func instantiateInitialViewController() -> UIViewController? {
        switch (self) {
        case Filter:
            return self.instance.instantiateInitialViewController() as! NavigationController
        default:
            return self.instance.instantiateInitialViewController() as? UIViewController
        }
        switch (self) {
        case Main:
            return self.instance.instantiateInitialViewController() as! LoginViewController
        default:
            return self.instance.instantiateInitialViewController() as? UIViewController
        }
        switch (self) {
        case Results:
            return self.instance.instantiateInitialViewController() as! NavigationController
        default:
            return self.instance.instantiateInitialViewController() as? UIViewController
        }
    }

    func instantiateViewControllerWithIdentifier(identifier: String) -> UIViewController {
        return self.instance.instantiateViewControllerWithIdentifier(identifier) as! UIViewController
    }
}

As you can see, for some reason, it is creating one switch for each Storyboard instead of just one switch. Because of this, only the first switch is executed.

While I have your attention, it would be cool, if it was possible to instantiate an initial VC and immediately get the correct VC type, at the moment, this method only returns a UIViewController?

I am using the latest version of the script.

Callum

krzyzanowskim / Natalie

Bug with multiple Storyboards #9