42sh/src/repl/repl.c

/*
** EPITECH PROJECT, 2025
** __
** File description:
** _
*/

#include <stdio.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>

#include "ast.h"
#include "common.h"
#include "debug.h"
#include "git.h"
#include "repl.h"
#include "repl/key_handler.h"
#include "u_str.h"
#include "vt100_esc_codes.h"

const key_handler_t KEY_HANDLERS[] = {
    {"\03", handle_key_ctrl_c},  // ^C
    {"\04", handle_key_ctrl_d},  // ^D
    {"\014", handle_key_ctrl_l}, // ^L
    {"\05", handle_key_ctrl_e}, // ^E
    {"\01", handle_key_ctrl_a}, // ^A
    {"\02", handle_key_ctrl_b}, // ^B
    {"\06", handle_key_ctrl_f}, // ^F
    {"\027", handle_key_ctrl_w}, // ^W
    {"\031", handle_key_ctrl_y}, // ^Y
    {"\022", handle_key_ctrl_r}, // ^R
    {"\013", handle_key_ctrl_k}, // ^K
    {ESC "[A", handle_key_arrow_up},
    {ESC "[B", handle_key_arrow_down},
    {ESC "[C", handle_key_arrow_right},
    {ESC "[D", handle_key_arrow_left},
    {"\t", handle_key_tab},
    {"\x7f", handle_backspace},
    {"\x1b[3~", handle_delete},
};

void print_second_shell_prompt(exec_ctx_t *ec)
{
    env_t *env_ptr = ec->env;
    char const *ps1 = get_env_value(env_ptr, "PS1");
    char const *username = get_env_value(env_ptr, "USER");

    if (username == nullptr)
        username = "?";
    if (ps1 == nullptr) {
        printf(BLUE "└─[" PURPLE "%s%s" BLUE "] " RESET,
            ec->history->last_exit_code == 0 ? "" : RED,
            strcmp(username, "root") == 0 ? "#" : "$");
        ec->prompt_len = 6;
    } else {
        printf("%s", ps1);
        ec->prompt_len = strlen(ps1);
    }
}

static
void print_git_prompt(git_status_t *gs)
{
    if (!gs->ahead && !gs->behind)
        printf(BLUE " [" RED "%s" BLUE "] " RESET "-", gs->branch);
    if (gs->ahead)
        printf(BLUE " [" RED "%s" GREEN " +" BLUE "] " RESET "-", gs->branch);
    else if (gs->behind)
        printf(BLUE " [" RED "%s" RED " -" BLUE "] " RESET "-", gs->branch);
}

static
void print_prompt(env_t *env_ptr, char *hostname, exec_ctx_t *ec)
{
    char const *username = get_env_value(env_ptr, "USER");
    git_status_t gs = {0};

    if (username == nullptr)
        username = "?";
    printf(BLUE PROMPT_HEADER GREEN "%s" RESET "@" CYAN "%s" BLUE "] "
        RESET "-" BLUE " [" RESET "%s" BLUE
        "] " RESET "-",
        username,
        hostname,
        get_env_value(env_ptr, "PWD"));
    if (get_git_status(&gs))
        print_git_prompt(&gs);
    printf(BLUE " [" YELLOW "%d" BLUE "]\n└─[" PURPLE "%s%s" BLUE "] " RESET,
        ec->history_command->sz + 1,
        ec->history->last_exit_code == 0 ? "" : RED,
        strcmp(username, "root") == 0 ? "#" : "$");
}

void print_shell_prompt(exec_ctx_t *ec)
{
    env_t *env_ptr = ec->env;
    char const *ps1 = get_env_value(env_ptr, "PS1");
    char hostname[64];

    if (ec->precmd != nullptr)
        visitor(ec->precmd, ec);
    if (ps1 == nullptr) {
        if (gethostname(hostname, 64) < 0)
            return;
        print_prompt(env_ptr, hostname, ec);
        ec->prompt_len = 6;
    } else {
        printf("%s", ps1);
        ec->prompt_len = strlen(ps1);
    }
}

bool is_sequence(char *read_buff)
{
    for (size_t i = 0; i < sizeof KEY_HANDLERS / sizeof *KEY_HANDLERS; i++) {
        if (strncmp(read_buff, KEY_HANDLERS[i].name,
            strlen(KEY_HANDLERS[i].name)) == 0)
            return true;
    }
    return false;
}

ssize_t handle_keys(
    readline_helper_t *rh,
    buff_t *buff,
    char const *read_buff,
    size_t)
{
    if (!rh->ec->isatty)
        return 0;
    for (size_t i = 0; i < sizeof KEY_HANDLERS / sizeof *KEY_HANDLERS; i++) {
        if (strncmp(read_buff, KEY_HANDLERS[i].name,
                strlen(KEY_HANDLERS[i].name)) != 0)
            continue;
        if (!KEY_HANDLERS[i].exec(rh, rh->ec, buff))
            return strlen(KEY_HANDLERS[i].name);
        return -1;
    }
    return 0;
}