InjectedBundleNodeHandle.cpp [plain text]
#include "config.h"
#include "InjectedBundleNodeHandle.h"
#include "ShareableBitmap.h"
#include "WebFrame.h"
#include "WebFrameLoaderClient.h"
#include "WebImage.h"
#include <JavaScriptCore/APICast.h>
#include <WebCore/Document.h>
#include <WebCore/Frame.h>
#include <WebCore/FrameLoader.h>
#include <WebCore/FrameView.h>
#include <WebCore/GraphicsContext.h>
#include <WebCore/HTMLFrameElement.h>
#include <WebCore/HTMLIFrameElement.h>
#include <WebCore/HTMLInputElement.h>
#include <WebCore/HTMLNames.h>
#include <WebCore/HTMLTableCellElement.h>
#include <WebCore/HTMLTextAreaElement.h>
#include <WebCore/IntRect.h>
#include <WebCore/JSNode.h>
#include <WebCore/Node.h>
#include <WebCore/Page.h>
#include <WebCore/RenderObject.h>
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/text/WTFString.h>
using namespace WebCore;
using namespace HTMLNames;
namespace WebKit {
typedef HashMap<Node*, InjectedBundleNodeHandle*> DOMHandleCache;
static DOMHandleCache& domHandleCache()
{
static NeverDestroyed<DOMHandleCache> cache;
return cache;
}
PassRefPtr<InjectedBundleNodeHandle> InjectedBundleNodeHandle::getOrCreate(JSContextRef, JSObjectRef object)
{
Node* node = toNode(toJS(object));
return getOrCreate(node);
}
PassRefPtr<InjectedBundleNodeHandle> InjectedBundleNodeHandle::getOrCreate(Node* node)
{
if (!node)
return 0;
DOMHandleCache::AddResult result = domHandleCache().add(node, nullptr);
if (!result.isNewEntry)
return PassRefPtr<InjectedBundleNodeHandle>(result.iterator->value);
RefPtr<InjectedBundleNodeHandle> nodeHandle = InjectedBundleNodeHandle::create(node);
result.iterator->value = nodeHandle.get();
return nodeHandle.release();
}
PassRefPtr<InjectedBundleNodeHandle> InjectedBundleNodeHandle::create(Node* node)
{
return adoptRef(new InjectedBundleNodeHandle(node));
}
InjectedBundleNodeHandle::InjectedBundleNodeHandle(Node* node)
: m_node(node)
{
}
InjectedBundleNodeHandle::~InjectedBundleNodeHandle()
{
domHandleCache().remove(m_node.get());
}
Node* InjectedBundleNodeHandle::coreNode() const
{
return m_node.get();
}
PassRefPtr<InjectedBundleNodeHandle> InjectedBundleNodeHandle::document()
{
return getOrCreate(&m_node->document());
}
IntRect InjectedBundleNodeHandle::elementBounds() const
{
if (!m_node->isElementNode())
return IntRect();
return toElement(m_node.get())->boundsInRootViewSpace();
}
IntRect InjectedBundleNodeHandle::renderRect(bool* isReplaced) const
{
return m_node.get()->pixelSnappedRenderRect(isReplaced);
}
static PassRefPtr<WebImage> imageForRect(FrameView* frameView, const IntRect& rect, SnapshotOptions options)
{
IntSize bitmapSize = rect.size();
float scaleFactor = frameView->frame().page()->deviceScaleFactor();
bitmapSize.scale(scaleFactor);
RefPtr<WebImage> snapshot = WebImage::create(bitmapSize, snapshotOptionsToImageOptions(options));
if (!snapshot->bitmap())
return 0;
auto graphicsContext = snapshot->bitmap()->createGraphicsContext();
graphicsContext->clearRect(IntRect(IntPoint(), bitmapSize));
graphicsContext->applyDeviceScaleFactor(scaleFactor);
graphicsContext->translate(-rect.x(), -rect.y());
FrameView::SelectionInSnapshot shouldPaintSelection = FrameView::IncludeSelection;
if (options & SnapshotOptionsExcludeSelectionHighlighting)
shouldPaintSelection = FrameView::ExcludeSelection;
frameView->paintContentsForSnapshot(graphicsContext.get(), rect, shouldPaintSelection, FrameView::DocumentCoordinates);
return snapshot.release();
}
PassRefPtr<WebImage> InjectedBundleNodeHandle::renderedImage(SnapshotOptions options)
{
Frame* frame = m_node->document().frame();
if (!frame)
return 0;
FrameView* frameView = frame->view();
if (!frameView)
return 0;
m_node->document().updateLayout();
RenderObject* renderer = m_node->renderer();
if (!renderer)
return 0;
LayoutRect topLevelRect;
IntRect paintingRect = pixelSnappedIntRect(renderer->paintingRootRect(topLevelRect));
frameView->setNodeToDraw(m_node.get());
RefPtr<WebImage> image = imageForRect(frameView, paintingRect, options);
frameView->setNodeToDraw(0);
return image.release();
}
void InjectedBundleNodeHandle::setHTMLInputElementValueForUser(const String& value)
{
if (!isHTMLInputElement(m_node.get()))
return;
toHTMLInputElement(m_node.get())->setValueForUser(value);
}
bool InjectedBundleNodeHandle::isHTMLInputElementAutofilled() const
{
if (!isHTMLInputElement(m_node.get()))
return false;
return toHTMLInputElement(m_node.get())->isAutofilled();
}
void InjectedBundleNodeHandle::setHTMLInputElementAutofilled(bool filled)
{
if (!isHTMLInputElement(m_node.get()))
return;
toHTMLInputElement(m_node.get())->setAutofilled(filled);
}
bool InjectedBundleNodeHandle::htmlInputElementLastChangeWasUserEdit()
{
if (!isHTMLInputElement(m_node.get()))
return false;
return toHTMLInputElement(m_node.get())->lastChangeWasUserEdit();
}
bool InjectedBundleNodeHandle::htmlTextAreaElementLastChangeWasUserEdit()
{
if (!isHTMLTextAreaElement(m_node.get()))
return false;
return toHTMLTextAreaElement(m_node.get())->lastChangeWasUserEdit();
}
bool InjectedBundleNodeHandle::isTextField() const
{
return isHTMLInputElement(m_node.get()) && toHTMLInputElement(m_node.get())->isText();
}
PassRefPtr<InjectedBundleNodeHandle> InjectedBundleNodeHandle::htmlTableCellElementCellAbove()
{
if (!m_node->hasTagName(tdTag))
return 0;
return getOrCreate(static_cast<HTMLTableCellElement*>(m_node.get())->cellAbove());
}
PassRefPtr<WebFrame> InjectedBundleNodeHandle::documentFrame()
{
if (!m_node->isDocumentNode())
return nullptr;
Frame* frame = static_cast<Document*>(m_node.get())->frame();
if (!frame)
return nullptr;
return WebFrame::fromCoreFrame(*frame);
}
PassRefPtr<WebFrame> InjectedBundleNodeHandle::htmlFrameElementContentFrame()
{
if (!m_node->hasTagName(frameTag))
return nullptr;
Frame* frame = static_cast<HTMLFrameElement*>(m_node.get())->contentFrame();
if (!frame)
return nullptr;
return WebFrame::fromCoreFrame(*frame);
}
PassRefPtr<WebFrame> InjectedBundleNodeHandle::htmlIFrameElementContentFrame()
{
if (!m_node->hasTagName(iframeTag))
return nullptr;
Frame* frame = toHTMLIFrameElement(m_node.get())->contentFrame();
if (!frame)
return nullptr;
return WebFrame::fromCoreFrame(*frame);
}
}