/*

 * Copyright (C) 2006, 2007 Apple Inc.  All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

 */

#include "config.h"

#include "WebKitDLL.h"

#include "WebLocalizableStrings.h"

#pragma warning(push, 0)

#include <WebCore/PlatformString.h>

#include <WebCore/StringHash.h>

#include <wtf/text/CString.h>

#pragma warning(pop)

#include <wtf/Assertions.h>

#include <wtf/HashMap.h>

#include <wtf/RetainPtr.h>

#include <wtf/StdLibExtras.h>

#include <CoreFoundation/CoreFoundation.h>

class LocalizedString;

using namespace WebCore;

WebLocalizableStringsBundle WebKitLocalizableStringsBundle = { "com.apple.WebKit", 0 };

typedef HashMap<String, LocalizedString*> LocalizedStringMap;

static Mutex& mainBundleLocStringsMutex()

{

    DEFINE_STATIC_LOCAL(Mutex, mutex, ());

    return mutex;

}

static LocalizedStringMap& mainBundleLocStrings()

{

    DEFINE_STATIC_LOCAL(LocalizedStringMap, map, ());

    return map;

}

static Mutex& frameworkLocStringsMutex()

{

    DEFINE_STATIC_LOCAL(Mutex, mutex, ());

    return mutex;

}

static LocalizedStringMap frameworkLocStrings()

{

    DEFINE_STATIC_LOCAL(LocalizedStringMap, map, ());

    return map;

}

class LocalizedString : public Noncopyable {

public:

    LocalizedString(CFStringRef string)

        : m_cfString(string)

    {

        ASSERT_ARG(string, string);

    }

    operator LPCTSTR() const;

    operator CFStringRef() const { return m_cfString; }

private:

    CFStringRef m_cfString;

    mutable String m_string;

};

LocalizedString::operator LPCTSTR() const

{

    if (!m_string.isEmpty())

        return m_string.charactersWithNullTermination();

    m_string = m_cfString;

    for (unsigned int i = 1; i < m_string.length(); i++)

        if (m_string[i] == '@' && (m_string[i - 1] == '%' || (i > 2 && m_string[i - 1] == '$' && m_string[i - 2] >= '1' && m_string[i - 2] <= '9' && m_string[i - 3] == '%')))

            m_string.replace(i, 1, "s");

    return m_string.charactersWithNullTermination();

}

static CFBundleRef createWebKitBundle()

{

    static CFBundleRef bundle;

    static bool initialized;

    if (initialized)

        return bundle;

    initialized = true;

    WCHAR pathStr[MAX_PATH];

    DWORD length = ::GetModuleFileNameW(gInstance, pathStr, MAX_PATH);

    if (!length || (length == MAX_PATH && GetLastError() == ERROR_INSUFFICIENT_BUFFER))

        return 0;

    bool found = false;

    for (int i = length - 1; i >= 0; i--) {

        // warning C6385: Invalid data: accessing 'pathStr', the readable size is '520' bytes, but '2000' bytes might be read

        #pragma warning(suppress: 6385)

        if (pathStr[i] == L'\\') {

            // warning C6386: Buffer overrun: accessing 'pathStr', the writable size is '520' bytes, but '1996' bytes might be written

            #pragma warning(suppress: 6386)

            pathStr[i] = 0;

            found = true;

            break;

        }

    }

    if (!found)

        return 0;

    if (wcscat_s(pathStr, MAX_PATH, L"\\WebKit.resources"))

        return 0;

    String bundlePathString(pathStr);

    CFStringRef bundlePathCFString = bundlePathString.createCFString();

    if (!bundlePathCFString)

        return 0;

    CFURLRef bundleURLRef = CFURLCreateWithFileSystemPath(0, bundlePathCFString, kCFURLWindowsPathStyle, true);

    CFRelease(bundlePathCFString);

    if (!bundleURLRef)

        return 0;

    bundle = CFBundleCreate(0, bundleURLRef);

    CFRelease(bundleURLRef);

    return bundle;

}

static CFBundleRef cfBundleForStringsBundle(WebLocalizableStringsBundle* stringsBundle)

{

    if (!stringsBundle) {

        static CFBundleRef mainBundle = CFBundleGetMainBundle();

        return mainBundle;

    }

    createWebKitBundle();

    if (!stringsBundle->bundle)

        stringsBundle->bundle = CFBundleGetBundleWithIdentifier(RetainPtr<CFStringRef>(AdoptCF, CFStringCreateWithCString(0, stringsBundle->identifier, kCFStringEncodingASCII)).get());

    return stringsBundle->bundle;

}

static CFStringRef copyLocalizedStringFromBundle(WebLocalizableStringsBundle* stringsBundle, const String& key)

{

    static CFStringRef notFound = CFSTR("localized string not found");

    CFBundleRef bundle = cfBundleForStringsBundle(stringsBundle);

    if (!bundle)

        return notFound;

    RetainPtr<CFStringRef> keyString(AdoptCF, key.createCFString());

    CFStringRef result = CFCopyLocalizedStringWithDefaultValue(keyString.get(), 0, bundle, notFound, 0);

    ASSERT_WITH_MESSAGE(result != notFound, "could not find localizable string %s in bundle", key);

    return result;

}

static LocalizedString* findCachedString(WebLocalizableStringsBundle* stringsBundle, const String& key)

{

    if (!stringsBundle) {

        MutexLocker lock(mainBundleLocStringsMutex());

        return mainBundleLocStrings().get(key);

    }

    if (stringsBundle->bundle == WebKitLocalizableStringsBundle.bundle) {

        MutexLocker lock(frameworkLocStringsMutex());

        return frameworkLocStrings().get(key);

    }

    return 0;

}

static void cacheString(WebLocalizableStringsBundle* stringsBundle, const String& key, LocalizedString* value)

{

    if (!stringsBundle) {

        MutexLocker lock(mainBundleLocStringsMutex());

        mainBundleLocStrings().set(key, value);

        return;

    }

    MutexLocker lock(frameworkLocStringsMutex());

    frameworkLocStrings().set(key, value);

}

static const LocalizedString& localizedString(WebLocalizableStringsBundle* stringsBundle, const String& key)

{

    LocalizedString* string = findCachedString(stringsBundle, key);

    if (string)

        return *string;

    string = new LocalizedString(copyLocalizedStringFromBundle(stringsBundle, key));

    cacheString(stringsBundle, key, string);

    return *string;

}

CFStringRef WebLocalizedStringUTF8(WebLocalizableStringsBundle* stringsBundle, LPCSTR key)

{

    if (!key)

        return 0;

    return localizedString(stringsBundle, String::fromUTF8(key));

}

LPCTSTR WebLocalizedLPCTSTRUTF8(WebLocalizableStringsBundle* stringsBundle, LPCSTR key)

{

    if (!key)

        return 0;

    return localizedString(stringsBundle, String::fromUTF8(key));

}

// These functions are deprecated.

CFStringRef WebLocalizedString(WebLocalizableStringsBundle* stringsBundle, LPCTSTR key)

{

    if (!key)

        return 0;

    return localizedString(stringsBundle, String(key));

}

LPCTSTR WebLocalizedLPCTSTR(WebLocalizableStringsBundle* stringsBundle, LPCTSTR key)

{

    if (!key)

        return 0;

    return localizedString(stringsBundle, String(key));

}

void SetWebLocalizedStringMainBundle(CFBundleRef)

{

}