#include "config.h"
#include "APIWebArchive.h"
#if PLATFORM(COCOA)
#include "APIArray.h"
#include "APIData.h"
#include "APIWebArchiveResource.h"
#include <WebCore/LegacyWebArchive.h>
#include <wtf/RetainPtr.h>
using namespace WebCore;
namespace API {
PassRefPtr<WebArchive> WebArchive::create(WebArchiveResource* mainResource, PassRefPtr<API::Array> subresources, PassRefPtr<API::Array> subframeArchives)
{
return adoptRef(new WebArchive(mainResource, subresources, subframeArchives));
}
PassRefPtr<WebArchive> WebArchive::create(API::Data* data)
{
return adoptRef(new WebArchive(data));
}
PassRefPtr<WebArchive> WebArchive::create(PassRefPtr<LegacyWebArchive> legacyWebArchive)
{
return adoptRef(new WebArchive(legacyWebArchive));
}
PassRefPtr<WebArchive> WebArchive::create(Range* range)
{
return adoptRef(new WebArchive(LegacyWebArchive::create(range)));
}
WebArchive::WebArchive(WebArchiveResource* mainResource, PassRefPtr<API::Array> subresources, PassRefPtr<API::Array> subframeArchives)
: m_cachedMainResource(mainResource)
, m_cachedSubresources(subresources)
, m_cachedSubframeArchives(subframeArchives)
{
RefPtr<ArchiveResource> coreMainResource = m_cachedMainResource->coreArchiveResource();
Vector<RefPtr<ArchiveResource>> coreArchiveResources;
coreArchiveResources.reserveInitialCapacity(m_cachedSubresources->size());
for (size_t i = 0; i < m_cachedSubresources->size(); ++i) {
RefPtr<WebArchiveResource> resource = m_cachedSubresources->at<WebArchiveResource>(i);
ASSERT(resource);
coreArchiveResources.uncheckedAppend(resource->coreArchiveResource());
}
Vector<RefPtr<LegacyWebArchive>> coreSubframeLegacyWebArchives;
coreSubframeLegacyWebArchives.reserveInitialCapacity(m_cachedSubframeArchives->size());
for (size_t i = 0; i < m_cachedSubframeArchives->size(); ++i) {
RefPtr<WebArchive> subframeWebArchive = m_cachedSubframeArchives->at<WebArchive>(i);
ASSERT(subframeWebArchive);
coreSubframeLegacyWebArchives.uncheckedAppend(subframeWebArchive->coreLegacyWebArchive());
}
m_legacyWebArchive = LegacyWebArchive::create(coreMainResource.release(), WTF::move(coreArchiveResources), WTF::move(coreSubframeLegacyWebArchives));
}
WebArchive::WebArchive(API::Data* data)
{
RefPtr<SharedBuffer> buffer = SharedBuffer::create(data->bytes(), data->size());
m_legacyWebArchive = LegacyWebArchive::create(buffer.get());
}
WebArchive::WebArchive(PassRefPtr<LegacyWebArchive> legacyWebArchive)
: m_legacyWebArchive(legacyWebArchive)
{
}
WebArchive::~WebArchive()
{
}
WebArchiveResource* WebArchive::mainResource()
{
if (!m_cachedMainResource)
m_cachedMainResource = WebArchiveResource::create(m_legacyWebArchive->mainResource());
return m_cachedMainResource.get();
}
API::Array* WebArchive::subresources()
{
if (!m_cachedSubresources) {
Vector<RefPtr<API::Object>> subresources;
subresources.reserveInitialCapacity(m_legacyWebArchive->subresources().size());
for (const auto& subresource : m_legacyWebArchive->subresources())
subresources.uncheckedAppend(WebArchiveResource::create(subresource));
m_cachedSubresources = API::Array::create(WTF::move(subresources));
}
return m_cachedSubresources.get();
}
API::Array* WebArchive::subframeArchives()
{
if (!m_cachedSubframeArchives) {
Vector<RefPtr<API::Object>> subframeWebArchives;
subframeWebArchives.reserveInitialCapacity(m_legacyWebArchive->subframeArchives().size());
for (const auto& subframeArchive : m_legacyWebArchive->subframeArchives())
subframeWebArchives.uncheckedAppend(WebArchive::create(static_cast<LegacyWebArchive*>(subframeArchive.get())));
m_cachedSubframeArchives = API::Array::create(WTF::move(subframeWebArchives));
}
return m_cachedSubframeArchives.get();
}
static void releaseCFData(unsigned char*, const void* data)
{
CFRelease(data);
}
PassRefPtr<API::Data> WebArchive::data()
{
RetainPtr<CFDataRef> rawDataRepresentation = m_legacyWebArchive->rawDataRepresentation();
CFRetain(rawDataRepresentation.get());
return API::Data::createWithoutCopying(CFDataGetBytePtr(rawDataRepresentation.get()), CFDataGetLength(rawDataRepresentation.get()), releaseCFData, rawDataRepresentation.get());
}
LegacyWebArchive* WebArchive::coreLegacyWebArchive()
{
return m_legacyWebArchive.get();
}
}
#endif // PLATFORM(COCOA)