1begin
 2  require 'open3_detach'
 3rescue LoadError
 4  require 'open3'
 5end
 6
 7module GitHub
 8  module Markup
 9    extend self
10    @@markups = {}
11
12    def render(filename, content = nil)
13      content ||= File.read(filename)
14
15      if proc = renderer(filename)
16        proc[content]
17      else
18        content
19      end
20    end
21
22    def markup(file, pattern, &block)
23      require file.to_s
24      add_markup(pattern, &block)
25      true
26    rescue LoadError
27      false
28    end
29
30    def command(command, regexp, &block)
31      command = command.to_s
32
33      if File.exists?(file = File.dirname(__FILE__) + "/commands/#{command}")
34        command = file
35      end
36
37      add_markup(regexp) do |content|
38        rendered = execute(command, content)
39        rendered = rendered.to_s.empty? ? content : rendered
40
41        if block && block.arity == 2
42          # If the block takes two arguments, pass new content and old
43          # content.
44          block.call(rendered, content)
45        elsif block
46          # One argument is just the new content.
47          block.call(rendered)
48        else
49          # No block? No problem!
50          rendered
51        end
52      end
53    end
54
55    def add_markup(regexp, &block)
56      @@markups[regexp] = block
57    end
58
59    def can_render?(filename)
60      !!renderer(filename)
61    end
62
63    def renderer(filename)
64      @@markups.each do |key, value|
65        if Regexp.compile("\\.(#{key})$") =~ filename
66          return value
67        end
68      end
69      nil
70    end
71
72    def renderer_name(filename)
73      @@markups.each do |key, value|
74        if Regexp.compile("\\.(#{key})$") =~ filename
75          return key
76        end
77      end
78      nil
79    end
80
81    def execute(command, target)
82      out = ''
83      Open3.popen3(command) do |stdin, stdout, _|
84        stdin.puts target
85        stdin.close
86        out = stdout.read
87      end
88      out.gsub("\r", '')
89    rescue Errno::EPIPE
90      ""
91    rescue Errno::ENOENT
92      ""
93    end
94
95    # Define markups
96    markups_rb = File.dirname(__FILE__) + '/markups.rb'
97    instance_eval File.read(markups_rb), markups_rb
98  end
99end