#!/usr/bin/env ruby
require "bundler/inline"
require "fileutils"
require "optparse"
require "securerandom"
gemfile do
source "https://rubygems.org"
gem "sinatra", require: "sinatra/base"
gem "sinatra-flash", require: "sinatra/flash"
gem "webrick"
end
options = {}
OptionParser.new do |op|
op.on "-p", "--port NUMBER", Integer do |v|
options[:port] = v
end
op.on "-u", "--upload DIRECTORY" do |v|
options[:upload] = v
end
end.parse!(ARGV)
Class.new(Sinatra::Base) do
TEMPLATE = DATA.read
UPLOAD = File.expand_path(options[:upload] || "upload")
FileUtils.mkdir_p(UPLOAD)
$stderr.puts "== Upload folder: #{UPLOAD}"
set :port, options[:port] if options[:port]
enable :sessions
register Sinatra::Flash
get "/" do
erb TEMPLATE
end
post "/upload" do
name = "%s%s" % [SecureRandom.uuid, File.extname(params[:file][:filename])]
file = params[:file][:tempfile]
path = File.join(UPLOAD, name)
File.open(path, "wb") do |f|
f.write(file.read)
end
flash[:notice] = "Uploaded to #{path}"
redirect "/"
end
end.run!
__END__
<%= File.basename(__FILE__, ".rb") %>
<% if flash[:notice] %>
<%= flash[:notice] %>
<% end %>