/* Copyright (C) 2016
 * swift project Community / Contributors
 *
 * This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution. No part of swift project, including this file, may be copied, modified, propagated,
 * or distributed except according to the terms contained in the LICENSE file.
 */

#include "blackmisc/processinfo.h"
#include <QFile>
#include <QFileInfo>
#include <type_traits>

#if defined(Q_OS_MACOS)
#include <libproc.h>
#elif defined(Q_OS_WIN)
#ifndef NOMINMAX
#define NOMINMAX
#endif
#include <windows.h>
#include <psapi.h>
#endif

BLACK_DEFINE_VALUEOBJECT_MIXINS(BlackMisc, CProcessInfo)

namespace BlackMisc
{

    QString CProcessInfo::convertToQString(bool) const
    {
        return QStringLiteral("{ %1, %2 }").arg(QString::number(m_pid), m_name);
    }

#if defined(Q_OS_LINUX)
    QString CProcessInfo::processNameFromId(qint64 pid)
    {
        QString path = QFileInfo(QStringLiteral("/proc/%1/exe").arg(pid)).symLinkTarget();
        return QFileInfo(path).fileName();
    }
#elif defined(Q_OS_MACOS)
    QString CProcessInfo::processNameFromId(qint64 pid)
    {
        char name[1024];
        proc_name(pid, name, std::extent_v<decltype(name)>);
        return name;
    }
#elif defined(Q_OS_WIN)
    QString CProcessInfo::processNameFromId(qint64 pid)
    {
        HANDLE proc = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, static_cast<DWORD>(pid));
        if (! proc) { return {}; }
        wchar_t path[1024];
        auto len = GetModuleFileNameEx(proc, nullptr, path, std::extent_v<decltype(path)>);
        CloseHandle(proc);
        if (len <= 0) { return {}; }
        return QFileInfo(QString::fromWCharArray(path)).completeBaseName();
    }
#else
    QString CProcessInfo::processNameFromId(qint64)
    {
        qFatal("Not implemented");
        return {};
    }
#endif

}