//
//  UIViewController+UIBarButtonItem.swift
//  PaiAi
//
//  Created by mac on 16/7/21.
//  Copyright © 2016年 FFIB. All rights reserved.
//

import UIKit

public extension UIViewController {
    func setRightBarButtonItem(image: UIImage?, action: Selector) {
        let button = UIButton(type: .custom)
        button.setImage(image, for: .normal)
        button.addTarget(self, action: action, for: .touchDown)
        navigationItem.rightBarButtonItem = UIBarButtonItem(customView: button)
    }
}