From b17e92e07fc7edc28317e824dc516eaaacf4e8d5 Mon Sep 17 00:00:00 2001 From: nixzhu Date: Fri, 3 Apr 2015 12:16:59 +0800 Subject: [PATCH] fix save in downloadAttachmentOfMessage --- Yep/Services/YepServiceSync.swift | 37 +++++++++++++++++-------------- 1 file changed, 20 insertions(+), 17 deletions(-) diff --git a/Yep/Services/YepServiceSync.swift b/Yep/Services/YepServiceSync.swift index d409cfae..dd9ab97d 100644 --- a/Yep/Services/YepServiceSync.swift +++ b/Yep/Services/YepServiceSync.swift @@ -30,25 +30,28 @@ func downloadAttachmentOfMessage(message: Message) { dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)) { let data = NSData(contentsOfURL: url) - let fileName = NSUUID().UUIDString + if let data = data { + let fileName = NSUUID().UUIDString - dispatch_async(dispatch_get_main_queue()) { - switch message.mediaType { - case MessageMediaType.Image.rawValue: - if let fileURL = NSFileManager.yepMessageImageURLWithName(fileName) { - updateMessage(message, withAttachmentFileName: fileName) + dispatch_async(dispatch_get_main_queue()) { + switch message.mediaType { + case MessageMediaType.Image.rawValue: + if let fileURL = NSFileManager.saveMessageImageData(data, withName: fileName) { + updateMessage(message, withAttachmentFileName: fileName) + } + + case MessageMediaType.Video.rawValue: + break // TODO: download video + + case MessageMediaType.Audio.rawValue: + if let fileURL = NSFileManager.saveMessageAudioData(data, withName: fileName) { + + updateMessage(message, withAttachmentFileName: fileName) + } + + default: + break } - - case MessageMediaType.Video.rawValue: - break // TODO: download video - - case MessageMediaType.Audio.rawValue: - if let fileURL = NSFileManager.yepMessageAudioURLWithName(fileName) { - updateMessage(message, withAttachmentFileName: fileName) - } - - default: - break } } }