InjectedBundleRangeHandle.cpp [plain text]
#include "config.h"
#include "InjectedBundleRangeHandle.h"
#include "InjectedBundleNodeHandle.h"
#include "ShareableBitmap.h"
#include "WebImage.h"
#include <JavaScriptCore/APICast.h>
#include <JavaScriptCore/HeapInlines.h>
#include <WebCore/Document.h>
#include <WebCore/FloatRect.h>
#include <WebCore/Frame.h>
#include <WebCore/FrameSelection.h>
#include <WebCore/FrameView.h>
#include <WebCore/GeometryUtilities.h>
#include <WebCore/GraphicsContext.h>
#include <WebCore/IntRect.h>
#include <WebCore/JSRange.h>
#include <WebCore/Page.h>
#include <WebCore/Range.h>
#include <WebCore/RenderView.h>
#include <WebCore/SimpleRange.h>
#include <WebCore/VisibleSelection.h>
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
#if PLATFORM(MAC)
#include <WebCore/LocalDefaultSystemAppearance.h>
#endif
namespace WebKit {
using namespace WebCore;
using DOMRangeHandleCache = HashMap<Range*, InjectedBundleRangeHandle*>;
static DOMRangeHandleCache& domRangeHandleCache()
{
static NeverDestroyed<DOMRangeHandleCache> cache;
return cache;
}
RefPtr<InjectedBundleRangeHandle> InjectedBundleRangeHandle::getOrCreate(JSContextRef context, JSObjectRef object)
{
return getOrCreate(JSRange::toWrapped(toJS(context)->vm(), toJS(object)));
}
RefPtr<InjectedBundleRangeHandle> InjectedBundleRangeHandle::getOrCreate(Range* range)
{
if (!range)
return nullptr;
auto result = domRangeHandleCache().add(range, nullptr);
if (!result.isNewEntry)
return result.iterator->value;
auto rangeHandle = adoptRef(*new InjectedBundleRangeHandle(*range));
result.iterator->value = rangeHandle.ptr();
return rangeHandle;
}
InjectedBundleRangeHandle::InjectedBundleRangeHandle(Range& range)
: m_range(range)
{
}
InjectedBundleRangeHandle::~InjectedBundleRangeHandle()
{
domRangeHandleCache().remove(m_range.ptr());
}
Range& InjectedBundleRangeHandle::coreRange() const
{
return m_range.get();
}
Ref<InjectedBundleNodeHandle> InjectedBundleRangeHandle::document()
{
return InjectedBundleNodeHandle::getOrCreate(m_range->ownerDocument());
}
WebCore::IntRect InjectedBundleRangeHandle::boundingRectInWindowCoordinates() const
{
Frame* frame = m_range->ownerDocument().frame();
return frame->view()->contentsToWindow(enclosingIntRect(unionRectIgnoringZeroRects(RenderObject::absoluteBorderAndTextRects(makeSimpleRange(m_range)))));
}
RefPtr<WebImage> InjectedBundleRangeHandle::renderedImage(SnapshotOptions options)
{
Document& ownerDocument = m_range->ownerDocument();
Frame* frame = ownerDocument.frame();
if (!frame)
return nullptr;
FrameView* frameView = frame->view();
if (!frameView)
return nullptr;
#if PLATFORM(MAC)
LocalDefaultSystemAppearance localAppearance(frameView->useDarkAppearance());
#endif
Ref<Frame> protector(*frame);
VisibleSelection oldSelection = frame->selection().selection();
frame->selection().setSelection(makeSimpleRange(m_range));
float scaleFactor = (options & SnapshotOptionsExcludeDeviceScaleFactor) ? 1 : frame->page()->deviceScaleFactor();
IntRect paintRect = enclosingIntRect(unionRectIgnoringZeroRects(RenderObject::absoluteBorderAndTextRects(makeSimpleRange(m_range))));
IntSize backingStoreSize = paintRect.size();
backingStoreSize.scale(scaleFactor);
auto backingStore = ShareableBitmap::createShareable(backingStoreSize, { });
if (!backingStore)
return nullptr;
auto graphicsContext = backingStore->createGraphicsContext();
if (!graphicsContext)
return nullptr;
graphicsContext->scale(scaleFactor);
paintRect.move(frameView->frameRect().x(), frameView->frameRect().y());
paintRect.moveBy(-frameView->scrollPosition());
graphicsContext->translate(-paintRect.location());
OptionSet<PaintBehavior> oldPaintBehavior = frameView->paintBehavior();
OptionSet<PaintBehavior> paintBehavior = oldPaintBehavior;
paintBehavior.add({ PaintBehavior::SelectionOnly, PaintBehavior::FlattenCompositingLayers, PaintBehavior::Snapshotting });
if (options & SnapshotOptionsForceBlackText)
paintBehavior.add(PaintBehavior::ForceBlackText);
if (options & SnapshotOptionsForceWhiteText)
paintBehavior.add(PaintBehavior::ForceWhiteText);
frameView->setPaintBehavior(paintBehavior);
ownerDocument.updateLayout();
frameView->paint(*graphicsContext, paintRect);
frameView->setPaintBehavior(oldPaintBehavior);
frame->selection().setSelection(oldSelection);
return WebImage::create(backingStore.releaseNonNull());
}
String InjectedBundleRangeHandle::text() const
{
return m_range->text();
}
RefPtr<InjectedBundleRangeHandle> createHandle(const Optional<WebCore::SimpleRange>& range)
{
return InjectedBundleRangeHandle::getOrCreate(createLiveRange(range).get());
}
}