require 'escape_utils'

module CommonMarker
  class HtmlRenderer < Renderer
    def render(node)
      result = super(node)
    end

    def header(node)
      block do
        out('<h', node.header_level, '>', :children,
                 '</h', node.header_level, ">")
      end
    end

    def paragraph(node)
      if @in_tight && node.parent.type != :blockquote
        out(:children)
      else
        block do
          container('<p>', '</p>') do
            out(:children)
          end
        end
      end
    end

    def list(node)
      old_in_tight = @in_tight
      @in_tight = node.list_tight

      block do
        if node.list_type == :bullet_list
          container("<ul>\n", '</ul>') do
            out(:children)
          end
        else
          start = node.list_start == 1 ? "<ol>\n" :
                  ('<ol start="' + node.list_start.to_s + "\">\n")
          container(start, '</ol>') do
            out(:children)
          end
        end
      end

      @in_tight = old_in_tight
    end

    def list_item(node)
      block do
        container('<li>', '</li>') do
          out(:children)
        end
      end
    end

    def blockquote(node)
      block do
        container("<blockquote>\n", '</blockquote>') do
          out(:children)
        end
      end
    end

    def hrule(node)
      block do
        out('<hr />')
      end
    end

    def code_block(node)
      block do
        out('<pre><code')
        if node.fence_info && node.fence_info.length > 0
          out(' class="language-', node.fence_info.split(/\s+/)[0], '">')
        else
          out(">")
        end
        out(escape_html(node.string_content))
        out('</code></pre>')
      end
    end

    def html(node)
      block do
        out(node.string_content)
      end
    end

    def inline_html(node)
      out(node.string_content)
    end

    def emph(node)
      out('<em>', :children, '</em>')
    end

    def strong(node)
      out('<strong>', :children, '</strong>')
    end

    def link(node)
      out('<a href="', node.url.nil? ? '' : escape_uri(node.url), '"')
      if node.title && node.title.length > 0
        out(' title="', escape_html(node.title), '"')
      end
      out('>', :children, '</a>')
    end

    def image(node)
      out('<img src="', escape_uri(node.url), '"')
      plain do
        out(' alt="', :children, '"')
      end
      if node.title && node.title.length > 0
        out(' title="', escape_html(node.title), '"')
      end
      out(' />')
    end

    def text(node)
      out(escape_html(node.string_content))
    end

    def code(node)
      out('<code>')
      out(escape_html(node.string_content))
      out('</code>')
    end

    def linebreak(node)
      out('<br />')
      softbreak(node)
    end

    def softbreak(node)
      out("\n")
    end

    # these next two methods are horrendous BS
    def escape_uri(str)
      EscapeUtils.escape_uri(str.gsub('%20', ' ')).gsub(']', '%5D').gsub('&', '&amp;').gsub('[', '%5B')
    end

    def escape_html(str)
      EscapeUtils.escape_html(str).gsub('&#39;', "'").gsub('&#47;', '/')
    end
  end
end