module Jekyll
  class SeoTag < Liquid::Tag

    attr_accessor :context

    def initialize(_, markup, _)
      super
      @options = {
        "title" => !(markup =~ /title\s*:\s*false/i)
      }
    end

    def render(context)
      @context = context
      output = template.render!(payload, info)

      output
    end

    private

    def payload
      {
        "page" => context.registers[:page],
        "site" => context.registers[:site].site_payload["site"],
        "seo" => @options
      }
    end

    def info
      {
        :registers => context.registers,
        :filters   => [Jekyll::Filters]
      }
    end

    def template
      @template ||= Liquid::Template.parse template_contents
    end

    def template_contents
      @template_contents ||= File.read(template_path).gsub(/(>\n|[%}]})\s+(<|{[{%])/,'\1\2').chomp
    end

    def template_path
      @template_path ||= File.expand_path "./template.html", File.dirname(__FILE__)
    end
  end
end

Liquid::Template.register_tag('seo', Jekyll::SeoTag)