#include <iostream>
#include <fstream>
#include <cstdlib>
#include <filesystem>
#include <unistd.h>

namespace fs = std::filesystem;

int compileKernel(const std::string& versionFolder, const std::string& versionFileName) {
    // 构建文件路径
    std::string fullPath = versionFolder + "/" + versionFileName;

    // 从文件读取内核下载链接
    std::ifstream file(fullPath);
    if (!file.is_open()) {
        std::cerr << "Error: Unable to open " << fullPath << std::endl;
        return 1;
    }

    std::string kernelUrl;
    std::getline(file, kernelUrl);
    file.close();

    // 下载内核源代码
    std::string downloadCommand = "curl -O " + kernelUrl;
    int downloadResult = system(downloadCommand.c_str());
    if (downloadResult != 0) {
        std::cerr << "Error: Unable to download kernel source code" << std::endl;
        return 1;
    }

    // 获取下载文件名
    size_t lastSlash = kernelUrl.find_last_of('/');
    std::string fileName = kernelUrl.substr(lastSlash + 1);

    // 输出一些信息以便调试
    std::cout << "Downloaded kernel source code: " << fileName << std::endl;

    // 解压缩内核源代码
    std::string extractCommand = "tar -xvf " + fileName;
    int extractResult = system(extractCommand.c_str());
    if (extractResult != 0) {
        std::cerr << "Error: Unable to extract kernel source code" << std::endl;
        return 1;
    }

    // 输出一些信息以便调试
    std::string kernelDir = fileName.substr(0, fileName.find_last_of('.'));
    std::cout << "Extracted kernel source code: " << kernelDir << std::endl;

    // 进入解压缩后的目录
    chdir(kernelDir.c_str());

    // 拷贝启动目录到源代码目录
    system("cp -v /boot/config-$(uname -r) .config");

    // 额外的配置命令
    system("scripts/config --set-str SYSTEM_TRUSTED_KEYS \"\"");
    system("scripts/config --set-str SYSTEM_REVOCATION_KEYS \"\"");
    system("scripts/config --undefine DEBUG_INFO");
    system("scripts/config --undefine DEBUG_INFO_COMPRESSED");
    system("scripts/config --undefine DEBUG_INFO_REDUCED");
    system("scripts/config --undefine DEBUG_INFO_SPLIT");
    system("scripts/config --undefine GDB_SCRIPTS");
    system("scripts/config --set-val DEBUG_INFO_DWARF5 n");
    system("scripts/config --set-val DEBUG_INFO_NONE y");

    // 使用 CPU 核心数量进行并行编译
    int cpuCores = std::atoi(std::getenv("CPU_CORES"));
    std::string compileCommand = "sudo make bindeb-pkg -j" + std::to_string(cpuCores);
    int compileResult = system(compileCommand.c_str());
    if (compileResult != 0) {
        std::cerr << "Error: Compilation failed" << std::endl;
        return 1;
    }

    // 安装当前目录下所有的 Debian 包
    for (const auto& entry : fs::directory_iterator(".")) {
        if (entry.path().extension() == ".deb") {
            std::string installCommand = "sudo dpkg -i " + entry.path().filename().string();
            int installResult = system(installCommand.c_str());
            if (installResult != 0) {
                std::cerr << "Error: Installation of Debian packages failed" << std::endl;
                return 1;
            }
        }
    }

    // 更新引导
    system("sudo update-grub");

    return 0;
}

int main() {
    // 执行 Python 脚本获取最新版本和 URL
    for (int i = 0; i <= 2; ++i) {
        std::string pythonScript = "python3 get-newest-version.py " + std::to_string(i);
        int pythonExitCode = system(pythonScript.c_str());
        if (pythonExitCode != 0) {
            std::cerr << "Error: Failed to run Python script" << std::endl;
            return 1;
        }
    }

    const std::vector<std::string> versionFileNames = {
        "longtermurl.txt",
        "mainlineurl.txt",
        "stableurl.txt"
    };

    for (const auto& versionFileName : versionFileNames) {
        int compileResult = compileKernel("/tmp", versionFileName);
        if (compileResult != 0) {
            std::cerr << "Error: Compilation failed for version: " << versionFileName << std::endl;
            return 1;
        }
    }

    return 0;
}