File: NGMimeTextBodyGenerator.m

package info (click to toggle)
sope 3.2.6-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 16,408 kB
  • ctags: 2,779
  • sloc: objc: 167,855; sh: 3,646; ansic: 3,399; python: 318; makefile: 181
file content (99 lines) | stat: -rw-r--r-- 3,195 bytes parent folder | download | duplicates (9)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*
  Copyright (C) 2000-2007 SKYRIX Software AG
  Copyright (C) 2007      Helge Hess

  This file is part of SOPE.

  SOPE is free software; you can redistribute it and/or modify it under
  the terms of the GNU Lesser General Public License as published by the
  Free Software Foundation; either version 2, or (at your option) any
  later version.

  SOPE 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 Lesser General Public
  License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with SOPE; see the file COPYING.  If not, write to the
  Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#include "NGMimeBodyGenerator.h"
#include "NGMimePartGenerator.h"
#include <NGExtensions/NSString+Encoding.h>
#include "common.h"

@implementation NGMimeTextBodyGenerator

- (NSStringEncoding)_encodingFromContentType:(NGMimeType *)_type {
  NSStringEncoding encoding;
  NSString *charset;

  encoding = [NSString defaultCStringEncoding];
  if (_type != nil) {
    charset = [_type valueOfParameter: @"charset"];
    if ([charset length] > 0) {
      if ([charset isEqualToString: @"utf-8"])
	encoding = NSUTF8StringEncoding;
      else if ([charset isEqualToString: @"iso-8859-1"])
	encoding = NSISOLatin1StringEncoding;
      /* more should be handled here */
    }
  }
    
  return encoding;
}

- (NSData *)generateBodyOfPart:(id<NGMimePart>)_part
  additionalHeaders:(NGMutableHashMap *)_addHeaders
  delegate:(id)_delegate
{
  NSData *data;
  id     body;
  
  body = [_part body];
  data = nil;
  
  if ([body isKindOfClass:[NSString class]]) {
    NSString *charset = [[_part contentType] valueOfParameter:@"charset"];
    if ([charset isNotEmpty])
      data = [body dataUsingEncodingNamed:charset];
    
    if (data == nil) /* either no charset given or the charset failed */
      data = [body dataUsingEncoding:[NSString defaultCStringEncoding]];
  }
  else if ([body respondsToSelector:@selector(bytes)]) {
    /* an NSData, but why not check the class?!, we can't just cast it */
    data = body;
  }
  else if ([body respondsToSelector:@selector(dataUsingEncoding:)]) {
    /* hm, whats that?, NSString is covered before */
    NSStringEncoding encoding = 0;
    NSString *charset = [[_part contentType] valueOfParameter:@"charset"];
    
    if ([charset isNotEmpty]) {
#if NeXT_Foundation_LIBRARY || COCOA_Foundation_LIBRARY
      encoding = [NSString stringEncodingForEncodingNamed:charset];
#else
      encoding = [self _encodingFromContentType:[_part contentType]];
#endif
    }
    
    data = [body dataUsingEncoding:
		   (encoding != 0 ? encoding : NSISOLatin1StringEncoding)];
  }
  else if (body != nil) {
    [self errorWithFormat:@"unexpected part body %@: %@", [body class], body];
    return nil;
  }
  
  if (data == nil) {
    [self warnWithFormat:@"%s: generate empty body", __PRETTY_FUNCTION__];
    data = [NSData data];
  }
  return [self encodeData:data forPart:_part additionalHeaders:_addHeaders];
}

@end /* NGMimeTextBodyGenerator */