1/* 2 * Copyright (C) 2005, 2008 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of 14 * its contributors may be used to endorse or promote products derived 15 * from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29#import <WebKit/WebNSDataExtras.h> 30 31#import <wtf/Assertions.h> 32 33@implementation NSDictionary (WebNSDictionaryExtras) 34-(NSNumber *)_webkit_numberForKey:(id)key 35{ 36 id object = [self objectForKey:key]; 37 return [object isKindOfClass:[NSNumber class]] ? object : nil; 38} 39 40-(int)_webkit_intForKey:(NSString *)key 41{ 42 NSNumber *number = [self _webkit_numberForKey:key]; 43 return number == nil ? 0 : [number intValue]; 44} 45 46-(NSString *)_webkit_stringForKey:(id)key 47{ 48 id object = [self objectForKey:key]; 49 return [object isKindOfClass:[NSString class]] ? object : nil; 50} 51 52-(NSArray *)_webkit_arrayForKey:(id)key 53{ 54 id object = [self objectForKey:key]; 55 return [object isKindOfClass:[NSArray class]] ? object : nil; 56} 57 58-(id)_webkit_objectForMIMEType:(NSString *)MIMEType 59{ 60 id result; 61 NSRange slashRange; 62 63 result = [self objectForKey:MIMEType]; 64 if (result) { 65 return result; 66 } 67 68 slashRange = [MIMEType rangeOfString:@"/"]; 69 if (slashRange.location == NSNotFound) { 70 return nil; 71 } 72 73 return [self objectForKey:[MIMEType substringToIndex:slashRange.location + 1]]; 74} 75 76- (BOOL)_webkit_boolForKey:(id)key 77{ 78 NSNumber *number = [self _webkit_numberForKey:key]; 79 return number && [number boolValue]; 80} 81 82@end 83 84@implementation NSMutableDictionary (WebNSDictionaryExtras) 85-(void)_webkit_setObject:(id)object forUncopiedKey:(id)key 86{ 87 CFDictionarySetValue((CFMutableDictionaryRef)self, key, object); 88} 89 90-(void)_webkit_setInt:(int)value forKey:(id)key 91{ 92 NSNumber *object = [[NSNumber alloc] initWithInt:value]; 93 [self setObject:object forKey:key]; 94 [object release]; 95} 96 97-(void)_webkit_setFloat:(float)value forKey:(id)key 98{ 99 NSNumber *object = [[NSNumber alloc] initWithFloat:value]; 100 [self setObject:object forKey:key]; 101 [object release]; 102} 103 104-(void)_webkit_setBool:(BOOL)value forKey:(id)key 105{ 106 NSNumber *object = [[NSNumber alloc] initWithBool:value]; 107 [self setObject:object forKey:key]; 108 [object release]; 109} 110 111- (void)_webkit_setUnsignedLongLong:(unsigned long long)value forKey:(id)key 112{ 113 NSNumber *object = [[NSNumber alloc] initWithUnsignedLongLong:value]; 114 [self setObject:object forKey:key]; 115 [object release]; 116} 117 118@end 119 120