seating.rb

require './env'
require './chooser'

Airrecord.api_key = ENV["AIRTABLE_KEY"]

class Choice < Airrecord::Table
  self.base_key = ENV["AIRTABLE_BASE_KEY"]
  self.table_name = "Choices"
end

class Desk < Airrecord::Table
  self.base_key = ENV["AIRTABLE_BASE_KEY"]
  self.table_name = "Desks"
end

# wrapper to turn an id into a desk name
def desk(id)
  @dnames[id]
end

# make some easier indexes for the desk info
@dlist = {}
@dnames = {}
Desk.all.each do |desk|
  number = desk["Name"].split("-").first.to_i
  @dlist[desk["Name"]] = {id: desk.id, number: number}
  @dnames[desk.id] = desk["Name"]
end

# construct the user choice data array from airtable choices
data = []
ulist = {}
Choice.all.each do |choice|
  next if choice["Desker Type"] != "Perma-desk"

  user_choices = {}
  email = choice["Email"]
  ulist[email] = {id: choice.id, email: email}
  user_choices[:email] = email
  user_choices[:choices] = []
  user_choices[:choices] << [desk(choice["First Choice"].first), choice["First Choice Weight"].to_f] if choice["First Choice"]
  user_choices[:choices] << [desk(choice["Second Choice"].first), choice["Second Choice Weight"].to_f] if choice["Second Choice"]
  user_choices[:choices] << [desk(choice["Third Choice"].first), choice["Third Choice Weight"].to_f] if choice["Third Choice"]
  user_choices[:choices] << [desk(choice["Fourth Choice"].first), choice["Fourth Choice Weight"].to_f] if choice["Fourth Choice"]
  data << user_choices
end

# construct a list of the possible choices
seats = @dlist.keys

# do the assignment
assigns, score = Chooser.new(data, seats).assign!

# debug data
# ap data     # all choices sent to chooser
# ap assigns  # the assignments that came back

# show the results
puts "SCORE: #{score}"

puts "UPLOADING..."
# actually go through the results and update the choices table with the assignments
assigns.each do |desk_name, user_data|
  # get the desk assigned
  desk_id = @dlist[desk_name][:id]
  desk = Desk.find(desk_id)

  # get the user's choice record
  user_id = ulist[user_data[:user]][:id]

  # update the record with the assignment
  choice = Choice.find(user_id)
  choice["Result Desk"] = [desk_id]
  choice["Result Score"] = user_data[:score]
  choice.save
end