/*
 * 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"
#import <MobileCoreServices/MobileCoreServices.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 {
  
  static NSDateFormatter *formatter;
  
  if(!formatter){
    formatter = [NSDateFormatter new];
    formatter.dateStyle = NSDateFormatterShortStyle;
    formatter.timeStyle = NSDateFormatterShortStyle;
    
  }
  
  BITFeedbackMessageAttachment *newAttachment = [BITFeedbackMessageAttachment new];
  newAttachment.contentType = contentType;
  newAttachment.data = data;
  newAttachment.originalFilename = [NSString stringWithFormat:@"Attachment: %@", [formatter stringFromDate:[NSDate date]]];
  return newAttachment;
}

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

-(void)setData:(NSData *)data {
  self->_internalData = data;
  self.filename = [self possibleFilename];
  [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];
}

-(BOOL)needsLoadingFromURL {
  return (self.sourceURL && ![[NSFileManager defaultManager] fileExistsAtPath:self.localURL.absoluteString]);
}

- (BOOL)isImage {
  return ([self.contentType rangeOfString:@"image"].location != NSNotFound);
}

- (NSURL *)localURL {
  if (self.filename){
    return [NSURL fileURLWithPath:self.filename];
  } else
  { return nil;}
}

#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"];
  [aCoder encodeObject:self.sourceURL forKey:@"url"];
  
  
}

- (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"];
    self.sourceURL = [aDecoder decodeObjectForKey:@"url"];
    
  }
  
  return self;
}

#pragma mark - Thubmnails / Image Representation

- (UIImage *)imageRepresentation {
  if ([self.contentType rangeOfString:@"image"].location != NSNotFound && self.filename ){
    return [UIImage imageWithData:self.data];
  } else {
    // Create a Icon ..
    UIDocumentInteractionController* docController = [[UIDocumentInteractionController alloc] init];
    docController.name = self.originalFilename;
    NSArray* icons = docController.icons;
    if (icons.count){
      return icons[0];
    } else {
      return nil;
    }
  }
}

- (UIImage *)thumbnailWithSize:(CGSize)size {
  id<NSCopying> cacheKey = [NSValue valueWithCGSize:size];
  
  if (!self.thumbnailRepresentations[cacheKey]){
    UIImage *image = self.imageRepresentation;
    // consider the scale.
    if (!image)
      return nil;
    
    CGFloat scale = [UIScreen mainScreen].scale;
    
    if (scale != image.scale){
      
      CGSize scaledSize = CGSizeApplyAffineTransform(size, CGAffineTransformMakeScale(scale, scale));
      UIImage *thumbnail = bit_imageToFitSize(image, scaledSize, YES) ;
      
      UIImage *scaledTumbnail = [UIImage imageWithCGImage:thumbnail.CGImage scale:scale orientation:thumbnail.imageOrientation];
      if (thumbnail){
        [self.thumbnailRepresentations setObject:scaledTumbnail forKey:cacheKey];
      }
      
      } else {
        UIImage *thumbnail = bit_imageToFitSize(image, size, YES) ;
        
        [self.thumbnailRepresentations setObject:thumbnail forKey:cacheKey];
        
      }
    
  }
  
    return self.thumbnailRepresentations[cacheKey];
  }

  
#pragma mark - Persistence Helpers
  
  - (NSString *)possibleFilename {
    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];
    
    // File extension that suits the Content type.
    
    CFStringRef mimeType = (__bridge CFStringRef)self.contentType;
    CFStringRef uti = UTTypeCreatePreferredIdentifierForTag(kUTTagClassMIMEType, mimeType, NULL);
    CFStringRef extension = UTTypeCopyPreferredTagWithClass(uti, kUTTagClassFilenameExtension);
    if (extension){
    cachePath = [cachePath stringByAppendingPathExtension:(__bridge NSString *)(extension)];
      CFRelease(extension);
 
    }
    
    CFRelease(uti);
    
    return  cachePath;
  }
  
  - (void)deleteContents {
    if (self.filename){
      [[NSFileManager defaultManager] removeItemAtPath:self.filename error:nil];
      self.filename = nil;
    }
  }
  
  
  @end