Integrity auto-refresher

gistfile1.rb

#!/usr/bin/env ruby
require "rubygems"
require "integrity"

class Refresher
  JS = <<-EOS
<script type="text/javascript">
//<![CDATA[
 setTimeout('location.reload()',  %s * 1000)
//]]>
</script>
EOS

  def initialize(app, interval)
    @app      = app
    @interval = interval
  end

  def call(env)
    response = @app.call(env)
    return response unless env["PATH_INFO"] == "/"

    body = ""; response.last.each { |s| body << s }
    body.gsub!(/<\/head>/, (JS % @interval) + "\n</head>")
    response[1]["Content-Length"] = body.size.to_s

    [response.first, response[1], [body]]
  end
end

Integrity.new
DataMapper.auto_migrate!

use Refresher, 5
run Integrity::App