ticketmaster.rb

require "sinatra/base"
require "yaml"
require "./helpers/unfuddle_api.rb"
require "./helpers/utest_aids.rb"

# ticketmaster requires the following to be set, either as environment
# variables or in ./config/credentials.yaml
# TM_USER
# TM_PASS - credetionals to access ticketmaster
# UNFUDDLE_USER
# UNFUDDLE_PASS - credentials to access unfuddle.
# UNFUDDLE_PROJECT_ID - unfuddle project ID, e.g. 313388
# UNFUDDLE_MILESTONE_ID - ID of milestone to associate with this project, e.g. 307163

class String
  def safe
    Rack::Utils.escape_html self
  end
end

class Ticketmaster < Sinatra::Base

  class << self
    attr_accessor :config
  end

  use Rack::Session::Pool

  self.config = YAML::load(File.open("./config/credentials.yaml")) rescue {}

  use Rack::Auth::Basic, "Restricted Area" do |user, pass|
    [user, pass] == [
      Ticketmaster.config["TM_USER"] || ENV["TM_USER"],
      Ticketmaster.config["TM_PASS"] || ENV["TM_PASS"]
    ]
  end

  get "/" do
    erb :index
  end

  post "/upyougo" do
    begin
      session[:tickets] = UtestAids::ParseCsv.fromfile params[:tickets][:tempfile]
      session[:notice] = "#{session[:tickets].count} ticket(s) uploaded."
      redirect "/verify"
    rescue
      session[:notice] = "Unrecognized file format."
      redirect "/"
    end
  end

  get "/verify" do
    @tickets = session[:tickets] || []
    redirect "/" if @tickets.empty?
    erb :verify
  end

  post "/awayyougo" do
   tickets = session[:tickets]
    if tickets.empty?
      session[:notice] = "No tickets."
      redirect "/"
    end
    submitted = 0
    failed = []
    errors = []
    fu = UnfuddleApi::Futicket.new(
      Ticketmaster.config["UNFUDDLE_USER"] || ENV["UNFUDDLE_USER"],
      Ticketmaster.config["UNFUDDLE_PASS"] || ENV["UNFUDDLE_PASS"],
      Ticketmaster.config["UNFUDDLE_PROJECT_ID"] || ENV["UNFUDDLE_PROJECT_ID"],
      Ticketmaster.config["UNFUDDLE_MILESTONE_ID"] || ENV["UNFUDDLE_MILESTONE_ID"]
    )
    begin
      tickets.each do |t|
        ok, message = fu.submit(t[:title], t[:description])
        if ok
          submitted += 1
        else
          failed << t
          errors << message
        end
      end
    rescue UnfuddleApi::Authentication
      failed += tickets[(submitted+failed.count)..-1]
      errors << "Authentication error."
    end
    errors.uniq!
    session[:notice] = "#{submitted} ticket(s) submitted."
    redirect "/" if failed.empty?
    session[:errors] = errors.join(", ")
    session[:tickets] = failed
    redirect "/verify"
  end

end