Swiftgram/Classes/BITFeedbackMessageAttachment.m

/*
 * Author: Moritz Haarmann <post@moritzhaarmann.de>
 *
 * Copyright (c) 2012-2014 HockeyApp, Bit Stadium GmbH.
 * All rights reserved.
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use,
 * copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following
 * conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */


#import "BITFeedbackMessageAttachment.h"
#import "BITHockeyHelper.h"
#import "HockeySDKPrivate.h"

#define kCacheFolderName @"hockey_attachments"

@interface BITFeedbackMessageAttachment()

@property (nonatomic, strong) NSMutableDictionary *thumbnailRepresentations;
@property (nonatomic, strong) NSData *internalData;
@property (nonatomic, copy) NSString *filename;

@end

@implementation BITFeedbackMessageAttachment

+ (BITFeedbackMessageAttachment *)attachmentWithData:(NSData *)data contentType:(NSString *)contentType {
  BITFeedbackMessageAttachment *newAttachment = [BITFeedbackMessageAttachment new];
  newAttachment.contentType = contentType;
  newAttachment.data = data;
  return newAttachment;
}

-(id)init {
  self = [super init];
  if (self){
    self.thumbnailRepresentations = [NSMutableDictionary new];
  }
  return self;
}

-(void)setData:(NSData *)data {
  self->_internalData = data;
  self.filename = [self createFilename];
  [self->_internalData writeToFile:self.filename atomically:NO];
}

-(NSData *)data {
  if (!self->_internalData && self.filename){
    self.internalData = [NSData dataWithContentsOfFile:self.filename];
  }
  
  if (self.internalData){
    return self.internalData;
  }
  
  return nil;
}

- (void)replaceData:(NSData *)data {
  self.data = data;
  self.thumbnailRepresentations = [NSMutableDictionary new];
}

#pragma mark NSCoding

- (void)encodeWithCoder:(NSCoder *)aCoder {
  [aCoder encodeObject:self.contentType forKey:@"contentType"];
  [aCoder encodeObject:self.filename forKey:@"filename"];
  [aCoder encodeObject:self.originalFilename forKey:@"originalFilename"];

}

- (id)initWithCoder:(NSCoder *)aDecoder {
  self = [super init];
  
  if (self){
    self.contentType = [aDecoder decodeObjectForKey:@"contentType"];
    self.filename = [aDecoder decodeObjectForKey:@"filename"];
    self.thumbnailRepresentations = [NSMutableDictionary new];
    self.originalFilename = [aDecoder decodeObjectForKey:@"originalFilename"];
  }
  
  return self;
}

#pragma mark - Thubmnails / Image Representation

- (UIImage *)imageRepresentation {
  if ([self.contentType rangeOfString:@"image"].location != NSNotFound){
    return [UIImage imageWithData:self.data];
  } else {
    return bit_imageNamed(@"feedbackActiviy.png", BITHOCKEYSDK_BUNDLE); // TODO add another placeholder.
  }
}

- (UIImage *)thumbnailWithSize:(CGSize)size {
  id<NSCopying> cacheKey = [NSValue valueWithCGSize:size];
  
  if (!self.thumbnailRepresentations[cacheKey]){
    UIImage *image = self.imageRepresentation;
    UIImage *thumbnail = bit_imageToFitSize(image, size, NO);
    if (thumbnail){
      [self.thumbnailRepresentations setObject:thumbnail forKey:cacheKey];
    }
  }
  
  return self.thumbnailRepresentations[cacheKey];
}

#pragma mark - Persistence Helpers

- (NSString *)createFilename {
  NSArray* cachePathArray = NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES);
  NSString* cachePath = [cachePathArray lastObject];
  cachePath = [cachePath stringByAppendingPathComponent:kCacheFolderName];
  
  BOOL isDirectory;
  
  if (![[NSFileManager defaultManager] fileExistsAtPath:cachePath isDirectory:&isDirectory]){
    [[NSFileManager defaultManager] createDirectoryAtPath:cachePath withIntermediateDirectories:YES attributes:nil error:nil];
  }
  
  NSString *uniqueString = bit_UUID();
  cachePath = [cachePath stringByAppendingPathComponent:uniqueString];
  
  return  cachePath;
}

- (void)deleteContents {
  if (self.filename){
    [[NSFileManager defaultManager] removeItemAtPath:self.filename error:nil];
    self.filename = nil;
  }
}


@end