module ActiveSeed

# File lib/active_seed/active_seed.rb, line 73
def self.print_errors(errors, line)
  puts "\nThere were errors on line " + line.to_s
  errors.each do |e|
    puts e.to_s
  end
end

# File lib/active_seed/active_seed.rb, line 80
def seed(set)
  set_file = File.join(::Rails.root.to_s, "db", "active_seed", set + ".yml")
  if !File.exists?(set_file)
    puts "Set file doesn't exist: " << set_file
    return
  end
  puts "Seeding from set '" + set + "'"
  fixture_list = YAML::load_file(set_file)
  fixture_list.each do |model, sf|
    seed_file = File.join(::Rails.root.to_s, "db", "active_seed", "data", sf + ".csv")
    if !File.exists?(seed_file)
      puts "Seed file doesn't exist: " << seed_file
    else
      puts "Seeding '" + seed_file + "'..."
      ActiveSeed::seed_csv(model, seed_file)
    end
  end
end

# File lib/active_seed/active_seed.rb, line 4
def self.seed_csv(modelname, filename)
  data = CSV.read(filename)
  if data.size < 2
    puts "No data found in file " << filename
    return
  end
  header = data[0];
  processed_header = Array.new
  evaluations = Array.new
  statics = Array.new
  header.each do |h|
    # Check if the header field has an assignment
    if h.include? "="
      c = h.split "="
      h = c[0] # Replace the header with just the field name
      c.delete_at(0) # Remove the field name
      evaluation = c.join("=") # Rejoin the rest of the evaluation (if there was another =)
      # If there is no question mark in the evaluation string then we set this column to be
      # static.
      unless evaluation.include? "?"
        statics.push(h + "=" + evaluation)
      else
        processed_header.push(h)
        evaluations.push(evaluation)
      end
    else
      evaluations.push(nil)
      processed_header.push(h)
    end
  end
  header = processed_header
  data.delete_at(0);
  puts "Seeding " + data.size.to_s + " record" + (data.size > 1 ? "s" : "")
  line = 1
  data.each do |d|
    line+=1
    if d.size == header.size
      code = "model = " + modelname + ".new\n"
      for count in 0..(header.size - 1) do
        unless (header[count].strip == "nil")
          value = d[count]
          value = "" if value.nil?
          value = "'" + value.strip.gsub(/'/, "\\\\'") + "'" 
          if evaluations[count].nil?
            assignment = value
          else
            assignment = evaluations[count].gsub(/\?+/) do |s|
              s.size == 2 ? "?" : value 
            end
          end
          code += "model." + header[count].strip + "=" + assignment + "\n" 
        end
      end
      # Add in the statics
      statics.each do |s|
        code += "model." + s + "\n"
      end
      code += "unless model.save\n"
      code += "ActiveSeed::print_errors(model.errors, " + line.to_s + ")"
      code += "end\n"
      eval code
      #puts (line - 1).to_s + "/" + data.size.to_s
    else
      puts "Skipping line " + line.to_s + " with mismatch in number of fields (" + d.size.to_s + ")"
    end
  end
  puts " Done"
end

# File lib/active_seed/active_seed.rb, line 80
def seed(set)
  set_file = File.join(::Rails.root.to_s, "db", "active_seed", set + ".yml")
  if !File.exists?(set_file)
    puts "Set file doesn't exist: " << set_file
    return
  end
  puts "Seeding from set '" + set + "'"
  fixture_list = YAML::load_file(set_file)
  fixture_list.each do |model, sf|
    seed_file = File.join(::Rails.root.to_s, "db", "active_seed", "data", sf + ".csv")
    if !File.exists?(seed_file)
      puts "Seed file doesn't exist: " << seed_file
    else
      puts "Seeding '" + seed_file + "'..."
      ActiveSeed::seed_csv(model, seed_file)
    end
  end
end