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