IOS中摇一摇实现截屏(可实现问题反馈的功能)

　　有一段时间没有更新博客了，今天更新一篇关于最近工作中用到的一个功能，先简单描述一下：我们知道，测试人员在测试客户端产品时，当出现问题或者BUG的时候，都得先对页面截图，然后从相册中选择截图，加上一段描述放到TD库或者直接通过邮件发给开发人员，以方便开发人员进行修改，过程繁琐。通过下面功能的介绍，您可以很方便的让测试人员遇到BUG的时候，摇一摇设备，然后自动截图并保存到应用的沙盒之中。截图保存到沙盒之后，您可以自定义一个问题反馈的页面，对问题进行描述(描述可使用第三方平台，如讯飞语音，实现语音录入的效果)之后，将描述信息和页面截图一起发送给开发人员(可以是邮件，也可以是直接发送到后台)。

　　下面来看看功能是怎么实现的。

　　首先，定义了一个category：ShakeAndCutter

　　UIViewController+ShakeAndCutter.h文件源码：

#import <UIKit/UIKit.h>

@interface UIViewController (ShakeAndCutter)

@end

　　UIViewController+ShakeAndCutter.m文件源码：

#import "UIViewController+ShakeAndCutter.h"
#import <QuartzCore/QuartzCore.h>
#import "TestFeedbackViewController.h"

@implementation UIViewController (ShakeAndCutter)

- (BOOL)canBecomeFirstResponder
{
    return YES;
}

#pragma mark - 摇一摇动作处理

- (void)motionBegan:(UIEventSubtype)motion withEvent:(UIEvent *)event
{
    NSLog(@"began");
}

- (void)motionCancelled:(UIEventSubtype)motion withEvent:(UIEvent *)event
{
    NSLog(@"cancel");
}

- (void)motionEnded:(UIEventSubtype)motion withEvent:(UIEvent *)event
{
    NSLog(@"end");
    [self cutterViewToDocument];
    
    // 这里是自定义的问题反馈页面,也可以直接跳转到系统邮件发送的页面
    TestFeedbackViewController *testFeedbackViewController = [[[TestFeedbackViewController alloc] initWithNibName:@"TestFeedbackViewController" bundle:nil] autorelease];
    [self.navigationController pushViewController:testFeedbackViewController animated:YES];
}

#pragma mark - 截屏处理

- (void)cutterViewToDocument
{
    UIWindow *screenWindow = [[UIApplication sharedApplication] keyWindow];
    
    UIGraphicsBeginImageContext(screenWindow.frame.size);
    [screenWindow.layer renderInContext:UIGraphicsGetCurrentContext()];
    UIImage *screenShot = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    NSData *screenShotPNG = UIImagePNGRepresentation(screenShot);
    NSError *error = nil;
    [screenShotPNG writeToFile:[[NSHomeDirectory() stringByAppendingPathComponent:@"Documents"] stringByAppendingPathComponent:@"error.png"] options:NSAtomicWrite error:&error];
}

@end

　　至此，摇一摇和截图功能就已经完成了，那么在应用中怎么使用呢？

• 对于IOS6.0以上的设备，只需要在项目的pch文件中引进头文件#import "UIViewController+ShakeAndCutter.h"即可(我在IOS6.0.1、IOS6.1.5、IOS7.0.6设备上已经测试过)，这样测试工作完成之后，您只需要将这行代码注释或者删除就可以了；
• 对于IOS6.0以下的设备，在需要实现摇一摇效果的页面，有可能要加入如下代码(手头没有IOS6.0以下的设备，如果大家有的话，帮忙验证一下，非常感谢)：

　　1.viewDidLoad方法中，添加两行代码：

[[UIApplication sharedApplication] setApplicationSupportsShakeToEdit:YES];
[self becomeFirstResponder];

        2.viewDidAppear方法中，添加一行代码：

[self becomeFirstResponder];

        3.viewDidDisappear方法中，添加一行代码：

[self resignFirstResponder];

　　当然，以上的功能，您也可以实现其他的效果：比如说用户问题的反馈等等，具体可以根据应用的需求而定。

　　

　　这里，再附上发送到邮件功能的示例代码：

　　MailViewController.h文件源码：

#import <UIKit/UIKit.h>
#import <MessageUI/MessageUI.h>
#import<MessageUI/MFMailComposeViewController.h>

@interface MailViewController : UIViewController<MFMailComposeViewControllerDelegate,
MFMessageComposeViewControllerDelegate>
{
    UIButton                    *shareToMailButton;
}

@end

　　MailViewController.m文件源码：

#import "MailViewController.h"
#import "Constants.h"

@interface MailViewController ()

@end

@implementation MailViewController

- (id)initWithNibName:(NSString *)nibNameOrNil bundle:(NSBundle *)nibBundleOrNil
{
    self = [super initWithNibName:nibNameOrNil bundle:nibBundleOrNil];
    if (self) {
        // Custom initialization
    }
    return self;
}

- (void)viewDidLoad
{
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    
    self.title = @"Bug反馈";
    
    shareToMailButton = [self buttonWithFrame:CGRectMake(20, 100, 280, 40) action:@selector(btnClicked:) withTag:1000];
    [shareToMailButton setTitle:@"Bug反馈" forState:UIControlStateNormal];
}

- (void)didReceiveMemoryWarning
{
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

#pragma mark - 自定义方法

- (void)btnClicked:(id)sender
{
    UIButton *btnSender = (UIButton *)sender;
    
    switch (btnSender.tag)
    {
        case 1000:  // 分享到邮箱
        {
            [self showMailPicker];
            break;
        }
        default:
            break;
    }
}

/*******************************************************************************
 * 方法名称：buttonWithFrame:action:
 * 功能描述：初始化页面按钮，并添加到页面视图
 * 输入参数：
 * 输出参数：
 ******************************************************************************/
- (UIButton *)buttonWithFrame:(CGRect)frame action:(SEL)action withTag:(int)tag
{
    UIImage *buttonBackgroundImage = [[UIImage imageNamed:@"button_background.png"] stretchableImageWithLeftCapWidth:5 topCapHeight:5];
    UIImage *disabledButtonBackgroundImage = [[UIImage imageNamed:@"button_background_disabled.png"] stretchableImageWithLeftCapWidth:5 topCapHeight:5];
    
    UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];
    button.frame = frame;
    [button setBackgroundImage:buttonBackgroundImage forState:UIControlStateNormal];
    [button setBackgroundImage:disabledButtonBackgroundImage forState:UIControlStateDisabled];
    [button setTitleColor:[UIColor whiteColor] forState:UIControlStateNormal];
    [button setTitleColor:[UIColor grayColor] forState:UIControlStateDisabled];
    [button addTarget:self action:action forControlEvents:UIControlEventTouchUpInside];
    button.tag = tag;
    [self.view addSubview:button];
    
    return button;
}

- (void)showMailPicker
{
    Class mailClass = (NSClassFromString(@"MFMailComposeViewController"));
    
    if (mailClass !=nil)
    {
        if ([mailClass canSendMail])
        {
            [self displayMailComposerSheet];
        }
        else
        {
            UIAlertView *alert=[[UIAlertView alloc] initWithTitle:@""message:@"设备不支持邮件功能" delegate:self cancelButtonTitle:@"确定" otherButtonTitles:nil];
            [alert show];
            [alert release];
        }
    }
    else
    {
        
    }
    
}

- (void)displayMailComposerSheet
{
    MFMailComposeViewController *picker = [[MFMailComposeViewController alloc] init];
    
    // 设置picker的委托方法，完成之后会自动调用成功或失败的方法
    picker.mailComposeDelegate = self;
    // 添加主题
    [picker setSubject:@"Bug反馈"];
    // 添加收件人
    NSArray *toRecipients = [NSArray arrayWithObject:@"279352257@qq.com"];
    // 说明：也可以添加多个收件人，代码如下所示：
//    NSArray *toRecipients = [NSArray arrayWithObjects:@"one@qq.com",@"two@qq.com",nil];
    // 添加抄送
//    NSArray *ccRecipients = [NSArray arrayWithObjects:@"second@qq.com",@"third@qq.com", nil];
    // 添加密送
//    NSArray *bccRecipients = [NSArray arrayWithObject:@"fourth@qq.com"];
    
    [picker setToRecipients:toRecipients];
//    [picker setCcRecipients:ccRecipients];
//    [picker setBccRecipients:bccRecipients];
    // 发送图片附件(其他格式的附件，可以都先转化称NSData类型，然后设置相应的mimeType即可，如txt类型为@"text/txt",doc类型为@"text/doc",pdf类型为@"file/pdf"等等)
    NSData *myData = [NSData dataWithContentsOfFile:[[NSHomeDirectory() stringByAppendingPathComponent:@"Documents"] stringByAppendingPathComponent:@"error.png"]];
    [picker addAttachmentData:myData mimeType:@"image/jpeg" fileName:@"error.png"];
    NSString *emailBody = [NSString stringWithFormat:@"<p>文字没有全部显示</p>"];
    
    // 直接在HTML代码中写入图片的地址
//    NSString *emailBody = [NSString stringWithFormat:@"<img src='http://p2.so.qhimg.com/t0130e3288d86929b97.jpg' /><p>我分享了图片</p>"];
    
    [picker setMessageBody:emailBody isHTML:YES];
    [self presentModalViewController:picker animated:YES];
    [picker release];
}


- (void)mailComposeController:(MFMailComposeViewController*)controller
          didFinishWithResult:(MFMailComposeResult)result error:(NSError*)error
{
    switch (result)
    {
        case MFMailComposeResultCancelled:
            NSLog(@"Result: Mail sending canceled");  // 邮件发送取消  
            break;
        case MFMailComposeResultSaved:
            NSLog(@"Result: Mail saved");  // 邮件保存成功
            break;
        case MFMailComposeResultSent:
            NSLog(@"Result: Mail sent");  // 邮件发送成功
            break;
        case MFMailComposeResultFailed:
            NSLog(@"Result: Mail sending failed");  // 邮件发送失败
            break;
        default:
            NSLog(@"Result: Mail not sent");
            break;
    }
    [self dismissModalViewControllerAnimated:YES];
}

- (void)messageComposeViewController:(MFMessageComposeViewController *)controller didFinishWithResult:(MessageComposeResult)result
{
    NSLog(@"messageComposeViewController");
}

@end