github-migration/rocco
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

first spike

Browse Source
This commit is contained in:
Ryan Tomayko 2010-03-08 09:56:18 -08:00
parent 9f3826d8e0
commit b69e02c802
2 changed files with 125 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bin/rocco Executable file
View File

@ -0,0 +1,13 @@
#!/usr/bin/env ruby
require 'rocco'
rocco =
Rocco.new(ARGV[0]) do
if $stdin.tty?
File.read(ARGV[0])
else
$stdin.read
end
end
$stdout.write(rocco.to_html)

112
lib/rocco.rb Normal file
View File

@ -0,0 +1,112 @@
# Rocco
# =====
#
# Rocco is a quick-and-dirty literate-programming-style documentation
# generator based *heavily* on [Docco](http://jashkenas.github.com/docco/).
# The RDiscount library is required for Markdown processing.
require 'rdiscount'
class Rocco
# `Rocco.new` takes a `filename` and an optional `block` used to read
# the file's contents. When `block` is given, it must read the contents
# of the file using whatever means necessary and return it as a string.
# With no `block`, the file is read to retrieve data.
def initialize(filename, &block)
@file = filename
@data =
if block_given?
yield
else
File.read(filename)
end
# Jump right into the parsing and highlighting phase.
@sections = highlight(parse(@data))
end
# Internal Parsing and Highlighting
# ---------------------------------
protected
# Parse the raw file data into a list of two-tuples.
def parse(data)
sections = []
docs, code = [], []
data.split("\n").each do |line|
case line
when /^\s*#/
if code.any?
sections << [docs, code]
docs, code = [], []
end
docs << line
when /^\s*$/
if code.any?
code << line
else
docs << line
end
else
code << line
end
end
sections << [docs, code] if docs.any? || code.any?
sections
end
# Take the raw section data and apply markdown formatting and syntax
# highlighting.
def highlight(sections)
# Start by splitting the docs and codes blocks into two separate lists.
docs_blocks, code_blocks = [], []
sections.each do |docs,code|
docs_blocks << docs.map { |line| line.sub(/^\s*#\s?/, '') }.join("\n")
code_blocks << code.join("\n")
end
# Combine all docs blocks into a single big markdown document and run
# through RDiscount. Then split it back out into separate sections.
markdown = docs_blocks.join("\n##### DIVIDER\n")
docs_html = Markdown.new(markdown, :smart).
to_html.
split("\n<h5>DIVIDER</h5>\n")
# Combine all code blocks into a single big stream and run through
# pygments. We `popen` a pygmentize process and then fork off a
# writer process.
open("|pygmentize -l ruby -f html", 'rw') do |fd|
fork {
fd.close_read
fd.write code_blocks.join("\n# DIVIDER\n")
fd.close_write
exit!
}
fd.close_write
code_html = fd.read.split(/\n*<span class="c1"># DIVIDER<\/span>\n*/m)
fd.close_read
end
# Combine the docs and code lists into the same sections style list we
# started with.
docs_html.zip(code_html)
end
public
def to_html
buf = []
buf << "<!DOCTYPE html>"
buf << "<table>"
@sections.each do |docs,code|
buf << "<tr>"
buf << "<td>#{docs}</td>"
buf << "<td>#{code}</td>"
buf << "</tr>"
end
buf << "</table>"
buf.join("\n")
end
end