#!/usr/bin/env ruby


#
#  Copyright 2017 Amazon.com
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.



require 'rubygems'  # not necessary for Ruby 1.9
require 'mongo'

port = ARGV.first || '27017'

# Connect to db. Located here to make it easy to change port etc...
db = Mongo::Client.new([ '127.0.0.1:' + port ], :database => 'dms_sample')
Mongo::Logger.logger       = ::Logger.new('./log/load_nfl_stadium_data.log')
Mongo::Logger.logger.level = ::Logger::INFO

# Load data from the file
fname = ARGV.first || './data/nfl_stadium_data.sql'

puts "loading NFL stadium data from: " + fname

cols = []
vals = []
stadiums = []

File.open(fname).each_with_index do |r,idx|

  # Get the attributes
  if idx == 0
    col_list = r.slice(0, r.rindex(') values ('))   # remove insert statement 
    col_list.slice!('Insert into nfl_stadium_data (')
    cols = col_list.split(',').map{ |c| c.to_sym }
  end

  # Get the values
  unless r.chomp.empty?
    r =  r.slice(r.rindex('values (') +8, r.length).strip.chomp(');')  
    r =  r.sub(/'/,'').gsub(/\'\'/,'').gsub(/','/,'~').gsub(/,'/,'~').gsub(/',/,'~').gsub(/,null/,'~null').chomp("'")

    vals = r.split('~')

    stadium = {}
    cols.each_with_index do |c,i|
      stadium[c] = vals[i]
    end

    stadium[:seating_capacity] = stadium[:seating_capacity].to_i
    stadium[:sport_location_id] = stadium[:sport_location_id].to_i

    stadiums << stadium   # Add to hash
  end
end


# Load Mongo collection
nfl_stadium_data = db[:nfl_stadium_data]
result = nfl_stadium_data.drop
result = nfl_stadium_data.insert_many(stadiums)
puts "inserted: " + result.inserted_count.to_s + " nfl stadium records"