Swift - 自定义UIActivity分享

UIActivity 可以十分方便地将文字、图片等内容进行分享，比如分享到微信、微博、发送邮件、短信等等。我们不仅可以分享内容出来，也可以在自己的 App 里添加自己的分享按钮或隐藏已有的分享按钮来实现定制的功能。

import UIKit

class ViewController: UIViewController {
    
    override func viewDidLoad() {
        super.viewDidLoad()
    }
    
    override func didReceiveMemoryWarning() {
        super.didReceiveMemoryWarning()
    }
    
    //点击按钮弹出分享框
    @IBAction func btnClicked(_ sender: AnyObject) {
        //准备分享内容，文本图片链接
        let items:[Any] = ["航歌", UIImage(named:"toSendImage.png")!,
                     URL(fileURLWithPath:"http://hangge.com")]
        //新建自定义的分享对象数组
        let acts = [WeiXinActivity(),HanggeActivity()]
        //根据分享内容和自定义的分享按钮调用分享视图
        let actView:UIActivityViewController =
            UIActivityViewController(activityItems: items, applicationActivities: acts)
        //要排除的分享按钮，不显示在分享框里
        actView.excludedActivityTypes = [.mail,.copyToPasteboard,.print,
                                         .assignToContact,.saveToCameraRoll]
        //显示分享视图
        self.present(actView, animated:true, completion:nil)
    }
}

import UIKit

class HanggeActivity : UIActivity {
    //用于保存传递过来的要分享的数据
    var text:String!
    var url:URL!
    var image:UIImage!
    
    //显示在分享框里的名称
    override var activityTitle: String?  {
        return "航歌"
    }
    
    //分享框的图片
    override var activityImage: UIImage? {
        return UIImage(named:"hangge.png")
    }
    
    //分享类型，在UIActivityViewController.completionHandler回调里可以用于判断，一般取当前类名
    override var activityType: UIActivityType? {
        return UIActivityType(rawValue: HanggeActivity.self.description())
    }
    
    //按钮类型（分享按钮：在第一行，彩色，动作按钮：在第二行，黑白）
    override class var activityCategory: UIActivityCategory {
        return .action
    }
    
    //是否显示分享按钮，这里一般根据用户是否授权,或分享内容是否正确等来决定是否要隐藏分享按钮
    override func canPerform(withActivityItems activityItems: [Any]) -> Bool {
        for item in activityItems {
            if item is UIImage {
                return true
            }
            if item is String {
                return true
            }
            if item is URL {
                return true
            }
        }
        return false
    }
    
    //解析分享数据时调用，可以进行一定的处理
    override func prepare(withActivityItems activityItems: [Any]) {
        print("prepareWithActivityItems：hangge")
        for item in activityItems {
            if item is UIImage {
                image = item as! UIImage
            }
            if item is String {
                text = item as! String
            }
            if item is URL {
                url = item as! URL
            }
        }
    }
    
    //执行分享行为
    //这里根据自己的应用做相应的处理
    //例如你可以分享到另外的app例如微信分享，也可以保存数据到照片或其他地方，甚至分享到网络
    override func perform() {
        print("performActivity：hangge")
        //具体的执行代码这边先省略
        //......
        activityDidFinish(true)
    }
    
    //分享时调用
    override var activityViewController: UIViewController? {
        print("activityViewController：hangge")
        return nil
    }
    
    //完成分享后调用
    override func activityDidFinish(_ completed: Bool) {
        print("activitydidfinish：hangge")
    }
}

import UIKit

class WeiXinActivity : UIActivity {
    //用于保存传递过来的要分享的数据
    var text:String!
    var url:URL!
    var image:UIImage!
    
    //显示在分享框里的名称
    override var activityTitle: String?  {
        return "微信"
    }
    
    //分享框的图片
    override var activityImage: UIImage? {
        return UIImage(named:"wechat_session")
    }
    
    //分享类型，在UIActivityViewController.completionHandler回调里可以用于判断，一般取当前类名
    override var activityType: UIActivityType? {
        return UIActivityType(rawValue: WeiXinActivity.self.description())
    }
    
    //按钮类型（分享按钮：在第一行，彩色，动作按钮：在第二行，黑白）
    override class var activityCategory: UIActivityCategory {
        return .share
    }
    
    //是否显示分享按钮，这里一般根据用户是否授权,或分享内容是否正确等来决定是否要隐藏分享按钮
     override func canPerform(withActivityItems activityItems: [Any]) -> Bool {
        for item in activityItems {
            if item is UIImage {
                return true
            }
            if item is String {
                return true
            }
            if item is URL {
                return true
            }
        }
        return false
    }
    
    //解析分享数据时调用，可以进行一定的处理
    override func prepare(withActivityItems activityItems: [Any]) {
        print("prepareWithActivityItems：weixin")
        for item in activityItems {
            if item is UIImage {
                image = item as! UIImage
            }
            if item is String {
                text = item as! String
            }
            if item is URL {
                url = item as! URL
            }
        }
    }
    
    //执行分享行为
    //这里根据自己的应用做相应的处理
    //例如你可以分享到另外的app例如微信分享，也可以保存数据到照片或其他地方，甚至分享到网络
    override func perform() {
        print("performActivity：weixin")
        //具体的执行代码这边先省略
        //......
        activityDidFinish(true)
    }
    
    //分享时调用
    override var activityViewController: UIViewController? {
        print("activityViewController：weixin")
        return nil
    }
    
    //完成分享后调用
    override func activityDidFinish(_ completed: Bool) {
        print("activitydidfinish：weixin")
    }
}