Make sure that .find() with multiple direct child selectors is handled correctly...
[jquery.git] / src / traversing.js
1 (function( jQuery ) {
2
3 var runtil = /Until$/,
4         rparentsprev = /^(?:parents|prevUntil|prevAll)/,
5         // Note: This RegExp should be improved, or likely pulled from Sizzle
6         rmultiselector = /,/,
7         rchild = /(^|,)\s*>/g,
8         isSimple = /^.[^:#\[\.,]*$/,
9         slice = Array.prototype.slice,
10         POS = jQuery.expr.match.POS;
11
12 jQuery.fn.extend({
13         find: function( selector ) {
14                 // Handle "> div" child selectors and pass them to .children()
15                 if ( typeof selector === "string" && rchild.test( selector ) ) {
16                         return this.children( selector.replace( rchild, "$1" ) );
17                 }
18
19                 var ret = this.pushStack( "", "find", selector ), length = 0;
20
21                 for ( var i = 0, l = this.length; i < l; i++ ) {
22                         length = ret.length;
23                         jQuery.find( selector, this[i], ret );
24
25                         if ( i > 0 ) {
26                                 // Make sure that the results are unique
27                                 for ( var n = length; n < ret.length; n++ ) {
28                                         for ( var r = 0; r < length; r++ ) {
29                                                 if ( ret[r] === ret[n] ) {
30                                                         ret.splice(n--, 1);
31                                                         break;
32                                                 }
33                                         }
34                                 }
35                         }
36                 }
37
38                 return ret;
39         },
40
41         has: function( target ) {
42                 var targets = jQuery( target );
43                 return this.filter(function() {
44                         for ( var i = 0, l = targets.length; i < l; i++ ) {
45                                 if ( jQuery.contains( this, targets[i] ) ) {
46                                         return true;
47                                 }
48                         }
49                 });
50         },
51
52         not: function( selector ) {
53                 return this.pushStack( winnow(this, selector, false), "not", selector);
54         },
55
56         filter: function( selector ) {
57                 return this.pushStack( winnow(this, selector, true), "filter", selector );
58         },
59         
60         is: function( selector ) {
61                 return !!selector && jQuery.filter( selector, this ).length > 0;
62         },
63
64         closest: function( selectors, context ) {
65                 var ret = [], i, l, cur = this[0];
66
67                 if ( jQuery.isArray( selectors ) ) {
68                         var match, matches = {}, selector, level = 1;
69
70                         if ( cur && selectors.length ) {
71                                 for ( i = 0, l = selectors.length; i < l; i++ ) {
72                                         selector = selectors[i];
73
74                                         if ( !matches[selector] ) {
75                                                 matches[selector] = jQuery.expr.match.POS.test( selector ) ? 
76                                                         jQuery( selector, context || this.context ) :
77                                                         selector;
78                                         }
79                                 }
80
81                                 while ( cur && cur.ownerDocument && cur !== context ) {
82                                         for ( selector in matches ) {
83                                                 match = matches[selector];
84
85                                                 if ( match.jquery ? match.index(cur) > -1 : jQuery(cur).is(match) ) {
86                                                         ret.push({ selector: selector, elem: cur, level: level });
87                                                 }
88                                         }
89
90                                         cur = cur.parentNode;
91                                         level++;
92                                 }
93                         }
94
95                         return ret;
96                 }
97
98                 var pos = POS.test( selectors ) ? 
99                         jQuery( selectors, context || this.context ) : null;
100
101                 for ( i = 0, l = this.length; i < l; i++ ) {
102                         cur = this[i];
103
104                         while ( cur ) {
105                                 if ( pos ? pos.index(cur) > -1 : jQuery.find.matchesSelector(cur, selectors) ) {
106                                         ret.push( cur );
107                                         break;
108
109                                 } else {
110                                         cur = cur.parentNode;
111                                         if ( !cur || !cur.ownerDocument || cur === context ) {
112                                                 break;
113                                         }
114                                 }
115                         }
116                 }
117
118                 ret = ret.length > 1 ? jQuery.unique(ret) : ret;
119                 
120                 return this.pushStack( ret, "closest", selectors );
121         },
122         
123         // Determine the position of an element within
124         // the matched set of elements
125         index: function( elem ) {
126                 if ( !elem || typeof elem === "string" ) {
127                         return jQuery.inArray( this[0],
128                                 // If it receives a string, the selector is used
129                                 // If it receives nothing, the siblings are used
130                                 elem ? jQuery( elem ) : this.parent().children() );
131                 }
132                 // Locate the position of the desired element
133                 return jQuery.inArray(
134                         // If it receives a jQuery object, the first element is used
135                         elem.jquery ? elem[0] : elem, this );
136         },
137
138         add: function( selector, context ) {
139                 var set = typeof selector === "string" ?
140                                 jQuery( selector, context || this.context ) :
141                                 jQuery.makeArray( selector ),
142                         all = jQuery.merge( this.get(), set );
143
144                 return this.pushStack( isDisconnected( set[0] ) || isDisconnected( all[0] ) ?
145                         all :
146                         jQuery.unique( all ) );
147         },
148
149         andSelf: function() {
150                 return this.add( this.prevObject );
151         }
152 });
153
154 // A painfully simple check to see if an element is disconnected
155 // from a document (should be improved, where feasible).
156 function isDisconnected( node ) {
157         return !node || !node.parentNode || node.parentNode.nodeType === 11;
158 }
159
160 jQuery.each({
161         parent: function( elem ) {
162                 var parent = elem.parentNode;
163                 return parent && parent.nodeType !== 11 ? parent : null;
164         },
165         parents: function( elem ) {
166                 return jQuery.dir( elem, "parentNode" );
167         },
168         parentsUntil: function( elem, i, until ) {
169                 return jQuery.dir( elem, "parentNode", until );
170         },
171         next: function( elem ) {
172                 return jQuery.nth( elem, 2, "nextSibling" );
173         },
174         prev: function( elem ) {
175                 return jQuery.nth( elem, 2, "previousSibling" );
176         },
177         nextAll: function( elem ) {
178                 return jQuery.dir( elem, "nextSibling" );
179         },
180         prevAll: function( elem ) {
181                 return jQuery.dir( elem, "previousSibling" );
182         },
183         nextUntil: function( elem, i, until ) {
184                 return jQuery.dir( elem, "nextSibling", until );
185         },
186         prevUntil: function( elem, i, until ) {
187                 return jQuery.dir( elem, "previousSibling", until );
188         },
189         siblings: function( elem ) {
190                 return jQuery.sibling( elem.parentNode.firstChild, elem );
191         },
192         children: function( elem ) {
193                 return jQuery.sibling( elem.firstChild );
194         },
195         contents: function( elem ) {
196                 return jQuery.nodeName( elem, "iframe" ) ?
197                         elem.contentDocument || elem.contentWindow.document :
198                         jQuery.makeArray( elem.childNodes );
199         }
200 }, function( name, fn ) {
201         jQuery.fn[ name ] = function( until, selector ) {
202                 var ret = jQuery.map( this, fn, until );
203                 
204                 if ( !runtil.test( name ) ) {
205                         selector = until;
206                 }
207
208                 if ( selector && typeof selector === "string" ) {
209                         ret = jQuery.filter( selector, ret );
210                 }
211
212                 ret = this.length > 1 ? jQuery.unique( ret ) : ret;
213
214                 if ( (this.length > 1 || rmultiselector.test( selector )) && rparentsprev.test( name ) ) {
215                         ret = ret.reverse();
216                 }
217
218                 return this.pushStack( ret, name, slice.call(arguments).join(",") );
219         };
220 });
221
222 jQuery.extend({
223         filter: function( expr, elems, not ) {
224                 if ( not ) {
225                         expr = ":not(" + expr + ")";
226                 }
227
228                 return elems.length === 1 ?
229                         jQuery.find.matchesSelector(elems[0], expr) ? [ elems[0] ] : [] :
230                         jQuery.find.matches(expr, elems);
231         },
232         
233         dir: function( elem, dir, until ) {
234                 var matched = [], cur = elem[dir];
235                 while ( cur && cur.nodeType !== 9 && (until === undefined || cur.nodeType !== 1 || !jQuery( cur ).is( until )) ) {
236                         if ( cur.nodeType === 1 ) {
237                                 matched.push( cur );
238                         }
239                         cur = cur[dir];
240                 }
241                 return matched;
242         },
243
244         nth: function( cur, result, dir, elem ) {
245                 result = result || 1;
246                 var num = 0;
247
248                 for ( ; cur; cur = cur[dir] ) {
249                         if ( cur.nodeType === 1 && ++num === result ) {
250                                 break;
251                         }
252                 }
253
254                 return cur;
255         },
256
257         sibling: function( n, elem ) {
258                 var r = [];
259
260                 for ( ; n; n = n.nextSibling ) {
261                         if ( n.nodeType === 1 && n !== elem ) {
262                                 r.push( n );
263                         }
264                 }
265
266                 return r;
267         }
268 });
269
270 // Implement the identical functionality for filter and not
271 function winnow( elements, qualifier, keep ) {
272         if ( jQuery.isFunction( qualifier ) ) {
273                 return jQuery.grep(elements, function( elem, i ) {
274                         var retVal = !!qualifier.call( elem, i, elem );
275                         return retVal === keep;
276                 });
277
278         } else if ( qualifier.nodeType ) {
279                 return jQuery.grep(elements, function( elem, i ) {
280                         return (elem === qualifier) === keep;
281                 });
282
283         } else if ( typeof qualifier === "string" ) {
284                 var filtered = jQuery.grep(elements, function( elem ) {
285                         return elem.nodeType === 1;
286                 });
287
288                 if ( isSimple.test( qualifier ) ) {
289                         return jQuery.filter(qualifier, filtered, !keep);
290                 } else {
291                         qualifier = jQuery.filter( qualifier, filtered );
292                 }
293         }
294
295         return jQuery.grep(elements, function( elem, i ) {
296                 return (jQuery.inArray( elem, qualifier ) >= 0) === keep;
297         });
298 }
299
300 })( jQuery );