/*

 * 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 "MIMETypeRegistry.h"

#include <shlwapi.h>

#include <wtf/HashMap.h>

namespace WebCore 

{

String getMIMETypeForUTI(const String & uti)

{

    String mimeType;

    // FIXME: This is an ugly hack: public.type -> image/type mimetype

    if (int dotLocation = uti.reverseFind('.')) {

        mimeType = String("image/")+uti.substring(dotLocation + 1);

    }

    return mimeType;

}

static String mimeTypeForExtension(const String& extension)

{

    String ext = "." + extension;

    WCHAR contentTypeStr[256];

    DWORD contentTypeStrLen = sizeof(contentTypeStr);

    DWORD keyType;

    HRESULT result = SHGetValue(HKEY_CLASSES_ROOT, ext.charactersWithNullTermination(), L"Content Type", &keyType, (LPVOID)contentTypeStr, &contentTypeStrLen);

    if (result == ERROR_SUCCESS && keyType == REG_SZ) 

        return String(contentTypeStr, contentTypeStrLen / sizeof(contentTypeStr[0]) - 1);

    return String();

}

String MIMETypeRegistry::getPreferredExtensionForMIMEType(const String& type)

{

    String mimeType;

    int semiColonPos = type.find(';');

    if (semiColonPos < 0)

        mimeType = type;

    else

        mimeType = type.substring(0, semiColonPos);

    String path = "MIME\\Database\\Content Type\\" + type;

    WCHAR extStr[MAX_PATH];

    DWORD extStrLen = sizeof(extStr);

    DWORD keyType;

    HRESULT result = SHGetValueW(HKEY_CLASSES_ROOT, path.charactersWithNullTermination(), L"Extension", &keyType, (LPVOID)extStr, &extStrLen);

    if (result == ERROR_SUCCESS && keyType == REG_SZ) 

        return String(extStr + 1, extStrLen / sizeof(extStr[0]) - 2);

    return String();

}

String MIMETypeRegistry::getMIMETypeForExtension(const String &ext)

{

    if (ext.isEmpty())

        return String();

    static HashMap<String, String> mimetypeMap;

    if (mimetypeMap.isEmpty()) {

        //fill with initial values

        mimetypeMap.add("txt", "text/plain");

        mimetypeMap.add("pdf", "application/pdf");

        mimetypeMap.add("ps", "application/postscript");

        mimetypeMap.add("html", "text/html");

        mimetypeMap.add("htm", "text/html");

        mimetypeMap.add("xml", "text/xml");

        mimetypeMap.add("xsl", "text/xsl");

        mimetypeMap.add("js", "application/x-javascript");

        mimetypeMap.add("xhtml", "application/xhtml+xml");

        mimetypeMap.add("rss", "application/rss+xml");

        mimetypeMap.add("webarchive", "application/x-webarchive");

        mimetypeMap.add("svg", "image/svg+xml");

        mimetypeMap.add("svgz", "image/svg+xml");

        mimetypeMap.add("jpg", "image/jpeg");

        mimetypeMap.add("jpeg", "image/jpeg");

        mimetypeMap.add("png", "image/png");

        mimetypeMap.add("tif", "image/tiff");

        mimetypeMap.add("tiff", "image/tiff");

        mimetypeMap.add("ico", "image/ico");

        mimetypeMap.add("cur", "image/ico");

        mimetypeMap.add("bmp", "image/bmp");

    }

    String result = mimetypeMap.get(ext);

    if (result.isEmpty()) {

        result = mimeTypeForExtension(ext);

        if (!result.isEmpty())

            mimetypeMap.add(ext, result);

    }

    return result;

}

}