WebNSDictionaryExtras.m 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  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. #import <WebKit/WebNSDataExtras.h>
  29. #import <wtf/Assertions.h>
  30. @implementation NSDictionary (WebNSDictionaryExtras)
  31. -(NSNumber *)_webkit_numberForKey:(id)key
  32. {
  33. id object = [self objectForKey:key];
  34. return [object isKindOfClass:[NSNumber class]] ? object : nil;
  35. }
  36. -(int)_webkit_intForKey:(NSString *)key
  37. {
  38. NSNumber *number = [self _webkit_numberForKey:key];
  39. return number == nil ? 0 : [number intValue];
  40. }
  41. -(NSString *)_webkit_stringForKey:(id)key
  42. {
  43. id object = [self objectForKey:key];
  44. return [object isKindOfClass:[NSString class]] ? object : nil;
  45. }
  46. -(NSArray *)_webkit_arrayForKey:(id)key
  47. {
  48. id object = [self objectForKey:key];
  49. return [object isKindOfClass:[NSArray class]] ? object : nil;
  50. }
  51. -(id)_webkit_objectForMIMEType:(NSString *)MIMEType
  52. {
  53. id result;
  54. NSRange slashRange;
  55. result = [self objectForKey:MIMEType];
  56. if (result) {
  57. return result;
  58. }
  59. slashRange = [MIMEType rangeOfString:@"/"];
  60. if (slashRange.location == NSNotFound) {
  61. return nil;
  62. }
  63. return [self objectForKey:[MIMEType substringToIndex:slashRange.location + 1]];
  64. }
  65. - (BOOL)_webkit_boolForKey:(id)key
  66. {
  67. NSNumber *number = [self _webkit_numberForKey:key];
  68. return number && [number boolValue];
  69. }
  70. @end
  71. @implementation NSMutableDictionary (WebNSDictionaryExtras)
  72. -(void)_webkit_setInt:(int)value forKey:(id)key
  73. {
  74. NSNumber *object = [[NSNumber alloc] initWithInt:value];
  75. [self setObject:object forKey:key];
  76. [object release];
  77. }
  78. -(void)_webkit_setFloat:(float)value forKey:(id)key
  79. {
  80. NSNumber *object = [[NSNumber alloc] initWithFloat:value];
  81. [self setObject:object forKey:key];
  82. [object release];
  83. }
  84. -(void)_webkit_setBool:(BOOL)value forKey:(id)key
  85. {
  86. NSNumber *object = [[NSNumber alloc] initWithBool:value];
  87. [self setObject:object forKey:key];
  88. [object release];
  89. }
  90. - (void)_webkit_setLongLong:(long long)value forKey:(id)key
  91. {
  92. NSNumber *object = [[NSNumber alloc] initWithLongLong:value];
  93. [self setObject:object forKey:key];
  94. [object release];
  95. }
  96. - (void)_webkit_setUnsignedLongLong:(unsigned long long)value forKey:(id)key
  97. {
  98. NSNumber *object = [[NSNumber alloc] initWithUnsignedLongLong:value];
  99. [self setObject:object forKey:key];
  100. [object release];
  101. }
  102. @end