1 /*

     2  * Copyright (C) 2005 Apple Computer, 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 <JavaScriptCore/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 -(id)_webkit_objectForMIMEType:(NSString *)MIMEType

    53 {

    54     id result;

    55     NSRange slashRange;

    56     

    57     result = [self objectForKey:MIMEType];

    58     if (result) {

    59         return result;

    60     }

    61     

    62     slashRange = [MIMEType rangeOfString:@"/"];

    63     if (slashRange.location == NSNotFound) {

    64         return nil;

    65     }

    66     

    67     return [self objectForKey:[MIMEType substringToIndex:slashRange.location + 1]];

    68 }

    69 

    70 @end

    71 

    72 @implementation NSMutableDictionary (WebNSDictionaryExtras)

    73 -(void)_webkit_setObject:(id)object forUncopiedKey:(id)key

    74 {

    75     CFDictionarySetValue((CFMutableDictionaryRef)self, key, object);

    76 }

    77 

    78 -(void)_webkit_setInt:(int)value forKey:(id)key

    79 {

    80     NSNumber *object = [[NSNumber alloc] initWithInt:value];

    81     [self setObject:object forKey:key];

    82     [object release];

    83 }

    84 

    85 -(void)_webkit_setFloat:(float)value forKey:(id)key

    86 {

    87     NSNumber *object = [[NSNumber alloc] initWithFloat:value];

    88     [self setObject:object forKey:key];

    89     [object release];

    90 }

    91 

    92 -(void)_webkit_setBool:(BOOL)value forKey:(id)key

    93 {

    94     NSNumber *object = [[NSNumber alloc] initWithBool:value];

    95     [self setObject:object forKey:key];

    96     [object release];

    97 }

    98 

    99 @end

   100