#import "KeywordSaveController.h"
#import "KeywordMapper.h"
#import "KeywordMapping.h"
#import "KeywurlPlugin.h"
@interface KeywordSaveController (Internal)
- (void) collectFields: (NSMutableArray*) fields
fromElement: (DOMElement*) element;
- (void) collectButtons: (NSMutableDictionary*) buttons
fromElement: (DOMElement*) element;
@end
@interface FormField : NSObject {
NSString* name;
NSString* value;
}
- initWithName: (NSString*) theName value: (NSString*) theValue;
- (NSString*) name;
- (NSString*) value;
@end
@implementation FormField
- initWithName: (NSString*) theName value: (NSString*) theValue {
name = [theName copy];
value = [theValue copy];
return self;
}
- (NSString*) name {
return name;
}
- (NSString*) value {
return value;
}
@end
@implementation KeywordSaveController
- (id) initWithUrl: (NSString*) theUrl
inputElement: (DOMElement*) theInputElement
formElement: (DOMElement*) theFormElement {
url = [theUrl copy];
inputElement = [theInputElement retain];
formElement = [theFormElement retain];
if (![NSBundle loadNibNamed: @"SaveKeywordWindow" owner: self]) {
NSLog(@"Could not load resource.");
return nil;
}
{
[buttonPopupButton removeAllItems];
NSMutableDictionary* buttons = [NSMutableDictionary new];
[self collectButtons: buttons fromElement: formElement];
if ([buttons count] <= 1) {
[buttonPopupButton setHidden: YES];
[buttonPopupLabel setHidden: YES];
} else {
NSEnumerator* keyEumerator = [buttons keyEnumerator];
NSString* name;
int index = 0;
while (name = [keyEumerator nextObject]) {
NSString* value = (NSString*) [buttons objectForKey: name];
if ([value length] == 0) {
value = @"Default button";
}
[buttonPopupButton insertItemWithTitle: value atIndex: index];
NSMenuItem* item = [buttonPopupButton itemAtIndex: index];
[item setRepresentedObject: name];
index++;
}
[buttonPopupButton selectItemAtIndex: 0];
[buttonPopupButton synchronizeTitleAndSelectedItem];
}
}
[NSApp beginSheet: window
modalForWindow: [NSApp keyWindow]
modalDelegate: self
didEndSelector: @selector(didEndSheet:returnCode:contextInfo:)
contextInfo: nil];
return self;
}
- (void)dealloc
{
[url release];
[inputElement release];
[formElement release];
[super dealloc];
}
- (void) saveKeyword: (id) sender {
NSMutableArray* fields = [NSMutableArray new];
[self collectFields: fields fromElement: formElement];
NSURL* absoluteUrl = [NSURL URLWithString: url];
NSString* action = [formElement action];
NSURL* actionUrl = [NSURL URLWithString: action relativeToURL: absoluteUrl];
NSMutableString* expansion = [NSMutableString new];
[expansion appendString: [actionUrl absoluteString]];
[expansion appendString: @"?"];
for (int i = 0; i < [fields count]; i++) {
FormField* field = (FormField*) [fields objectAtIndex: i];
[expansion appendString: [[field name] stringByAddingPercentEscapesUsingEncoding: NSUTF8StringEncoding]];
[expansion appendString: @"="];
[expansion appendString: [[field value] stringByAddingPercentEscapesUsingEncoding: NSUTF8StringEncoding]];
[expansion appendString: @"&"];
}
NSMenuItem* buttonItem = [buttonPopupButton selectedItem];
if (buttonItem) {
[expansion appendString: [[buttonItem representedObject] stringByAddingPercentEscapesUsingEncoding: NSUTF8StringEncoding]];
[expansion appendString: @"="];
[expansion appendString: [[buttonItem title] stringByAddingPercentEscapesUsingEncoding: NSUTF8StringEncoding]];
[expansion appendString: @"&"];
}
[expansion appendString: [inputElement name]];
[expansion appendString: @"={query}"];
NSString* keyword = [nameTextField stringValue];
KeywordMapping* mapping = [[KeywordMapping alloc] initWithKeyword: keyword
expansion: expansion];
KeywordMapper* mapper = [[KeywurlPlugin sharedInstance] keywordMapper];
[mapper addKeyword: mapping];
[mapper saveMappings];
[NSApp endSheet: window];
NSAlert *alert = [[NSAlert alloc] init];
[alert addButtonWithTitle: @"OK"];
[alert setMessageText: @"Keyword added"];
[alert setInformativeText: [NSString stringWithFormat:
@"You can now type '%@ something' in Safari's address field to do a search for 'something'", keyword]];
[alert setAlertStyle: NSInformationalAlertStyle];
[alert beginSheetModalForWindow: [NSApp keyWindow]
modalDelegate: nil
didEndSelector: nil
contextInfo: nil];
}
- (void) cancel: (id) sender {
[NSApp endSheet: window];
}
- (void) didEndSheet: (NSWindow*) sheet returnCode: (int) returnCode
contextInfo: (void*) contextInfo
{
[window orderOut: self];
}
- (void) collectFields: (NSMutableArray*) fields
fromElement: (DOMElement*) element {
DOMNodeList* children = [element childNodes];
for (int i = 0; i < [children length]; i++) {
DOMNode* node = [children item: i];
if ([node isKindOfClass: [DOMElement class]]) {
[self collectFields: fields fromElement: (DOMElement*) node];
}
}
if (element != inputElement && [element respondsToSelector: @selector(disabled)] &&
![element disabled]) {
NSString* tag = [[element tagName] lowercaseString];
if ([tag isEqualToString: @"textarea"] || (
[tag isEqualToString: @"input"] &&
([[element getAttribute: @"type"] isEqualToString: @"hidden"] ||
([[element getAttribute: @"type"] isEqualToString: @"radio"] && [element checked]) ||
([[element getAttribute: @"type"] isEqualToString: @"checkbox"] && [element checked])))) {
[fields addObject: [[FormField alloc] initWithName: [element name] value: [element value]]];
} else if ([tag isEqualToString: @"select"] &&
![element multiple] && [element selectedIndex] != -1) {
[fields addObject: [[FormField alloc] initWithName: [element name] value: [element value]]];
} else if ([tag isEqualToString: @"select"] && [element multiple]) {
DOMHTMLOptionsCollection* options = [element options];
for (int i = 0; i < [options length]; i++) {
DOMHTMLOptionElement* option = [options item: i];
if ([option selected]) {
[fields addObject: [[FormField alloc] initWithName: [element name]
value: [option value]]];
}
}
}
}
}
- (void) collectButtons: (NSMutableDictionary*) buttons
fromElement: (DOMElement*) element {
DOMNodeList* children = [element childNodes];
for (int i = 0; i < [children length]; i++) {
DOMNode* node = [children item: i];
if ([node isKindOfClass: [DOMElement class]]) {
[self collectButtons: buttons fromElement: (DOMElement*) node];
}
}
NSString* tag = [[element tagName] lowercaseString];
if ([tag isEqualToString: @"input"] &&
([[element getAttribute: @"type"] isEqualToString: @"submit"] ||
[[element getAttribute: @"type"] isEqualToString: @"image"])) {
[buttons setObject: [element value] forKey: [element name]];
}
}
@end
