#!/usr/local/bin/ruby

require 'cgi'
require 'cgi/session'

class SessionDemo

  ##
  # Initializes the session

  def initialize
    @cgi = CGI.new
    File.umask 0066
    @session = CGI::Session.new @cgi
    @cmd = @cgi.params['cmd'].first || 'start'
    @header = { "type" => "text/html", "charset" => "euc-jp" }
    send "cmd_#{@cmd}"
  end

  ##
  # Displays form for getting name

  def cmd_start
    @cgi.out @header do
      <<-END
<title>CGI::Session Demo</title>
<form action="#{ENV['SCRIPT_NAME']}" method="get">
<div>
<p>Insert your name!
<input type="text" name="name">
<input type="hidden" name="cmd" value="hello">
<input type="submit" value="hello">
</div>
</form>
      END
    end
  end

  ##
  # Puts name in session

  def cmd_hello
    name = @cgi.params['name'].first
    @session['name'] = name
    @cgi.out @header do
      <<-END
<title>CGI::Session Demo</title>
<p>Welcome #{name}
<p><a href="#{ENV['SCRIPT_NAME']}?cmd=bye">[bye]
      END
    end
  end

  ##
  # Pulls name from session

  def cmd_bye
    name = @session['name']
    @cgi.out @header do
      <<-END
<title>CGI::Session Demo</title>
<p>bye! #{name}
<p><a href="#{ENV['SCRIPT_NAME']}">[start]
      END
    end
  end

end

SessionDemo.new