#include <stdio.h> /* fopen, fprintf, fread, printf */
#include <string.h> /* memcpy */

#include <cairo.h>
#include <gdk/gdk.h>
#include <gdk-pixbuf/gdk-pixbuf.h> /* includes glib.h */
#include <svgtiny.h>

/*
 * The width and height of the viewport that we'll render the SVG
 * into. The final "picture" may not actually be this size; based on
 * the height, width, viewBox, and preserveAspectRatio attributes in
 * the SVG itself, libsvgtiny may scale, stretch, offset, etc. the
 * paths to make them fit nicely into the viewport.
 */
#define VIEWPORT_WIDTH 512
#define VIEWPORT_HEIGHT 512

#define SVG_BUFFER_INCREMENT (size_t)4194304

/* Convenient typedefs for libsvgtiny */
typedef struct svgtiny_diagram diagram_t;
typedef struct svgtiny_shape shape_t;


typedef struct {
  GdkPixbufModuleUpdatedFunc  updated_func;
  GdkPixbufModulePreparedFunc prepared_func;
  GdkPixbufModuleSizeFunc     size_func;
  gpointer                    user_data;

  /* The "file" */
  char*                       svg_data;
  size_t                      svg_data_size;
  size_t                      svg_data_max;

} SvgTinyContext;


/**
 * @brief Render an svgtiny path using cairo.
 *
 * @param cr
 *   A pointer to a valid cairo context.
 *
 * @param path
 *   A pointer to an svgtiny shape that will be rendered on the
 *   cairo context's target surface.
 */
static void render_path(cairo_t* cr, shape_t* path) {
  unsigned int j;

  cairo_new_path(cr);
  for (j = 0; j != path->path_length; ) {
    switch ((int) path->path[j]) {
    case svgtiny_PATH_MOVE:
      cairo_move_to(cr,
		    path->path[j + 1],
		    path->path[j + 2]);
      j += 3;
      break;
    case svgtiny_PATH_CLOSE:
      cairo_close_path(cr);
      j += 1;
      break;
    case svgtiny_PATH_LINE:
      cairo_line_to(cr,
                    path->path[j + 1],
                    path->path[j + 2]);
      j += 3;
      break;
    case svgtiny_PATH_BEZIER:
      cairo_curve_to(cr,
                     path->path[j + 1],
                     path->path[j + 2],
                     path->path[j + 3],
                     path->path[j + 4],
                     path->path[j + 5],
                     path->path[j + 6]);
      j += 7;
    }
  }
  if (path->fill != svgtiny_TRANSPARENT) {
    cairo_set_source_rgba(cr,
                          svgtiny_RED(path->fill) / 255.0,
                          svgtiny_GREEN(path->fill) / 255.0,
                          svgtiny_BLUE(path->fill) / 255.0,
                          1);
    cairo_fill_preserve(cr);
  }
  if (path->stroke != svgtiny_TRANSPARENT) {
    cairo_set_source_rgba(cr,
                          svgtiny_RED(path->stroke) / 255.0,
                          svgtiny_GREEN(path->stroke) / 255.0,
                          svgtiny_BLUE(path->stroke) / 255.0,
                          1);
    cairo_set_line_width(cr, path->stroke_width);
    cairo_stroke_preserve(cr);
  }
}

/**
 * @brief Parse a buffer of SVG data into a diagram_t structure.
 *
 * @param buffer
 *   The buffer containing the SVG document.
 *
 * @param bytecount
 *   The number of bytes in @c buffer.
 *
 * @return If successful, a pointer to a @c diagram_t structure is
 *   returned; if not, @c NULL is returned. You are expected to @c
 *   svgtiny_free the result if it is valid.
 */
static diagram_t* svgtiny_diagram_from_buffer(char* buffer,
					      size_t bytecount,
					      int width,
					      int height,
					      GError** error) {
  diagram_t* diagram;
  svgtiny_code code;

  diagram = svgtiny_create();
  if (!diagram) {
    g_set_error_literal(error,
                        GDK_PIXBUF_ERROR,
                        GDK_PIXBUF_ERROR_FAILED,
                        "svgtiny_create() failed");
    return NULL;
  }

  code = svgtiny_parse(diagram, buffer, bytecount, "", width, height);
  free(buffer);

  if (code != svgtiny_OK) {
    switch (code) {
    case svgtiny_OUT_OF_MEMORY:
      g_set_error_literal(error,
                          G_FILE_ERROR,
                          G_FILE_ERROR_NOMEM,
                          "out of memory in svgtiny_parse()");
      break;
    case svgtiny_LIBDOM_ERROR:
      g_set_error_literal(error,
                          GDK_PIXBUF_ERROR,
                          GDK_PIXBUF_ERROR_CORRUPT_IMAGE,
                          "libdom error in svgtiny_parse()");
      break;
    case svgtiny_NOT_SVG:
      g_set_error_literal(error,
                          GDK_PIXBUF_ERROR,
                          GDK_PIXBUF_ERROR_CORRUPT_IMAGE,
                          "encountered svgtiny_NOT_SVG in svgtiny_parse()");
      break;
    case svgtiny_SVG_ERROR:
      g_set_error(error,
                  GDK_PIXBUF_ERROR,
                  GDK_PIXBUF_ERROR_CORRUPT_IMAGE,
                  "SVG error in svgtiny_parse() on line %i: %s",
                  diagram->error_line,
                  diagram->error_message);
      break;
    }
    return NULL;
  }

  return diagram;
}

/**
 * @brief Create a cairo context from a libsvgtiny diagram.
 *
 * @param diagram
 *   A pointer to a valid libsvgtiny diagram.
 *
 * @return If successful, a pointer to a @c cairo_t context structure
 *   is returned; if not, @c NULL is returned. You are expected to @c
 *   cairo_destroy the result if it is valid.
 */
static cairo_t* cairo_context_from_diagram(diagram_t* diagram) {
  cairo_t* cr;
  cairo_surface_t* surface;
  cairo_status_t crs;
  unsigned int i;

  surface = cairo_image_surface_create(CAIRO_FORMAT_ARGB32,
                                       diagram->width,
                                       diagram->height);

  crs = cairo_surface_status(surface);
  if (crs != CAIRO_STATUS_SUCCESS) {
    fprintf(stderr,
            "cairo_image_surface_create failed: %s\n",
            cairo_status_to_string(crs));
    cairo_surface_destroy(surface);
    return NULL;
  }

  cr = cairo_create(surface);
  crs = cairo_status(cr);

  /* Immediately destroy the surface which is now accessible as
     cr->target */
  cairo_surface_destroy(surface);

  if (crs != CAIRO_STATUS_SUCCESS) {
    fprintf(stderr,
            "cairo_create failed: %s\n",
            cairo_status_to_string(crs));
    cairo_destroy(cr);
    return NULL;
  }

  cairo_set_source_rgba(cr, 0, 0, 0, 0);
  cairo_paint(cr);

  /* Loop through the shapes in the diagram... */
  for (i = 0; i != diagram->shape_count; i++) {

    /* If this shape is a path, just render it. */
    if (diagram->shape[i].path) {
      render_path(cr, &diagram->shape[i]);
    }

    /* If this shape is text... */
    if (diagram->shape[i].text) {
      /* Figure out what color to use from the R/G/B components of the
         shape's stroke. */
      cairo_set_source_rgba(cr,
                            svgtiny_RED(diagram->shape[i].stroke) / 255.0,
                            svgtiny_GREEN(diagram->shape[i].stroke) / 255.0,
                            svgtiny_BLUE(diagram->shape[i].stroke) / 255.0,
                            1);
      /* Then move to the actual position of the text within the
         shape... */
      cairo_move_to(cr,
                    diagram->shape[i].text_x,
                    diagram->shape[i].text_y);

      /* and draw it. */
      cairo_show_text(cr, diagram->shape[i].text);
    }
  }


  /* Check the status again just for good measure? */
  crs = cairo_status(cr);
  if (crs != CAIRO_STATUS_SUCCESS) {
    fprintf(stderr,
	    "cairo error: %s\n",
	    cairo_status_to_string(crs));
    cairo_destroy(cr);
    return NULL;
  }

  return cr;
}

/**
 * @brief Create a GdkPixbuf from a buffer of SVG data.
 *
 * @param buffer
 *   The buffer containing the SVG document.
 *
 * @param bytecount
 *   The number of bytes in @c buffer.
 *
 * @param error
 *   The address of a @c GError pointer that we use to return errors.
 *
 * @return If successful, a valid pointer to a @c GdkPixbuf is
 *   returned; if not, @c NULL is returned and @c error is populated.
 */
static GdkPixbuf* gdk_pixbuf_from_svg_buffer(char* buffer,
					     size_t bytecount,
					     GError** error) {
  diagram_t* diagram;
  cairo_t* cr = 0;
  GdkPixbuf* pb;
  GError* sub_error = NULL;

  diagram = svgtiny_diagram_from_buffer(buffer,
					bytecount,
					VIEWPORT_WIDTH,
					VIEWPORT_HEIGHT,
					&sub_error);
  if (!diagram) {
    g_propagate_error(error, sub_error);
    return NULL;
  }

  cr = cairo_context_from_diagram(diagram);
  if (!cr) {
    svgtiny_free(diagram);
    g_set_error_literal(error,
			GDK_PIXBUF_ERROR,
			GDK_PIXBUF_ERROR_CORRUPT_IMAGE,
			"could not create Cairo surface from SVG diagram");
    return NULL;
  }


  /* We're using the viewport width and height and not the diagram
   * width/height for the image. The diagram can be of a different
   * size and aspect ratio than the viewport, and our main use case is
   * for icons that are generally square and reasonably sized. If the
   * diagram is "small," then we want to scale it up until it fits
   * nicely in the viewport before rendering it. That's as opposed to
   * rendering the image small, and letting GDK scale it up. Of course
   * this reasoning makes the assumption that the viewport is usually
   * larger than the diagram.
  */
  pb = gdk_pixbuf_get_from_surface(cairo_get_target(cr),
				   0,
				   0,
				   VIEWPORT_WIDTH,
				   VIEWPORT_HEIGHT);


  if (!pb) {
    g_set_error_literal(error,
			GDK_PIXBUF_ERROR,
			GDK_PIXBUF_ERROR_FAILED,
			"failed to obtain a GdkPixbuf from Cairo surface");
  }

  return pb;
}


/**
 * @brief Create a GdkPixbuf from an SVG filestream.
 *
 * This is essentially a wrapper around @gdk_pixbuf_from_svg_buffer
 * that reads a @c FILE pointer into a buffer.
 *
 * @param fp
 *   A pointer to a @c FILE containing the SVG document.
 *
 * @param error
 *   The address of a @c GError pointer that we use to return errors.
 *
 * @return If successful, a valid pointer to a @c GdkPixbuf is
 *   returned; if not, @c NULL is returned and @c error is populated.
 */
static GdkPixbuf* gdk_pixbuf_from_svg_file_stream(FILE *fp, GError** error) {
  size_t bytecount, bytesread;
  char* buffer;

  /* Find the size of the file stream */
  fseek(fp, 0L, SEEK_END);
  bytecount = ftell(fp);
  rewind(fp);

  /* YOLO, no error checking */
  buffer = g_malloc(bytecount);

  bytesread = fread(buffer, 1, bytecount, fp);
  if (bytesread != bytecount) {
    g_set_error(error,
		G_FILE_ERROR,
		G_FILE_ERROR_FAILED,
		"read only %zd of %zd bytes from stream",
		bytesread,
		bytecount);
    return NULL;
  }

  return gdk_pixbuf_from_svg_buffer(buffer, bytecount, error);
}


static gpointer gdk_pixbuf_begin_load(GdkPixbufModuleSizeFunc size_func,
				      GdkPixbufModulePreparedFunc prep_func,
				      GdkPixbufModuleUpdatedFunc updated_func,
				      gpointer user_data,
				      GError **error) {

  SvgTinyContext* context = g_new(SvgTinyContext, 1);

  context->size_func      = size_func;
  context->prepared_func  = prep_func;
  context->updated_func   = updated_func;
  context->user_data      = user_data;

  /* YOLO, no error checking */
  context->svg_data       = g_malloc(SVG_BUFFER_INCREMENT);
  context->svg_data_size  = 0;

  return context;
}

static gboolean gdk_pixbuf_load_increment(gpointer data,
					  const guchar* buf,
					  guint size,
					  GError** error) {
  size_t increment = 0;
  SvgTinyContext* context = (SvgTinyContext*)data;

  if (context->svg_data_size + size > context->svg_data_max) {
    if (size > SVG_BUFFER_INCREMENT) {
      increment = size;
    }
    else {
      increment = SVG_BUFFER_INCREMENT;
    }

    /* YOLO, no error checking */
    context->svg_data = g_realloc(context->svg_data,
				  context->svg_data_max + increment);

    context->svg_data_max += increment;
  }

  memcpy(context->svg_data + context->svg_data_size, buf, size);
  context->svg_data_size += size;

  return TRUE;
}

static void emit_updated(SvgTinyContext* context, GdkPixbuf* pixbuf) {
  if (context->updated_func != NULL) {
    (*context->updated_func)(pixbuf,
			     0,
			     0,
			     gdk_pixbuf_get_width(pixbuf),
			     gdk_pixbuf_get_height(pixbuf),
			     context->user_data);
  }
}

static void emit_prepared(SvgTinyContext* context, GdkPixbuf* pixbuf) {
  if (context->prepared_func != NULL) {
    (*context->prepared_func)(pixbuf, NULL, context->user_data);
  }
}


/*
static void emit_size(SvgTinyContext* context, GdkPixbuf* pixbuf) {
  int w = gdk_pixbuf_get_width(pixbuf);
  int h = gdk_pixbuf_get_height(pixbuf);
  if (context->size_func != NULL) {
    (*context->size_func)(&w, &h, context->user_data);
  }
}
*/


static gboolean gdk_pixbuf_stop_load(gpointer data, GError **error) {
  SvgTinyContext* context = (SvgTinyContext*)data;
  GdkPixbuf* pixbuf = NULL;
  gboolean result = TRUE;
  GError* sub_error = NULL;

  pixbuf = gdk_pixbuf_from_svg_buffer(context->svg_data,
				      context->svg_data_size,
				      &sub_error);

  if (pixbuf != NULL) {
    /*emit_size(context, pixbuf);*/
    emit_prepared(context, pixbuf);
    emit_updated(context, pixbuf);
    g_object_unref(pixbuf);
  }
  else {
    g_propagate_error(error, sub_error);
    result = FALSE;
  }
  g_free(context);

  return result;
}


G_MODULE_EXPORT void fill_vtable(GdkPixbufModule* module);
void fill_vtable(GdkPixbufModule* module) {
  module->begin_load = gdk_pixbuf_begin_load;
  module->load_increment = gdk_pixbuf_load_increment;
  module->stop_load = gdk_pixbuf_stop_load;
  module->load = gdk_pixbuf_from_svg_file_stream;
}

G_MODULE_EXPORT void fill_info(GdkPixbufFormat *info);
void fill_info(GdkPixbufFormat* info) {
  /* Borrowed from librsvg-2.40.21 */
  static const GdkPixbufModulePattern signature[] = {
    {  " <svg",  "*    ", 100 },
    {  " <!DOCTYPE svg",  "*             ", 100 },
    { NULL, NULL, 0 }
  };

  /* I'm not sure if we should support gzipped svg here? */
  static const gchar *mime_types[] = {
    "image/svg+xml",
    "image/svg",
    "image/svg-xml",
    "image/vnd.adobe.svg+xml",
    "text/xml-svg",
    "image/svg+xml-compressed",
    NULL
  };

  static const gchar *extensions[] = {
    "svg",
    "svgz",
    "svg.gz",
    NULL
  };

  info->name        = "svg";
  info->signature   = (GdkPixbufModulePattern*)signature;
  info->description = "Scalable Vector Graphics";
  info->mime_types  = (gchar**)mime_types;
  info->extensions  = (gchar**)extensions;
  info->flags       = GDK_PIXBUF_FORMAT_SCALABLE;
  info->license     = "AGPL3";
}


int main(int argc, char** argv) {
  char* svgpath;
  char* pngpath;
  FILE* fp;
  GError* err = NULL;
  GdkPixbuf* pb;

  /* Parse arguments, and maybe print usage */
  if (argc < 3) {
    printf("Usage: %s INPUT OUTPUT\n", argv[0]);
    printf("Convert an SVG file (INPUT) to a PNG file (OUTPUT)\n");
    return 2;
  }

  svgpath = argv[1];
  pngpath = argv[2];

  fp = fopen(svgpath, "rb");
  if (!fp) {
    perror(svgpath);
    return 1;
  }

  pb = gdk_pixbuf_from_svg_file_stream(fp, &err);
  if (!pb) {
    fprintf(stderr,
            "Error %d in gdk_pixbuf_from_svg_file_stream: %s\n",
            err->code,
            err->message);
    g_error_free(err);
    return 1;
  }

  gdk_pixbuf_save(pb, pngpath, "png", NULL, NULL);
  g_object_unref(pb);
  return 0;
}