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 PassRefPtr<Archive> RawDataCreationFunction(const URL&, SharedBuffer*);
typedef HashMap<String, RawDataCreationFunction*, CaseFoldingHash> ArchiveMIMETypesMap;
template <typename ArchiveClass> static PassRefPtr<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", static_cast<RawDataCreationFunction*>(&archiveFactoryCreate<LegacyWebArchive>));
#endif
#if ENABLE(MHTML)
mimeTypes.get().set("multipart/related", static_cast<RawDataCreationFunction*>(&archiveFactoryCreate<MHTMLArchive>));
mimeTypes.get().set("application/x-mimearchive", static_cast<RawDataCreationFunction*>(&archiveFactoryCreate<MHTMLArchive>));
#endif
initialized = true;
return mimeTypes;
}
bool ArchiveFactory::isArchiveMimeType(const String& mimeType)
{
return !mimeType.isEmpty() && archiveMIMETypes().contains(mimeType);
}
PassRefPtr<Archive> ArchiveFactory::create(const URL& url, SharedBuffer* data, const String& mimeType)
{
RawDataCreationFunction* function = mimeType.isEmpty() ? 0 : archiveMIMETypes().get(mimeType);
return function ? function(url, data) : PassRefPtr<Archive>(0);
}
void ArchiveFactory::registerKnownArchiveMIMETypes()
{
HashSet<String>& mimeTypes = MIMETypeRegistry::getSupportedNonImageMIMETypes();
for (const auto& mimeType : archiveMIMETypes().keys())
mimeTypes.add(mimeType);
}
}