ArchiveFactory.cpp [plain text]
#include "config.h"
#include "ArchiveFactory.h"
#include "MIMETypeRegistry.h"
#if ENABLE(WEB_ARCHIVE) && USE(CF)
#include "LegacyWebArchive.h"
#endif
#if ENABLE(MHTML)
#include "MHTMLArchive.h"
#endif
#include <wtf/HashMap.h>
#include <wtf/HashSet.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/StdLibExtras.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
typedef RefPtr<Archive> RawDataCreationFunction(const URL&, SharedBuffer&);
typedef HashMap<String, RawDataCreationFunction*, ASCIICaseInsensitiveHash> ArchiveMIMETypesMap;
template <typename ArchiveClass> static RefPtr<Archive> archiveFactoryCreate(const URL& url, SharedBuffer& buffer)
{
return ArchiveClass::create(url, buffer);
}
static ArchiveMIMETypesMap& archiveMIMETypes()
{
static NeverDestroyed<ArchiveMIMETypesMap> mimeTypes;
static bool initialized = false;
if (initialized)
return mimeTypes;
#if ENABLE(WEB_ARCHIVE) && USE(CF)
mimeTypes.get().set("application/x-webarchive", archiveFactoryCreate<LegacyWebArchive>);
#endif
#if ENABLE(MHTML)
mimeTypes.get().set("multipart/related", archiveFactoryCreate<MHTMLArchive>);
mimeTypes.get().set("application/x-mimearchive", archiveFactoryCreate<MHTMLArchive>);
#endif
initialized = true;
return mimeTypes;
}
bool ArchiveFactory::isArchiveMimeType(const String& mimeType)
{
return !mimeType.isEmpty() && archiveMIMETypes().contains(mimeType);
}
RefPtr<Archive> ArchiveFactory::create(const URL& url, SharedBuffer* data, const String& mimeType)
{
RawDataCreationFunction* function = mimeType.isEmpty() ? 0 : archiveMIMETypes().get(mimeType);
if (!data)
return nullptr;
return function ? function(url, *data) : RefPtr<Archive>(nullptr);
}
void ArchiveFactory::registerKnownArchiveMIMETypes()
{
auto& mimeTypes = MIMETypeRegistry::getSupportedNonImageMIMETypes();
for (auto& mimeType : archiveMIMETypes().keys())
mimeTypes.add(mimeType);
}
}