InternetMessages/ATEncodedWord.m at master · Lessercomputer/InternetMessages · GitHub

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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
//
//  ATEncodedWord.m
//  ATMail
//
//  Created by 高田　明史 on 06/10/24.
//  Copyright 2006 __MyCompanyName__. All rights reserved.
//

#import "ATEncodedWord.h"
#import "ATCFWS.h"
#import "ATBase64Decoder.h"
#import "ATQuotedPrintableDecoder.h"

@implementation ATEncodedWord

+ (id)encodedWordWith:(NSString *)aCharset encoding:(NSString *)anEncoding encodedText:(NSString *)anEncodedText
{
	return [[[self alloc] initWith:aCharset encoding:anEncoding encodedText:anEncodedText] autorelease];
}

- (id)initWith:(NSString *)aCharset encoding:(NSString *)anEncoding encodedText:(NSString *)anEncodedText
{
	[super init];

	[self setCharset:aCharset];
	[self setEncoding:anEncoding];
	encodedText = [anEncodedText retain];

	return self;
}

- (void)dealloc
{
	[self setCharset:nil];
	[self setEncoding:nil];
	[encodedText release];
	[self setLeftCFWS:nil];
	[self setRight:nil];

	[super dealloc];
}

- (BOOL)isEncoded
{
	return YES;
}

- (void)setCharset:(NSString *)aCharset
{
	[charset release];
	charset = [aCharset copy];
}

- (void)setEncoding:(NSString *)anEncoding
{
	[encoding release];
	encoding = [anEncoding copy];
}

- (void)setLeftCFWS:(ATCFWS *)aLeftCFWS
{
	[leftCFWS release];
	leftCFWS = [aLeftCFWS retain];
}

- (void)setRight:(ATCFWS *)aRightCFWS
{
	[rightCFWS release];
	rightCFWS = [aRightCFWS retain];
}

- (NSString *)displayString
{
	return [[self class] decodeEncodedWord:charset encoding:encoding encodedText:encodedText leftCFWS:leftCFWS rightCFWS:rightCFWS includingRightCFWS:YES];
}

- (NSString *)displayStringExceptRightCFWS
{
	return [[self class] decodeEncodedWord:charset encoding:encoding encodedText:encodedText leftCFWS:leftCFWS rightCFWS:rightCFWS includingRightCFWS:NO];
}

+ (NSString *)decodeEncodedWord:(NSString *)aCharset encoding:(NSString *)anEncoding encodedText:(NSString *)anEncodedText leftCFWS:(ATCFWS *)aLeftCFWS rightCFWS:(ATCFWS *)aRightCFWS includingRightCFWS:(BOOL)anIncludingFlag
{
	NSData *aStringData = nil;
	CFStringEncoding aCFEncoding;
	NSString *anEncodedWord = nil;
	//NSString *aReturningWord = nil;
	NSString *aLeftString = @"";
	NSString *aRightString = @"";

	if ([[anEncoding lowercaseString] isEqualToString:@"b"])
	{
		ATBase64Decoder *aDecoder = [[[ATBase64Decoder alloc] initWith:[anEncodedText dataUsingEncoding:NSASCIIStringEncoding]] autorelease];
		aStringData = [aDecoder decode];
	} else if ([[anEncoding lowercaseString] isEqualToString:@"q"])
	{
		aStringData = [ATQuotedPrintableDecoder decodeWithQ:[anEncodedText dataUsingEncoding:NSASCIIStringEncoding]];
	}

	aCFEncoding = CFStringConvertIANACharSetNameToEncoding((CFStringRef)aCharset);

	if (aStringData && aCFEncoding != kCFStringEncodingInvalidId)
	{
		NSStringEncoding aNSEncoding = CFStringConvertEncodingToNSStringEncoding(aCFEncoding);

		anEncodedWord = [[[NSString alloc] initWithData:aStringData encoding:aNSEncoding] autorelease];
	}

	if (!anEncodedWord)
		anEncodedWord = [self encodedWord:aCharset encoding:anEncoding encodedText:anEncodedText];

	//return [NSString stringWithFormat:@"%@%@%@", (aLeftCFWS ? @" " : @""), anEncodedWord, ((aRightCFWS && anIncludingFlag) ? @" " : @"")];

	if (aLeftCFWS)
	{
		if ([aLeftCFWS isKindOfClass:[NSString class]])
			aLeftString = aLeftCFWS;
		else
			aLeftString = @" ";
	}

	if (anIncludingFlag && aRightCFWS)
	{
		if ([aRightCFWS isKindOfClass:[NSString class]])
			aRightString = aRightCFWS;
		else
			aRightString = @" ";
	}

	return [NSString stringWithFormat:@"%@%@%@", aLeftString, anEncodedWord, aRightString];
}

+ (NSString *)encodedWord:(NSString *)aCharset encoding:(NSString *)anEncoding encodedText:(NSString *)anEncodedText
{
	return [NSString stringWithFormat:@"=?%@?%@?%@?=", aCharset, anEncoding, anEncodedText];
}

@end