JSHTMLCollectionCustom.cpp [plain text]
#include "config.h"
#include "JSHTMLCollection.h"
#include "AtomicString.h"
#include "HTMLCollection.h"
#include "HTMLOptionsCollection.h"
#include "JSHTMLAllCollection.h"
#include "JSHTMLOptionsCollection.h"
#include "JSNamedNodesCollection.h"
#include "JSNode.h"
#include "Node.h"
#include "JSDOMBinding.h"
#include <wtf/Vector.h>
using namespace JSC;
namespace WebCore {
static JSValue getNamedItems(ExecState* exec, HTMLCollection* impl, const Identifier& propertyName)
{
Vector<RefPtr<Node> > namedItems;
impl->namedItems(propertyName, namedItems);
if (namedItems.isEmpty())
return jsUndefined();
if (namedItems.size() == 1)
return toJS(exec, namedItems[0].get());
return new (exec) JSNamedNodesCollection(exec, namedItems);
}
static JSValue JSC_HOST_CALL callHTMLCollection(ExecState* exec, JSObject* function, JSValue, const ArgList& args)
{
if (args.size() < 1)
return jsUndefined();
HTMLCollection* collection = static_cast<JSHTMLCollection*>(function)->impl();
if (args.size() == 1) {
bool ok;
UString string = args.at(0).toString(exec);
unsigned index = string.toUInt32(&ok, false);
if (ok)
return toJS(exec, collection->item(index));
return getNamedItems(exec, collection, Identifier(exec, string));
}
bool ok;
UString string = args.at(0).toString(exec);
unsigned index = args.at(1).toString(exec).toUInt32(&ok, false);
if (ok) {
String pstr = string;
Node* node = collection->namedItem(pstr);
while (node) {
if (!index)
return toJS(exec, node);
node = collection->nextNamedItem(pstr);
--index;
}
}
return jsUndefined();
}
CallType JSHTMLCollection::getCallData(CallData& callData)
{
callData.native.function = callHTMLCollection;
return CallTypeHost;
}
bool JSHTMLCollection::canGetItemsForName(ExecState* exec, HTMLCollection* thisObj, const Identifier& propertyName)
{
return !getNamedItems(exec, thisObj, propertyName).isUndefined();
}
JSValue JSHTMLCollection::nameGetter(ExecState* exec, const Identifier& propertyName, const PropertySlot& slot)
{
JSHTMLCollection* thisObj = static_cast<JSHTMLCollection*>(asObject(slot.slotBase()));
return getNamedItems(exec, thisObj->impl(), propertyName);
}
JSValue JSHTMLCollection::item(ExecState* exec, const ArgList& args)
{
bool ok;
uint32_t index = args.at(0).toString(exec).toUInt32(&ok, false);
if (ok)
return toJS(exec, impl()->item(index));
return getNamedItems(exec, impl(), Identifier(exec, args.at(0).toString(exec)));
}
JSValue JSHTMLCollection::namedItem(ExecState* exec, const ArgList& args)
{
return getNamedItems(exec, impl(), Identifier(exec, args.at(0).toString(exec)));
}
JSValue toJS(ExecState* exec, HTMLCollection* collection)
{
if (!collection)
return jsNull();
DOMObject* wrapper = getCachedDOMObjectWrapper(exec->globalData(), collection);
if (wrapper)
return wrapper;
switch (collection->type()) {
case SelectOptions:
wrapper = CREATE_DOM_OBJECT_WRAPPER(exec, HTMLOptionsCollection, collection);
break;
case DocAll:
typedef HTMLCollection HTMLAllCollection;
wrapper = CREATE_DOM_OBJECT_WRAPPER(exec, HTMLAllCollection, collection);
break;
default:
wrapper = CREATE_DOM_OBJECT_WRAPPER(exec, HTMLCollection, collection);
break;
}
return wrapper;
}
}