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", "Create sunzi project"
    def create(project = 'sunzi')
      empty_directory project
      empty_directory "#{project}/remote"
      empty_directory "#{project}/remote/recipes"
      template "templates/sunzi.yml",                 "#{project}/sunzi.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 example.com (or user@example.com:2222)", "Deploy sunzi project"
    def deploy(target)
      user, host, port = parse_target(target)
      endpoint = "#{user}@#{host}"

      # compile attributes and recipes
      compile

      # The host key might change when we instantiate a new VM, so
      # we remove (-R) the old host key from known_hosts.
      `ssh-keygen -R #{host} 2> /dev/null`

      commands = <<-EOS
      cd remote
      tar cz . | ssh -o 'StrictHostKeyChecking no' #{endpoint} -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 shell.set_color(line, :red, true)
          end
        end
        while (line = stdout.gets)
          print print shell.set_color(line, :green, true)
        end
        t.join
      end
    end

    desc "compile", "Compile sunzi project"
    def compile
      # Check if you're in the sunzi directory
      unless File.exists?('sunzi.yml')
        say shell.set_color("You must be in the sunzi folder", :red, true)
        abort
      end

      # Load sunzi.yml
      hash = YAML.load(File.read('sunzi.yml'))
      empty_directory 'remote/attributes'
      empty_directory 'remote/recipes'

      # Compile attributes.yml
      hash['attributes'].each do |key, value|
        File.open("remote/attributes/#{key}", 'w'){|file| file.write(value) }
      end
      # Compile recipes.yml
      hash['recipes'].each do |key, value|
        get value, "remote/recipes/#{key}.sh"
      end
    end

    no_tasks do
      def parse_target(target)
        target.match(/(.*@)?(.*?)(:.*)?$/)
        [ ($1 && $1.delete('@') || 'root'), $2, ($3 && $3.delete(':') || '22') ]
      end
    end
  end
end