#!/usr/bin/env ruby

require 'webrick'
require 'fileutils'

BIND_ADDRESS = "192.168.0.28"
PORT = 2626

BACKUP_DIR = 'bak'

if ARGV.length != 0
    root = ARGV.first.gsub('\\', '/')
else
    root = '.'
end

module WEBrick
    module HTTPServlet

        class FileHandler
            alias do_PUT do_GET
        end

        class DefaultFileHandler
            def do_PUT(req, res)
                file = "#{@config[:DocumentRoot]}#{req.path}"
                res.body = ''
                unless Dir.exist? BACKUP_DIR
                    Dir.mkdir BACKUP_DIR
                end
                FileUtils.cp(file, "#{BACKUP_DIR}/#{File.basename(file, '.html')}.#{Time.now.to_i.to_s}.html")
                File.open(file, "w+") {|f| f.puts(req.body)}
            end

            def do_OPTIONS(req, res)
                res['Allow'] = "GET,HEAD,OPTIONS,PUT"
                res['dav'] = 'anything' # TW looks for a 'dav' header, and ignores any value
            end

        end
    end
end

server = WEBrick::HTTPServer.new({:Port => PORT, :DocumentRoot => root, :BindAddress => BIND_ADDRESS})

# ctrl-c handler
trap "INT" do
    puts "Shutting down..."
    server.shutdown
end

puts "Serving on http://#{BIND_ADDRESS}:#{PORT}"

server.start