# frozen_string_literal: true

require 'test_helper'

class TestVariousTocHtml < Minitest::Test
  # ref. https://github.com/toshimaru/jekyll-toc/issues/45
  ANGLE_BRACKET_HTML = <<-HTML
<h1>h1</h1>
<h1>&lt;base href&gt;</h1>
<h1>&amp; &lt; &gt;</h1>
  HTML

  NO_TOC_HTML = <<-HTML
<h1>h1</h1>
<h1 class="no_toc">no_toc h1</h1>
<h2>h2</h2>
<h2 class="no_toc">no_toc h2</h2>
<h3>h3</h3>
<h3 class="no_toc">no_toc h3</h3>
<h4>h4</h4>
<h4 class="no_toc">no_toc h4</h4>
  HTML

  JAPANESE_HEADINGS_HTML = <<-HTML
<h1>あ</h1>
<h2>い</h2>
<h3>う</h3>
  HTML

  TAGS_INSIDE_HEADINGS_HTML = <<-HTML
<h2><strong>h2</strong></h2>
<h2><em>h2</em></h2>
  HTML

  TEST_HTML_1 = <<-HTML
<h1>h1</h1>
<h3>h3</h3>
<h6>h6</h6>
  HTML

  TEST_HTML_2 = <<-HTML
<h1>h1</h1>
<h3>h3</h3>
<h2>h2</h2>
<h6>h6</h6>
  HTML

  TEST_HTML_3 = <<-HTML
<h6>h6</h6>
<h5>h5</h5>
<h4>h4</h4>
<h3>h3</h3>
<h2>h2</h2>
<h1>h1</h1>
  HTML

  TEST_HTML_4 = <<-HTML
<h1>h1</h1>
<h3>h3</h3>
<h2>h2</h2>
<h4>h4</h4>
<h5>h5</h5>
  HTML

  def test_nested_toc
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_1)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#h1">h1</a>
<ul>
<li class="toc-entry toc-h3">
<a href="#h3">h3</a>
<ul>
<li class="toc-entry toc-h6"><a href="#h6">h6</a></li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_nested_toc_with_min_and_max
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_1, { 'min_level' => 2, 'max_level' => 5 })
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h3"><a href="#h3">h3</a></li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_complex_nested_toc
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_2)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#h1">h1</a>
<ul>
<li class="toc-entry toc-h3"><a href="#h3">h3</a></li>
<li class="toc-entry toc-h2">
<a href="#h2">h2</a>
<ul>
<li class="toc-entry toc-h6"><a href="#h6">h6</a></li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_decremental_headings1
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_3)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h6"><a href="#h6">h6</a></li>
<li class="toc-entry toc-h5"><a href="#h5">h5</a></li>
<li class="toc-entry toc-h4"><a href="#h4">h4</a></li>
<li class="toc-entry toc-h3"><a href="#h3">h3</a></li>
<li class="toc-entry toc-h2"><a href="#h2">h2</a></li>
<li class="toc-entry toc-h1"><a href="#h1">h1</a></li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_decremental_headings2
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_4)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#h1">h1</a>
<ul>
<li class="toc-entry toc-h3"><a href="#h3">h3</a></li>
<li class="toc-entry toc-h2">
<a href="#h2">h2</a>
<ul>
<li class="toc-entry toc-h4">
<a href="#h4">h4</a>
<ul>
<li class="toc-entry toc-h5"><a href="#h5">h5</a></li>
</ul>
</li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML

    assert_equal(expected, doc.css('ul.section-nav').to_s)
  end

  def test_no_toc
    parser = Jekyll::TableOfContents::Parser.new(NO_TOC_HTML)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#h1">h1</a>
<ul>
<li class="toc-entry toc-h2">
<a href="#h2">h2</a>
<ul>
<li class="toc-entry toc-h3">
<a href="#h3">h3</a>
<ul>
<li class="toc-entry toc-h4"><a href="#h4">h4</a></li>
</ul>
</li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_japanese_toc
    parser = Jekyll::TableOfContents::Parser.new(JAPANESE_HEADINGS_HTML)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#%E3%81%82">あ</a>
<ul>
<li class="toc-entry toc-h2">
<a href="#%E3%81%84">い</a>
<ul>
<li class="toc-entry toc-h3"><a href="#%E3%81%86">う</a></li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_angle_bracket
    parser = Jekyll::TableOfContents::Parser.new(ANGLE_BRACKET_HTML)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1"><a href="#h1">h1</a></li>
<li class="toc-entry toc-h1"><a href="#base-href">&lt;base href&gt;</a></li>
<li class="toc-entry toc-h1"><a href="#--">&amp; &lt; &gt;</a></li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  def test_tags_inside_heading
    parser = Jekyll::TableOfContents::Parser.new(TAGS_INSIDE_HEADINGS_HTML)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h2"><a href="#h2">h2</a></li>
<li class="toc-entry toc-h2"><a href="#h2-1">h2</a></li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s

    assert_equal(expected, actual)
  end

  TEST_HTML_IGNORE = <<-HTML
<h1>h1</h1>
<div class="no_toc_section">
<h2>h2</h2>
</div>
<h3>h3</h3>
<h6>h6</h6>
  HTML

  def test_nested_toc_with_no_toc_section_class
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_IGNORE)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#h1">h1</a>
<ul>
<li class="toc-entry toc-h3">
<a href="#h3">h3</a>
<ul>
<li class="toc-entry toc-h6"><a href="#h6">h6</a></li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s
    assert_equal(expected, actual)

    html = parser.inject_anchors_into_html
    assert_match(%r{<h1>.+</h1>}m, html)
    assert_match(%r{<h3>.+</h3>}m, html)
    assert_match(%r{<h6>.+</h6>}m, html)
    assert_includes(html, '<h2>h2</h2>')
  end

  TEST_HTML_IGNORE_2 = <<-HTML
<h1>h1</h1>
<div class="exclude">
<h2>h2</h2>
</div>
<h3>h3</h3>
<div class="exclude">
<h4>h4</h4>
<h5>h5</h5>
</div>
<h6>h6</h6>
  HTML

  def test_nested_toc_with_no_toc_section_class_option
    parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_IGNORE_2, { 'no_toc_section_class' => 'exclude' })
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1">
<a href="#h1">h1</a>
<ul>
<li class="toc-entry toc-h3">
<a href="#h3">h3</a>
<ul>
<li class="toc-entry toc-h6"><a href="#h6">h6</a></li>
</ul>
</li>
</ul>
</li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s
    assert_equal(expected, actual)

    html = parser.inject_anchors_into_html
    assert_match(%r{<h1>.+</h1>}m, html)
    assert_match(%r{<h3>.+</h3>}m, html)
    assert_match(%r{<h6>.+</h6>}m, html)
    assert_includes(html, '<h2>h2</h2>')
    assert_includes(html, '<h4>h4</h4>')
    assert_includes(html, '<h5>h5</h5>')
  end

  TEST_EXPLICIT_ID = <<-HTML
<h1>h1</h1>
<h1 id="second">h2</h1>
<h1 id="third">h3</h1>
  HTML

  def test_toc_with_explicit_id
    parser = Jekyll::TableOfContents::Parser.new(TEST_EXPLICIT_ID)
    doc = Nokogiri::HTML(parser.toc)
    expected = <<-HTML
<ul class="section-nav">
<li class="toc-entry toc-h1"><a href="#h1">h1</a></li>
<li class="toc-entry toc-h1"><a href="#second">h2</a></li>
<li class="toc-entry toc-h1"><a href="#third">h3</a></li>
</ul>
    HTML
    actual = doc.css('ul.section-nav').to_s
    assert_equal(expected, actual)

    html = parser.inject_anchors_into_html
    assert_includes(html, %(<a id="h1" class="anchor" href="#h1" aria-hidden="true">))
    assert_includes(html, %(<a id="second" class="anchor" href="#second" aria-hidden="true">))
    assert_includes(html, %(<a id="third" class="anchor" href="#third" aria-hidden="true">))
  end
end

def test_custom_css_classes
  parser = Jekyll::TableOfContents::Parser.new(TEST_HTML_1, { "item_class" => "custom-item", "list_class" => "custom-list", "sublist_class" => "custom-sublist", "item_prefix" => "custom-prefix-" })
  doc = Nokogiri::HTML(parser.toc)
  expected = <<-HTML
<ul class="custom-list">
<li class="custom-item custom-prefix-h1">
<a href="#h1">h1</a>
<ul class="custom-sublist">
<li class="custom-item custom-prefix-h3">
<a href="#h3">h3</a>
<ul class="custom-sublist">
<li class="custom-item custom-prefix-h6"><a href="#h6">h6</a></li>
</ul>
</li>
</ul>
</li>
</ul>
  HTML

  assert_equal(expected, doc.css('ul.custom-list').to_s)
end