/*

 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

#include <algorithm>

#include <fstream>

#include "ResourceEditor.h"

#include "WinErrorHandling.h"

#include "Log.h"

ResourceEditor::FileLock::FileLock(const std::wstring& binaryPath) {

    h = BeginUpdateResource(binaryPath.c_str(), FALSE);

    if (NULL == h) {

        JP_THROW(SysError(tstrings::any() << "BeginUpdateResource(" 

                    << binaryPath << ") failed", BeginUpdateResource));

    }

    discard(false);

}

ResourceEditor::FileLock::~FileLock() {

    if (!EndUpdateResource(h, theDiscard)) {

        JP_NO_THROW(JP_THROW(SysError(tstrings::any() 

            << "EndUpdateResource(" << h << ") failed.", EndUpdateResource)));

    }

}

ResourceEditor::ResourceEditor() {

    language(MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL)).type(unsigned(0)).id(unsigned(0));

}

ResourceEditor& ResourceEditor::type(unsigned v) {

    return type(MAKEINTRESOURCE(v));

}

ResourceEditor& ResourceEditor::type(LPCWSTR v) {

    if (IS_INTRESOURCE(v)) {

        std::wostringstream printer;

        printer << L"#" << reinterpret_cast<size_t>(v);

        theType = printer.str();

        theTypePtr = MAKEINTRESOURCE(static_cast<DWORD>(reinterpret_cast<DWORD_PTR>(v)));

    } else {

        theType = v;

        theTypePtr = theType.c_str();

    }

    return *this;

}

ResourceEditor& ResourceEditor::id(unsigned v) {

    return id(MAKEINTRESOURCE(v));

}

ResourceEditor& ResourceEditor::id(LPCWSTR v) {

    if (IS_INTRESOURCE(v)) {

        std::wostringstream printer;

        printer << L"#" << reinterpret_cast<size_t>(v);

        theId = printer.str();

    } else {

        theId = v;

        theIdPtr = theId.c_str();

    }

    return *this;

}

ResourceEditor& ResourceEditor::apply(const FileLock& dstBinary, 

                            std::istream& srcStream, std::streamsize size) {

    typedef std::vector<BYTE> ByteArray;

    ByteArray buf;

    if (size <= 0) {

        // Read the entire stream.

        buf = ByteArray((std::istreambuf_iterator<char>(srcStream)),

                                            std::istreambuf_iterator<char>());

    } else {

        buf.resize(size_t(size));

        srcStream.read(reinterpret_cast<char*>(buf.data()), size); 

    }

    auto reply = UpdateResource(dstBinary.get(), theTypePtr, theIdPtr, lang, 

                                buf.data(), static_cast<DWORD>(buf.size()));

    if (reply == FALSE) {

        JP_THROW(SysError("UpdateResource() failed", UpdateResource));

    }

    return *this;

}

ResourceEditor& ResourceEditor::apply(const FileLock& dstBinary, 

                                                const std::wstring& srcFile) {

    std::ifstream input(srcFile, std::ios_base::binary);

    input.exceptions(std::ios::failbit | std::ios::badbit);

    return apply(dstBinary, input);

}