ec2-scp

#! /usr/bin/env ruby
require "rubygems"
require "bundler/setup"
require "fog"
require "fileutils"
require 'trollop'
require 'yaml'

require_relative 'environment.rb'
require_relative 'menu.rb'
require_relative 'ssh_instance.rb'
require_relative 'scp_instance.rb'
include Environment
include Menu

SUB_COMMANDS = %w(get put)
opts = Trollop::options do
	opt :environment, "", :type => :string, :default => ""
  opt :regex, "", :type => :string
  opt :user, "", :type => :string
	stop_on SUB_COMMANDS
end	

cmd = ARGV.shift # get the subcommand
cmd_opts = case cmd.downcase
  when "get" # parse delete options
    Trollop::options do
      opt :get, "get the file", :default => true
    end
  when "put"  # parse copy options
    Trollop::options do
      opt :put, "puts the file", :default => true
    end
  else
    Trollop::die "unknown subcommand #{cmd.inspect}"
  end


environment = opts[:environment] || get_environment
settings = set_environment(environment)
regex = opts[:regex]

instance = ScpInstance.get(environment, settings, regex)

instance.user = opts[:user]

instance.remote_path = get_path('remote')
instance.local_path = get_path('local')

ssh_cmd = instance.get_cmd if !cmd_opts[:get].nil?
ssh_cmd = instance.put_cmd if !cmd_opts[:put].nil?
puts ssh_cmd
exec(ssh_cmd)