/* Copyright 2013-2021 MultiMC Contributors
 *
 * 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 <fstream>
#include <string>

#include <launch/LaunchTask.h>
#include "PrintInstanceInfo.h"

#if defined(Q_OS_LINUX) || defined(Q_OS_FREEBSD)
namespace {
#if defined(Q_OS_LINUX)
void probeProcCpuinfo(QStringList& log)
{
    std::ifstream cpuin("/proc/cpuinfo");
    for (std::string line; std::getline(cpuin, line);) {
        if (strncmp(line.c_str(), "model name", 10) == 0) {
            log << QString::fromStdString(line.substr(13, std::string::npos));
            break;
        }
    }
}

void runLspci(QStringList& log)
{
    // FIXME: fixed size buffers...
    char buff[512];
    int gpuline = -1;
    int cline = 0;
    FILE* lspci = popen("lspci -k", "r");

    if (!lspci)
        return;

    while (fgets(buff, 512, lspci) != NULL) {
        std::string str(buff);
        if (str.length() < 9)
            continue;
        if (str.substr(8, 3) == "VGA") {
            gpuline = cline;
            log << QString::fromStdString(str.substr(35, std::string::npos));
        }
        if (gpuline > -1 && gpuline != cline) {
            if (cline - gpuline < 3) {
                log << QString::fromStdString(str.substr(1, std::string::npos));
            }
        }
        cline++;
    }
    pclose(lspci);
}
#elif defined(Q_OS_FREEBSD)
void runSysctlHwModel(QStringList& log)
{
    char buff[512];
    FILE* hwmodel = popen("sysctl hw.model", "r");
    while (fgets(buff, 512, hwmodel) != NULL) {
        log << QString::fromUtf8(buff);
        break;
    }
    pclose(hwmodel);
}

void runPciconf(QStringList& log)
{
    char buff[512];
    std::string strcard;
    FILE* pciconf = popen("pciconf -lv -a vgapci0", "r");
    while (fgets(buff, 512, pciconf) != NULL) {
        if (strncmp(buff, "    vendor", 10) == 0) {
            std::string str(buff);
            strcard.append(str.substr(str.find_first_of("'") + 1, str.find_last_not_of("'") - (str.find_first_of("'") + 2)));
            strcard.append(" ");
        } else if (strncmp(buff, "    device", 10) == 0) {
            std::string str2(buff);
            strcard.append(str2.substr(str2.find_first_of("'") + 1, str2.find_last_not_of("'") - (str2.find_first_of("'") + 2)));
        }
        log << QString::fromStdString(strcard);
        break;
    }
    pclose(pciconf);
}
#endif
void runGlxinfo(QStringList& log)
{
    // FIXME: fixed size buffers...
    char buff[512];
    FILE* glxinfo = popen("glxinfo", "r");
    if (!glxinfo)
        return;

    while (fgets(buff, 512, glxinfo) != NULL) {
        if (strncmp(buff, "OpenGL version string:", 22) == 0) {
            log << QString::fromUtf8(buff);
            break;
        }
    }
    pclose(glxinfo);
}

}  // namespace
#endif

void PrintInstanceInfo::executeTask()
{
    auto instance = m_parent->instance();
    QStringList log;

#if defined(Q_OS_LINUX)
    ::probeProcCpuinfo(log);
    ::runLspci(log);
    ::runGlxinfo(log);
#elif defined(Q_OS_FREEBSD)
    ::runSysctlHwModel(log);
    ::runPciconf(log);
    ::runGlxinfo(log);
#endif

    logLines(log, MessageLevel::Launcher);
    logLines(instance->verboseDescription(m_session, m_serverToJoin), MessageLevel::Launcher);
    emitSucceeded();
}