ruby 为Github降价文档生成TOC

Posted 2021-05-15

#!/usr/bin/env ruby

# Generate a correct table of contents for Github markdown documents
# See https://gist.github.com/asabaylus/3071099#gistcomment-1593627

doc = IO.readlines(ARGV[0])
lines = doc.select { |l| l.start_with? "#" }.map do |l|
  indent = l.chomp.gsub(/^# .*/, '* ').gsub(/^## .*/, '    * ').gsub(/^### .*/, '        * ')
  link = l.gsub(/[^a-zA-Z\- ]/u,'').strip().downcase.gsub(' ','-')
  indent + l.chomp.gsub(/^([\# ]*)(.*)$/, '[\2]') + "(##{link})"
end

puts lines.join("\n")