gspell/NSStringAdditions.m at master · natan/gspell · 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
//
//  NSStringAdditions.m
//  gSpell
//
//  Created by Nathan Spindel on 5/17/08.
//

#import "NSStringAdditions.h"


@implementation NSString (GoogleSpellingAdditions)

/* request/response learned from http://weblogs.asp.net/pwelter34/archive/2005/07/19/419838.aspx */
- (NSString *)googleSpellingSuggestion {
	NSString *preferredLanguage = [[[NSUserDefaults standardUserDefaults] objectForKey: @"AppleLanguages"] objectAtIndex: 0];

	NSURL *updateURL = [NSURL URLWithString: [NSString stringWithFormat: @"https://www.google.com/tbproxy/spell?lang=%@&hl=%@", preferredLanguage, preferredLanguage]];
	NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL: updateURL];
	[request setHTTPMethod: @"POST"];

	NSString *body = [NSString stringWithFormat: @"<?xml version=\"1.0\" encoding=\"utf-8\" ?><spellrequest textalreadyclipped=\"0\" ignoredups=\"0\" ignoredigits=\"1\" ignoreallcaps=\"0\"><text>%@</text></spellrequest>", self];
	[request setHTTPBody: [body dataUsingEncoding: NSUTF8StringEncoding allowLossyConversion: YES]];

	NSURLResponse *urlResponse = nil;
	NSData *responseData = [NSURLConnection sendSynchronousRequest: request returningResponse: &urlResponse error: nil];

	NSMutableString *suggestionString = nil;

	if (responseData && ([(NSHTTPURLResponse *)urlResponse statusCode] == 200)) {
		suggestionString = [[self mutableCopy] autorelease];

		NSXMLDocument *responseXMLDocument = [[[NSXMLDocument alloc] initWithData: responseData options: nil error: nil] autorelease];

		NSArray *suggestions = [responseXMLDocument objectsForXQuery: @".//c" error: nil];

		int offsetModifier = 0;
		NSXMLElement *currentSuggestion = nil;
		NSEnumerator *suggestionsEnumerator = [suggestions objectEnumerator];
		while (currentSuggestion = [suggestionsEnumerator nextObject]) {
			int offset = [[[currentSuggestion attributeForName: @"o"] objectValue] intValue];
			int length = [[[currentSuggestion attributeForName: @"l"] objectValue] intValue];

			NSArray *suggestedSpellings = [[currentSuggestion stringValue] componentsSeparatedByString: @"\t"];
			if ([suggestedSpellings count]) {
				NSString *firstSuggestion = [suggestedSpellings objectAtIndex: 0];
				[suggestionString replaceCharactersInRange: NSMakeRange(offset + offsetModifier, length) withString: firstSuggestion];
				offsetModifier += ([firstSuggestion length] - length);
			}
		}
	}

	return suggestionString;
}

@end