• В TextField и TextView свойство textBox сделать публичным
• В TextBox свойство state сделать публичным
• Добавить делегат (или KVO) для возможности отслеживания изменения свойства TextBox.state

Примерно такой протокол:

public protocol TextBoxDelegate: class {
    func willChangeState(newValue: TextInputState, animated: Bool)
    func didChangeState(newValue: TextInputState, animated: Bool)
}