#include <serial.h>
#include <common.h>

extern EFI_SYSTEM_TABLE *ST;

serial_context g_serial;

void serial_init(EFI_SERIAL_IO_PROTOCOL *SerialIo) {
    g_serial.SerialIo = SerialIo;
    uefi_call_wrapper(SerialIo->Reset, 1, SerialIo);
}

void serial_write_char(char c) {
    if (!g_serial.SerialIo) {
        uefi_call_wrapper(ST->ConOut->OutputString, 2, ST->ConOut, (CHAR16*)L"serial: null io\n");
        return;
    }
    UINTN size = 1;
    EFI_STATUS status = uefi_call_wrapper(g_serial.SerialIo->Write, 3, g_serial.SerialIo, &size, &c);
    if (status != EFI_SUCCESS) {
        uefi_call_wrapper(ST->ConOut->OutputString, 2, ST->ConOut, (CHAR16*)L"serial: write failed\n");
    }
}

void serial_write(const char *str) {
    if (!g_serial.SerialIo) {
        uefi_call_wrapper(ST->ConOut->OutputString, 2, ST->ConOut, (CHAR16*)L"serial: null io\n");
        return;
    }
    while (*str) {
        serial_write_char(*str++);
    }
}

void serial_write_hex(UINTN val) {
    char buf[19];
    buf[0] = '0'; buf[1] = 'x';
    for (SSINT32 i = 17; i >= 2; i--) {
        UINTN digit = val & 0xF;
        buf[i] = digit < 10 ? '0' + digit : 'A' + digit - 10;
        val >>= 4;
    }
    buf[18] = '\0';
    serial_write(buf);
}

char serial_read_char() {
    // 后面可能用的上，比如远程调试？
    if (!g_serial.SerialIo) return 0;
    char c = 0;
    UINTN size = 1;
    EFI_STATUS status = uefi_call_wrapper(g_serial.SerialIo->Read, 3, g_serial.SerialIo, &size, &c);
    if (status != EFI_SUCCESS || size == 0) return 0;
    return c;
}