#include "pgzint.h"

PG_MODULE_MAGIC;

struct png_memory
{
  char *buffer;
  size_t size;
};

void write_png_data(png_structp png_ptr, png_bytep data, png_size_t length)
{
    struct png_memory* p = (struct png_memory *)png_get_io_ptr(png_ptr);
    size_t newsize = p->size + length;
    if(p->buffer) {
        char *tmpptr = palloc0(newsize);
        memcpy(tmpptr, p->buffer, p->size);
        p->buffer = tmpptr;
    }
    else {
        p->buffer = palloc0(newsize);
    }
  
    if(!p->buffer) {
        ereport(ERROR, (errmsg("Error allocating PNG write buffer.")));
    }
    memcpy(p->buffer + p->size, data, length);
    p->size += length;
}

unsigned char *uchar_p_from_text_p(text *input)
{
    size_t len = VARSIZE(input) - VARHDRSZ;
    unsigned char *new = palloc0(len + 1);
    memcpy(new, VARDATA(input), len);
    new[len] = '\0';
    return new;
}

void validate_and_set_input(text *input, char *dest)
{
    size_t len = VARSIZE(input) - VARHDRSZ;
    if (len > strlen(dest))
      len = strlen(dest);
    memcpy(dest, VARDATA(input), len);
    dest[len] = '\0';
}

bytea *png_from_barcode(struct zint_symbol *input)
{
    bytea *result;
    png_structp png_ptr = NULL;
    png_infop info_ptr = NULL;
    int x, y, pos = 0;
    png_byte ** row_pointers = NULL;
    int pixel_size = 3;
    int depth = 8;
    struct png_memory image;
    image.buffer = 0;
    image.size = 0;

    png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
    if (png_ptr == NULL) {
        return 0;
    }

    info_ptr = png_create_info_struct (png_ptr);
    if (info_ptr == NULL) {
        png_destroy_write_struct(&png_ptr, &info_ptr);
        return 0;
    }

    if (setjmp(png_jmpbuf(png_ptr))) {
        png_destroy_write_struct(&png_ptr, &info_ptr);
        return 0;
    }

    png_set_IHDR(png_ptr,
                 info_ptr,
                 input->bitmap_width,
                 input->bitmap_height,
                 depth,
                 PNG_COLOR_TYPE_RGB,
                 PNG_INTERLACE_NONE,
                 PNG_COMPRESSION_TYPE_DEFAULT,
                 PNG_FILTER_TYPE_DEFAULT);
    
    row_pointers = png_malloc(png_ptr, input->bitmap_height * sizeof(png_byte *));
    for (y = 0; y < input->bitmap_height; y++) {
        png_byte *row = png_malloc(png_ptr, sizeof(uint8_t) * input->bitmap_width * pixel_size);
        row_pointers[y] = row;
        for (x = 0; x < input->bitmap_width; x++) {
            *row++ = input->bitmap[pos];
            *row++ = input->bitmap[pos + 1];
            *row++ = input->bitmap[pos + 2];
            pos += 3;
        }
    }

    png_set_write_fn(png_ptr, &image, write_png_data, NULL);
    png_set_rows(png_ptr, info_ptr, row_pointers);
    png_write_png(png_ptr, info_ptr, PNG_TRANSFORM_IDENTITY, NULL);

    // free up the memory used by libpng
    for (y = 0; y < input->bitmap_height; y++) {
        png_free(png_ptr, row_pointers[y]);
    }
    png_free(png_ptr, row_pointers);

    result = (bytea *) palloc0(VARHDRSZ + image.size);
    SET_VARSIZE(result, VARHDRSZ + image.size);
    // need to skip the header..
    memcpy(&result[1], image.buffer, image.size);

    return result;
}

PG_FUNCTION_INFO_V1(bc_generate);
Datum
bc_generate(PG_FUNCTION_ARGS)
{
    bytea *result;
    struct zint_symbol *barcode;
    unsigned char *input;
    int error = 0;
    int rotation_angle = 0;
    barcode = ZBarcode_Create();
    barcode->input_mode = UNICODE_MODE;

    if (PG_ARGISNULL(0)) {
        ZBarcode_Delete(barcode);
        ereport(ERROR, (errmsg("Error creating barcode"),
                        errdetail("No input text provided"),
                        errhint("Input text is required to generate a barcode")));
    }
    else
       input = uchar_p_from_text_p(PG_GETARG_TEXT_P(0));

    if (!PG_ARGISNULL(1)) {
        int symbology = PG_GETARG_INT32(1);
        if (symbology >= 1) {
            barcode->symbology = symbology;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid symbology provided: %d", symbology),
                            errhint("Symbology must be greater than or equal to 1. "
                                    "This is not checked against the list of symbologies so the caller must ensure validity.")));
        }
    }

    if (!PG_ARGISNULL(2)) {
        int height = PG_GETARG_INT32(2);
        if (height >= 0) {
            barcode->height = height;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid height provided: %d", height),
                            errhint("Scale must be greater than or equal to 0")));
        }
    }

    if (!PG_ARGISNULL(3)) {
        float8 scale = PG_GETARG_FLOAT8(3);
        if (scale >= 0.01) {
            barcode->scale = scale;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid scale provided: %f", scale),
                            errhint("Scale must be greater than or equal to 0.01")));
        }
    }

    if (!PG_ARGISNULL(4)) {
        int whitespace_width = PG_GETARG_INT32(4);
        if (whitespace_width >= 0 && whitespace_width <= 1000) {
            barcode->whitespace_width = whitespace_width;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid whitespace width provided: %d", whitespace_width),
                            errhint("Whitespace width must be between 0 to 1000")));
        }
    }

    if (!PG_ARGISNULL(5)) {
        int border_width = PG_GETARG_INT32(5);
        if (border_width >= 0 && border_width <= 1000) {
            barcode->border_width = border_width;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid border width provided: %d", border_width),
                            errhint("Border width must be between 0 to 1000")));
        }
    }

    if (!PG_ARGISNULL(6)) {
        int output_options = PG_GETARG_INT32(6);
        if (output_options >= 0) {
            barcode->output_options = output_options;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid output option provided: %d", output_options),
                            errhint("Output options must be greater than or equal to 0")));
        }
    }

    if (!PG_ARGISNULL(7))
        validate_and_set_input(PG_GETARG_TEXT_P(7), barcode->fgcolour);
    else
        strcpy(barcode->fgcolour, "000000");

    if (!PG_ARGISNULL(8))
        validate_and_set_input(PG_GETARG_TEXT_P(8), barcode->bgcolour);
    else
        strcpy(barcode->bgcolour, "FFFFFF");

    if (!PG_ARGISNULL(9) && PG_GETARG_BOOL(9))
        barcode->show_hrt = 1;
    else
        barcode->show_hrt = 0;

    if (!PG_ARGISNULL(10)) {
        int option_1 = PG_GETARG_INT32(10);
        if (option_1 >= 0) {
            barcode->option_1 = option_1;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid option provided: %d", option_1),
                            errhint("Option 1 must be greater than or equal to 0")));
        }
    }

    if (!PG_ARGISNULL(11)) {
        int option_2 = PG_GETARG_INT32(11);
        if (option_2 >= 0) {
            barcode->option_2 = option_2;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid option provided: %d", option_2),
                            errhint("Option 2 must be greater than or equal to 0")));
        }
    }

    if (!PG_ARGISNULL(12)) {
        int option_3 = PG_GETARG_INT32(12);
        if (option_3 >= 0) {
            barcode->option_3 = option_3;
        }
        else {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid option provided: %d", option_3),
                            errhint("Option 3 must be greater than or equal to 0")));
        }
    }

    if (!PG_ARGISNULL(13)) {
        rotation_angle = PG_GETARG_INT32(13);
        if (rotation_angle != 0 && rotation_angle != 90 && rotation_angle != 180 && rotation_angle != 270) {
            ZBarcode_Delete(barcode);
            ereport(ERROR, (errmsg("Error creating barcode"),
                            errdetail("Invalid rotation angle provided: %d", rotation_angle),
                            errhint("Rotation angle must be either 0, 90, 180, or 270, defaults to 0")));
        }
    }

    error = ZBarcode_Encode_and_Buffer(barcode, input, 0, rotation_angle);
    // zint changed ZWARN_INVALID_OPTION to ZINT_WARN_INVALID_OPTION
    // both values are 2, so to keep compatible I just use the magic number
    if (error >= 2) {
        ZBarcode_Delete(barcode);
        ereport(ERROR, (errmsg("Zint Error: %s", barcode->errtxt)));
    }

    result = png_from_barcode(barcode);
    ZBarcode_Delete(barcode);
    PG_RETURN_BYTEA_P(result);
}