export_submissions

#!/usr/bin/env ruby
# frozen_string_literal: true

root_dir = File.expand_path('../', File.dirname(__FILE__))
require root_dir + '/config/environment'
require 'fileutils'

if ![1, 2].include?(ARGV.length) || ['-h', '--help'].include?(ARGV[0])
  puts 'Usage: script/export_submissions dest_dir [course_name]'
  puts
  exit(false)
end

dest_dir = ARGV[0]

def write_file(filename, data)
  File.open(filename, 'wb') { |f| f.write(data) }
end

query = Submission.order(:id)
if ARGV[1]
  begin
    course = Course.find_by!(name: ARGV[1])
  rescue StandardError
    puts "Couldn't find course: #{ARGV[1]}"
    exit(false)
  end
  query = query.where(course_id: course.id)
end

total = query.count

if total == 0
  puts '0 submissions found'
  exit(true)
end

i = 1
query.each do |sub|
  puts "Exporting submission id=#{sub.id} for #{sub.exercise_name}  (#{i}/#{total})"
  FileUtils.mkdir_p("#{dest_dir}/#{sub.exercise_name}")
  dest_file = "#{dest_dir}/#{sub.exercise_name}/#{sub.id}.zip"
  write_file(dest_file, sub.return_file)
  i += 1
end