lib/banzai/filter/front_matter_filter.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

# frozen_string_literal: true

module Banzai
  module Filter
    class FrontMatterFilter < HTML::Pipeline::Filter
      def call
        lang_mapping = Gitlab::FrontMatter::DELIM_LANG

        html.sub(Gitlab::FrontMatter::PATTERN) do |_match|
          lang = $~[:lang].presence || lang_mapping[$~[:delim]]

          before = $~[:before]
          before = "\n#{before}" if $~[:encoding].presence

          "#{before}```#{lang}:frontmatter\n#{$~[:front_matter]}```\n"
        end
      end
    end
  end
end