#include "moar.h"

/* Decodes the specified number of bytes of latin1 into an NFG string,

 * creating a result of the specified type. The type must have the MVMString

 * REPR. */

MVMString * MVM_string_latin1_decode(MVMThreadContext *tc, const MVMObject *result_type,

                                     char *latin1_c, size_t bytes) {

    MVMuint8  *latin1 = (MVMuint8 *)latin1_c;

    MVMString *result = (MVMString *)REPR(result_type)->allocate(tc, STABLE(result_type));

    size_t i, k, result_graphs;

    MVMuint8 writing_32bit = 0;

    result->body.storage_type   = MVM_STRING_GRAPHEME_8;

    result->body.storage.blob_8 = MVM_malloc(sizeof(MVMint8) * bytes);

    result_graphs = 0;

    for (i = 0; i < bytes; i++) {

        if (latin1[i] == '\r' && i + 1 < bytes && latin1[i + 1] == '\n') {

            if (writing_32bit)

                result->body.storage.blob_32[result_graphs++] = MVM_nfg_crlf_grapheme(tc);

            else

                result->body.storage.blob_8[result_graphs++] = MVM_nfg_crlf_grapheme(tc);

            i++;

        }

        else {

            if (latin1[i] > 127 && !writing_32bit) {

                MVMGrapheme8 *old_storage = result->body.storage.blob_8;

                result->body.storage.blob_32 = MVM_malloc(sizeof(MVMGrapheme32) * bytes);

                result->body.storage_type = MVM_STRING_GRAPHEME_32;

                writing_32bit = 1;

                for (k = 0; k < i; k++)

                    result->body.storage.blob_32[k] = old_storage[k];

                MVM_free(old_storage);

            }

            if (writing_32bit)

                result->body.storage.blob_32[result_graphs++] = latin1[i];

            else

                result->body.storage.blob_8[result_graphs++] = latin1[i];

        }

    }

    result->body.num_graphs = result_graphs;

    return result;

}

/* Decodes using a decodestream. Decodes as far as it can with the input

 * buffers, or until a stopper is reached. */

MVMuint32 MVM_string_latin1_decodestream(MVMThreadContext *tc, MVMDecodeStream *ds,

                                    const MVMint32 *stopper_chars,

                                    MVMDecodeStreamSeparators *seps) {

    MVMint32 count = 0, total = 0;

    MVMint32 bufsize;

    MVMGrapheme32 *buffer;

    MVMDecodeStreamBytes *cur_bytes;

    MVMDecodeStreamBytes *last_accept_bytes = ds->bytes_head;

    MVMint32 last_accept_pos, last_was_cr;

    MVMuint32 reached_stopper;

    /* If there's no buffers, we're done. */

    if (!ds->bytes_head)

        return 0;

    last_accept_pos = ds->bytes_head_pos;

    /* If we're asked for zero chars, also done. */

    if (stopper_chars && *stopper_chars == 0)

        return 1;

    /* Take length of head buffer as initial guess. */

    bufsize = ds->bytes_head->length;

    buffer = MVM_malloc(bufsize * sizeof(MVMGrapheme32));

    /* Decode each of the buffers. */

    cur_bytes = ds->bytes_head;

    last_was_cr = 0;

    reached_stopper = 0;

    while (cur_bytes) {

        /* Process this buffer. */

        MVMint32  pos = cur_bytes == ds->bytes_head ? ds->bytes_head_pos : 0;

        unsigned char *bytes = (unsigned char *)cur_bytes->bytes;

        while (pos < cur_bytes->length) {

            MVMCodepoint codepoint = bytes[pos++];

            MVMGrapheme32 graph;

            if (last_was_cr) {

                if (codepoint == '\n') {

                    graph = MVM_nfg_crlf_grapheme(tc);

                }

                else {

                    graph = '\r';

                    pos--;

                }

                last_was_cr = 0;

            }

            else if (codepoint == '\r') {

                last_was_cr = 1;

                continue;

            }

            else {

                graph = codepoint;

            }

            if (count == bufsize) {

                /* We filled the buffer. Attach this one to the buffers

                 * linked list, and continue with a new one. */

                MVM_string_decodestream_add_chars(tc, ds, buffer, bufsize);

                buffer = MVM_malloc(bufsize * sizeof(MVMGrapheme32));

                count = 0;

            }

            buffer[count++] = graph;

            last_accept_bytes = cur_bytes;

            last_accept_pos = pos;

            total++;

            if (stopper_chars && *stopper_chars == total) {

                reached_stopper = 1;

                goto done;

            }

            if (MVM_string_decode_stream_maybe_sep(tc, seps, codepoint)) {

                reached_stopper = 1;

                goto done;

            }

        }

        cur_bytes = cur_bytes->next;

    }

  done:

    /* Attach what we successfully parsed as a result buffer, and trim away

     * what we chewed through. */

    if (count) {

        MVM_string_decodestream_add_chars(tc, ds, buffer, count);

    }

    else {

        MVM_free(buffer);

    }

    MVM_string_decodestream_discard_to(tc, ds, last_accept_bytes, last_accept_pos);

    return reached_stopper;

}

/* Encodes the specified substring to latin-1. Anything outside of latin-1 range

 * will become a ?. The result string is NULL terminated, but the specified

 * size is the non-null part. */

char * MVM_string_latin1_encode_substr(MVMThreadContext *tc, MVMString *str, MVMuint64 *output_size, MVMint64 start, MVMint64 length,

        MVMString *replacement, MVMint32 translate_newlines) {

    /* Latin-1 is a single byte encoding, but \r\n is a 2-byte grapheme, so we

     * may have to resize as we go. */

    MVMuint32 startu = (MVMuint32)start;

    MVMStringIndex strgraphs = MVM_string_graphs(tc, str);

    MVMuint32 lengthu = (MVMuint32)(length == -1 ? strgraphs - startu : length);

    MVMuint8 *result;

    size_t result_alloc;

    MVMuint8 *repl_bytes = NULL;

    MVMuint64 repl_length;

    /* must check start first since it's used in the length check */

    if (start < 0 || start > strgraphs)

        MVM_exception_throw_adhoc(tc, "start out of range");

    if (length < -1 || start + lengthu > strgraphs)

        MVM_exception_throw_adhoc(tc, "length out of range");

    if (replacement)

        repl_bytes = (MVMuint8 *) MVM_string_latin1_encode_substr(tc,

            replacement, &repl_length, 0, -1, NULL, translate_newlines);

    result_alloc = lengthu;

    result = MVM_malloc(result_alloc + 1);

    if (str->body.storage_type == MVM_STRING_GRAPHEME_ASCII) {

        /* No encoding needed; directly copy. */

        memcpy(result, str->body.storage.blob_ascii, lengthu);

        result[lengthu] = 0;

        if (output_size)

            *output_size = lengthu;

    }

    else {

        MVMuint32 i = 0;

        MVMCodepointIter ci;

        MVM_string_ci_init(tc, &ci, str, translate_newlines);

        while (MVM_string_ci_has_more(tc, &ci)) {

            MVMCodepoint ord = MVM_string_ci_get_codepoint(tc, &ci);

            if (i == result_alloc) {

                result_alloc += 8;

                result = MVM_realloc(result, result_alloc + 1);

            }

            if (ord >= 0 && ord <= 255) {

                result[i] = (MVMuint8)ord;

                i++;

            }

            else if (replacement) {

                if (repl_length >= result_alloc || i >= result_alloc - repl_length) {

                    result_alloc += repl_length;

                    result = MVM_realloc(result, result_alloc + 1);

                }

                memcpy(result + i, repl_bytes, repl_length);

                i += repl_length;

            }

            else {

                MVM_free(result);

                MVM_free(repl_bytes);

                MVM_exception_throw_adhoc(tc,

                    "Error encoding Latin-1 string: could not encode codepoint %d",

                    ord);

            }

        }

        result[i] = 0;

        if (output_size)

            *output_size = i;

    }

    MVM_free(repl_bytes);

    return (char *)result;

}

/* Encodes the specified string to latin-1. Anything outside of latin-1 range

 * will become a ?. The result string is NULL terminated, but the specified

 * size is the non-null part. */

char * MVM_string_latin1_encode(MVMThreadContext *tc, MVMString *str, MVMuint64 *output_size,

        MVMint32 translate_newlines) {

    return MVM_string_latin1_encode_substr(tc, str, output_size, 0, -1, NULL, translate_newlines);

}