root/lang/ruby/ekfloras/ekfloras.rb @ 1023

#!/usr/bin/env ruby
#

$LOAD_PATH.unshift "lib"

require "cgi"
require "pathname"
require "ostruct"

begin
        begin
                require "rubygems"
        rescue LoadError
        end
        require "erubis"
rescue LoadError
        require "erb"
end

require "module/pluggable"

class Ekfloras
        DEFAULT_OPTS = {
                :title    => "Ekfloras test",
                :flavour  => ".html",
                :data_dir => "data",
                :data_ext => ".txt",
        }.freeze

        class Entry < OpenStruct; end

        class Plugin
                def init(ekfloras)
                        @ekfloras = ekfloras
                        # priority を返す
                        # 数字が小さいほど先に実行される
                        100
                end

                # 本体の entries のうわがき
                # 上書きしないなら定義してはいけない
                # 定義されているプラグインのうち、
                # 最も priority が高いもののみ使用される。
                def entries
                end
                undef entries

                # ソートしたりフィルタしたりする
                def filter(entries)
                        entries
                end

                # 出力加工用
                def last(out)
                        out
                end

                # あと処理用
                def finish
                end
        end

        pluggable :plugins, :base_class => Plugin

        attr_accessor :path_info, :flavour, :entries
        attr_accessor :home, :server_root, :opts, :cgi
        attr_accessor :stash

        def initialize(cgi, opts=DEFAULT_OPTS.dup)
                @cgi             = cgi
                @opts            = opts
                @opts[:data_dir] = Pathname.new(@opts[:data_dir])
                @data            = Pathname.new(opts[:data_dir])
                @home            = @cgi.script_name
                @server_root     = @cgi.server_name
                @stash           = {}
        end

        def run
                @path_info = Pathname.new(@cgi.path_info.to_s)
                @flavour   = @path_info.extname
                @flavour   = @opts[:flavour] if @flavour.empty?
                @path_info = @path_info.parent if @path_info.basename(".*") == "index"
                @path_info = @path_info.dirname + @path_info.basename(".*")

                # initialize plugins and get priorities
                @priorities = plugins.init(self).sort_by {|k,v| v}

                entries_plugin = @priorities.find {|n,q| plugins[n].respond_to? :entries }
                @entries   = entries_plugin ? plugins[entries_plugin[0]].entries : list_entries()

                @entries = @entries.sort_by {|i|
                        i.time
                }.reverse


                case @path_info.to_s
                when %r|^/(\d{4})(?:/(\d\d)(?:/(\d\d))?)?|
                        _, year, month, day = *Regexp.last_match
                        @entries.reject! {|i|
                                ![["%Y", year], ["%m", month], ["%d", day]].all? {|f,k|
                                        !k || i.time.strftime(f) == k
                                }
                        }
                else
                        @entries.reject! {|i|
                                i.name !~ /#{@path_info}/
                        }
                end

                # call plugins' filters
                @entries = call_plugins(:filter, @entries)

                show({
                        :title       => @opts[:title],
                        :path_info   => @path_info,
                        :flavour     => @flavour,
                        :entries     => @entries,
                        :home        => @home,
                        :server_root => @server_root,
                }.merge(@stash))

                call_plugins :finish
        rescue SystemExit
        rescue Exception => e
                @flavour = ".html"
                $stderr.puts e
                $stderr.puts e.backtrace.join("\n\t")
                show({
                        :error => e,
                        :status => "500 Exception raised",
                })
        end

        def list_entries
                Pathname.glob(@data + "**/*#{@opts[:data_ext]}").map {|f|
                        content = f.readlines
                        name = f.relative_path_from(@opts[:data_dir])
                        Entry.new({
                                :title => content.first,
                                :body  => content[1..-1].join,
                                :time  => f.mtime,
                                :path  => f,
                                :name  => "/#{name.dirname + name.basename(".*")}"
                        })
                }
        end

        def call_plugins(name, arg=nil)
                @priorities.each do |n,q|
                        next unless plugins[n].respond_to? name
                        if arg
                                arg = plugins[n].send(name, arg)
                        else
                                plugins[n].send(name)
                        end
                end
                arg
        end


        def show(stash)
                stash = OpenStruct.new(stash)

                erb = (defined? Erubis) ? Erubis::Eruby : ERB
                ret = erb.new(File.read("template#{@flavour}")).result(binding)

                ret = call_plugins(:last, ret) unless stash.error

                puts "Status: #{stash.status}" if stash.status
                puts ret
                exit
        end
end

if $stdout.tty?
        Ekfloras.new(OpenStruct.new({
                #:path_info => "/test/index.html"
                #:path_info => "/2007/"
                #:path_info => "/test01"
                :path_info => "/intro.html"
        })).run
else
        Ekfloras.new(CGI.new).run
end