ObsUtils.cpp

#include <QtCore/QDebug>

#include "ObsUtils.hpp"

#ifdef Q_OS_OSX
#include "ObsUtilsOSX.hpp"
#endif

namespace ObsUtils
{

    QString getOpenApp(QHash<QString, QString> &appsToWatch)
    {
        QSet<QString> appsOpen;
#ifdef Q_OS_WIN
        EnumWindows(EnumWindowsProcOpenApps, reinterpret_cast<LPARAM>(&appsOpen));
#endif
#ifdef Q_OS_OSX
        ObsUtilsOSX::setOpenApps(appsOpen);
#endif
        auto end = appsToWatch.cend();
        for (auto it = appsToWatch.cbegin(); it != end; ++it)
        {
            if (appsOpen.contains(it.key()))
            {
                return it.value();
            }
        }
        return "";
    }

    QString getFilteredNameFromAppPath(const QString &appPath) {
        QString appName = getNameFromAppPath(appPath);
        appName.replace("|", "");
        appName.replace(":", "");
        appName.replace("<", "");
        appName.replace(">", "");
        appName.replace("?", "");
        appName.replace("*", "");
        appName.replace("\"", "");
        return appName;
    }

    QString getNameFromAppPath(const QString &appPath) {
#ifdef Q_OS_WIN
        const char *exe = appPath.toStdString().c_str();
        DWORD dwHandle;
        DWORD dwLen = GetFileVersionInfoSizeA(exe, &dwHandle);
        if (dwLen != 0)
        {
            std::vector<unsigned char> data(dwLen);
            if (GetFileVersionInfoA(exe, dwHandle, dwLen, &data[0]))
            {
                // catch default information
                VS_FIXEDFILEINFO fileInfo;
                LPVOID lpInfo;
                UINT unInfoLen;
                if (VerQueryValueA(&data[0], "\\", &lpInfo, &unInfoLen))
                {
                    memcpy(&fileInfo, lpInfo, unInfoLen);
                }

                // find best matching language and codepage
                VerQueryValueA(&data[0], "\\VarFileInfo\\Translation", &lpInfo, &unInfoLen);

                DWORD dwLangCode = 0;
                if (!GetTranslationId(lpInfo, unInfoLen, GetUserDefaultLangID(), dwLangCode, FALSE))
                {
                    if (!GetTranslationId(lpInfo, unInfoLen, GetUserDefaultLangID(), dwLangCode, TRUE))
                    {
                        if (!GetTranslationId(lpInfo, unInfoLen, MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), dwLangCode, TRUE))
                        {
                            if (!GetTranslationId(lpInfo, unInfoLen, MAKELANGID(LANG_ENGLISH, SUBLANG_NEUTRAL), dwLangCode, TRUE))
                                // use the first one we can get
                                dwLangCode = *((DWORD*)lpInfo);
                        }
                    }
                }

                CHAR buffer[1024];
                std::sprintf(buffer, "\\StringFileInfo\\%04X%04X\\FileDescription", dwLangCode & 0x0000FFFF, (dwLangCode & 0xFFFF0000) >> 16);
                if (VerQueryValueA(&data[0], buffer, &lpInfo, &unInfoLen))
                {
                    return QString((char*)lpInfo);
                }
                else
                {
                    std::sprintf(buffer, "\\StringFileInfo\\%04X%04X\\ProductName", dwLangCode & 0x0000FFFF, (dwLangCode & 0xFFFF0000) >> 16);
                    if (VerQueryValueA(&data[0], buffer, &lpInfo, &unInfoLen))
                    {
                        return QString((char*)lpInfo);
                    }
                }
            }
        }
        QString appFile = appPath.mid(appPath.lastIndexOf("\\") + 1);
#else
        QString appFile = appPath.mid(appPath.lastIndexOf("/") + 1);
#endif
        if (appFile.endsWith(".exe") || appFile.endsWith(".app"))
        {
            return appFile.chopped(4);
        }
        else
        {
            return appFile;
        }
    }

#ifdef Q_OS_WIN
    BOOL GetTranslationId(LPVOID lpData, UINT unBlockSize, WORD wLangId, DWORD &dwId, BOOL bPrimaryEnough/*= FALSE*/)
    {
        LPWORD lpwData;
        for (lpwData = (LPWORD)lpData; (LPBYTE)lpwData < ((LPBYTE)lpData) + unBlockSize; lpwData += 2)
        {
            if (*lpwData == wLangId)
            {
                dwId = *((DWORD*)lpwData);
                return TRUE;
            }
        }

        if (!bPrimaryEnough)
            return FALSE;

        for (lpwData = (LPWORD)lpData; (LPBYTE)lpwData < ((LPBYTE)lpData) + unBlockSize; lpwData += 2)
        {
            if (((*lpwData) & 0x00FF) == (wLangId & 0x00FF))
            {
                dwId = *((DWORD*)lpwData);
                return TRUE;
            }
        }

        return FALSE;
    }

    BOOL CALLBACK EnumWindowsProcOpenApps(HWND hwnd, LPARAM lParam)
    {
        if ((!IsWindowVisible(hwnd) && !IsIconic(hwnd)))
            return TRUE;

        DWORD pid = 0;
        GetWindowThreadProcessId(hwnd, &pid);
        HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, pid);
        if (hProcess == 0)
            return TRUE;
        DWORD exe_size = 1024;
        CHAR exe[1024];
        QueryFullProcessImageNameA(hProcess, 0, exe, &exe_size);
        QSet<QString>& appsOpen =
            *reinterpret_cast<QSet<QString>*>(lParam);
        QString filename(&exe[0]);
        appsOpen.insert(filename.mid(filename.lastIndexOf("\\") + 1));
        CloseHandle(hProcess);

        return TRUE;
    }
#endif

}