给微信聊天记录添加截图功能_知识小集的专栏文章_微信文章

有时候，知识小集群里讨论的技术问题，比较有价值，我们会把有价值的内容整理出来供大家查阅。但为了保护群友隐私，需要把昵称和头像都打码，如果碰到几百条聊天记录，这样做简直要吐血。而且也不能截一张长图，只能一张一张截取，然后拼接起来。群聊记录只能在微信内分享，这也限制了传播的渠道。为了提高小集成员工作效率，想着能不能给微信做个插件，解决这些问题。我们一直在追求如何更有效率开展我们的工作，比如使用脚本自动整理每周小集内容，使用微信小程序给读者更好阅读体验。（呀，还有脚本，如果你还不知道，那肯定没有点 star 吧，传送门）

提出问题

通过以上痛点，可以确定我们要解决的问题主要有：

截图：把所有的聊天记录截一张长图并保存到相册；
截图（马赛克）：把所有的聊天记录截一张长图并保存到相册，需要把群友的头像和昵称打码；
预览：预览打码后的效果。

如何解决

下面这张图是聊天记录页面，点击导航右边按钮，会弹出 ActionSheet。从图中可以看出，添加截图功能，在 ActionSheet 上添加是不错的选择。

我们下面主要的工作是：

1.拿到 ActionSheet 并添加3个菜单（截图，截图（马赛克），预览）
2.找到聊天记录所在的页面，找到展示消息的视图；
3.获取 ActionSheet 点击菜单对应的事件；
4.实现截取长图并保存到相册功能；
5.对头像和昵称打马赛克；
6.添加版权信息。

实现

本文使用 MonkeyDev 工具开发（无需越狱），重点是教你如何开发一个微信插件，并不打算介绍工具。关注我们的朋友应该都知道，以往的 #iOS知识小集中我已经介绍了三个工具的使用。这三个工具在下面会用到。

给 ActionSheet 添加3个菜单

使用 Reveal 工具查看聊天记录页面对应的 VC，ActionSheet 对应的类名。如何使用 Reveal 调试第三方 APP，网上有很多教程。使用 MonkeyDev 无需越狱。

通过上图可以看到聊天记录所对应的VC是 MsgRecordDetailViewController ，使用 MMTableView 展示聊天内容。弹出的 ActionSheet 对应的类为 WCActionSheet , 可以发现它是一个 UIWindow 。那么我们看看这几个类中的内容吧。

使用 class-dump 查看第三方 APP 的头文件。在 #iOS知识小集中已经介绍过这个工具的使用。

在 MsgRecordDetailViewController 的头文件中发现有一个 WCActionSheet *favImgLongPressAction; 我们可以断定出 WCActionSheet 就是我们要找的 ActionSheet。好了，接下来主要就是看 WCActionSheet 的头文件，挖掘有用的信息。

WCActionSheet头文件

@property(strong, nonatomic) NSMutableArray *buttonTitleList;
- (void)showInView:(id)arg1;
- (long long)addButtonWithItem:(id)arg1 atIndex:(unsigned long long)arg2;
- (long long)addButtonWithTitle:(id)arg1 atIndex:(unsigned long long)arg2;
- (long long)addButtonWithTitle:(id)arg1;

我们的目标是给 WCActionSheet 添加3个菜单。下面这些方法似乎对我们有用。目前想到有两种方案：

1.在 buttonTitleList 中添加一个对象

我们所关心的最主要的问题是 buttonTitleList 中存放的的对象是什么？需要使用Cycript工具，这个工具在以往的 #iOS知识小集介绍过，想了解的朋友可以在知识小集小程序中搜索 Cycript调试第三方APP 。

通过 Cycript 可以看到 buttonTitleList 中存放的对象是 WCActionSheetItem 。我们看看 WCActionSheetItem 的头文件，发现其实就是一个 Model 对象，用来表示菜单的标题，颜色等等。

@interface WCActionSheetItem : NSObject 
@property(copy, nonatomic) NSString *titleColor;
@property(copy, nonatomic) NSString *title; 
- (id)initWithTitle:(id)arg1 fontSize:(long long)arg2 fontColor:(id)arg3 WithDesc:(id)arg4 descFontSize:(long long)arg5 descFontColor:(id)arg6 enable:(_Bool)arg7;
- (id)initWithTitle:(id)arg1;

看到这里，我们可以直接在 buttonTitleList 中添加 WCActionSheetItem 实例即可。

2.直接调用 addButtonWithTitle: 方法

从上图可以看出直接调用 addButtonWithTitle: 这个方法，返回一个 Index 为 3，说明可以直接调用这个方法。

接下来主要的问题是，找到添加菜单的时机。第一想到的是在 WCActionSheetDelegate 的代理中添加菜单，果断在 MsgRecordDetailViewController 中 Hook 下面这3个代理方法，但是通过实验证明，发现最后两个方法并没有被调用，因为 MsgRecordDetailViewController 并没有实现这两个代理，只好放弃了这种思路。

- (void)actionSheet:(id)arg1 clickedButtonAtIndex:(long long)arg2;
- (void)didPresentActionSheet:(WCActionSheet *)arg1;
- (void)willPresentActionSheet:(WCActionSheet *)arg1;

无奈之下看到 WCActionSheet 中有个 showInView: 方法，可以直接 Hook 这个方法。但这样导致所有的 WCActionSheet 都会被添加了额外的菜单。而我们的目的只是在聊天记录页中的 WCActionSheet 显示截图菜单。所以用 [WeChatSaveData defaultSaveData].isNeedAddMenu 加了一个判断，isNeedAddMenu 在 MsgRecordDetailViewController 页面将要出现的时候，设置成 YES，在页面将要消失的时候，设置成 NO。所以需要 Hook MsgRecordDetailViewController 的 viewWillAppear: 和 viewWillDisappear: 方法。

CHOptimizedMethod1(self, void, WCActionSheet, showInView, UIView *, view){
    if ([WeChatSaveData defaultSaveData].isNeedAddMenu) {
        // 方案一
        [self addButtonWithTitle:@""]; // 填坑
        [self addButtonWithTitle:kScreenshotTitle];
        [self addButtonWithTitle:kScreenshotTitleMask];
        
        // 方案二
       WCActionSheetItem *shotItem = [[objc_getClass("WCActionSheetItem") alloc] initWithTitle:kScreenshotTitle];
        WCActionSheetItem *shotItem2 = [[objc_getClass("WCActionSheetItem") alloc] initWithTitle:kScreenshotTitleMask];
        [self.buttonTitleList addObject:shotItem];
        [self.buttonTitleList addObject:shotItem2];
    }
    CHSuper1(WCActionSheet, showInView, view);
}

执行结果如下图：

找到聊天记录页面和展示消息的视图

MsgRecordDetailViewController头文件

@interface MsgRecordDetailViewController: UIViewController
{
    MMTableView *m_tableView;
}
- (void)viewWillAppear:(_Bool)arg1;
- (void)viewWillDisappear:(_Bool)arg1;
- (void)actionSheet:(id)arg1 clickedButtonAtIndex:(long long)arg2;
- (UITableViewCell *)tableView:(id)arg1 cellForRowAtIndexPath:(id)arg2;
@end

通过对 MsgRecordDetailViewController 头文件分析，可以达到截图功能，只需要截取 TableView 为一张长图即可。

获取到 MsgRecordDetailViewController 实例，使用 KVC 的方式即可获取到 MMTableView

MMTableView *tableView = [viewController valueForKeyPath:@"m_tableView"];

实现截取长图的功能，保存到相册

首先需要 Hook actionSheet: clickedButtonAtIndex: 捕获菜单的点击事件，做截图功能。

CHOptimizedMethod2(self, void, MsgRecordDetailViewController, actionSheet, WCActionSheet*, sheet, clickedButtonAtIndex, int, index){
    CHSuper2(MsgRecordDetailViewController, actionSheet, sheet, clickedButtonAtIndex, index);

    [WeChatCapture saveCaptureImageWithSheet:sheet index:index viewController:self];
}

saveCaptureImageWithSheet 这个方法中主要获取到 MMTableView 并截图保存到相册。有兴趣可以看源码。

对头像和昵称打马赛克

为了保护用户的隐私，需要对用户的头像和昵称做保护，那么我们可以在 TableView 的代理中获取头像和昵称对应的 View，然后替换 View 的内容即可。需要 Hook cellForRowAtIndexPath 这个方法。

CHOptimizedMethod2(self, UITableViewCell *, MsgRecordDetailViewController, tableView, MMTableView *, tableViewArg, cellForRowAtIndexPath, NSIndexPath, *indexPath){
    UITableViewCell *cell = CHSuper2(MsgRecordDetailViewController, tableView, tableViewArg, cellForRowAtIndexPath, indexPath);
    [WeChatCapture updateCellDataWithCell:cell indexPath:indexPath];
    return cell;
}

获取到 Cell 如果当前是要打码截图，需要对头像和昵称的内容做处理。这里做一个特殊的处理，头像和昵称我们换成三国人物的头像的名字。

给微信聊天记录添加截图功能

正文