InternetMessages/ATBodyTextRepresentation.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
139
140
141
142
143
144
145
146
147
148
149
//
//  ATBodyTextRepresentation.m
//  ATMail
//
//  Created by 高田　明史 on 07/07/15.
//  Copyright 2007 __MyCompanyName__. All rights reserved.
//

#import "ATBodyTextRepresentation.h"
#import "ATHeaderField.h"
#import "ATContentTypeFieldBody.h"
#import "ATContentTransferEncodingFieldBody.h"
#import "ATQuotedPrintableDecoder.h"
#import "ATBase64Decoder.h"
#import "ATMessageEntity.h"

@implementation ATBodyTextRepresentation

+ (id)textRepresentationWithTextData:(NSData *)aTextData parentEntity:(ATMessageEntity *)anEntity
{
	return [self textRepresentationWithTextData:aTextData contentType:[anEntity contentType] contentTransferEncoding:[anEntity contentTransferEncoding]];
}

+ (id)textRepresentationWithTextData:(NSData *)aTextData contentType:(ATHeaderField *)aContentType contentTransferEncoding:(ATHeaderField *)aContentTransferEncoding
{
	return [[[self alloc] initWithTextData:aTextData contentType:aContentType contentTransferEncoding:aContentTransferEncoding] autorelease];
}

- (id)initWithTextData:(NSData *)aTextData contentType:(ATHeaderField *)aContentType contentTransferEncoding:(ATHeaderField *)aContentTransferEncoding;
{
	[super init];

	textData = [aTextData retain];
	text = @"";

	contentType = [aContentType retain];
	contentTransferEncoding = [aContentTransferEncoding retain];

	[self interpret];

	return self;
}

- (void)dealloc
{
	[contentType release];
	[contentTransferEncoding release];
	[textData release];
	[text release];
	[attributedString release];

	[super dealloc];
}

- (void)interpret
{
	if ([contentType typeIs:@"text"] && textData)
	{
		NSString *aCharset = [contentType charset];
		CFStringEncoding aCFEncoding = CFStringConvertIANACharSetNameToEncoding((CFStringRef)aCharset);

		if (aCFEncoding != kCFStringEncodingInvalidId)
		{
			NSStringEncoding aNSEncoding = CFStringConvertEncodingToNSStringEncoding(aCFEncoding);
			NSString *aText = [[NSString alloc] initWithData:textData encoding:aNSEncoding];

			if (aText)
				text = aText;
			else if ([aCharset isEqualToString:@"Shift_JIS"])
			{
				aText = [[NSString alloc] initWithData:textData encoding:NSShiftJISStringEncoding];

				if (aText)
					text = aText;
			}
		}
	}
}

- (NSString *)text
{
	return text;
}

- (NSString *)stringValue
{
	return [self text];
}

- (void)addAttributesToURIsInText
{
	NSMutableAttributedString *anAttrStr = [[NSMutableAttributedString alloc] initWithString:[self text]];
	NSScanner *aScanner = [NSScanner scannerWithString:[self text]];
	BOOL aScannerIsAtEnd = NO;
	NSArray *aSchemeNames = [NSArray arrayWithObjects:@"http", @"https", @"mailto", nil];

	[aScanner setCharactersToBeSkipped:nil];

	while (!(aScannerIsAtEnd = [aScanner isAtEnd]))
	{
		unsigned aSchemeStatingLocation = 0;
		NSString *aURIString = nil;
		NSURL *aURL = nil;
		NSEnumerator *anEnumerator = [aSchemeNames objectEnumerator];
		NSString *aSchemeName = nil;
		BOOL aSchemeNameFound = NO;

		[aScanner scanCharactersFromSet:[NSCharacterSet whitespaceAndNewlineCharacterSet] intoString:nil];

		aSchemeStatingLocation = [aScanner scanLocation];
		anEnumerator = [aSchemeNames objectEnumerator];

		while ((aSchemeName = [anEnumerator nextObject]) && !(aSchemeNameFound = [aScanner scanString:[aSchemeName stringByAppendingString:@":"] intoString:nil]))
			;

		if (aSchemeNameFound)
		{
			[aScanner setScanLocation:aSchemeStatingLocation];
			[aScanner scanUpToCharactersFromSet:[NSCharacterSet whitespaceAndNewlineCharacterSet] intoString:&aURIString];

			if (aURIString && (aURL = [NSURL URLWithString:aURIString]))
			{
				NSRange aRange = NSMakeRange(aSchemeStatingLocation, [aURIString length]);

				[anAttrStr addAttribute:NSLinkAttributeName value:aURL  range:aRange];
				[anAttrStr addAttribute:NSCursorAttributeName value:[NSCursor pointingHandCursor] range:aRange];
			}
		}
		else
			[aScanner scanUpToCharactersFromSet:[NSCharacterSet whitespaceAndNewlineCharacterSet] intoString:nil];
	}

	attributedString = anAttrStr;
}

- (NSMutableAttributedString *)attributedString
{
	if (!attributedString)
		[self addAttributesToURIsInText];

	return attributedString;
}

- (BOOL)saveTo:(NSString *)aFilePath
{
	return [textData writeToFile:aFilePath atomically:YES];
}

@end