syntax_highlight.rb

# frozen_string_literal: true

module Kramdown
  module Converter
    class Html # rubocop:disable Style/Documentation
      alias original_convert_codeblock convert_codeblock

      def convert_codeblock(el, indent)
        render_shiki(el, indent)
      end

      def render_shiki(el, _indent) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength,Naming/MethodParameterName
        language = extract_code_language!(el.attr) || 'text'
        data = data_attributes(el.attr)
        copy = copy?(el.attr)
        code = el.value
        id = SecureRandom.uuid

        snippet = CodeHighlighter.new.highlight(code, language, id)

        Liquid::Template.parse(template, { line_numbers: true }).render(
          {
            'codeblock' => {
              'copy' => copy,
              'css_classes' => el.attr['class'],
              'collapsible' => el.attr.fetch('class', '').include?('collapsible'),
              'ask_kai' => ask_kai(data, code),
              'render_header' => !data['data-file'].nil? || !ask_kai(data, code).nil?,
              'id' => id,
              'data' => data,
              'snippet' => snippet
            }
          },
          context
        )
      end

      def template
        @template ||= File.read(File.expand_path('app/_includes/syntax_highlighting.html'))
      end

      def context
        @context = Liquid::Context.new(site, {}, {})
      end

      def site
        @site ||= Jekyll.sites.first
      end

      def copy?(attr)
        !attr.fetch('class', '').include?('no-copy-code')
      end

      def data_attributes(attr)
        @data_attributes = attr.each_with_object({}) do |(key, value), data|
          data[key] = value if key.start_with?('data-')
        end
      end

      def ask_kai(data, code)
        return nil if data['data-ask-kai'].nil?

        "https://cloud.konghq.com/?agent=true&agent-prompt=#{URI.encode_www_form_component(code)}"
      end
    end
  end
end