Swift + Add UIToolbar or Done Button on Keyboard

 override func viewDidLoad()

    {

        super.viewDidLoad()

     

        //--- add UIToolBar on keyboard and Done button on UIToolBar ---//

        self.addDoneButtonOnKeyboard()

        

    }

    //--- *** ---//

    

    func addDoneButtonOnKeyboard()

    {

        var doneToolbar: UIToolbar = UIToolbar(frame: CGRectMake(0, 0, 320, 50))

        doneToolbar.barStyle = UIBarStyle.BlackTranslucent

       

        var flexSpace = UIBarButtonItem(barButtonSystemItem:UIBarButtonSystemItem.FlexibleSpace, target: nil, action: nil)

        var done: UIBarButtonItem = UIBarButtonItem(title: "Done", style:UIBarButtonItemStyle.Done, target: self, action: Selector("doneButtonAction"))

        

        var items = NSMutableArray()

        items.addObject(flexSpace)

        items.addObject(done)

      

        doneToolbar.items = items

        doneToolbar.sizeToFit()

        

        self.textView.inputAccessoryView = doneToolbar

        self.textField.inputAccessoryView = doneToolbar

        

    }

    

    func doneButtonAction()

    {

        self.textViewDescription.resignFirstResponder()

        self.textViewDescription.resignFirstResponder()

    }