[Ruby][CGI] ファイルをアップロードする。

Rubyでファイルアップロード機能を実装する機会がありまして、載せておいてみます。

HTML form はこんな感じ↓

<form method="post" enctype="multipart/form-data" action="/app/upload">
    upload<input type="file" name="file"><input type="submit" value="アップロード">
</form>

で、肝心のソースはこちら↓

#!/usr/local/bin/ruby
require "cgi"

cgi = CGI.new
print cgi.header("type" => "text/html", "charset" => "UTF-8")

# ファイル名取得、いろいろ考えるのが面倒だったのでこんな方法↓で。しょぼ。
if cgi['file'].original_filename =~ /\\([^\\]*)$/
  # Windows + IE の場合、ローカルのフルパスで渡されやがるんで...
  filename = $1
else
  # それ以外はたぶんこれでいいはず
  filename = File.basename(cgi['file'].original_filename)
end

open("/home/fager/#{filename}","w") do |fh|
  fh.binmode
  fh.write cgi['file'].read
end
 

あとはやるならファイルサイズチェックとかかな。