require 'thor'
require 'yaml'
require 'fileutils'
require 'open3'

module Sunzi
  CONFIG_DIR = File.join(ENV['HOME'],'.config','sunzi')

  class Cli < Thor
    include Thor::Actions

    class << self
      def source_root
        File.expand_path('../../',__FILE__)
      end
    end

    map "c" => :create
    map "d" => :deploy

    desc "create [PROJECT]", "Create sunzi project (Shortcut: c)"
    def create(project = 'sunzi')
      empty_directory project
      empty_directory "#{project}/remote"
      empty_directory "#{project}/remote/recipes"
      template "templates/attributes.yml",            "#{project}/attributes.yml"
      template "templates/remote/install.sh",         "#{project}/remote/install.sh"
      template "templates/remote/recipes/ssh_key.sh", "#{project}/remote/recipes/ssh_key.sh"
    end

    desc "deploy [USER@HOST] [PORT]", "Deploy sunzi project (Shortcut: d)"
    def deploy(*target)
      if target.empty? or !target.first.match(/@/)
        puts "Usage: sunzi deploy root@example.com"
        abort
      end

      # Compile attributes.yml
      unless File.exists?('attributes.yml')
        puts "You must be in the sunzi folder"
        abort
      end
      hash = YAML.load(File.read('attributes.yml'))
      FileUtils.mkdir_p('remote/attributes')
      hash.each do |key, value|
        File.open("remote/attributes/#{key}", 'w'){|file| file.write(value) }
      end

      host, port = target
      port ||= 22
      user, domain = host.split('@')

      commands = <<-EOS
      ssh-keygen -R #{domain}
      cd remote
      tar cz . | ssh -o 'StrictHostKeyChecking no' #{host} -p #{port} '
      rm -rf ~/sunzi &&
      mkdir ~/sunzi &&
      cd ~/sunzi &&
      tar xz &&
      bash install.sh'
      EOS

      Open3.popen3(commands) do |stdin, stdout, stderr|
        stdin.close
        t = Thread.new(stderr) do |terr|
          while (line = terr.gets)
            print "\e[31m#{line}\e[0m"
          end
        end
        while (line = stdout.gets)
          print "\e[32m#{line}\e[0m"
        end
        t.join
      end
    end

  end
end