Неправильная высота расширения клавиатуры после поворота

У меня есть специальная клавиатура iOS, которая меняет высоту при вращении.

Мой код работает нормально в 95% случаев... Но в некоторых случаях (см. ниже) высота не меняется при повороте в альбомную - сохраняется портретная высота.

Проблема может быть воспроизведена с помощью этого (почти) минимального кода — создайте новую цель расширения клавиатуры и скопируйте этот код в KeyboardViewController:

class KeyboardViewController: UIInputViewController {
    private var orangeView = UIView()
    private var heightConstraint: NSLayoutConstraint!
    @IBOutlet var nextKeyboardButton: UIButton!

    override func updateViewConstraints() {
        super.updateViewConstraints()
        let screenSize = UIScreen.mainScreen().bounds.size
        let screenH = screenSize.height
        let screenW = screenSize.width
        let isLandscape = !(self.view.frame.size.width == screenW * ((screenW < screenH) ? 1 : 0) + screenH * ((screenW > screenH) ? 1 : 0))
        let desiredHeight: CGFloat = isLandscape ? 193 : 252
        if heightConstraint.constant != desiredHeight {
            heightConstraint!.constant = desiredHeight
            orangeView.frame = CGRect(x: 0, y: 0, width: screenW, height: isLandscape ? 193 : 252)
        }
    }

    override func viewDidLoad() {
        super.viewDidLoad()
        nextKeyboardButton = UIButton(type: .System)
        nextKeyboardButton.setTitle("Next Keyboard", forState: .Normal)
        nextKeyboardButton.sizeToFit()
        nextKeyboardButton.translatesAutoresizingMaskIntoConstraints = false
        nextKeyboardButton.addTarget(self, action: "advanceToNextInputMode", forControlEvents: .TouchUpInside)
        heightConstraint = NSLayoutConstraint(item:self.inputView!, attribute:.Height, relatedBy:.Equal, toItem:nil, attribute:.NotAnAttribute, multiplier: 0.0, constant: 0) //preparing heightConstraint
        heightConstraint?.priority = 999
        orangeView.backgroundColor = UIColor.orangeColor()
        view.addSubview(orangeView)
        view.addSubview(self.nextKeyboardButton)
        let nextKeyboardButtonLeftSideConstraint = NSLayoutConstraint(item: self.nextKeyboardButton, attribute: .Left, relatedBy: .Equal, toItem: self.view, attribute: .Left, multiplier: 1.0, constant: 0.0)
        let nextKeyboardButtonBottomConstraint = NSLayoutConstraint(item: self.nextKeyboardButton, attribute: .Bottom, relatedBy: .Equal, toItem: self.view, attribute: .Bottom, multiplier: 1.0, constant: 0.0)
        view.addConstraints([nextKeyboardButtonLeftSideConstraint, nextKeyboardButtonBottomConstraint])
    }

    override func viewWillAppear(animated: Bool) {
        if self.view.constraints.filter({c in c == self.heightConstraint}).isEmpty {
            self.view.addConstraint(heightConstraint)
        }
        view.setNeedsUpdateConstraints() //ensures that updateViewConstraints always gets called at least once
        super.viewWillAppear(animated)
    }
}

Для получения дополнительной информации о том, как это работает, см. это и этот ответ.

Когда вы запускаете клавиатуру, она, вероятно, будет работать нормально — она немного выше, и после поворота оранжевый вид покрывает всю клавиатуру:

Однако вы также можете получить это (высота портрета сохраняется после поворота):

Действия для воспроизведения проблемы (с кодом выше):

1. Откройте клавиатуру в приложении «Сообщения», поверните альбомную ориентацию и обратно
2. Если вы не увидели проблему, выключите приложение «Сообщения» (нажмите кнопку «Домой» 2 раза -> проведите пальцем вверх по сообщениям).
3. Открывайте сообщения снова, поворачивайте альбомную ориентацию и обратно
4. Возможно, вам придется повторить шаги 1–3 несколько раз, чтобы увидеть проблему (обычно не более 2–4 раз).

Что я знаю:

• Если вы видите проблему, вы будете видеть ее до тех пор, пока клавиатура не будет скрыта и снова не отобразится.
• Если вы видите, что проблема скрывает и повторно отображает клавиатуру в том же приложении - проблема всегда исчезает.
• Проблема может появиться снова только после закрытия и перезапуска приложения хостинга.
• Отладчик показывает, что heightConstraint!.constant равно 193, но и view.frame.height, и view.window?.frame.height по-прежнему 253 (изменение кадра напрямую не исправляет это)

Что я пробовал:

• Вместо того, чтобы просто установить ограничение heightConstraint!.constant = desiredHeight, сначала удалите его из view, установите новое значение, а затем снова добавьте его.
• Я проверил, что heightConstraint!.constant всегда меняется, когда это должно быть

Как исправить или обойти эту проблему? Должно быть решение, потому что SwiftKey не имеет этой проблемы.