133 lines
2.5 KiB
Ruby
133 lines
2.5 KiB
Ruby
|
|
require 'sinatra'
|
|
require 'erb'
|
|
require 'nokogiri'
|
|
require 'redcarpet'
|
|
require 'tomlib'
|
|
|
|
$root = "/home/pranshu/Documents/proj/perc/test"
|
|
Dir.chdir $root;
|
|
|
|
|
|
class Config
|
|
attr_reader :approved_regex, :layout
|
|
def initialize
|
|
approved_files = [/\.md$/, /\.erb$/]
|
|
@approved_regex = Regexp.union(approved_files)
|
|
toml = Tomlib.load(File.read "config.toml")
|
|
|
|
@layout = toml["layout"] || "layout"
|
|
@layout = @layout.to_sym
|
|
# @layout = :layout
|
|
|
|
end
|
|
end
|
|
|
|
$config = Config.new();
|
|
|
|
class Sidebar
|
|
FileNode = Struct.new(:type, :path, :name)
|
|
DirNode = Struct.new(:type, :path, :name, :files)
|
|
|
|
def initialize
|
|
@tree = process_dir "."
|
|
end
|
|
|
|
def show(n=@tree, indent="")
|
|
if n.type == :dir
|
|
print indent, n.path, "\n"
|
|
n.files.each do |r|
|
|
show r, " " + indent
|
|
end
|
|
else
|
|
print indent, n.path, "\n"
|
|
end
|
|
end
|
|
|
|
def html
|
|
builder = Nokogiri::HTML::Builder.new do |doc|
|
|
def deal_with_dir (doc, dir, path)
|
|
doc.li {
|
|
doc.text dir.path
|
|
doc.ul {
|
|
dir.files.each do |f|
|
|
fpath = File.join(path, f.path)
|
|
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.path
|
|
}
|
|
end
|
|
builder.to_html().split(/\n/)[1..-1].join
|
|
end
|
|
|
|
private
|
|
|
|
def getname(file)
|
|
File.basename(file, File.extname(file))
|
|
end
|
|
|
|
def dir_files()
|
|
# Remove "." and ".." from file list
|
|
Dir.entries(".").filter {|f| f !~ /^\./ }.filter do |f|
|
|
if File.file? f
|
|
f =~ $config.approved_regex
|
|
else
|
|
true
|
|
end
|
|
end
|
|
end
|
|
|
|
def process_dir(dir)
|
|
DirNode.new(:dir, dir, getname(dir),
|
|
Dir.chdir(dir) do
|
|
dir_files.map do |f|
|
|
if File.file? f
|
|
FileNode.new(:file, f, getname(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, layout: $config.layout
|
|
end
|
|
|
|
get "/*.*" do |path, ext|
|
|
if ext == "md"
|
|
md = markdown File.read(path + ".md"), layout: $config.layout
|
|
|
|
else
|
|
erb path.to_sym, layout: $config.layout
|
|
end
|
|
end
|
|
|
|
helpers do
|
|
def tree
|
|
$sidebar.html
|
|
end
|
|
end
|
|
|
|
end
|
|
|