#!/usr/bin/env ruby
require 'date'

models_dir = 'models'
model_name = (ARGV[0] or "default")
model_path = File.join(models_dir, model_name)

episode_count = (File.read File.join(model_path, 'episodes_trained')).to_i

puts "Found model #{model_name} with episodes #{episode_count} trained!"

file_name = "model-#{model_name}-#{episode_count}-#{Time.now.strftime('%Y%m%d-%H%M%S')}.tar.gz"
save_path = File.join(models_dir, 'saves', file_name)
puts "Saving to #{save_path}"

`tar -cvzf #{save_path} -C #{models_dir} #{model_name}`