highlighter.rb

# frozen_string_literal: true

require 'rouge'

# While Rouge is downlevel (Rouge PR#1715 unreleased)
module Rouge
  module Lexers
    class Swift
      prepend :root do
        rule(/\b(?:async|await|isolated)\b/, Keyword)
        rule(/\b(?:actor|nonisolated)\b/, Keyword::Declaration)
      end
    end
  end
end

module Jazzy
  # This module helps highlight code
  module Highlighter
    SWIFT = 'swift'
    OBJC = 'objective_c'

    class Formatter < Rouge::Formatters::HTML
      def initialize(language)
        @language = language
        super()
      end

      def stream(tokens, &block)
        yield "<pre class=\"highlight #{@language}\"><code>"
        super
        yield "</code></pre>\n"
      end
    end

    def self.highlight_swift(source)
      highlight(source, SWIFT)
    end

    def self.highlight_objc(source)
      highlight(source, OBJC)
    end

    def self.highlight(source, language)
      source && Rouge.highlight(source, language, Formatter.new(language))
    end
  end
end