lib/sunzi/cli.rb in sunzi-0.0.1 vs lib/sunzi/cli.rb in sunzi-0.1.0

@@ -1,6 +1,9 @@
 require 'thor'
+require 'yaml'
+require 'fileutils'
+require 'open3'
 
 module Sunzi
   CONFIG_DIR = File.join(ENV['HOME'],'.config','sunzi')
 
   class Cli < Thor
@@ -11,20 +14,66 @@
         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}/here"
-      empty_directory "#{project}/there"
-      empty_directory "#{project}/there/recipes"
-      template "templates/here/attributes.yml",       "#{project}/here/attributes.yml"
-      template "templates/here/compile.rb",           "#{project}/here/compile.rb"
-      template "templates/here/deploy.sh",            "#{project}/here/deploy.sh"
-      template "templates/there/install.sh",          "#{project}/there/install.sh"
-      template "templates/there/recipes/ssh_key.sh",  "#{project}/there/recipes/ssh_key.sh"
+      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 [e-mail]"
+        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