// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <printf.h>
#include <xefi.h>

#define PCBUFMAX 126
// buffer is two larger to leave room for a \0 and room
// for a \r that may be added after a \n

typedef struct {
    int off;
    char16_t buf[PCBUFMAX + 2];
} _pcstate;

static int _printf_console_out(const char* str, size_t len, void* _state) {
    _pcstate* state = _state;
    char16_t* buf = state->buf;
    int i = state->off;
    int n = len;
    while (n > 0) {
        if (*str == '\n') {
            buf[i++] = '\r';
        }
        buf[i++] = *str++;
        if (i >= PCBUFMAX) {
            buf[i] = 0;
            gConOut->OutputString(gConOut, buf);
            i = 0;
        }
        n--;
    }
    state->off = i;
    return len;
}

int _printf(const char* fmt, ...) {
    va_list ap;
    _pcstate state;
    int r;
    state.off = 0;
    va_start(ap, fmt);
    r = _printf_engine(_printf_console_out, &state, fmt, ap);
    va_end(ap);
    if (state.off) {
        state.buf[state.off] = 0;
        gConOut->OutputString(gConOut, state.buf);
    }
    return r;
}

int puts16(char16_t* str) {
    return gConOut->OutputString(gConOut, str) == EFI_SUCCESS ? 0 : -1;
}