#include "config.h"
#include "FontDatabase.h"
#include "CString.h"
#include "FileSystem.h"
#include "PlatformString.h"
#include <WebKitSystemInterface/WebKitSystemInterface.h>
#include <shlobj.h>
#include <wtf/RetainPtr.h>
namespace WebCore {
static String systemFontsDirectory()
{
static bool initialized;
static String directory;
if (!initialized) {
initialized = true;
Vector<UChar> buffer(MAX_PATH);
if (FAILED(SHGetFolderPath(0, CSIDL_FONTS | CSIDL_FLAG_CREATE, 0, 0, buffer.data())))
return directory;
buffer.resize(wcslen(buffer.data()));
directory = String::adopt(buffer);
}
return directory;
}
static String fontsPlistPath()
{
static String path = pathByAppendingComponent(localUserSpecificStorageDirectory(), "FontsList.plist");
return path;
}
static bool systemHasFontsNewerThanFontsPlist()
{
WIN32_FILE_ATTRIBUTE_DATA plistAttributes = {0};
if (!GetFileAttributesEx(fontsPlistPath().charactersWithNullTermination(), GetFileExInfoStandard, &plistAttributes))
return true;
WIN32_FILE_ATTRIBUTE_DATA fontsDirectoryAttributes = {0};
if (!GetFileAttributesEx(systemFontsDirectory().charactersWithNullTermination(), GetFileExInfoStandard, &fontsDirectoryAttributes))
return true;
return CompareFileTime(&plistAttributes.ftLastWriteTime, &fontsDirectoryAttributes.ftLastWriteTime) < 0;
}
static RetainPtr<CFPropertyListRef> readFontPlist()
{
CString plistPath = fontsPlistPath().utf8();
RetainPtr<CFURLRef> url(AdoptCF, CFURLCreateFromFileSystemRepresentation(0, reinterpret_cast<const UInt8*>(plistPath.data()), plistPath.length(), false));
if (!url)
return 0;
RetainPtr<CFReadStreamRef> stream(AdoptCF, CFReadStreamCreateWithFile(0, url.get()));
if (!stream)
return 0;
if (!CFReadStreamOpen(stream.get()))
return 0;
CFPropertyListFormat format = kCFPropertyListBinaryFormat_v1_0 | kCFPropertyListXMLFormat_v1_0;
RetainPtr<CFPropertyListRef> plist(AdoptCF, CFPropertyListCreateFromStream(0, stream.get(), 0, kCFPropertyListMutableContainersAndLeaves, &format, 0));
CFReadStreamClose(stream.get());
return plist;
}
static bool populateFontDatabaseFromPlist()
{
RetainPtr<CFPropertyListRef> plist = readFontPlist();
if (!plist)
return false;
RetainPtr<CFDataRef> data(AdoptCF, CFPropertyListCreateXMLData(0, plist.get()));
if (!data)
return false;
wkAddFontsFromPlistRepresentation(data.get());
return true;
}
static bool populateFontDatabaseFromFileSystem()
{
RetainPtr<CFStringRef> directory(AdoptCF, systemFontsDirectory().createCFString());
if (!directory)
return false;
wkAddFontsInDirectory(directory.get());
return true;
}
static void writeFontDatabaseToPlist()
{
RetainPtr<CFDataRef> data(AdoptCF, wkCreateFontsPlistRepresentation());
if (!data)
return;
safeCreateFile(fontsPlistPath(), data.get());
}
void populateFontDatabase()
{
static bool initialized;
if (initialized)
return;
initialized = true;
if (!systemHasFontsNewerThanFontsPlist())
if (populateFontDatabaseFromPlist())
return;
if (populateFontDatabaseFromFileSystem())
writeFontDatabaseToPlist();
}
}