require "octopress-escape-code/version"
require 'octopress-hooks'

module Octopress
  module EscapeCode

    class EscapePage < Octopress::Hooks::Page
      priority :lowest

      def pre_render(page)
        if Octopress::EscapeCode.escape_enabled?(page)
          page.content = Octopress::EscapeCode.escape(page)
        end
      end

      def post_render(page)
        Octopress::EscapeCode.unescape_raw(page)
      end
    end

    class EscapePost < Octopress::Hooks::Post
      priority :lowest

      def pre_render(page)
        if Octopress::EscapeCode.escape_enabled?(page)
          page.content = Octopress::EscapeCode.escape(page)
        end
      end

      def post_render(page)
        Octopress::EscapeCode.unescape_raw(page)
      end
    end

    def self.unescape_raw(page)
      page.output.gsub!(/\*-(end)?raw-\*/, '% \1raw %')
    end

    def self.escape_enabled?(page)
      get_config(page, 'escape_code', false)
    end

    def self.get_config(page, config, default)
      site_config = page.site.config[config]
      site_config = default if site_config.nil?

      page_config = page.data[config]
      page_config = site_config if page_config.nil?

      page_config
    end

    def self.escape(page)
      ext = page.ext.downcase
      content = page.content.encode!("UTF-8")
      md_ext = %w{.markdown .mdown .mkdn .md .mkd .mdwn .mdtxt .mdtext}

      # Escape existing raw tags
      #
      content = content.gsub(/% (end)?raw %/, '*-\1raw-*')

      # Escape markdown style code blocks
      if md_ext.include?(ext)

        # Escape four tab or space indented code blocks
        content = content.gsub /^((\t| {4})[^\n].+?)\n($|\S)/m do
          "{% raw %}#{$1}\n{% endraw %}#{$3}"
        end

        # Escape in-line code backticks
        content = content.gsub /(`[^`\n]+?`)/ do
          "{% raw %}#{$1}{% endraw %}"
        end

        # Escape in-line code double backticks
        content = content.gsub /(``[^\n]+?``)/ do
          "{% raw %}#{$1}{% endraw %}"
        end

        # Remove internal raw tags within tab or space intented codeblocks
        content = content.gsub /^({% raw %})?((\t| {4})[^\n].+?)\n($|\S)/m do
          c1 = $1
          c2 = $2
          c4 = $4

          "#{c1}#{c2.gsub(/{% (end)?raw %}/, '')}\n#{c4}"
        end
      end

      # Escape codeblock tag contents
      content = content.gsub /^({%\s*codeblock.+?%})(.+?){%\s*endcodeblock\s*%}/m do
        "#{$1}{% raw %}#{$2.gsub /{% (end)?raw %}/, ''}{% endraw %}{% endcodeblock %}"
      end

      # Escape highlight tag contents
      content = content.gsub /^({%\s*highlight.+?%})(.+?){%\s*endhighlight\s*%}/m do
        "#{$1}{% raw %}#{$2.gsub(/{% (end)?raw %}/, '')}{% endraw %}{% endhighlight %}"
      end

      # Escape codefenced codeblocks
      content = content.gsub /^(`{3}.+?`{3})/m do
        
        # Replace any raw/endraw tags inside of codefence block
        # as some of the regex above may have escaped contents
        # of the codefence block
        #
        code = $1.gsub(/{% raw %}/, '').gsub(/{% endraw %}/, '') 

        # Wrap codefence content in raw tags
        "{% raw %}\n#{code}\n{% endraw %}"
      end

      content
    end
  end
end

# Add documentation for this plugin
if defined? Octopress::Docs
  Octopress::Docs.add({
    name:        "Octopress Escape Code",
    description: "Automatically escape code with {% raw %} blocks, preventing internal Liquid processing.",
    source_url:  "https://github.com/octopress/escape-code",
    path:         File.expand_path(File.join(File.dirname(__FILE__), "../"))
  })
end