Class: JekyllTitlesFromHeadings::Generator

Inherits:
Jekyll::Generator
  • Object
show all
Defined in:
lib/jekyll-titles-from-headings/generator.rb

Constant Summary collapse

TITLE_REGEX = 
%r!
  \A\s*                   # Beginning and whitespace
    (?:                   # either
      \#{1,3}\s+(.*)      # atx-style header
      |                   # or
      (.*)\r?\n[-=]+\s*   # Setex-style header
    )$                    # end of line
!x
CONVERTER_CLASS = 
Jekyll::Converters::Markdown
STRIP_MARKUP_FILTERS = 
%i[
  markdownify strip_html normalize_whitespace
].freeze
EXTRA_MARKUP_REGEX =

Regex to strip extra markup still present after markdownify (footnotes at the moment).

%r!\[\^[^\]]*\]!

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(site) ⇒ Generator

Returns a new instance of Generator.



26
27
28
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 26

def initialize(site)
  @site = site
end

Instance Attribute Details

#site ⇒ Object

Returns the value of attribute site.



3
4
5
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 3

def site
  @site
end

Instance Method Details

#generate(site) ⇒ Object 



30
31
32
33
34
35
36
37
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 30

def generate(site)
  @site = site

  site.pages.each do |document|
    next unless should_add_title?(document)
    document.data["title"] = title_for(document)
  end
end

#markdown?(document) ⇒ Boolean

Returns:

  • (Boolean) 


47
48
49
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 47

def markdown?(document)
  markdown_converter.matches(document.extname)
end

#markdown_converter ⇒ Object 



51
52
53
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 51

def markdown_converter
  @markdown_converter ||= site.find_converter_instance(CONVERTER_CLASS)
end

#should_add_title?(document) ⇒ Boolean

Returns:

  • (Boolean) 


39
40
41
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 39

def should_add_title?(document)
  markdown?(document) && !title?(document)
end

#title?(document) ⇒ Boolean

Returns:

  • (Boolean) 


43
44
45
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 43

def title?(document)
  !document.data["title"].nil?
end

#title_for(document) ⇒ Object 



55
56
57
58
59
60
61
 
# File 'lib/jekyll-titles-from-headings/generator.rb', line 55

def title_for(document)
  return document.data["title"] if title?(document)
  matches = document.content.match(TITLE_REGEX)
  strip_markup(matches[1] || matches[2]) if matches
rescue ArgumentError => e
  raise e unless e.to_s.start_with?("invalid byte sequence in UTF-8")
end