summaryrefslogtreecommitdiff
path: root/lib/rdoc/markup/preprocess.rb
blob: 760351d38605e1e71b47d78a2ea5b49734617fdd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
require 'rdoc/markup'

##
# Handle common directives that can occur in a block of text:
#
# : include : filename

class RDoc::Markup::PreProcess

  def initialize(input_file_name, include_path)
    @input_file_name = input_file_name
    @include_path = include_path
  end

  ##
  # Look for common options in a chunk of text. Options that we don't handle
  # are passed back to our caller as |directive, param|

  def handle(text)
    text.gsub!(/^([ \t#]*):(\w+):\s*(.+)?\n/) do
      prefix    = $1
      directive = $2.downcase
      param     = $3

      case directive
      when "include"
        filename = param.split[0]
        include_file(filename, prefix)

      else
        yield(directive, param)
      end
    end
  end

  private

  ##
  # Include a file, indenting it correctly.

  def include_file(name, indent)
    if full_name = find_include_file(name) then
      content = File.open(full_name) {|f| f.read}
      # strip leading '#'s, but only if all lines start with them
      if content =~ /^[^#]/
        content.gsub(/^/, indent)
      else
        content.gsub(/^#?/, indent)
      end
    else
      $stderr.puts "Couldn't find file to include: '#{name}'"
      ''
    end
  end

  ##
  # Look for the given file in the directory containing the current file,
  # and then in each of the directories specified in the RDOC_INCLUDE path

  def find_include_file(name)
    to_search = [ File.dirname(@input_file_name) ].concat @include_path
    to_search.each do |dir|
      full_name = File.join(dir, name)
      stat = File.stat(full_name) rescue next
      return full_name if stat.readable?
    end
    nil
  end

end