module Jekyll
  class EnunciationTagBlock < Liquid::Block

    def initialize(tag_name, arg, parse_context)
      super
      @arg = arg.strip
    end

    def render(context)
      text = super
      # If the enunciation ends with a KaTeX displayed equation,
      # then we remove the bottom margin.
      if text[-3,3] == "$$\n"
        text += '{: .katex-display .mb-0 }'
      end
      # Check if a description was given.
      if @arg.length < 1
        "{: .#{block_name} }\n #{text.gsub!(/^/,'> ')}"
      else
        "{: .#{block_name}-title }\n> #{block_name.capitalize} (#{@arg})\n>\n#{text.gsub!(/^/,'> ')}"
      end
    end

  end
end

Liquid::Template.register_tag('definition', Jekyll::EnunciationTagBlock)
Liquid::Template.register_tag('theorem', Jekyll::EnunciationTagBlock)
Liquid::Template.register_tag('proposition', Jekyll::EnunciationTagBlock)
Liquid::Template.register_tag('lemma', Jekyll::EnunciationTagBlock)
Liquid::Template.register_tag('corollary', Jekyll::EnunciationTagBlock)
Liquid::Template.register_tag('axiom', Jekyll::EnunciationTagBlock)