klasses = [ Project, Document, User, Subscription, Karma, Plan, Article, Event, EventInstance, Karma ]
old_counts = klasses.map(&:count)
should_prompt = old_counts.min > 0

def get_country
  country = File.readlines(Rails.root + 'spec/fixtures/country_codes.txt').sample
  code, name = country.chomp.split('|')
  @country = {country_name: name, country_code: code}
end

if should_prompt
  puts 'Would you like to ' + 'delete'.red.bold + ' all the existing projects and documents from the database?'
end

while true
  if should_prompt
    print 'yes(y)/no(n): '
    response = STDIN.gets.downcase.chomp
  else
    response = 'y'
  end

  if response == 'y' || response == 'yes'
    puts 'Clearing existing projects and documents'
    klasses.each(&:delete_all)

    pw = 'randomrandom'
    u = User.create(first_name: 'Random', last_name: 'Guy', email: 'random@random.com', password: pw)
    puts 'Added default user with email: ' + u.email.bold.blue + ' and password: ' + pw.bold.red

    autograder = u.projects.create!(
      title: 'Autograder',
      description: 'Autograder for the EdX CS169.x SaaS course',
      status: 'Active',
      commit_count: 200
    )
    autograder.source_repositories.create!(url: 'https://github.com/saasbook/rag')

    websiteone = u.projects.create!(
      title: 'WebsiteOne',
      description: 'The AgileVentures site - a platform for online collaboration and crowdsourced project development.',
      status: 'Active',
      commit_count: 190
    )
    websiteone.source_repositories.create!(url: 'https://github.com/AgileVentures/WebsiteOne')

    localsupport = u.projects.create!(
      title: 'LocalSupport',
      description: 'The open source project Local Support is a directory of local charity and non-profit organisations for a small geographical area.
Our customer is the non-profit organization Voluntary Action Harrow.
The mission is to support members of the public searching for support groups for things like helping care for an elderly or sick relative; and also to help charities and non-profits find each other and network.',
      status: 'Active',
      commit_count: 100
    )
    localsupport.source_repositories.create!(url: 'https://github.com/AgileVentures/LocalSupport')

    u.projects.create!(
      title: 'EduChat',
      description: 'Supporting real time synchronous chat in online classes',
      status: 'Inactive',
      commit_count: 100
    )

    u.projects.create!(
      title: 'PP Scheduler',
      description: "Problem: Lots of people want to pair, but they don't know when each other are available
Solution: is something that requires absolutely minimal effort on their part to use in order to let them pair",
      status: 'Active'
    )

    u.projects.create!(
      title: 'Funniest Computer Ever',
      description: "Can YOU write a program to make humans laugh? Get your editors fired up and your coding caps ready because you've arrived at the Funniest Computer Ever competition!",
      status: 'Active'
    )

    puts 'Created default projects'
    break
  elsif response == 'n' || response == 'no'
    break
  end
end

u ||= User.last
3.times do
  p = u.projects.create(
    title: Faker::Lorem.words(number: 3).join(' '),
    description: Faker::Lorem.paragraph,
    status: 'active',
    created_at: 1.month.ago
  )
  3.times do
    d = p.documents.create(
      title: Faker::Lorem.words(number: 3).join(' '),
      body: Faker::Lorem.paragraph,
      created_at: 1.month.ago,
      user_id: p.user_id
    )
    3.times do
      d.children.create(
        title: Faker::Lorem.words(number: 3).join(' '),
        body: Faker::Lorem.paragraph,
        project_id: p.id,
        created_at: 1.month.ago,
        user_id: p.user_id
      )
    end
  end
end

# Premium Users
Rake::Task['db:create_plans'].invoke

plans = ["premium", "premiummob", "premiumf2f", "premiumplus"]
plans.each  do |plan|
  get_country
  u = User.create!(
    first_name: Faker::Name.first_name,
    last_name: Faker::Name.last_name,
    email: "#{plan}@premi.um",
    password: "premium123",
    country_name: @country[:country_name],
    country_code: @country[:country_code]
  )
  Subscription.create(
    started_at: DateTime.now,
    user_id: u.id,
    plan_id: Plan.find_by_third_party_identifier(plan).id
  )
end

100.times do |counter|
  get_country
  u = User.create(
    first_name: Faker::Name.first_name,
    last_name: Faker::Name.last_name,
    email: Faker::Internet.email,
    password: Faker::Lorem.characters(number: 10),
    country_name: @country[:country_name],
    country_code: @country[:country_code],
  )
  Karma.create(
    user_id: u.id,
    total: counter,
    activity: Faker::Number.digit,
    event_participation: Faker::Number.digit,
    membership_length: Faker::Number.digit,
    profile_completeness: Faker::Number.digit,
    number_github_contributions: Faker::Number.digit,
  )

  Project.all.sample(3).each do |p|
    u.follow p
  end
end

4.times do
  User.all.sample(3).each do |u|
    u.articles.create!(
      title: Faker::Lorem.words(number: 3).join(' '),
      content: Faker::Lorem.paragraphs(number: 3).join('\n')
    )
  end
end

if StaticPage.count == 0
  Rake::Task['db:import_pages'].invoke
end

# Seed data from "db/seeds" folder
Dir[File.join(Rails.root, 'db', 'seeds', '*.rb')].sort.each do |seed|
  puts "seeding from : #{seed}"
  load seed
end

# Create ghost user
Rake::Task['user:create_anonymous'].invoke

klasses.each_with_index do |klass, i|
  puts "#{klass.name}.count " + old_counts[i].to_s.bold.red + ' -> ' + klass.count.to_s.bold.green
end
