- require 'ffi'
- require 'stringio'
- require 'cgi'
- module CMark
- extend FFI::Library
- ffi_lib ['libcmark', 'cmark']
- typedef :pointer, :node
- enum :node_type, [:document, :blockquote, :list, :list_item,
- :fenced_code, :indented_code, :html, :paragraph,
- :atx_header, :setext_header, :hrule, :reference_def,
- :str, :softbreak, :linebreak, :code, :inline_html,
- :emph, :strong, :link, :image]
- attach_function :cmark_free_nodes, [:node], :void
- attach_function :cmark_node_unlink, [:node], :void
- attach_function :cmark_markdown_to_html, [:string, :int], :string
- attach_function :cmark_parse_document, [:string, :int], :node
- attach_function :cmark_node_first_child, [:node], :node
- attach_function :cmark_node_parent, [:node], :node
- attach_function :cmark_node_next, [:node], :node
- attach_function :cmark_node_previous, [:node], :node
- attach_function :cmark_node_get_type, [:node], :node_type
- attach_function :cmark_node_get_string_content, [:node], :string
- end
- class Node
- attr_accessor :type, :children, :string_content, :pointer
- def initialize(pointer)
- if pointer.null?
- return nil
- end
- @pointer = pointer
- @type = CMark::cmark_node_get_type(pointer)
- @children = []
- first_child = CMark::cmark_node_first_child(pointer)
- b = first_child
- while !b.null?
- @children << Node.new(b)
- b = CMark::cmark_node_next(b)
- end
- @string_content = CMark::cmark_node_get_string_content(pointer)
- if @type == :document
- self.free
- end
- end
- def self.from_markdown(s)
- len = s.bytes.length
- Node.new(CMark::cmark_parse_document(s, len))
- end
- def free
- CMark::cmark_free_nodes(@pointer)
- end
- end
- class Renderer
- def initialize(stream = nil)
- if stream
- @stream = stream
- @stringwriter = false
- else
- @stringwriter = true
- @stream = StringIO.new
- end
- end
- def outf(format, *args)
- @stream.printf(format, *args)
- end
- def out(arg)
- @stream.puts(arg)
- end
- def render(node)
- case node.type
- when :document
- self.document(node.children)
- if @stringwriter
- @stream.string
- end
- when :paragraph
- self.paragraph(node.children)
- when :str
- self.str(node.string_content)
- else
- # raise "unimplemented " + node.type.to_s
- end
- end
- def document(children)
- children.each { |x| render(x) }
- end
- def paragraph(children)
- children.each { |x| render(x) }
- end
- def str(content)
- self.out(content)
- end
- end
- class HtmlRenderer < Renderer
- def paragraph(children)
- self.out("<p>")
- children.each { |x| render(x) }
- self.out("</p>\n")
- end
- def str(content)
- self.out(CGI.escapeHTML(content))
- end
- end
- doc = Node.from_markdown(STDIN.read())
- renderer = HtmlRenderer.new(STDOUT)
- renderer.render(doc)
- # def markdown_to_html(s)
- # len = s.bytes.length
- # CMark::cmark_markdown_to_html(s, len)
- # end
- # print markdown_to_html(STDIN.read())
|