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_setLongLong:(long long)value forKey:(id)key 112 { 113 NSNumber *object = [[NSNumber alloc] initWithLongLong:value]; 114 [self setObject:object forKey:key]; 115 [object release]; 116 } 117 118 - (void)_webkit_setUnsignedLongLong:(unsigned long long)value forKey:(id)key 119 { 120 NSNumber *object = [[NSNumber alloc] initWithUnsignedLongLong:value]; 121 [self setObject:object forKey:key]; 122 [object release]; 123 } 124 125 @end 126 127