require 'rack'
require 'bcat/reader'
require 'bcat/ansi'
require 'bcat/html'
require 'bcat/server'
require 'bcat/browser'
class Bcat
VERSION = '0.6.2'
include Rack::Utils
attr_reader :format
def initialize(args=[], config={})
@config = {:Host => '127.0.0.1', :Port => 8091}.merge(config)
@reader = Bcat::Reader.new(@config[:command], args)
@format = @config[:format]
end
def [](key)
@config[key]
end
def to_app
app = self
Rack::Builder.new do
use Rack::Chunked
run app
end
end
def serve!(&bk)
Bcat::Server.run to_app, @config, &bk
end
def call(env)
notice "#{env['REQUEST_METHOD']} #{env['PATH_INFO'].inspect}"
[200, {"Content-Type" => "text/html;charset=utf-8"}, self]
end
def assemble
@reader.open
@format = @reader.sniff if @format.nil?
@filter = @reader
@filter = TeeFilter.new(@filter) if @config[:tee]
@filter = TextFilter.new(@filter) if @format == 'text'
@filter = ANSI.new(@filter) if @format == 'text' || @config[:ansi]
end
def each
assemble
head_parser = Bcat::HeadParser.new
@filter.each do |buf|
if head_parser.nil?
yield buf
elsif head_parser.feed(buf)
yield content_for_head(inject=head_parser.head)
yield "\n"
yield head_parser.body
head_parser = nil
end
end
if head_parser
yield content_for_head(inject=head_parser.head) +
"\n" +
head_parser.body
end
yield foot
rescue Errno::EINVAL
# socket was closed
notice "browser client went away"
rescue => boom
notice "boom: #{boom.class}: #{boom.to_s}"
raise
end
def content_for_head(inject='')
[
"\n" * 1000,
"",
"",
"",
"",
inject.to_s,
"",
"#{self[:title] || 'bcat'}",
""
].join("\n")
end
def foot
"\n