/fe/fcgi/handler.rb

require 'pstore'
require 'stringio'
require 'socket'

class Stop < RuntimeError
  def initialize(s)
    super(s)
  end
end

class PStore
  def get(*k)
    if k.size == 1
      transaction(true) do
        self[k[0]]
      end
    else
      transaction(true) do
        k.map{|v|self[v]}
      end
    end
  end
end

class Hash
  def method_missing(s, *a)
    if key?(s)
      val_(self[s])
    elsif key?(t=s.to_s)
      val_(self[t])
    else
      super(s, *a)
    end
  end
  def val_(r)
    if r.class == Array && r.size == 1
      r[0]
    else
      r
    end
  end
end

class Handler
  @@serv = '192.168.36.2'
  @@port = 9999
  @@maintenance = File.exist?('maintenance')

  @@eol = "\r\n"

  def root_url
    if develop?
      'http://127.0.0.1:81'
    else
      'http://golf.shinh.org'
    end
  end

  def handle(req)
    @req = req
    @i = req.in
    @o = req.out
    @e = req.env
    @headered = nil

    if @e['SERVER_NAME'] == 'www.shinh.org'
    #if false
      print "Status Code: 301 Moved Permanently\r\n"
      l = @e['REQUEST_URI']
      print "Location: #{root_url}#{l}\r\n"
      print "\r\n"
      @req.finish
      return
    end

    begin
      handle_
    rescue
      if @orig_o
        puts end_buffering
      end

      if $!.class == Stop
        html_header if !@headered
        puts "Error:"
        puts $!
      else
        html_header if !@headered
        print "<p>#{$!}:</p><pre>#{$!.backtrace*"\n"}</pre>"
      end
    end

    @req.finish
  end

  def html_header(h={})
    @headered = true
    print "Content-Type: text/html; charset=UTF-8\r\n"
    h.each do |k, v|
      print "#{k}: #{v}\r\n"
    end
    print "\r\n"
  end
  def text_header(h={})
    @headered = true
    print "Content-Type: text/plain\r\n"
    h.each do |k, v|
      print "#{k}: #{v}\r\n"
    end
    print "\r\n"
  end
  def redirect(l)
    print "Status Code: 300 Found\r\n"
    print "Location: #{root_url}#{l}\r\n"
    print "\r\n"
  end

  def user_name
    @e['HTTP_COOKIE'] =~ /un=(.*)/
    $1 ? CGI.unescape($1) : ''
  end

  def err(e)
    raise Stop.new(e)
  end

  def store(d, q)
    db = PStore.new('db/' + d + '.db')
    db.transaction do
      q.each do |k, x|
        x = x[0] if x.class == Array && x.size == 1
        db[k] = x
      end
    end
  end

  def file_types
    [
     'rb','pl','py','php','scm','l','arc','clj',
     'ly','io','js','lua','tcl','xtal','kt','cy',
     'st', 'pro','for','bas',
     'pl6', 'erl', 'ijs', 'a+', 'mind',
     'c','cpp','d','go',
     'ml',
     'hs','adb','m','java','pas','f95','cs','n','cob','curry','lmn','max','oct','reb','asy',
     'awk','sed','sh','zsh','fish','wake','bc','dc',
     'xgawk','m4','ps','r','vhdl','qcl',
     'bf','ws','bef', 'pef', 'ms', 'gs', 'flog', 'nand',
     'unl', 'lazy', 'grass', 'lamb', 'wr', 'di',
     's','out','z8b','com','class','vi','grb',
     'groovy', 'scala', 'logo',
    ]
  end
  def file_langs
    [
     'Ruby','Perl','Python','PHP','Scheme',
     'Common LISP','Arc','Clojure',
     'LilyPond','Io','JavaScript','Lua','Tcl','Xtal','Kite','Cyan',
     'Smalltalk', 'Prolog','Forth','BASIC',
     'Perl6', 'Erlang', 'J', 'A+', 'Mind',
     'C','C++','D','Go',
     'OCaml',
     'Haskell',
     'Ada','ObjC','Java','Pascal','Fortran','C#','Nemerle','COBOL','Curry','LMNtal','Maxima','Octave','REBOL','Asymptote',
     'AWK','sed','Bash','Zsh','fish','wake','bc','dc',
     'xgawk','m4','Postscript','R','VHDL','QCL',
     'Brainfuck','Whitespace','Befunge','Pefunge','Minus','GolfScript', 'FlogScript', 'FerNANDo',
     'Unlambda','Lazy-K','Grass','Universal Lambda','Whirl','D-compile-time',
     'gas','x86','z80','DOS','JVM','vi','goruby',
     'Groovy', 'Scala', 'Logo',
    ]
  end

  def ext2lang(e)
    file_langs[file_types.index(e)]
  end

  def get_db(d)
    PStore.new('db/' + d + '.db')
  end

  def develop?
    @e['REMOTE_ADDR'] == '127.0.0.1'
  end

  def escape_binary(s)
    s.gsub(/[\x00-\x09\x0b-\x1f\x7f-\xff]/){'<span class="gray">\x%02x</span>'%$&[0]}
    #s
  end

  def print(*a)
    @o.print(*a)
  end
  def puts(*a)
    @o.puts(*a)
  end
  def p(*a)
    a.each do |x|
      @o.puts(x.inspect)
    end
  end

  def tr(*a)
    r = '<tr>'
    a.each do |x|
      r += "<td>#{x}</td>"
    end
    r += '</tr>'
  end

  def start_buffering
    @orig_o = @o
    @o = StringIO.new
  end

  def end_buffering
    @o.rewind
    r = @o.read
    @o = @orig_o
    @orig_o = nil
    r
  end

  def query
    CGI::parse(@i.read)
  end

  def problem_db
    PStore.new('db/problem.db')
  end

  def a(h, t, n=nil)
    if n
      %Q(<a href="#{h}" name="#{n}">#{t}</a>)
    else
      %Q(<a href="#{h}">#{t}</a>)
    end
  end

  def tag(t, c)
    %Q(<#{t}>#{c}</#{t}>)
  end

  def problem_url(x)
    '/p.rb?' + x.tr(' ','+')
  end

  def problem_summary(x, d, now)
    anchor = a(problem_url(x), x)
    if !d || d == 0
      anchor + " (endless)"
    elsif d > now   
      anchor + " #{time_diff(d-now)} left (#{Time.at(d).strftime('%m/%d %H:%M:%S JST')})"
    else
      anchor + " (post mortem)"
    end
  end

  def lang_url(x)
    '/l.rb?' + x
  end

  def page
    q = @e['QUERY_STRING']
    err('no page') if !q
    [q.tr('+',' '), q]
  end

  def time_diff(dl)
    dls = "%02d" % (dl % 60)
    dlm = "%02d" % ((dl / 60) % 60)
    dlh = (dl / 60 / 60) % 24
    dld = (dl / 60 / 60 / 24)
    dld = dld > 0 ? "#{dld}day(s) and " : ""
    "#{dld}#{dlh}:#{dlm}:#{dls}"
  end

  def title(t)
    print %Q(<!DOCTYPE html>
<html>

<head>
 <meta http-equiv="CONTENT-TYPE" content="text/html; charset=UTF-8">
 <title>#{t}</title>
 <link rev="MADE" href="mailto:shinichiro.hamaji _at_ gmail.com">
 <link rel="INDEX" href=".">
 <link rel="stylesheet" type="text/css" href="/site.css">
</head>

<body>)
  end

  def foot
    print %Q(<p><a href="/">return top</a></p></body></html>)
  end

  def mircbot(msg)
    begin
      sock = TCPSocket.new('localhost',9999)
      sock.print(msg)
      sock.close
    rescue
    end
  end

  def lranking(ext, l, pn, del=nil, pm=0)
    ret = %Q(<table border="1"><tr><th>Rank</th><th>User</th><th>Size</th><th>Time</th><th>Date</th><th><a href="bas.html">Statistics</a></th>)
    if del
      ret += del[0]
    end
    ret += %Q(</tr>)
    l.each_with_index do |v, i|
      st = v[4]
      if st
        if st[1]
          st = [st[0], st[2], st[3]]
        else
          st = [st[0], '?', '?']
        end
      else
        st = ['?','?','?']
      end
      un = CGI.escapeHTML(v[0])
      if pm == 1
        un = %Q(<a href="reveal.rb?#{CGI.escape(pn)}/#{CGI.escape(v[0])}_#{v[3].to_i}&#{ext}">#{un != '' ? un : '_'}</a>)
      end
      date = v[3].strftime('%y/%m/%d %H:%M:%S')
      if v[5] == 1
        date = %Q(<em class="pm">#{date}</em>)
      end
      ret += "<tr><td>#{i+1}</td><td>#{un}</td><td>#{v[1]}</td><td>#{"%.4f"%v[2]}</td><td>#{date}</td><td>#{st.map{|x|"#{x}B"}*' / '}</td>"
      if del
        del[1].call(v)
      end
      ret += %Q(</tr>)
    end
    ret + %Q(</table>)
  end

  def sorted_langs
    file_types.zip(file_langs).sort_by{|a|a[1].downcase}
  end

  def put_by_languages(pa, base, overall = [])
    puts "<p>|"
    types = overall + sorted_langs
    types.each do |la, ln|
      if pa == la
        puts ln, "|"
      else
        l = la.empty? ? base : "#{base}?#{la}"
        puts a(l, ln), "|"
      end
    end
    puts "</p>"
  end

  def ranking(pn, del, pm)
    ret=''
    lr = {}
    ret += tag('h2', 'Ranking')
    ldb = PStore.new("db/#{pn}/_ranks.db")
    ldb.transaction(true) do
      file_types.zip(file_langs).each do |ext, lang|
        next if !ldb.root?(ext)
        ret += tag('h3', a(lang_url(ext),lang)+' '+a("##{lang}",'_',"#{lang}"))
        l = ldb[ext]
        next if l.empty?

        min = l.sort{|a,b|a[1]<=>b[1]}[0]
        lr[ext] = [min[1],min[3],min[0]]
        #lr[ext] = [l[0][1],l[0][0]]

        ret += lranking(ext, l, pn, del, pm)
      end
    end
    [ret, lr]
  end

  def read_multipart(boundary, content_length)
    params = Hash.new([])
    boundary = "--" + boundary
    quoted_boundary = Regexp.quote(boundary, "n")
    buf = ""
    bufsize = 10 * 1024

    # start multipart/form-data
    @i.binmode if defined? @i.binmode
    boundary_size = boundary.size + @@eol.size
    content_length -= boundary_size
    status = @i.read(boundary_size)
    if nil == status
      raise EOFError, "no content body"
    elsif boundary + @@eol != status
      raise EOFError, "bad content body"
    end

    loop do
      head = nil
      if 10240 < content_length
        require "tempfile"
        body = Tempfile.new("CGI")
      else
        begin
          require "stringio"
          body = StringIO.new
        rescue LoadError
          require "tempfile"
          body = Tempfile.new("CGI")
        end
      end
      body.binmode if defined? body.binmode

      until head and /#{quoted_boundary}(?:#{@@eol}|--)/n.match(buf)

          if (not head) and /#{@@eol}#{@@eol}/n.match(buf)
              buf = buf.sub(/\A((?:.|\n)*?#{@@eol})#{@@eol}/n) do
            head = $1.dup
            ""
          end
            next
          end

        if head and ( (@@eol + boundary + @@eol).size < buf.size )
          body.print buf[0 ... (buf.size - (@@eol + boundary + @@eol).size)]
          buf[0 ... (buf.size - (@@eol + boundary + @@eol).size)] = ""
        end

        c = if bufsize < content_length
              @i.read(bufsize)
            else
              @i.read(content_length)
            end
        if c.nil? || c.empty?
          raise EOFError, "bad content body"
        end
        buf.concat(c)
        content_length -= c.size
      end

      buf = buf.sub(/\A((?:.|\n)*?)(?:[\r\n]{1,2})?#{quoted_boundary}([\r\n]{1,2}|--)/n) do
        body.print $1
        if "--" == $2
          content_length = -1
        end
        ""
      end

      body.rewind

      /Content-Disposition:.* filename="?([^\";]*)"?/ni.match(head)
      filename = ($1 or "")
      if /Mac/ni.match(@e['HTTP_USER_AGENT']) and
          /Mozilla/ni.match(@e['HTTP_USER_AGENT']) and
          (not /MSIE/ni.match(@e['HTTP_USER_AGENT']))
        filename = CGI::unescape(filename)
      end

      /Content-Type: (.*)/ni.match(head)
      content_type = ($1 or "")

      (class << body; self; end).class_eval do
        alias local_path path
        define_method(:original_filename) {filename.dup.taint}
        define_method(:content_type) {content_type.dup.taint}
      end

      /Content-Disposition:.* name="?([^\";]*)"?/ni.match(head)
      name = $1.dup

      if params.has_key?(name)
        params[name].push(body)
      else
        params[name] = [body]
      end
      break if buf.size == 0
      break if content_length === -1
    end

    params
  end # read_multipart

  def execute
    begin
      return TCPSocket.open(@@serv, @@port)
    rescue
      if @@maintenance
        puts %Q(now maintenance. it will be back soon. please try again later.)
      else
        puts %Q(something are broken. please send an email to shinichiro.hamaji@gmail.com if this issue stays long)
      end
      raise
    end
  end

  def execute2(filename, code, inputs, testing=false)
    modified_inputs = inputs.map do |input|
      if /\.sed$/ =~ filename && (!input || input.size == 0)
        input = "\n"
      end
      if input
        input.gsub!("\r\n","\n")
      else
        input = ''
      end
      input
    end

    payload = {
      :filename => filename,
      :code => code,
      :inputs => modified_inputs,
    }
    if testing
      payload[:testing] = true
    end

    encoded_payload = Marshal.dump(payload)

    s = execute
    s.puts(encoded_payload.size)
    s.print(encoded_payload)
    s.close_write

    s
  end

end