/**************************************************************************
**
** This file is part of Installer Framework
**
** Copyright (c) 2011-2012 Nokia Corporation and/or its subsidiary(-ies).
**
** Contact: Nokia Corporation (qt-info@nokia.com)
**
**
** GNU Lesser General Public License Usage
**
** This file may be used under the terms of the GNU Lesser General Public
** License version 2.1 as published by the Free Software Foundation and
** appearing in the file LICENSE.LGPL included in the packaging of this file.
** Please review the following information to ensure the GNU Lesser General
** Public License version 2.1 requirements will be met:
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** In addition, as a special exception, Nokia gives you certain additional
** rights. These rights are described in the Nokia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** Other Usage
**
** Alternatively, this file may be used in accordance with the terms and
** conditions contained in a signed written agreement between you and Nokia.
**
** If you have questions regarding the use of this file, please contact
** Nokia at qt-info@nokia.com.
**
**************************************************************************/

#include "registerfiletypeoperation.h"

#include "qsettingswrapper.h"

using namespace QInstaller;

RegisterFileTypeOperation::RegisterFileTypeOperation()
{
    setName(QLatin1String("RegisterFileType"));
}

void RegisterFileTypeOperation::backup()
{
}

bool RegisterFileTypeOperation::performOperation()
{
    // extension
    // command
    // (description)
    // (content type)
    // (icon)
#ifdef Q_OS_WIN
    const QStringList args = arguments();
    if (args.count() < 2 || args.count() > 5) {
        setError(InvalidArguments);
        setErrorString(tr("Invalid arguments in %0").arg(name()));
        return false;
    }

    const QString extension = args.at(0);
    const QString command = args.at(1);
    const QString description = args.value(2); // optional
    const QString contentType = args.value(3); // optional
    const QString icon = args.value(4); // optional

    const QString className = QString::fromLatin1("%1_auto_file").arg(extension);
    const QString settingsPrefix = QString::fromLatin1("Software/Classes/");

    //QSettingsWrapper settings(QLatin1String("HKEY_CLASSES_ROOT"), QSettingsWrapper::NativeFormat);
    QSettingsWrapper settings(QLatin1String("HKEY_CURRENT_USER"), QSettingsWrapper::NativeFormat);
    // first backup the old values
    setValue(QLatin1String("oldClass"), settings.value(QString::fromLatin1("%1.%2/Default").arg(settingsPrefix,
        extension)));
    setValue(QLatin1String("oldContentType"), settings.value(QString::fromLatin1("%1.%2/Content Type")
        .arg(settingsPrefix, extension)));

    // the file extension was not yet registered. Let's do so now
    settings.setValue(QString::fromLatin1("%1.%2/Default").arg(settingsPrefix, extension), className);

    // register the content type, if given
    if (!contentType.isEmpty())
        settings.setValue(QString::fromLatin1("%1.%2/Content Type").arg(settingsPrefix, extension), contentType);

    setValue(QLatin1String("className"), className);
    // register the description, if given
    if (!description.isEmpty())
        settings.setValue(QString::fromLatin1("%1%2/Default").arg(settingsPrefix, className), description);

    // register the icon, if given
    if (!icon.isEmpty()) {
        settings.setValue(QString::fromLatin1("%1%2/DefaultIcon/Default").arg(settingsPrefix,
            className), icon);
    }

    // register the command to open the file
    settings.setValue(QString::fromLatin1("%1%2/shell/Open/Command/Default").arg(settingsPrefix,
        className), command);

    return true;
#else
    setError(UserDefinedError);
    setErrorString(QObject::tr("Registering file types in only supported on Windows."));
    return false;
#endif
}

bool RegisterFileTypeOperation::undoOperation()
{
    // extension
    // command
    // (description)
    // (content type)
    // (icon)
#ifdef Q_OS_WIN
    const QStringList args = arguments();
    if (args.count() < 2 || args.count() > 5) {
        setErrorString(tr("Register File Type: Invalid arguments"));
        return false;
    }
    const QString extension = args.at(0);
    const QString command = args.at(1);
    const QString description = args.value(2); // optional
    const QString contentType = args.value(3); // optional
    const QString icon = args.value(4); // optional

    const QString className = QString::fromLatin1("%1_auto_file").arg(extension);
    const QString classes = QString::fromLatin1("Software/Classes/%1").arg(className);
    const QString extensions = QString::fromLatin1("Software/Classes/.%1").arg(extension);

    QSettingsWrapper settings(QLatin1String("HKEY_CURRENT_USER"), QSettingsWrapper::NativeFormat);

    const QString restoredClassName = value(QLatin1String("oldClassName")).toString();
    // register the command to open the file
    if (settings.value(QString::fromLatin1("%1/shell/Open/Command/Default").arg(classes)).toString() != command)
        return false;
    if (settings.value(QString::fromLatin1("%1/DefaultIcon/Default").arg(classes)).toString() != icon)
        return false;
    if (settings.value(QString::fromLatin1("%1/Default").arg(classes)).toString() != description)
        return false;
    if (settings.value(QString::fromLatin1("%1/Content Type").arg(extensions)).toString() != contentType)
        return false;
    if (settings.value(QString::fromLatin1("%1/Default").arg(extensions)).toString() != className)
        return false;

    const QLatin1String settingsPrefix("Software/Classes/");
    const QVariant oldCommand = value(QLatin1String("oldCommand"));
    if (!oldCommand.isNull()) {
        settings.setValue(QString::fromLatin1("%1%2/shell/Open/Command/Default").arg(settingsPrefix,
            restoredClassName), oldCommand);
    } else {
        settings.remove(QString::fromLatin1("%1/shell/Open/Command/Default").arg(classes));
    }

    // register the icon, if given
    const QVariant oldIcon = value(QLatin1String("oldIcon"));
    if (!oldIcon.isNull()) {
        settings.setValue(QString::fromLatin1("%1%2/DefaultIcon/Default").arg(settingsPrefix,
            restoredClassName), oldIcon);
    } else {
        settings.remove(QString::fromLatin1("%1%2/DefaultIcon/Default").arg(classes));
    }

    // register the description, if given
    const QVariant oldDescription = value(QLatin1String("oldDescription"));
    if (!oldDescription.isNull()) {
        settings.setValue(QString::fromLatin1("%1%2/Default").arg(settingsPrefix, restoredClassName),
            oldDescription);
    } else {
        settings.remove(QString::fromLatin1("%1%2/Default").arg(classes));
    }

    // content type
    settings.remove(classes);

    const QVariant oldContentType = value(QLatin1String("oldContentType"));
    if(!oldContentType.isNull())
        settings.setValue(QString::fromLatin1("%1/Content Type").arg(extensions), oldContentType);
    else
        settings.remove(QString::fromLatin1("%1/Content Type").arg(extensions));

    // class

    const QVariant oldClass = value(QLatin1String("oldClass"));
    if(!oldClass.isNull())
        settings.setValue(QString::fromLatin1("%1/Default").arg(extensions), oldClass);
    else
        settings.remove(extensions);

    return true;
#else
    setErrorString(QObject::tr("Registering file types in only supported on Windows."));
    return false;
#endif
}

bool RegisterFileTypeOperation::testOperation()
{
    return true;
}

Operation *RegisterFileTypeOperation::clone() const
{
    return new RegisterFileTypeOperation();
}