/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "library_windows.h"

#pragma warning(push)
// C5039	'function': pointer or reference to potentially throwing function passed to extern C function under -EHc.
// Undefined behavior may occur if this function throws an exception.
#pragma warning(disable : 5039)
#include <windows.h>
#pragma warning(pop)

#include <base/containers/iterator.h>
#include <base/containers/string.h>
#include <base/containers/string_view.h>
#include <base/namespace.h>
#include <core/log.h>
#include <core/namespace.h>

CORE_BEGIN_NAMESPACE()
using BASE_NS::string;
using BASE_NS::string_view;

LibraryWindows::LibraryWindows(const string_view filename)
{
    string tmp(filename);
    // fix the slashes. (without the correct backslashes the LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR wont work)
    for (auto& t : tmp) {
        if (t == '/') {
            t = '\\';
        }
    }
    libraryHandle_ =
        LoadLibraryEx(tmp.c_str(), nullptr, LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR | LOAD_LIBRARY_SEARCH_DEFAULT_DIRS);
    if (!libraryHandle_) {
        DWORD errorCode = GetLastError();
        CORE_LOG_E("Loading dynamic library '%s' failed: 0x%lx", tmp.c_str(), errorCode);
    }
}

LibraryWindows::~LibraryWindows()
{
    if (libraryHandle_) {
        FreeLibrary(libraryHandle_);
        libraryHandle_ = nullptr;
    }
}

IPlugin* LibraryWindows::GetPlugin() const
{
    if (!libraryHandle_) {
        return nullptr;
    }

    return reinterpret_cast<IPlugin*>(GetProcAddress(libraryHandle_, "gPluginData"));
}

void LibraryWindows::Destroy()
{
    delete this;
}

ILibrary::Ptr ILibrary::Load(const string_view filePath)
{
    return ILibrary::Ptr { new LibraryWindows(filePath) };
}

string_view ILibrary::GetFileExtension()
{
    return ".dll";
}
CORE_END_NAMESPACE()