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 "kjs_binding.h"
#include "kjs_html.h"
#include <wtf/Vector.h>
using namespace KJS;
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 JSNamedNodesCollection(exec->lexicalGlobalObject()->objectPrototype(), namedItems);
}
JSValue* JSHTMLCollection::callAsFunction(ExecState* exec, JSObject*, const List& args)
{
if (args.size() < 1)
return jsUndefined();
HTMLCollection* collection = impl();
if (args.size() == 1) {
bool ok;
UString string = args[0]->toString(exec);
unsigned index = string.toUInt32(&ok, false);
if (ok)
return toJS(exec, collection->item(index));
return getNamedItems(exec, collection, Identifier(string));
}
bool ok;
UString string = args[0]->toString(exec);
unsigned index = args[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();
}
bool JSHTMLCollection::implementsCall() const
{
return true;
}
bool JSHTMLCollection::canGetItemsForName(ExecState* exec, HTMLCollection* thisObj, const Identifier& propertyName)
{
return !getNamedItems(exec, thisObj, propertyName)->isUndefined();
}
JSValue* JSHTMLCollection::nameGetter(ExecState* exec, JSObject* originalObject, const Identifier& propertyName, const PropertySlot& slot)
{
JSHTMLCollection* thisObj = static_cast<JSHTMLCollection*>(slot.slotBase());
return getNamedItems(exec, thisObj->impl(), propertyName);
}
JSValue* JSHTMLCollection::item(ExecState* exec, const List& args)
{
bool ok;
uint32_t index = args[0]->toString(exec).toUInt32(&ok, false);
if (ok)
return toJS(exec, impl()->item(index));
return getNamedItems(exec, impl(), Identifier(args[0]->toString(exec)));
}
JSValue* JSHTMLCollection::namedItem(ExecState* exec, const List& args)
{
return getNamedItems(exec, impl(), Identifier(args[0]->toString(exec)));
}
JSValue* toJS(ExecState* exec, HTMLCollection* collection)
{
if (!collection)
return jsNull();
DOMObject* ret = ScriptInterpreter::getDOMObject(collection);
if (ret)
return ret;
switch (collection->type()) {
case HTMLCollection::SelectOptions:
ret = new JSHTMLOptionsCollection(JSHTMLOptionsCollectionPrototype::self(exec), static_cast<HTMLOptionsCollection*>(collection));
break;
case HTMLCollection::DocAll:
ret = new JSHTMLAllCollection(JSHTMLCollectionPrototype::self(exec), static_cast<HTMLCollection*>(collection));
break;
default:
ret = new JSHTMLCollection(JSHTMLCollectionPrototype::self(exec), static_cast<HTMLCollection*>(collection));
break;
}
ScriptInterpreter::putDOMObject(collection, ret);
return ret;
}
}