//
// Created by Michael on 25/03/2024.
//

#include <windows.h>
#include <string>
#include <vector>
#include <locale>

#include "application.h"

int main() {
    std::string libraryPath = "library.dll";

    // find the "library.dll" either in "./" or in "./lib" or in "./cmake-build-release"
    std::vector<std::string> libraryPaths = {
            libraryPath,
            "lib\\" + libraryPath,
            "cmake-build-release\\" + libraryPath,
            "../" + libraryPath,
            "../lib/" + libraryPath,
            "../cmake-build-release/" + libraryPath
    };

    std::string libraryFullPath;
    for(const std::string& path : libraryPaths) {
        if(GetFileAttributesA(path.c_str()) != INVALID_FILE_ATTRIBUTES) {
            libraryFullPath = path;
            break;
        }
    }

    if(libraryFullPath.empty()) {
        printf("Library not found\n");
        return 1;
    }


    // In this directory, find all directories that start with "rekordbox" and end with a number.
    std::string programPath = R"(C:\Program Files\Pioneer\)";

    // Find all matching directories
    std::vector<std::string> directories;
    WIN32_FIND_DATAA findData;
    HANDLE hFind = FindFirstFileA((programPath + "rekordbox*").c_str(), &findData);
    if(hFind != INVALID_HANDLE_VALUE) {
        do {
            if(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
                std::string directoryName = findData.cFileName;
                if(directoryName.find("rekordbox") == 0 && isdigit(directoryName.back())) {
                    directories.push_back(directoryName);
                }
            }
        } while(FindNextFileA(hFind, &findData));
        FindClose(hFind);
    }

    // If there are no directories, exit
    if(directories.empty()) {
        MessageBoxA(NULL, "No rekordbox installation found.", "Error", MB_OK | MB_ICONERROR);
        return 1;
    }

    // Print all directories to stdout

    printf("Found rekordbox installations: \n");

    for(const std::string& directory : directories) {
        printf("    %s\n", directory.c_str());
    }

    // Out of all the directories, find the latest version
    // Versions are semver
    int latestVersion[3] = {0, 0, 0};

    for(const std::string& directory : directories) {
        // Extract the version from the directory name
        std::string version_str = directory.substr(sizeof("rekordbox ")-1);


        // Split into major, minor, patch
        int version[3];
        sscanf_s(version_str.c_str(), "%d.%d.%d", &version[0], &version[1], &version[2]);

        // Compare to latest version
        // If it's a newer major version, replace
        if(version[0] > latestVersion[0]) {
            latestVersion[0] = version[0];
            latestVersion[1] = version[1];
            latestVersion[2] = version[2];
            continue;
        }

        // If it's the same major, but newer minor, replace
        if(version[0] == latestVersion[0] && version[1] > latestVersion[1]) {
            latestVersion[1] = version[1];
            latestVersion[2] = version[2];
            continue;
        }

        // If it's the same major and minor, but newer patch, replace
        if(version[0] == latestVersion[0] && version[1] == latestVersion[1] && version[2] > latestVersion[2]) {
            latestVersion[2] = version[2];
            continue;
        }
    }


    // Convert the latest version back to a string like "1.2.3"
    std::string latestVersion_str = std::to_string(latestVersion[0]) + "." + std::to_string(latestVersion[1]) + "." + std::to_string(latestVersion[2]);


    // Get the full path to the latest version
    std::string latestVersionPath = programPath + "rekordbox " + latestVersion_str + "\\rekordbox.exe";
    printf("Choosing latest version (v%s) at: %s\n", latestVersion_str.c_str(), latestVersionPath.c_str());



    // Start the executable with the library injected
    PROCESS_INFORMATION pi = { 0 };
    STARTUPINFOW si = { 0 };
    si.cb = sizeof( si );


    std::wstring latestVersionPathw = std::wstring(latestVersionPath.begin(), latestVersionPath.end());
    LPCWSTR latestVersionPathW = latestVersionPathw.c_str();


    DWORD flags = CREATE_SUSPENDED;

    if (!CreateProcessW(latestVersionPathW,  nullptr, nullptr, nullptr, FALSE, flags, nullptr, nullptr, &si, &pi )) {
        printf("Failed to start rekordbox\n");
        // Get error
        DWORD error = GetLastError();
        LPSTR messageBuffer = nullptr;
        size_t size = FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
                                     NULL, error, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPSTR)&messageBuffer, 0, NULL);
        printf("Error %lu: %s\n", error, messageBuffer);
        return 1;
    }

    // Make the stdout of the process the same as the current stdout


    LPVOID loc = VirtualAllocEx(pi.hProcess, nullptr, libraryFullPath.size() + 1, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
    BOOL result = WriteProcessMemory(pi.hProcess, loc, libraryFullPath.c_str(), libraryFullPath.size() + 1, nullptr);

    if(!result) {
        printf("Failed to inject DLL. Couldn't write process memory\n");
        return 1;
    }

    HANDLE hThread = CreateRemoteThread(pi.hProcess, nullptr, 0, (LPTHREAD_START_ROUTINE) LoadLibraryA, loc, 0, nullptr);

    if(hThread == nullptr) {
        printf("Failed to inject DLL. Couldn't create remote thread\n");
        return 1;
    }

    ResumeThread(pi.hThread);

    printf("Started rekordbox, goodbye!\n");


    // Cleanup
    CloseHandle(pi.hProcess);
    CloseHandle(pi.hThread);


    // Sleep for 5 seconds
    Sleep(5000);

    return 0;
}