105 lines
1.9 KiB
Ruby
105 lines
1.9 KiB
Ruby
|
|
require 'sinatra'
|
|
require 'erb'
|
|
require 'nokogiri'
|
|
require 'redcarpet'
|
|
|
|
$root = "/home/pranshu/Documents/proj/perc/test"
|
|
Dir.chdir $root;
|
|
|
|
class Sidebar
|
|
FileNode = Struct.new(:type, :name)
|
|
DirNode = Struct.new(:type, :name, :files)
|
|
|
|
def initialize
|
|
@tree = process_dir "."
|
|
end
|
|
|
|
def show(n=@tree, indent="")
|
|
if n.type == :dir
|
|
print indent, n.name, "\n"
|
|
n.files.each do |r|
|
|
show r, " " + indent
|
|
end
|
|
else
|
|
print indent, n.name, "\n"
|
|
end
|
|
end
|
|
|
|
def html
|
|
builder = Nokogiri::HTML::Builder.new do |doc|
|
|
def deal_with_dir (doc, dir, path)
|
|
doc.li {
|
|
doc.text dir.name
|
|
doc.ul {
|
|
dir.files.each do |f|
|
|
fpath = File.join(path, f.name)
|
|
if f.type == :dir
|
|
deal_with_dir doc, f, fpath
|
|
else
|
|
doc.a(:href => fpath) {
|
|
doc.li { doc.text f.name }
|
|
}
|
|
end
|
|
end
|
|
}
|
|
}
|
|
end
|
|
doc.div {
|
|
deal_with_dir doc, @tree, @tree.name
|
|
}
|
|
end
|
|
builder.to_html().split(/\n/)[1..-1].join
|
|
end
|
|
|
|
private
|
|
|
|
def dir_files()
|
|
# Remove "." and ".." from file list
|
|
Dir.entries(".").filter {|f| f !~ /^\.\.?$/ }
|
|
end
|
|
|
|
def process_dir(dir)
|
|
DirNode.new(:dir, dir,
|
|
Dir.chdir(dir) do
|
|
dir_files.map do |f|
|
|
if File.file? f
|
|
FileNode.new(:file, f)
|
|
else
|
|
process_dir f
|
|
end
|
|
end
|
|
end)
|
|
end
|
|
|
|
end
|
|
|
|
$sidebar = Sidebar.new
|
|
|
|
|
|
class Perc < Sinatra::Application
|
|
set :views, "."
|
|
set :markdown, :layout_engine => :erb
|
|
# @sidebar = Sidebar.new
|
|
|
|
get "/" do
|
|
erb :main
|
|
end
|
|
|
|
get "/*.*" do |path, ext|
|
|
if ext == "md"
|
|
md = markdown File.read(path + ".md")
|
|
|
|
else
|
|
erb path.to_sym
|
|
end
|
|
end
|
|
|
|
helpers do
|
|
def tree
|
|
$sidebar.html
|
|
end
|
|
end
|
|
|
|
end
|
|
|