markly 0.1.0

data/ext/markly/markly.c

@@ -0,0 +1,1226 @@
+#include "markly.h"
+#include "cmark-gfm.h"
+#include "houdini.h"
+#include "node.h"
+#include "registry.h"
+#include "parser.h"
+#include "syntax_extension.h"
+#include "cmark-gfm-core-extensions.h"
+
+static VALUE rb_Markly;
+static VALUE rb_Markly_Error;
+static VALUE rb_Markly_Node;
+static VALUE rb_Markly_Parser;
+
+static VALUE sym_document;
+static VALUE sym_blockquote;
+static VALUE sym_list;
+static VALUE sym_list_item;
+static VALUE sym_code_block;
+static VALUE sym_html;
+static VALUE sym_paragraph;
+static VALUE sym_header;
+static VALUE sym_hrule;
+static VALUE sym_text;
+static VALUE sym_softbreak;
+static VALUE sym_linebreak;
+static VALUE sym_code;
+static VALUE sym_inline_html;
+static VALUE sym_emph;
+static VALUE sym_strong;
+static VALUE sym_link;
+static VALUE sym_image;
+static VALUE sym_footnote_reference;
+static VALUE sym_footnote_definition;
+
+static VALUE sym_bullet_list;
+static VALUE sym_ordered_list;
+
+static VALUE sym_left;
+static VALUE sym_right;
+static VALUE sym_center;
+
+static void rb_Markly_Node_free(void *data) {
+	// It's important that the `free` function does not inspect the node data, as it may be part of a tree that was already freed.
+	cmark_node *node = (cmark_node*)data;
+	
+	if (cmark_node_parent(node) == NULL) {
+		cmark_node_free(node);
+	}
+}
+
+static void rb_Markly_Node_mark(void *data) {
+	cmark_node *node = data;
+	
+	// Mark the parent to make sure that the tree won't be freed as long as a child node is referenced.
+	cmark_node *parent = cmark_node_parent(node);
+	if (parent) {
+		void *user_data = cmark_node_get_user_data(parent);
+		rb_gc_mark((VALUE)user_data);
+	}
+	
+	// Mark all children to make sure their cached Ruby objects won't be freed.
+	for (cmark_node *child = cmark_node_first_child(node); child != NULL; child = cmark_node_next(child)) {
+		void *user_data = cmark_node_get_user_data(child);
+		
+		if (user_data) {
+			rb_gc_mark((VALUE)user_data);
+		}
+	}
+}
+
+static const rb_data_type_t rb_Markly_Node_Type = {
+	.wrap_struct_name = "Markly::Node",
+	.function = {
+		.dmark = rb_Markly_Node_mark,
+		.dfree = rb_Markly_Node_free,
+	},
+	.data = NULL,
+	.flags = RUBY_TYPED_FREE_IMMEDIATELY,
+};
+
+static VALUE rb_Markly_Node_wrap(cmark_node *node) {
+	if (node == NULL)
+		return Qnil;
+	
+	void * user_data = cmark_node_get_user_data(node);
+	
+	if (user_data) {
+		return (VALUE)user_data;
+	}
+	
+	VALUE self = TypedData_Wrap_Struct(rb_Markly_Node, &rb_Markly_Node_Type, node);
+	cmark_node_set_user_data(node, (void *)self);
+	
+	return self;
+}
+
+static void rb_Markly_Parser_free(void *data) {
+	cmark_parser_free(data);
+}
+
+static void rb_Markly_Parser_mark(void *data) {
+	cmark_parser *parser = data;
+	
+	// Mark the parent to make sure that the tree won't be freed as long as a child node is referenced.
+	cmark_node *root = parser->root;
+	
+	if (root) {
+		void *user_data = cmark_node_get_user_data(root);
+		rb_gc_mark((VALUE)user_data);
+	}
+}
+
+static const rb_data_type_t rb_Markly_Parser_Type = {
+	.wrap_struct_name = "Markly::Parser",
+	.function = {
+		.dmark = rb_Markly_Parser_mark,
+		.dfree = rb_Markly_Parser_free,
+	},
+	.data = NULL,
+	.flags = RUBY_TYPED_FREE_IMMEDIATELY,
+};
+
+static VALUE rb_Markly_Parser_root(VALUE self) {
+	cmark_parser *parser = NULL;
+	
+	TypedData_Get_Struct(self, cmark_parser, &rb_Markly_Parser_Type, parser);
+	
+	assert(parser != NULL);
+	
+	return rb_Markly_Node_wrap(parser->root);
+}
+
+static VALUE rb_Markly_Parser_alloc(VALUE self) {
+	return TypedData_Wrap_Struct(self, &rb_Markly_Parser_Type, NULL);
+}
+
+static VALUE rb_Markly_Parser_initialize(VALUE self, VALUE flags) {
+	Check_Type(flags, T_FIXNUM);
+	
+	cmark_mem *mem = cmark_get_default_mem_allocator();
+	cmark_parser *parser = cmark_parser_new_with_mem(NUM2INT(flags), mem);
+	
+	RTYPEDDATA_DATA(self) = parser;
+	
+	return self;
+}
+
+static VALUE rb_Markly_Parser_enable(VALUE self, VALUE extension) {
+	cmark_parser *parser = NULL;
+	
+	Check_Type(extension, T_SYMBOL);
+	
+	VALUE extension_name = rb_sym2str(extension);
+	
+	TypedData_Get_Struct(self, cmark_parser, &rb_Markly_Parser_Type, parser);
+	
+	cmark_syntax_extension *syntax_extension =
+		cmark_find_syntax_extension(StringValueCStr(extension_name));
+		
+	if (!syntax_extension) {
+		rb_raise(rb_eArgError, "extension %s not found", StringValueCStr(extension_name));
+	}
+	
+	cmark_parser_attach_syntax_extension(parser, syntax_extension);
+	
+	return Qnil;
+}
+
+static VALUE rb_Markly_Parser_parse(VALUE self, VALUE text) {
+	cmark_parser *parser = NULL;
+	
+	StringValue(text);
+	
+	TypedData_Get_Struct(self, cmark_parser, &rb_Markly_Parser_Type, parser);
+	
+	cmark_parser_feed(parser, RSTRING_PTR(text), RSTRING_LEN(text));
+	
+	cmark_node *root = cmark_parser_finish(parser);
+	
+	return rb_Markly_Node_wrap(root);
+}
+
+/*
+ * Internal: Creates a node based on a node type.
+ *
+ * type -  A {Symbol} representing the node to be created. Must be one of the
+ * following:
+ * - `:document`
+ * - `:blockquote`
+ * - `:list`
+ * - `:list_item`
+ * - `:code_block`
+ * - `:html`
+ * - `:paragraph`
+ * - `:header`
+ * - `:hrule`
+ * - `:text`
+ * - `:softbreak`
+ * - `:linebreak`
+ * - `:code`
+ * - `:inline_html`
+ * - `:emph`
+ * - `:strong`
+ * - `:link`
+ * - `:image`
+ */
+static VALUE rb_node_new(VALUE self, VALUE type) {
+  cmark_node_type node_type = 0;
+  cmark_node *node;
+
+  Check_Type(type, T_SYMBOL);
+
+  if (type == sym_document)
+    node_type = CMARK_NODE_DOCUMENT;
+  else if (type == sym_blockquote)
+    node_type = CMARK_NODE_BLOCK_QUOTE;
+  else if (type == sym_list)
+    node_type = CMARK_NODE_LIST;
+  else if (type == sym_list_item)
+    node_type = CMARK_NODE_ITEM;
+  else if (type == sym_code_block)
+    node_type = CMARK_NODE_CODE_BLOCK;
+  else if (type == sym_html)
+    node_type = CMARK_NODE_HTML;
+  else if (type == sym_paragraph)
+    node_type = CMARK_NODE_PARAGRAPH;
+  else if (type == sym_header)
+    node_type = CMARK_NODE_HEADER;
+  else if (type == sym_hrule)
+    node_type = CMARK_NODE_HRULE;
+  else if (type == sym_text)
+    node_type = CMARK_NODE_TEXT;
+  else if (type == sym_softbreak)
+    node_type = CMARK_NODE_SOFTBREAK;
+  else if (type == sym_linebreak)
+    node_type = CMARK_NODE_LINEBREAK;
+  else if (type == sym_code)
+    node_type = CMARK_NODE_CODE;
+  else if (type == sym_inline_html)
+    node_type = CMARK_NODE_INLINE_HTML;
+  else if (type == sym_emph)
+    node_type = CMARK_NODE_EMPH;
+  else if (type == sym_strong)
+    node_type = CMARK_NODE_STRONG;
+  else if (type == sym_link)
+    node_type = CMARK_NODE_LINK;
+  else if (type == sym_image)
+    node_type = CMARK_NODE_IMAGE;
+  else if (type == sym_footnote_reference)
+    node_type = CMARK_NODE_FOOTNOTE_REFERENCE;
+  else if (type == sym_footnote_definition)
+    node_type = CMARK_NODE_FOOTNOTE_DEFINITION;
+  else
+    rb_raise(rb_Markly_Error, "invalid node of type %d", node_type);
+
+  node = cmark_node_new(node_type);
+  if (node == NULL) {
+    rb_raise(rb_Markly_Error, "could not create node of type %d", node_type);
+  }
+
+  return rb_Markly_Node_wrap(node);
+}
+
+static VALUE encode_utf8_string(const char *c_string) {
+  VALUE string = rb_str_new2(c_string);
+  int enc = rb_enc_find_index("UTF-8");
+  rb_enc_associate_index(string, enc);
+  return string;
+}
+
+/*
+ * Public: Fetch the string contents of the node.
+ *
+ * Returns a {String}.
+ */
+static VALUE rb_node_get_string_content(VALUE self) {
+  const char *text;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  text = cmark_node_get_literal(node);
+  if (text == NULL) {
+    rb_raise(rb_Markly_Error, "could not get string content");
+  }
+
+  return encode_utf8_string(text);
+}
+
+/*
+ * Public: Sets the string content of the node.
+ *
+ * string - A {String} containing new content.
+ *
+ * Raises Error if the string content can't be set.
+ */
+static VALUE rb_node_set_string_content(VALUE self, VALUE s) {
+  char *text;
+  cmark_node *node;
+  Check_Type(s, T_STRING);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  text = StringValueCStr(s);
+
+  if (!cmark_node_set_literal(node, text)) {
+    rb_raise(rb_Markly_Error, "could not set string content");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Fetches the list type of the node.
+ *
+ * Returns a {Symbol} representing the node's type.
+ */
+static VALUE rb_node_get_type(VALUE self) {
+  int node_type;
+  cmark_node *node;
+  VALUE symbol;
+  const char *s;
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  node_type = cmark_node_get_type(node);
+  symbol = Qnil;
+
+  switch (node_type) {
+  case CMARK_NODE_DOCUMENT:
+    symbol = sym_document;
+    break;
+  case CMARK_NODE_BLOCK_QUOTE:
+    symbol = sym_blockquote;
+    break;
+  case CMARK_NODE_LIST:
+    symbol = sym_list;
+    break;
+  case CMARK_NODE_ITEM:
+    symbol = sym_list_item;
+    break;
+  case CMARK_NODE_CODE_BLOCK:
+    symbol = sym_code_block;
+    break;
+  case CMARK_NODE_HTML:
+    symbol = sym_html;
+    break;
+  case CMARK_NODE_PARAGRAPH:
+    symbol = sym_paragraph;
+    break;
+  case CMARK_NODE_HEADER:
+    symbol = sym_header;
+    break;
+  case CMARK_NODE_HRULE:
+    symbol = sym_hrule;
+    break;
+  case CMARK_NODE_TEXT:
+    symbol = sym_text;
+    break;
+  case CMARK_NODE_SOFTBREAK:
+    symbol = sym_softbreak;
+    break;
+  case CMARK_NODE_LINEBREAK:
+    symbol = sym_linebreak;
+    break;
+  case CMARK_NODE_CODE:
+    symbol = sym_code;
+    break;
+  case CMARK_NODE_INLINE_HTML:
+    symbol = sym_inline_html;
+    break;
+  case CMARK_NODE_EMPH:
+    symbol = sym_emph;
+    break;
+  case CMARK_NODE_STRONG:
+    symbol = sym_strong;
+    break;
+  case CMARK_NODE_LINK:
+    symbol = sym_link;
+    break;
+  case CMARK_NODE_IMAGE:
+    symbol = sym_image;
+    break;
+  case CMARK_NODE_FOOTNOTE_REFERENCE:
+    symbol = sym_footnote_reference;
+    break;
+  case CMARK_NODE_FOOTNOTE_DEFINITION:
+    symbol = sym_footnote_definition;
+    break;
+  default:
+    if (node->extension) {
+      s = node->extension->get_type_string_func(node->extension, node);
+      return ID2SYM(rb_intern(s));
+    }
+    rb_raise(rb_Markly_Error, "invalid node type %d", node_type);
+  }
+
+  return symbol;
+}
+
+/*	
+ * Public: Fetches the source_position of the node.
+ *
+ * Returns a {Hash} containing {Symbol} keys of the positions.
+ */
+static VALUE rb_node_get_source_position(VALUE self) {
+  int start_line, start_column, end_line, end_column;
+  VALUE result;
+
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  start_line = cmark_node_get_start_line(node);
+  start_column = cmark_node_get_start_column(node);
+  end_line = cmark_node_get_end_line(node);
+  end_column = cmark_node_get_end_column(node);
+
+  result = rb_hash_new();
+  rb_hash_aset(result, CSTR2SYM("start_line"), INT2NUM(start_line));
+  rb_hash_aset(result, CSTR2SYM("start_column"), INT2NUM(start_column));
+  rb_hash_aset(result, CSTR2SYM("end_line"), INT2NUM(end_line));
+  rb_hash_aset(result, CSTR2SYM("end_column"), INT2NUM(end_column));
+
+  return result;
+}
+
+/*
+ * Public: Returns the type of the current pointer as a string.
+ *
+ * Returns a {String}.
+ */
+static VALUE rb_node_get_type_string(VALUE self) {
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  return rb_str_new2(cmark_node_get_type_string(node));
+}
+
+/*
+ * Internal: Unlinks the node from the tree (fixing pointers in
+ * parents and siblings appropriately).
+ */
+static VALUE rb_node_unlink(VALUE self) {
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  cmark_node_unlink(node);
+
+  return Qnil;
+}
+
+/* Public: Fetches the first child of the node.
+ *
+ * Returns a {Node} if a child exists, `nil` otherise.
+ */
+static VALUE rb_node_first_child(VALUE self) {
+  cmark_node *node, *child;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  child = cmark_node_first_child(node);
+
+  return rb_Markly_Node_wrap(child);
+}
+
+/* Public: Fetches the next sibling of the node.
+ *
+ * Returns a {Node} if a sibling exists, `nil` otherwise.
+ */
+static VALUE rb_node_next(VALUE self) {
+  cmark_node *node, *next;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  next = cmark_node_next(node);
+
+  return rb_Markly_Node_wrap(next);
+}
+
+/*
+ * Public: Inserts a node as a sibling before the current node.
+ *
+ * sibling - A sibling {Node} to insert.
+ *
+ * Returns `true` if successful.
+ * Raises Error if the node can't be inserted.
+ */
+static VALUE rb_node_insert_before(VALUE self, VALUE sibling) {
+  cmark_node *node1, *node2;
+	TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node1);
+	TypedData_Get_Struct(sibling, cmark_node, &rb_Markly_Node_Type, node2);
+
+  if (!cmark_node_insert_before(node1, node2)) {
+    rb_raise(rb_Markly_Error, "could not insert before");
+  }
+
+  return Qtrue;
+}
+
+/* Internal: Convert the node to an HTML string.
+ *
+ * Returns a {String}.
+ */
+static VALUE rb_render_html(VALUE self, VALUE rb_options, VALUE rb_extensions) {
+  int options, extensions_len;
+  VALUE rb_ext_name;
+  int i;
+  cmark_node *node;
+  cmark_llist *extensions = NULL;
+  cmark_mem *mem = cmark_get_default_mem_allocator();
+  Check_Type(rb_options, T_FIXNUM);
+  Check_Type(rb_extensions, T_ARRAY);
+
+  options = FIX2INT(rb_options);
+  extensions_len = RARRAY_LEN(rb_extensions);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  for (i = 0; i < extensions_len; ++i) {
+    rb_ext_name = RARRAY_PTR(rb_extensions)[i];
+
+    if (!SYMBOL_P(rb_ext_name)) {
+      cmark_llist_free(mem, extensions);
+      rb_raise(rb_eTypeError, "extension names should be Symbols; got a %"PRIsVALUE"", rb_obj_class(rb_ext_name));
+    }
+
+    cmark_syntax_extension *syntax_extension =
+      cmark_find_syntax_extension(rb_id2name(SYM2ID(rb_ext_name)));
+
+    if (!syntax_extension) {
+      cmark_llist_free(mem, extensions);
+      rb_raise(rb_eArgError, "extension %s not found\n", rb_id2name(SYM2ID(rb_ext_name)));
+    }
+
+    extensions = cmark_llist_append(mem, extensions, syntax_extension);
+  }
+
+  char *html = cmark_render_html(node, options, extensions);
+  VALUE ruby_html = rb_str_new2(html);
+
+  cmark_llist_free(mem, extensions);
+  free(html);
+
+  return ruby_html;
+}
+
+/* Internal: Convert the node to a CommonMark string.
+ *
+ * Returns a {String}.
+ */
+static VALUE rb_render_commonmark(int argc, VALUE *argv, VALUE self) {
+  VALUE rb_options, rb_width;
+  rb_scan_args(argc, argv, "11", &rb_options, &rb_width);
+
+  int width = 120;
+  if (!NIL_P(rb_width)) {
+    Check_Type(rb_width, T_FIXNUM);
+    width = FIX2INT(rb_width);
+  }
+
+  int options;
+  cmark_node *node;
+  Check_Type(rb_options, T_FIXNUM);
+
+  options = FIX2INT(rb_options);
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  char *cmark = cmark_render_commonmark(node, options, width);
+  VALUE ruby_cmark = rb_str_new2(cmark);
+  free(cmark);
+
+  return ruby_cmark;
+}
+
+/* Internal: Convert the node to a plain textstring.
+ *
+ * Returns a {String}.
+ */
+static VALUE rb_render_plaintext(int argc, VALUE *argv, VALUE self) {
+  VALUE rb_options, rb_width;
+  rb_scan_args(argc, argv, "11", &rb_options, &rb_width);
+
+  int width = 120;
+  if (!NIL_P(rb_width)) {
+    Check_Type(rb_width, T_FIXNUM);
+    width = FIX2INT(rb_width);
+  }
+
+  int options;
+  cmark_node *node;
+  Check_Type(rb_options, T_FIXNUM);
+
+  options = FIX2INT(rb_options);
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  char *text = cmark_render_plaintext(node, options, width);
+  VALUE ruby_text = rb_str_new2(text);
+  free(text);
+
+  return ruby_text;
+}
+
+/*
+ * Public: Inserts a node as a sibling after the current node.
+ *
+ * sibling - A sibling {Node} to insert.
+ *
+ * Returns `true` if successful.
+ * Raises Error if the node can't be inserted.
+ */
+static VALUE rb_node_insert_after(VALUE self, VALUE sibling) {
+  cmark_node *node1, *node2;
+	TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node1);
+	TypedData_Get_Struct(sibling, cmark_node, &rb_Markly_Node_Type, node2);
+
+  if (!cmark_node_insert_after(node1, node2)) {
+    rb_raise(rb_Markly_Error, "could not insert after");
+  }
+
+  return Qtrue;
+}
+
+/*
+ * Public: Inserts a node as the first child of the current node.
+ *
+ * child - A child {Node} to insert.
+ *
+ * Returns `true` if successful.
+ * Raises Error if the node can't be inserted.
+ */
+static VALUE rb_node_prepend_child(VALUE self, VALUE child) {
+  cmark_node *node1, *node2;
+	TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node1);
+	TypedData_Get_Struct(child, cmark_node, &rb_Markly_Node_Type, node2);
+
+  if (!cmark_node_prepend_child(node1, node2)) {
+    rb_raise(rb_Markly_Error, "could not prepend child");
+  }
+
+  return Qtrue;
+}
+
+/*
+ * Public: Inserts a node as the last child of the current node.
+ *
+ * child - A child {Node} to insert.
+ *
+ * Returns `true` if successful.
+ * Raises Error if the node can't be inserted.
+ */
+static VALUE rb_node_append_child(VALUE self, VALUE child) {
+  cmark_node *node1, *node2;
+	TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node1);
+	TypedData_Get_Struct(child, cmark_node, &rb_Markly_Node_Type, node2);
+
+  if (!cmark_node_append_child(node1, node2)) {
+    rb_raise(rb_Markly_Error, "could not append child");
+  }
+
+  return Qtrue;
+}
+
+/* Public: Fetches the first child of the current node.
+ *
+ * Returns a {Node} if a child exists, `nil` otherise.
+ */
+static VALUE rb_node_last_child(VALUE self) {
+  cmark_node *node, *child;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  child = cmark_node_last_child(node);
+
+  return rb_Markly_Node_wrap(child);
+}
+
+/* Public: Fetches the parent of the current node.
+ *
+ * Returns a {Node} if a parent exists, `nil` otherise.
+ */
+static VALUE rb_node_parent(VALUE self) {
+  cmark_node *node, *parent;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  parent = cmark_node_parent(node);
+
+  return rb_Markly_Node_wrap(parent);
+}
+
+/* Public: Fetches the previous sibling of the current node.
+ *
+ * Returns a {Node} if a parent exists, `nil` otherise.
+ */
+static VALUE rb_node_previous(VALUE self) {
+  cmark_node *node, *previous;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  previous = cmark_node_previous(node);
+
+  return rb_Markly_Node_wrap(previous);
+}
+
+/*
+ * Public: Gets the URL of the current node (must be a `:link` or `:image`).
+ *
+ * Returns a {String}.
+ * Raises a Error if the URL can't be retrieved.
+ */
+static VALUE rb_node_get_url(VALUE self) {
+  const char *text;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  text = cmark_node_get_url(node);
+  if (text == NULL) {
+    rb_raise(rb_Markly_Error, "could not get url");
+  }
+
+  return rb_str_new2(text);
+}
+
+/*
+ * Public: Sets the URL of the current node (must be a `:link` or `:image`).
+ *
+ * url - A {String} representing the new URL
+ *
+ * Raises a Error if the URL can't be set.
+ */
+static VALUE rb_node_set_url(VALUE self, VALUE url) {
+  cmark_node *node;
+  char *text;
+  Check_Type(url, T_STRING);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  text = StringValueCStr(url);
+
+  if (!cmark_node_set_url(node, text)) {
+    rb_raise(rb_Markly_Error, "could not set url");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Gets the title of the current node (must be a `:link` or `:image`).
+ *
+ * Returns a {String}.
+ * Raises a Error if the title can't be retrieved.
+ */
+static VALUE rb_node_get_title(VALUE self) {
+  const char *text;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  text = cmark_node_get_title(node);
+  if (text == NULL) {
+    rb_raise(rb_Markly_Error, "could not get title");
+  }
+
+  return rb_str_new2(text);
+}
+
+/*
+ * Public: Sets the title of the current node (must be a `:link` or `:image`).
+ *
+ * title - A {String} representing the new title
+ *
+ * Raises a Error if the title can't be set.
+ */
+static VALUE rb_node_set_title(VALUE self, VALUE title) {
+  char *text;
+  cmark_node *node;
+  Check_Type(title, T_STRING);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  text = StringValueCStr(title);
+
+  if (!cmark_node_set_title(node, text)) {
+    rb_raise(rb_Markly_Error, "could not set title");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Gets the header level of the current node (must be a `:header`).
+ *
+ * Returns a {Number} representing the header level.
+ * Raises a Error if the header level can't be retrieved.
+ */
+static VALUE rb_node_get_header_level(VALUE self) {
+  int header_level;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  header_level = cmark_node_get_header_level(node);
+
+  if (header_level == 0) {
+    rb_raise(rb_Markly_Error, "could not get header_level");
+  }
+
+  return INT2NUM(header_level);
+}
+
+/*
+ * Public: Sets the header level of the current node (must be a `:header`).
+ *
+ * level - A {Number} representing the new header level
+ *
+ * Raises a Error if the header level can't be set.
+ */
+static VALUE rb_node_set_header_level(VALUE self, VALUE level) {
+  int l;
+  cmark_node *node;
+  Check_Type(level, T_FIXNUM);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  l = FIX2INT(level);
+
+  if (!cmark_node_set_header_level(node, l)) {
+    rb_raise(rb_Markly_Error, "could not set header_level");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Gets the list type of the current node (must be a `:list`).
+ *
+ * Returns a {Symbol}.
+ * Raises a Error if the title can't be retrieved.
+ */
+static VALUE rb_node_get_list_type(VALUE self) {
+  int list_type;
+  cmark_node *node;
+  VALUE symbol;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  list_type = cmark_node_get_list_type(node);
+
+  if (list_type == CMARK_BULLET_LIST) {
+    symbol = sym_bullet_list;
+  } else if (list_type == CMARK_ORDERED_LIST) {
+    symbol = sym_ordered_list;
+  } else {
+    rb_raise(rb_Markly_Error, "could not get list_type");
+  }
+
+  return symbol;
+}
+
+/*
+ * Public: Sets the list type of the current node (must be a `:list`).
+ *
+ * level - A {Symbol} representing the new list type
+ *
+ * Raises a Error if the list type can't be set.
+ */
+static VALUE rb_node_set_list_type(VALUE self, VALUE list_type) {
+  int type = 0;
+  cmark_node *node;
+  Check_Type(list_type, T_SYMBOL);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  if (list_type == sym_bullet_list) {
+    type = CMARK_BULLET_LIST;
+  } else if (list_type == sym_ordered_list) {
+    type = CMARK_ORDERED_LIST;
+  } else {
+    rb_raise(rb_Markly_Error, "invalid list_type");
+  }
+
+  if (!cmark_node_set_list_type(node, type)) {
+    rb_raise(rb_Markly_Error, "could not set list_type");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Gets the starting number the current node (must be an
+ * `:ordered_list`).
+ *
+ * Returns a {Number} representing the starting number.
+ * Raises a Error if the starting number can't be retrieved.
+ */
+static VALUE rb_node_get_list_start(VALUE self) {
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  if (cmark_node_get_type(node) != CMARK_NODE_LIST ||
+      cmark_node_get_list_type(node) != CMARK_ORDERED_LIST) {
+    rb_raise(rb_Markly_Error, "can't get list_start for non-ordered list %d",
+             cmark_node_get_list_type(node));
+  }
+
+  return INT2NUM(cmark_node_get_list_start(node));
+}
+
+/*
+ * Public: Sets the starting number of the current node (must be an
+ * `:ordered_list`).
+ *
+ * level - A {Number} representing the new starting number
+ *
+ * Raises a Error if the starting number can't be set.
+ */
+static VALUE rb_node_set_list_start(VALUE self, VALUE start) {
+  int s;
+  cmark_node *node;
+  Check_Type(start, T_FIXNUM);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  s = FIX2INT(start);
+
+  if (!cmark_node_set_list_start(node, s)) {
+    rb_raise(rb_Markly_Error, "could not set list_start");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Gets the tight status the current node (must be a `:list`).
+ *
+ * Returns a `true` if the list is tight, `false` otherwise.
+ * Raises a Error if the starting number can't be retrieved.
+ */
+static VALUE rb_node_get_list_tight(VALUE self) {
+  int flag;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  if (cmark_node_get_type(node) != CMARK_NODE_LIST) {
+    rb_raise(rb_Markly_Error, "can't get list_tight for non-list");
+  }
+
+  flag = cmark_node_get_list_tight(node);
+
+  return flag ? Qtrue : Qfalse;
+}
+
+/*
+ * Public: Sets the tight status of the current node (must be a `:list`).
+ *
+ * tight - A {Boolean} representing the new tightness
+ *
+ * Raises a Error if the tightness can't be set.
+ */
+static VALUE rb_node_set_list_tight(VALUE self, VALUE tight) {
+  int t;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  t = RTEST(tight);
+
+  if (!cmark_node_set_list_tight(node, t)) {
+    rb_raise(rb_Markly_Error, "could not set list_tight");
+  }
+
+  return Qnil;
+}
+
+/*
+ * Public: Gets the fence info of the current node (must be a `:code_block`).
+ *
+ * Returns a {String} representing the fence info.
+ * Raises a Error if the fence info can't be retrieved.
+ */
+static VALUE rb_node_get_fence_info(VALUE self) {
+  const char *fence_info;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  fence_info = cmark_node_get_fence_info(node);
+
+  if (fence_info == NULL) {
+    rb_raise(rb_Markly_Error, "could not get fence_info");
+  }
+
+  return rb_str_new2(fence_info);
+}
+
+/*
+ * Public: Sets the fence info of the current node (must be a `:code_block`).
+ *
+ * info - A {String} representing the new fence info
+ *
+ * Raises a Error if the fence info can't be set.
+ */
+static VALUE rb_node_set_fence_info(VALUE self, VALUE info) {
+  char *text;
+  cmark_node *node;
+  Check_Type(info, T_STRING);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  text = StringValueCStr(info);
+
+  if (!cmark_node_set_fence_info(node, text)) {
+    rb_raise(rb_Markly_Error, "could not set fence_info");
+  }
+
+  return Qnil;
+}
+
+static VALUE rb_node_get_tasklist_item_checked(VALUE self) {
+  int tasklist_state;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  tasklist_state = cmark_gfm_extensions_get_tasklist_item_checked(node);
+
+  if (tasklist_state == 1) {
+    return Qtrue;
+  } else {
+    return Qfalse;
+  }
+}
+
+/*
+ * Public: Sets the checkbox state of the current node (must be a `:tasklist`).
+ *
+ * item_checked - A {Boolean} representing the new checkbox state
+ *
+ * Returns a {Boolean} representing the new checkbox state.
+ * Raises a Error if the checkbox state can't be set.
+ */
+static VALUE rb_node_set_tasklist_item_checked(VALUE self, VALUE item_checked) {
+  int tasklist_state;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+  tasklist_state = RTEST(item_checked);
+
+  if (!cmark_gfm_extensions_set_tasklist_item_checked(node, tasklist_state)) {
+    rb_raise(rb_Markly_Error, "could not set tasklist_item_checked");
+  };
+
+  if (tasklist_state) {
+    return Qtrue;
+  } else {
+    return Qfalse;
+  }
+}
+
+// TODO: remove this, superseded by the above method
+static VALUE rb_node_get_tasklist_state(VALUE self) {
+  int tasklist_state;
+  cmark_node *node;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  tasklist_state = cmark_gfm_extensions_get_tasklist_item_checked(node);
+
+  if (tasklist_state == 1) {
+    return rb_str_new2("checked");
+  } else {
+    return rb_str_new2("unchecked");
+  }
+}
+
+static VALUE rb_node_get_table_alignments(VALUE self) {
+  uint16_t column_count, i;
+  uint8_t *alignments;
+  cmark_node *node;
+  VALUE ary;
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  column_count = cmark_gfm_extensions_get_table_columns(node);
+  alignments = cmark_gfm_extensions_get_table_alignments(node);
+
+  if (!column_count || !alignments) {
+    rb_raise(rb_Markly_Error, "could not get column_count or alignments");
+  }
+
+  ary = rb_ary_new();
+  for (i = 0; i < column_count; ++i) {
+    if (alignments[i] == 'l')
+      rb_ary_push(ary, sym_left);
+    else if (alignments[i] == 'c')
+      rb_ary_push(ary, sym_center);
+    else if (alignments[i] == 'r')
+      rb_ary_push(ary, sym_right);
+    else
+      rb_ary_push(ary, Qnil);
+  }
+  return ary;
+}
+
+/* Internal: Escapes href URLs safely. */
+static VALUE rb_html_escape_href(VALUE self, VALUE rb_text) {
+  char *result;
+  cmark_node *node;
+  Check_Type(rb_text, T_STRING);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  cmark_mem *mem = cmark_node_mem(node);
+  cmark_strbuf buf = CMARK_BUF_INIT(mem);
+
+  if (houdini_escape_href(&buf, (const uint8_t *)RSTRING_PTR(rb_text),
+                          RSTRING_LEN(rb_text))) {
+    result = (char *)cmark_strbuf_detach(&buf);
+    return rb_str_new2(result);
+  }
+
+  return rb_text;
+}
+
+/* Internal: Escapes HTML content safely. */
+static VALUE rb_html_escape_html(VALUE self, VALUE rb_text) {
+  char *result;
+  cmark_node *node;
+  Check_Type(rb_text, T_STRING);
+
+  TypedData_Get_Struct(self, cmark_node, &rb_Markly_Node_Type, node);
+
+  cmark_mem *mem = cmark_node_mem(node);
+  cmark_strbuf buf = CMARK_BUF_INIT(mem);
+
+  if (houdini_escape_html0(&buf, (const uint8_t *)RSTRING_PTR(rb_text),
+                           RSTRING_LEN(rb_text), 0)) {
+    result = (char *)cmark_strbuf_detach(&buf);
+    return rb_str_new2(result);
+  }
+
+  return rb_text;
+}
+
+VALUE rb_Markly_extensions(VALUE self) {
+  cmark_llist *exts, *it;
+  cmark_syntax_extension *ext;
+  VALUE ary = rb_ary_new();
+
+  cmark_mem *mem = cmark_get_default_mem_allocator();
+  exts = cmark_list_syntax_extensions(mem);
+  for (it = exts; it; it = it->next) {
+    ext = it->data;
+    rb_ary_push(ary, rb_str_new2(ext->name));
+  }
+	
+  cmark_llist_free(mem, exts);
+
+  return ary;
+}
+
+__attribute__((visibility("default"))) void Init_markly() {
+  sym_document = ID2SYM(rb_intern("document"));
+  sym_blockquote = ID2SYM(rb_intern("blockquote"));
+  sym_list = ID2SYM(rb_intern("list"));
+  sym_list_item = ID2SYM(rb_intern("list_item"));
+  sym_code_block = ID2SYM(rb_intern("code_block"));
+  sym_html = ID2SYM(rb_intern("html"));
+  sym_paragraph = ID2SYM(rb_intern("paragraph"));
+  sym_header = ID2SYM(rb_intern("header"));
+  sym_hrule = ID2SYM(rb_intern("hrule"));
+  sym_text = ID2SYM(rb_intern("text"));
+  sym_softbreak = ID2SYM(rb_intern("softbreak"));
+  sym_linebreak = ID2SYM(rb_intern("linebreak"));
+  sym_code = ID2SYM(rb_intern("code"));
+  sym_inline_html = ID2SYM(rb_intern("inline_html"));
+  sym_emph = ID2SYM(rb_intern("emph"));
+  sym_strong = ID2SYM(rb_intern("strong"));
+  sym_link = ID2SYM(rb_intern("link"));
+  sym_image = ID2SYM(rb_intern("image"));
+  sym_footnote_reference = ID2SYM(rb_intern("footnote_reference"));
+  sym_footnote_definition = ID2SYM(rb_intern("footnote_definition"));
+  
+  sym_bullet_list = ID2SYM(rb_intern("bullet_list"));
+  sym_ordered_list = ID2SYM(rb_intern("ordered_list"));
+  
+  sym_left = ID2SYM(rb_intern("left"));
+  sym_right = ID2SYM(rb_intern("right"));
+  sym_center = ID2SYM(rb_intern("center"));
+  
+  rb_Markly = rb_define_module("Markly");
+  rb_define_singleton_method(rb_Markly, "extensions", rb_Markly_extensions, 0);
+  
+  rb_Markly_Error = rb_define_class_under(rb_Markly, "Error", rb_eStandardError);
+  rb_define_singleton_method(rb_Markly_Node, "parse", rb_Markly_Parser_parse, 1);
+  
+  rb_Markly_Parser = rb_define_class_under(rb_Markly, "Parser", rb_cObject);
+	rb_define_alloc_func(rb_Markly_Parser, rb_Markly_Parser_alloc);
+	rb_define_method(rb_Markly_Parser, "initialize", rb_Markly_Parser_initialize, 1);
+	rb_define_method(rb_Markly_Parser, "enable", rb_Markly_Parser_enable, 1);
+	rb_define_method(rb_Markly_Parser, "parse", rb_Markly_Parser_parse, 1);
+	rb_define_method(rb_Markly_Parser, "root", rb_Markly_Parser_root, 0);
+	
+  rb_Markly_Node = rb_define_class_under(rb_Markly, "Node", rb_cObject);
+  rb_define_singleton_method(rb_Markly_Node, "new", rb_node_new, 1);
+  rb_define_method(rb_Markly_Node, "string_content", rb_node_get_string_content, 0);
+  rb_define_method(rb_Markly_Node, "string_content=", rb_node_set_string_content, 1);
+  rb_define_method(rb_Markly_Node, "type", rb_node_get_type, 0);
+  rb_define_method(rb_Markly_Node, "type_string", rb_node_get_type_string, 0);
+  rb_define_method(rb_Markly_Node, "source_position", rb_node_get_source_position, 0);
+  rb_define_method(rb_Markly_Node, "delete", rb_node_unlink, 0);
+  rb_define_method(rb_Markly_Node, "first_child", rb_node_first_child, 0);
+  rb_define_method(rb_Markly_Node, "next", rb_node_next, 0);
+  rb_define_method(rb_Markly_Node, "insert_before", rb_node_insert_before, 1);
+  rb_define_method(rb_Markly_Node, "_render_html", rb_render_html, 2);
+  rb_define_method(rb_Markly_Node, "_render_commonmark", rb_render_commonmark, -1);
+  rb_define_method(rb_Markly_Node, "_render_plaintext", rb_render_plaintext, -1);
+  rb_define_method(rb_Markly_Node, "insert_after", rb_node_insert_after, 1);
+  rb_define_method(rb_Markly_Node, "prepend_child", rb_node_prepend_child, 1);
+  rb_define_method(rb_Markly_Node, "append_child", rb_node_append_child, 1);
+  rb_define_method(rb_Markly_Node, "last_child", rb_node_last_child, 0);
+  rb_define_method(rb_Markly_Node, "parent", rb_node_parent, 0);
+  rb_define_method(rb_Markly_Node, "previous", rb_node_previous, 0);
+  rb_define_method(rb_Markly_Node, "url", rb_node_get_url, 0);
+  rb_define_method(rb_Markly_Node, "url=", rb_node_set_url, 1);
+  rb_define_method(rb_Markly_Node, "title", rb_node_get_title, 0);
+  rb_define_method(rb_Markly_Node, "title=", rb_node_set_title, 1);
+  rb_define_method(rb_Markly_Node, "header_level", rb_node_get_header_level, 0);
+  rb_define_method(rb_Markly_Node, "header_level=", rb_node_set_header_level, 1);
+  rb_define_method(rb_Markly_Node, "list_type", rb_node_get_list_type, 0);
+  rb_define_method(rb_Markly_Node, "list_type=", rb_node_set_list_type, 1);
+  rb_define_method(rb_Markly_Node, "list_start", rb_node_get_list_start, 0);
+  rb_define_method(rb_Markly_Node, "list_start=", rb_node_set_list_start, 1);
+  rb_define_method(rb_Markly_Node, "list_tight", rb_node_get_list_tight, 0);
+  rb_define_method(rb_Markly_Node, "list_tight=", rb_node_set_list_tight, 1);
+  rb_define_method(rb_Markly_Node, "fence_info", rb_node_get_fence_info, 0);
+  rb_define_method(rb_Markly_Node, "fence_info=", rb_node_set_fence_info, 1);
+  rb_define_method(rb_Markly_Node, "table_alignments", rb_node_get_table_alignments, 0);
+  rb_define_method(rb_Markly_Node, "tasklist_state", rb_node_get_tasklist_state, 0);
+  rb_define_method(rb_Markly_Node, "tasklist_item_checked?", rb_node_get_tasklist_item_checked, 0);
+  rb_define_method(rb_Markly_Node, "tasklist_item_checked=", rb_node_set_tasklist_item_checked, 1);
+
+  rb_define_method(rb_Markly_Node, "html_escape_href", rb_html_escape_href, 1);
+  rb_define_method(rb_Markly_Node, "html_escape_html", rb_html_escape_html, 1);
+
+  cmark_gfm_core_extensions_ensure_registered();
+}