require 'rubygems'
require 'bonethug/conf'

module Bonethug

  class Syncer

    def self.sync(pull, pull_env, push, push_env)

      # exec env
      exec_path   = File.expand_path('.')

      # load config
      conf = Bonethug::Conf.new.add(exec_path + '/config/cnf.yml')
      conf.add(exec_path + '/config/database.yml' => { root: 'dbs.default' }) if File.exist? exec_path + '/config/database.yml'

      # extract some data
      cnf  = conf.to_hash
      envs = conf.get('deploy.environments').to_hash

      # validate
      unless pull and pull_env and push and push_env
        puts 'Usage: syncer.rb [pull target] [pull environment] [push target] [push environment]'
        exit
      end

      unless envs.has_key? pull_env
        puts 'could not find pull environment'
        exit
      end

      unless envs.has_key? push_env
        puts 'could not find push environment'
        exit
      end

      if pull == 'local' and push == 'local'
        puts 'local to local sync is not supported at this time'
        exit
      end

      if pull == 'remote' and push == 'remote'
        puts 'remote to remote sync is not supported via this interface'
        exit
      end

      # identify the local and the remote
      env_local  = pull == 'local' ? pull_env : push_env
      env_remote = pull == 'local' ? push_env : pull_env

      # build config
      dbs           = conf.get 'dbs'
      remote_deploy = conf.node_merge 'deploy.common', 'deploy.environments.' + env_remote
      local_deploy  = conf.node_merge 'deploy.common', 'deploy.environments.' + env_local
      resources     = conf.get('resources','Array') || []
      log_dirs      = conf.get('log_dirs','Array') || []
      remote_vhost  = remote_deploy.get('project_slug') + '_' + env_remote

      # directories we need to track
      resources += ['backups']

      # logs
      log_dirs.push 'log' unless log_dirs.include? 'log'

      # do the common work
      remote_path = remote_deploy.get('base_dir') + '/' + remote_vhost
      remote_ssh  = "ssh -p #{remote_deploy.get 'port'} #{remote_deploy.get 'user'}@#{remote_deploy.get 'domain'}"

      # output
      puts "Cloning Databases... "

      # output
      dbs.each do |index, db|

        db_remote = db.get env_remote
        db_local  = db.get env_local

        if push == 'local' and pull == 'remote'
          cmd = "#{remote_ssh} \"mysqldump -u #{db_remote.get 'user'} -p#{db_remote.get 'pass'} #{db_remote.get 'name'} --verbose --single-transaction -CceKq | bzip2 -c\" | bunzip2 -c | mysql -u #{db_local.get 'user'} -p#{db_local.get 'pass'} #{db_local.get 'name'}"
        elsif pull == 'local' and push == 'remote'
          cmd = "mysqldump -u #{db_local.get 'user'} -p#{db_local.get 'pass'} #{db_local.get 'name'} --verbose --single-transaction -CceKq | bzip2 -c | #{remote_ssh} \"bunzip2 -c | mysql -u #{db_remote.get 'user'} -p#{db_remote.get 'pass'} #{db_remote.get 'name'}\""
        end

        puts cmd
        system cmd

      end

      puts "Done."
      puts "Syncing Files... "

      # sync the files
      (resources + log_dirs).each do |item|

        if push == 'local' and pull == 'remote'
          cmd = "rsync -avz -e \"ssh -p #{remote_deploy.get('port')} -l #{remote_deploy.get('user')}\" --delete --progress #{remote_deploy.get('domain')}:#{remote_path}/current/#{item}/ #{exec_path}/#{item}/"
        elsif pull == 'local' and push == 'remote'
          cmd = "rsync -avz -e \"ssh -p #{remote_deploy.get('port')} -l #{remote_deploy.get('user')}\" --delete --progress #{exec_path}/#{item}/ #{remote_deploy.get('domain')}:#{remote_path}/current/#{item}/"
        end

        puts cmd
        system cmd

      end

      puts "Done."

    end

  end

end