limine/stage2/menu.c

#include <stdint.h>
#include <stddef.h>
#include <stdbool.h>
#include <limine.h>
#include <menu.h>
#include <lib/print.h>
#include <lib/blib.h>
#include <lib/libc.h>
#include <lib/config.h>
#include <lib/term.h>
#include <mm/pmm.h>
#include <drivers/vbe.h>

static char *cmdline;
#define CMDLINE_MAX 1024

static char config_entry_name[1024];

char *menu(int boot_drive) {
    cmdline = conv_mem_alloc(CMDLINE_MAX);

    char buf[16];

    int selected_entry = 0;
    if (config_get_value(buf, 0, 16, "DEFAULT_ENTRY")) {
        selected_entry = (int)strtoui(buf);
    }

    int timeout = 5;
    if (config_get_value(buf, 0, 16, "TIMEOUT")) {
        timeout = (int)strtoui(buf);
    }

    if (!timeout)
        goto autoboot;

    // If there is GRAPHICS config key and the value is "yes", enable graphics
    if (config_get_value(buf, 0, 16, "GRAPHICS") && !strcmp(buf, "yes")) {
        // default scheme
        int margin = 64;
        int margin_gradient = 20;
        uint32_t colourscheme[] = {
            0x00000000, // black
            0x00aa0000, // red
            0x0000aa00, // green
            0x00aa5500, // brown
            0x000000aa, // blue
            0x00aa00aa, // magenta
            0x0000aaaa, // cyan
            0x00aaaaaa  // grey
        };

        if (config_get_value(buf, 0, 16, "THEME_BLACK")) {
            colourscheme[0] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_RED")) {
            colourscheme[1] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_GREEN")) {
            colourscheme[2] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_BROWN")) {
            colourscheme[3] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_BLUE")) {
            colourscheme[4] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_MAGENTA")) {
            colourscheme[5] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_CYAN")) {
            colourscheme[6] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_GREY")) {
            colourscheme[7] = (int)strtoui16(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_MARGIN")) {
            margin = (int)strtoui(buf);
        }

        if (config_get_value(buf, 0, 16, "THEME_MARGIN_GRADIENT")) {
            margin_gradient = (int)strtoui(buf);
        }

        int bg_drive;
        if (!config_get_value(buf, 0, 16, "BACKGROUND_DRIVE")) {
            bg_drive = boot_drive;
        } else {
            bg_drive = (int)strtoui(buf);
        }
        int bg_part;
        if (!config_get_value(buf, 0, 16, "BACKGROUND_PARTITION")) {
            goto nobg;
        } else {
            bg_part = (int)strtoui(buf);
        }
        if (!config_get_value(cmdline, 0, CMDLINE_MAX, "BACKGROUND_PATH"))
            goto nobg;

        struct file_handle *bg_file = conv_mem_alloc(sizeof(struct file_handle));
        if (fopen(bg_file, bg_drive, bg_part, cmdline))
            goto nobg;

        struct image *bg = conv_mem_alloc(sizeof(struct image));
        if (open_image(bg, bg_file))
            goto nobg;

        term_vbe(colourscheme, margin, margin_gradient, bg);
        goto yesbg;

    nobg:
        term_vbe(colourscheme, margin, margin_gradient, NULL);

    yesbg:;
    }

    disable_cursor();
    bool skip_timeout = false;

refresh:
    clear(true);
    print("\n\n  \e[36m Limine " LIMINE_VERSION " \e[37m\n\n\n");

    print("Select an entry:\n\n");

    int max_entries;
    for (max_entries = 0; ; max_entries++) {
        if (config_get_entry_name(config_entry_name, max_entries, 1024) == -1)
            break;
        if (max_entries == selected_entry)
            print("  \e[47m\e[30m %s \e[40m\e[37m\n", config_entry_name);
        else
            print("   %s\n", config_entry_name);
    }

    if (max_entries == 0)
        panic("Config contains no entries.");

    print("\nArrows to choose, enter to select, 'e' to edit command line.");

    int c;

    if (skip_timeout == false) {
        print("\n\n");
        for (int i = timeout; i; i--) {
            print("\rBooting automatically in %u, press any key to stop the countdown...", i);
            if ((c = pit_sleep_and_quit_on_keypress(18))) {
                skip_timeout = true;
                print("\e[2K\r\e[2A");
                goto timeout_aborted;
            }
        }
        goto autoboot;
    }

    for (;;) {
        c = getchar();
timeout_aborted:
        switch (c) {
            case GETCHAR_CURSOR_UP:
                if (--selected_entry == -1)
                    selected_entry = max_entries - 1;
                goto refresh;
            case GETCHAR_CURSOR_DOWN:
                if (++selected_entry == max_entries)
                    selected_entry = 0;
                goto refresh;
            case '\r':
            autoboot:
                config_set_entry(selected_entry);
                enable_cursor();
                if (!config_get_value(cmdline, 0, CMDLINE_MAX, "KERNEL_CMDLINE")) {
                    if (!config_get_value(cmdline, 0, CMDLINE_MAX, "CMDLINE")) {
                        cmdline[0] = '\0';
                    }
                }
                clear(true);
                return cmdline;
            case 'e':
                config_set_entry(selected_entry);
                enable_cursor();
                if (!config_get_value(cmdline, 0, CMDLINE_MAX, "KERNEL_CMDLINE")) {
                    if (!config_get_value(cmdline, 0, CMDLINE_MAX, "CMDLINE")) {
                        cmdline[0] = '\0';
                    }
                }
                print("\n\n> ");
                gets(cmdline, cmdline, CMDLINE_MAX);
                clear(true);
                return cmdline;
        }
    }
}