/*
 * Copyright (C) 2025 Mohamad Al-Jaf
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

#include <stdarg.h>

#define COBJMACROS
#include "initguid.h"
#include "dxcore.h"

#include "wine/test.h"

static HRESULT (WINAPI *pDXCoreCreateAdapterFactory)(REFIID riid, void **out);

#define check_interface(iface, riid, supported) check_interface_(__LINE__, iface, riid, supported)
static void check_interface_(unsigned int line, void *iface, REFIID riid, BOOL supported)
{
    IUnknown *unknown = iface, *out;
    HRESULT hr, expected_hr;

    expected_hr = supported ? S_OK : E_NOINTERFACE;

    hr = IUnknown_QueryInterface(unknown, riid, (void **)&out);
    ok_(__FILE__, line)(hr == expected_hr, "got hr %#lx.\n", hr);
    if (SUCCEEDED(hr))
        IUnknown_Release(out);
}

static BOOL is_any_display_available(void)
{
    DISPLAY_DEVICEW display_device;
    display_device.cb = sizeof(display_device);
    return EnumDisplayDevicesW(NULL, 0, &display_device, 0);
}

static void test_DXCoreCreateAdapterFactory(void)
{
    IDXCoreAdapterFactory *factory2 = (void *)0xdeadbeef;
    IDXCoreAdapterFactory *factory = (void *)0xdeadbeef;
    IDXCoreAdapterList *list2 = (void *)0xdeadbeef;
    IDXCoreAdapterList *list = (void *)0xdeadbeef;
    IDXCoreAdapter *adapter2 = (void *)0xdeadbeef;
    IDXCoreAdapter *adapter = (void *)0xdeadbeef;
    uint32_t adapter_count = 0;
    LONG refcount;
    HRESULT hr;

    if (0) /* Crashes on w1064v1909 */
    {
        hr = pDXCoreCreateAdapterFactory(NULL, NULL);
        ok(hr == E_POINTER, "got hr %#lx.\n", hr);
    }
    hr = pDXCoreCreateAdapterFactory(&IID_IDXCoreAdapterFactory, NULL);
    ok(hr == E_POINTER || broken(hr == E_NOINTERFACE) /* w1064v1909 */, "got hr %#lx.\n", hr);
    hr = pDXCoreCreateAdapterFactory(&DXCORE_ADAPTER_ATTRIBUTE_D3D11_GRAPHICS, (void **)&factory);
    ok(hr == E_NOINTERFACE, "got hr %#lx.\n", hr);
    ok(factory == NULL || broken(factory == (void *)0xdeadbeef) /* w1064v1909 */, "got factory %p.\n", factory);

    hr = pDXCoreCreateAdapterFactory(&IID_IDXCoreAdapterFactory, (void **)&factory);
    ok(hr == S_OK || broken(hr == E_NOINTERFACE) /* w1064v1909 */, "got hr %#lx.\n", hr);
    if (FAILED(hr))
        return;

    hr = pDXCoreCreateAdapterFactory(&IID_IDXCoreAdapterFactory, (void **)&factory2);
    ok(hr == S_OK, "got hr %#lx.\n", hr);
    ok(factory == factory2, "got factory %p, factory2 %p.\n", factory, factory2);
    refcount = IDXCoreAdapterFactory_Release(factory2);
    ok(refcount == 1, "got refcount %ld.\n", refcount);

    check_interface(factory, &IID_IUnknown, TRUE);
    check_interface(factory, &IID_IDXCoreAdapterFactory, TRUE);
    check_interface(factory, &IID_IAgileObject, FALSE);
    check_interface(factory, &IID_IDXCoreAdapter, FALSE);
    check_interface(factory, &IID_IDXCoreAdapterList, FALSE);

    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 0, NULL, &IID_IDXCoreAdapterList, (void **)&list);
    ok(hr == E_INVALIDARG, "got hr %#lx.\n", hr);
    ok(list == NULL, "got list %p.\n", list);
    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 0, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS, &IID_IDXCoreAdapterList, (void **)&list);
    ok(hr == E_INVALIDARG, "got hr %#lx.\n", hr);
    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS, &IID_IDXCoreAdapterFactory, (void **)&list);
    ok(hr == E_NOINTERFACE, "got hr %#lx.\n", hr);
    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, NULL, &IID_IDXCoreAdapterFactory, (void **)&list);
    ok(hr == E_INVALIDARG, "got hr %#lx.\n", hr);
    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS, &IID_IDXCoreAdapterFactory, NULL);
    ok(hr == E_POINTER, "got hr %#lx.\n", hr);

    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS, &IID_IDXCoreAdapterList, (void **)&list);
    ok(hr == S_OK, "got hr %#lx.\n", hr);
    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS, &IID_IDXCoreAdapterList, (void **)&list2);
    ok(hr == S_OK, "got hr %#lx.\n", hr);
    ok(list != list2, "got same list %p, list2 %p.\n", list, list);
    refcount = IDXCoreAdapterList_Release(list2);
    ok(refcount == 0, "got refcount %ld.\n", refcount);

    check_interface(list, &IID_IUnknown, TRUE);
    check_interface(list, &IID_IDXCoreAdapterList, TRUE);
    check_interface(list, &IID_IAgileObject, FALSE);
    check_interface(list, &IID_IDXCoreAdapter, FALSE);
    check_interface(list, &IID_IDXCoreAdapterFactory, FALSE);

    adapter_count = IDXCoreAdapterList_GetAdapterCount(list);
    ok(adapter_count != 0, "got adapter_count 0.\n");

    hr = IDXCoreAdapterList_GetAdapter(list, 0xdeadbeef, &IID_IDXCoreAdapter, NULL);
    ok(hr == E_POINTER, "got hr %#lx.\n", hr);
    hr = IDXCoreAdapterList_GetAdapter(list, adapter_count, &IID_IDXCoreAdapter, (void **)&adapter);
    ok(hr == E_INVALIDARG, "got hr %#lx.\n", hr);
    ok(adapter == NULL, "got adapter %p.\n", adapter);
    hr = IDXCoreAdapterList_GetAdapter(list, 0, &IID_IDXCoreAdapterList, (void **)&adapter);
    ok(hr == E_NOINTERFACE, "got hr %#lx.\n", hr);

    hr = IDXCoreAdapterList_GetAdapter(list, 0, &IID_IDXCoreAdapter, (void **)&adapter);
    ok(hr == S_OK, "got hr %#lx.\n", hr);
    hr = IDXCoreAdapterList_GetAdapter(list, 0, &IID_IDXCoreAdapter, (void **)&adapter2);
    ok(hr == S_OK, "got hr %#lx.\n", hr);
    ok(adapter == adapter2, "got adapter %p, adapter2 %p.\n", adapter, adapter2);
    refcount = IDXCoreAdapter_Release(adapter2);
    todo_wine
    ok(refcount == 3, "got refcount %ld.\n", refcount);

    check_interface(adapter, &IID_IUnknown, TRUE);
    check_interface(adapter, &IID_IDXCoreAdapter, TRUE);
    check_interface(adapter, &IID_IAgileObject, FALSE);
    check_interface(adapter, &IID_IDXCoreAdapterList, FALSE);
    check_interface(adapter, &IID_IDXCoreAdapterFactory, FALSE);

    refcount = IDXCoreAdapter_Release(adapter);
    todo_wine
    ok(refcount == 2, "got refcount %ld.\n", refcount);
    refcount = IDXCoreAdapterList_Release(list);
    ok(refcount == 0, "got refcount %ld.\n", refcount);
    refcount = IDXCoreAdapterFactory_Release(factory);
    ok(refcount == 0, "got refcount %ld.\n", refcount);
}

static void test_GetProperty(void)
{
    IDXCoreAdapterFactory *factory;
    IDXCoreAdapterList *list;
    DXCoreHardwareID hwid[2];
    IDXCoreAdapter *adapter;
    uint32_t count, dummy;
    uint64_t memory;
    LARGE_INTEGER version;
    BYTE is_hardware;
    LUID luid[2];
    size_t size;
    HRESULT hr;
    char *str;

    if (FAILED(pDXCoreCreateAdapterFactory(&IID_IDXCoreAdapterFactory, (void **)&factory)))
        return;

    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS,
            &IID_IDXCoreAdapterList, (void **)&list);
    ok(hr == S_OK, "Got hr %#lx.\n", hr);

    count = IDXCoreAdapterList_GetAdapterCount(list);

    for (uint32_t i = 0; i < count; ++i)
    {
        hr = IDXCoreAdapterList_GetAdapter(list, i, &IID_IDXCoreAdapter, (void **)&adapter);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetProperty(adapter, 0xdeadbeef, 0, hwid);
        ok(hr == DXGI_ERROR_INVALID_CALL, "Got hr %#lx.\n", hr);

        /* InstanceLuid */
        hr = IDXCoreAdapter_GetProperty(adapter, InstanceLuid, 0, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, InstanceLuid, 0, luid);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, InstanceLuid, sizeof(luid[0]) - 1, luid);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, InstanceLuid, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, InstanceLuid, &size);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(size == sizeof(*luid), "Got size %Iu.\n", size);

        hr = IDXCoreAdapter_GetProperty(adapter, InstanceLuid, sizeof(luid[0]), luid);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetProperty(adapter, InstanceLuid, sizeof(luid[0]) + 1, luid);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        /* HardwareID */
        hr = IDXCoreAdapter_GetProperty(adapter, HardwareID, 0, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, HardwareID, 0, hwid);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, HardwareID, sizeof(hwid[0]) - 1, hwid);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, HardwareID, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, HardwareID, &size);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(size == sizeof(*hwid), "Got size %Iu.\n", size);

        memset(hwid, 0, sizeof(hwid));
        hr = IDXCoreAdapter_GetProperty(adapter, HardwareID, sizeof(hwid[0]), hwid);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(!!hwid[0].vendorID, "Expected vendorID.\n");

        memset(hwid, 0, sizeof(hwid));
        hr = IDXCoreAdapter_GetProperty(adapter, HardwareID, sizeof(hwid[0]) + 1, hwid);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(!!hwid[0].vendorID, "Expected vendorID.\n");
        ok(!hwid[1].vendorID, "Expected no vendorID.\n");
        ok(!hwid[1].deviceID, "Expected no deviceID.\n");

        /* DedicatedAdapterMemory */
        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedAdapterMemory, 0, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedAdapterMemory, 0, &memory);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedAdapterMemory, sizeof(memory) - 1, &memory);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, DedicatedAdapterMemory, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, DedicatedAdapterMemory, &size);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(size == sizeof(memory), "Got size %Iu.\n", size);

        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedAdapterMemory, sizeof(memory), &memory);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        /* DedicatedSystemMemory */
        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedSystemMemory, 0, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedSystemMemory, 0, &memory);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedSystemMemory, sizeof(memory) - 1, &memory);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, DedicatedSystemMemory, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, DedicatedSystemMemory, &size);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(size == sizeof(memory), "Got size %Iu.\n", size);

        hr = IDXCoreAdapter_GetProperty(adapter, DedicatedSystemMemory, sizeof(memory), &memory);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        /* SharedSystemMemory */
        hr = IDXCoreAdapter_GetProperty(adapter, SharedSystemMemory, 0, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, SharedSystemMemory, 0, &memory);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, SharedSystemMemory, sizeof(memory) - 1, &memory);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, SharedSystemMemory, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, SharedSystemMemory, &size);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(size == sizeof(memory), "Got size %Iu.\n", size);

        hr = IDXCoreAdapter_GetProperty(adapter, SharedSystemMemory, sizeof(memory), &memory);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        /* IsHardware */
        hr = IDXCoreAdapter_GetProperty(adapter, IsHardware, 0, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);
        hr = IDXCoreAdapter_GetProperty(adapter, IsHardware, 0, &is_hardware);
        ok(hr == E_INVALIDARG, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, IsHardware, NULL);
        ok(hr == E_POINTER, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetPropertySize(adapter, IsHardware, &size);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(size == sizeof(is_hardware), "Got property size.\n");

        is_hardware = 3;
        hr = IDXCoreAdapter_GetProperty(adapter, IsHardware, sizeof(is_hardware), &is_hardware);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(is_hardware == 0 || is_hardware == 1, "Got value %d.\n", is_hardware);

        dummy = 0;
        hr = IDXCoreAdapter_GetProperty(adapter, IsHardware, sizeof(dummy), &dummy);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        ok(dummy == is_hardware, "Got value %#x.\n", dummy);

        /* DriverDescription */
        size = 0;
        hr = IDXCoreAdapter_GetPropertySize(adapter, DriverDescription, &size);
        ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
        ok(!!size, "Unexpected property size.\n");

        str = malloc(size);

        hr = IDXCoreAdapter_GetProperty(adapter, DriverDescription, size, str);
        ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetProperty(adapter, DriverDescription, size, NULL);
        ok(hr == E_POINTER, "Unexpected hr %#lx.\n", hr);

        *str = 0x1;
        hr = IDXCoreAdapter_GetProperty(adapter, DriverDescription, size - 1, str);
        ok(hr == E_INVALIDARG, "Unexpected hr %#lx.\n", hr);
        ok(!*str, "Unexpected buffer contents %s.\n", wine_dbgstr_a(str));

        *str = 0x1;
        hr = IDXCoreAdapter_GetProperty(adapter, DriverDescription, 0, str);
        ok(hr == E_INVALIDARG, "Unexpected hr %#lx.\n", hr);
        ok(*str == 0x1, "Unexpected buffer contents %s.\n", wine_dbgstr_a(str));

        free(str);

        /* DriverVersion */
        size = 0;
        hr = IDXCoreAdapter_GetPropertySize(adapter, DriverVersion, &size);
        ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);
        ok(size == sizeof(LARGE_INTEGER), "Unexpected property size.\n");

        hr = IDXCoreAdapter_GetProperty(adapter, DriverVersion, size, &version);
        ok(hr == S_OK, "Unexpected hr %#lx.\n", hr);

        IDXCoreAdapter_Release(adapter);
    }

    IDXCoreAdapterList_Release(list);

    IDXCoreAdapterFactory_Release(factory);
}

static void test_GetAdapterByLuid(void)
{
    IDXCoreAdapter *adapter, *adapter2;
    IDXCoreAdapterFactory *factory;
    IDXCoreAdapterList *list;
    uint32_t count;
    HRESULT hr;
    LUID luid;

    if (FAILED(pDXCoreCreateAdapterFactory(&IID_IDXCoreAdapterFactory, (void **)&factory)))
        return;

    hr = IDXCoreAdapterFactory_CreateAdapterList(factory, 1, &DXCORE_ADAPTER_ATTRIBUTE_D3D12_GRAPHICS,
            &IID_IDXCoreAdapterList, (void **)&list);
    ok(hr == S_OK, "Got hr %#lx.\n", hr);

    count = IDXCoreAdapterList_GetAdapterCount(list);

    for (uint32_t i = 0; i < count; ++i)
    {
        hr = IDXCoreAdapterList_GetAdapter(list, i, &IID_IDXCoreAdapter, (void **)&adapter);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapter_GetProperty(adapter, InstanceLuid, sizeof(luid), &luid);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);

        hr = IDXCoreAdapterFactory_GetAdapterByLuid(factory, &luid, &IID_IDXCoreAdapter, (void **)&adapter2);
        ok(hr == S_OK, "Got hr %#lx.\n", hr);
        IDXCoreAdapter_Release(adapter2);

        IDXCoreAdapter_Release(adapter);
    }

    IDXCoreAdapterList_Release(list);

    IDXCoreAdapterFactory_Release(factory);
}

START_TEST(dxcore)
{
    HMODULE dxcore_handle = LoadLibraryA("dxcore.dll");
    if (!dxcore_handle)
    {
        win_skip("Could not load dxcore.dll\n");
        return;
    }
    pDXCoreCreateAdapterFactory = (void *)GetProcAddress(dxcore_handle, "DXCoreCreateAdapterFactory");
    if (!pDXCoreCreateAdapterFactory)
    {
        win_skip("Failed to get DXCoreCreateAdapterFactory address, skipping dxcore tests\n");
        FreeLibrary(dxcore_handle);
        return;
    }

    if (!is_any_display_available())
    {
        skip("No display available.\n");
        return;
    }

    test_DXCoreCreateAdapterFactory();
    test_GetProperty();
    test_GetAdapterByLuid();

    FreeLibrary(dxcore_handle);
}
