markdown_json.c

/**********************************************************************

  markdown_json.c - JSON output for Markdown parser.
  Copyright (c) 2012 Fedor Sheremetyev

  This program is free software; you can redistribute it and/or modify
  it under the terms of the MIT license. See LICENSE for details.

 ***********************************************************************/

#include <assert.h>
#include "markdown_json.h"

void print_json_string(GString *out, char* str) {
  while (*str != '\0') {
    switch (*str) {
      case '"':
        g_string_append_printf(out, "\\\"");
        break;
      case '\\':
        g_string_append_printf(out, "\\\\");
        break;
      case '\n':
        g_string_append_printf(out, "\\n");
        break;
      case '\r':
        g_string_append_printf(out, "\\r");
        break;
      default:
        g_string_append_c(out, *str);
    }
    str++;
  }  
}

void print_indent(GString *out, int level) {
  int i;
  for (i = 0; i < level * 4; i++)
    g_string_append_printf(out, " ");
}

// we return constant strings only
char* key_name(int key) {
  switch (key) {

    #define KEY_CASE(name) case name: return #name;
    ELEMENT_LIST(KEY_CASE)
    #undef KEY_CASE

    default: return "?";
  }
}

void print_json_list(GString *out, element *elt, int level);

void print_json_element(GString *out, element *elt, int level) {
  char * key = key_name(elt->key);
  print_indent(out, level);
  if ( elt->key == LINK || elt->key == IMAGE || elt->key == IMAGEBLOCK ) {
    g_string_append_printf(out, "[\"%s\", { \"URL\": \"", key);
    print_json_string(out, elt->contents.link->url);
    g_string_append_printf(out, "\" }");
    print_json_list(out, elt->contents.link->label, level + 1);
    g_string_append_printf(out, "\n");
    print_indent(out, level);
    g_string_append_printf(out, "]");
  } else if ( elt->contents.str != NULL ) {
    g_string_append_printf(out, "[\"%s\", \"", key);
    print_json_string(out, elt->contents.str);
    g_string_append_printf(out, "\"]");
    assert(elt->children == NULL);
  } else {
    g_string_append_printf(out, "[\"%s\"", key);
    if (elt->children) {
      print_json_list(out, elt->children, level + 1);
      g_string_append_printf(out, "\n");
      print_indent(out, level);
    }
    g_string_append_printf(out, "]", key);
  }
}

void print_json_list(GString *out, element *elt, int level) {
  while (elt != NULL) {
    if (elt->key == NOTELABEL || elt->key == REFERENCE ||
        (elt->key == NOTE && elt->contents.str != NULL)) {
      elt = elt->next;
      continue;
    }

    g_string_append_printf(out, ",\n");
    print_json_element(out, elt, level);
    elt = elt->next;
  }
}

void print_json_tree(GString *out, element *root) {
  g_string_append_printf(out, "[\"MARKDOWN\"");
  print_json_list(out, root, 1);
  g_string_append_printf(out, "\n]\n");
}