1 /* 2 * Copyright (C) 2007 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 "WebNSArrayExtras.h" 30 31 #import <wtf/Assertions.h> 32 33 @implementation NSArray (WebNSArrayExtras) 34 35 -(NSNumber *)_webkit_numberAtIndex:(NSUInteger)index 36 { 37 id object = [self objectAtIndex:index]; 38 return [object isKindOfClass:[NSNumber class]] ? object : nil; 39 } 40 41 -(NSString *)_webkit_stringAtIndex:(NSUInteger)index 42 { 43 id object = [self objectAtIndex:index]; 44 return [object isKindOfClass:[NSString class]] ? object : nil; 45 } 46 47 @end 48 49 @implementation NSMutableArray (WebNSArrayExtras) 50 51 - (void)_webkit_removeUselessMenuItemSeparators 52 { 53 // Starting with a mutable array of NSMenuItems, removes any separators at the start, 54 // removes any separators at the end, and collapses any other adjacent separators to 55 // a single separator. 56 57 int index; 58 // Start this with YES so very last item will be removed if it's a separator. 59 BOOL removePreviousItemIfSeparator = YES; 60 for (index = [self count] - 1; index >= 0; --index) { 61 NSMenuItem *item = [self objectAtIndex:index]; 62 ASSERT([item isKindOfClass:[NSMenuItem class]]); 63 BOOL itemIsSeparator = [item isSeparatorItem]; 64 if (itemIsSeparator && (removePreviousItemIfSeparator || index == 0)) 65 [self removeObjectAtIndex:index]; 66 removePreviousItemIfSeparator = itemIsSeparator; 67 } 68 69 // This could leave us with one initial separator; kill it off too 70 if ([self count] && [[self objectAtIndex:0] isSeparatorItem]) 71 [self removeObjectAtIndex:0]; 72 } 73 74 @end 75