Home | History | Annotate | Download | only in Misc
      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