/* NSData+Mail.m - this file is part of SOGo
 *
 * Copyright (C) 2007-2008 Inverse inc.
 *
 * Author: Wolfgang Sourdeau <wsourdeau@inverse.ca>
 *         Ludovic Marcotte <lmarcotte@inverse.ca>
 *
 * This file is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This file is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#import <Foundation/NSString.h>

#import <NGExtensions/NGBase64Coding.h>
#import <NGExtensions/NGQuotedPrintableCoding.h>
#import <NGExtensions/NSString+Encoding.h>

#import "NSData+Mail.h"

@implementation NSData (SOGoMailUtilities)

- (NSData *) bodyDataFromEncoding: (NSString *) encoding
{
  NSString *realEncoding;
  NSData *decodedData;

  realEncoding = [encoding lowercaseString];

  if ([realEncoding isEqualToString: @"7bit"]
      || [realEncoding isEqualToString: @"8bit"])
    decodedData = self;
  else if ([realEncoding isEqualToString: @"base64"])
    decodedData = [self dataByDecodingBase64];
  else if ([realEncoding isEqualToString: @"quoted-printable"])
    decodedData = [self dataByDecodingQuotedPrintableTransferEncoding];
  else
    {
      decodedData = nil;
      NSLog (@"encoding '%@' unknown, returning nil data", realEncoding);
    }

  return decodedData;
}

- (NSString *) decodedSubject
{
  const char *cData, *endFlag;
  unsigned int len;
  NSString *decodedSubject;

  cData = [self bytes];
  len = [self length];

  if (len)
    {
      decodedSubject = nil;
      if (len > 6)
	{
	  endFlag = cData + len - 2;
	  if (*cData == '=' && *(cData + 1) == '?'
	      && *endFlag == '?' && *(endFlag + 1) == '=')
	    {
	      NSString *enc;
	      int i;

	      cData += 2;
	      i = 2;

	      while (*cData != '?' && i < len)
		{
		  cData++;
		  i++;
		}

	      enc = [[[NSString alloc] initWithData:[self subdataWithRange: NSMakeRange(2, i-2)]
				       encoding: NSASCIIStringEncoding] autorelease];

	      if (i+3 < len)
		{
		  NSData *d;
		  
		  d = [self subdataWithRange: NSMakeRange(i+3, len-i-5)];
		  
		  // We check if we have a QP or Base64 encoding
		  if (*(cData+1) == 'q' || *(cData+1) == 'Q')
		    d = [d dataByDecodingQuotedPrintable];
		  else
		    d = [d dataByDecodingBase64];
		  
		  decodedSubject = [NSString stringWithData: d  usingEncodingNamed: enc];
		}
	      else
		decodedSubject = nil;
	    }
	}
      if (!decodedSubject)
	{
	  decodedSubject
	    = [[NSString alloc] initWithData: self
				encoding: NSUTF8StringEncoding];
	  if (!decodedSubject)
	    decodedSubject
	      = [[NSString alloc] initWithData: self
				  encoding: NSISOLatin1StringEncoding];
	  [decodedSubject autorelease];
	}
    }
  else
    decodedSubject = @"";

  return decodedSubject;
}

@end