sunzi 0.4.0 → 0.4.1

data/README.md

@@ -5,13 +5,13 @@ Sunzi
 "The supreme art of war is to subdue the enemy without fighting." - Sunzi
 ```
 
-Sunzi is the easiest server provisioning utility designed for mere mortals. If Chef or Puppet is driving you nuts, try Sunzi!
+Sunzi is the easiest [server provisioning](http://en.wikipedia.org/wiki/Provisioning#Server_provisioning) utility designed for mere mortals. If Chef or Puppet is driving you nuts, try Sunzi!
 
 Sunzi assumes that modern Linux distributions have (mostly) sane defaults and great package managers.
 
 Its design goals are:
 
-* **It's just shell script.** No clunky Ruby DSL involved. Sunzi recipes are written in a plain shell script. Why? Because, most of the information about server configuration on the web is written in shell commands. Just copy-paste them, why should you translate it into a proprietary, inconvenient DSL? Also, shell script is the greatest common denominator on minimum Linux installs.
+* **It's just shell script.** No clunky Ruby DSL involved. Sunzi recipes are written in a plain shell script. Most of the information about server configuration on the web is written in shell commands. Just copy-paste them, rather than translate it into an arbitrary DSL. Also, Bash is the greatest common denominator on minimum Linux installs.
 * **Focus on diff from default.** No big-bang overwriting. Append or replace the smallest possible piece of data in a config file. Loads of custom configurations make it difficult to understand what you are really doing.
 * **Always use the root user.** Think twice before blindly assuming you need a regular user - it doesn't add any security benefit for server provisioning, it just adds extra verbosity for nothing. However, it doesn't mean that you shouldn't create regular users with Sunzi - feel free to write your own recipes.
 * **Minimum dependencies.** No configuration server required. You don't even need a Ruby runtime on the remote server.
@@ -21,26 +21,46 @@ Quickstart
 
 Install:
 
-    $ gem install sunzi
+```bash
+gem install sunzi
+```
 
 Go to your project directory, then:
 
-    $ sunzi create
+```bash
+sunzi create
+```
 
 It generates a `sunzi` folder along with subdirectories and templates. Inside `sunzi`, there's `sunzi.yml`, which defines dynamic attributes to be used from recipes. Also there's the `remote` folder, which will be transferred to the remote server, that contains recipes and dynamic variables compiled from `sunzi.yml`.
 
-Go into the `sunzi` directory, then run the `sunzi deploy`:
+Go into the `sunzi` directory, then run `sunzi deploy`:
 
-    $ cd sunzi
-    $ sunzi deploy example.com
+```bash
+cd sunzi
+sunzi deploy example.com
+```
 
 Now, what it actually does is:
 
+1. Compile sunzi.yml to generate attributes and retrieve remote recipes
 1. SSH to `example.com` and login as `root`
 1. Transfer the content of the `remote` directory to the remote server and extract in `$HOME/sunzi`
 1. Run `install.sh` on the remote server
 
-As you can see, what you need to do is edit `install.sh` and add some shell commands. That's it.
+As you can see, all you need to do is edit `install.sh` and add some shell commands. That's it.
+
+A Sunzi project with no recipes is totally fine, so that you can start small, go big later.
+
+Commands
+--------
+
+```bash
+sunzi               # Show command help
+sunzi create        # Create a new Sunzi project
+sunzi deploy        # Deploy Sunzi project
+sunzi setup         # Setup a new VM on the Cloud services
+sunzi teardown      # Teardown an existing VM on the Cloud services
+```
 
 Directory structure
 -------------------
@@ -75,11 +95,12 @@ For instance, given a recipe `greeting.sh`:
 echo "Goodbye $1, Hello $2!"
 ```
 
-With `attributes.yml`:
+With `sunzi.yml`:
 
 ```yaml
-goodbye: Chef
-hello: Sunzi
+attributes:
+  goodbye: Chef
+  hello: Sunzi
 ```
 
 Then, include the recipe in `install.sh`:
@@ -111,12 +132,16 @@ recipes:
 Cloud Support
 -------------
 
-You can setup a new VM / teardown an existing VM interactively. Use `sunzi setup` and `sunzi teardown` for that.
+You can setup a new VM, or teardown an existing VM interactively. Use `sunzi setup` and `sunzi teardown` for that.
 
 The following screenshot says it all.
 
 ![Sunzi for Linode](http://farm8.staticflickr.com/7210/6783789868_ab89010d5c.jpg)
 
+Right now, only [Linode](http://www.linode.com/) is supported, but EC2 and Rackspace are coming.
+
+For DNS, Linode and [Amazon Route 53](http://aws.amazon.com/route53/) are supported.
+
 Vagrant
 -------
 
@@ -144,4 +169,6 @@ and now run `vagrant up`, it will change the root password to `vagrant`.
 
 Also keep in mind that you need to specify the port number 2222.
 
-    $ sunzi deploy localhost:2222
+```bash
+sunzi deploy localhost:2222
+```

data/bin/sunzi

@@ -1,4 +1,8 @@
 #!/usr/bin/env ruby
+
+# Abort beautifully with ctrl+c.
 Signal.trap(:INT) { abort "\nAborting." }
+
+# Load the main lib and invoke CLI.
 require File.expand_path('../../lib/sunzi',__FILE__)
 Sunzi::Cli.start

data/lib/sunzi.rb

@@ -1,11 +1,14 @@
 LIB_PATH = File.join(File.dirname(__FILE__), 'sunzi')
 
 require 'thor'
+require 'rainbow'
 require 'yaml'
 
 module Sunzi
   autoload :Cli,        File.join(LIB_PATH, 'cli')
   autoload :Dependency, File.join(LIB_PATH, 'dependency')
+  autoload :Logger,     File.join(LIB_PATH, 'logger')
+  autoload :Utility,    File.join(LIB_PATH, 'utility')
   autoload :Version,    File.join(LIB_PATH, 'version')
 
   module Cloud

data/lib/sunzi/cli.rb

@@ -1,100 +1,106 @@
 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/create/sunzi.yml",                 "#{project}/sunzi.yml"
-      template "templates/create/remote/install.sh",         "#{project}/remote/install.sh"
-      template "templates/create/remote/recipes/ssh_key.sh", "#{project}/remote/recipes/ssh_key.sh"
+      do_create(project)
     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
+      do_deploy(target)
     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
+      do_compile
     end
 
     desc "setup [linode|ec2]", "Setup a new VM"
     def setup(target)
-      Cloud::Base.choose(target).setup
+      Cloud::Base.choose(self, target).setup
     end
 
     desc "teardown [linode|ec2] [name]", "Teardown an existing VM"
     def teardown(target, name)
-      Cloud::Base.choose(target).teardown(name)
+      Cloud::Base.choose(self, target).teardown(name)
     end
 
     no_tasks do
+      include Sunzi::Utility
+
+      def self.source_root
+        File.expand_path('../../',__FILE__)
+      end
+
+      def do_create(project)
+        empty_directory project
+        empty_directory "#{project}/remote"
+        empty_directory "#{project}/remote/recipes"
+        template "templates/create/sunzi.yml",                 "#{project}/sunzi.yml"
+        template "templates/create/remote/install.sh",         "#{project}/remote/install.sh"
+        template "templates/create/remote/recipes/ssh_key.sh", "#{project}/remote/recipes/ssh_key.sh"
+      end
+
+      def do_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
+
+      def do_compile
+        # Check if you're in the sunzi directory
+        unless File.exists?('sunzi.yml')
+          abort_with "You must be in the sunzi folder"
+        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
+
       def parse_target(target)
         target.match(/(.*@)?(.*?)(:.*)?$/)
         [ ($1 && $1.delete('@') || 'root'), $2, ($3 && $3.delete(':') || '22') ]

data/lib/sunzi/cloud/base.rb

@@ -2,36 +2,27 @@ Sunzi::Dependency.load('highline')
 
 module Sunzi
   module Cloud
-    class Base < Thor
-      include Thor::Actions
+    class Base
+      include Sunzi::Utility
 
-      class << self
-        def source_root
-          File.expand_path('../../../',__FILE__)
-        end
-
-        def choose(target)
-          case target
-          when 'linode'
-            Cloud::Linode.new
-          when 'ec2'
-            Cloud::EC2.new
-          else
-            say shell.set_color("#{target} is not valid!", :red, true)
-            abort
-          end
+      def self.choose(cli, target)
+        case target
+        when 'linode'
+          Cloud::Linode.new(cli)
+        when 'ec2'
+          Cloud::EC2.new(cli)
+        else
+          abort_with "#{target} is not valid!"
         end
       end
 
-      def initialize(*args)
+      def initialize(cli)
+        @cli = cli
         @ui = HighLine.new
-        super
       end
 
-      no_tasks do
-        def ask(question, answer_type, &details)
-          @ui.ask(@ui.color(question, :green, :bold), answer_type, &details)
-        end
+      def ask(question, answer_type, &details)
+        @ui.ask(@ui.color(question, :green, :bold), answer_type, &details)
       end
     end
   end

data/lib/sunzi/cloud/ec2.rb

@@ -1,14 +1,12 @@
 module Sunzi
   module Cloud
     class EC2 < Base
-      no_tasks do
-        def setup
-          say shell.set_color('EC2 is not implemented yet!', :red, true)
-        end
+      def setup
+        Logger.error 'EC2 is not implemented yet!'
+      end
 
-        def teardown(target)
-          say shell.set_color('EC2 is not implemented yet!', :red, true)
-        end
+      def teardown(target)
+        Logger.error 'EC2 is not implemented yet!'
       end
     end
   end

data/lib/sunzi/cloud/linode.rb

@@ -6,243 +6,236 @@ YAML::ENGINE.yamler = 'syck'
 module Sunzi
   module Cloud
     class Linode < Base
-      no_tasks do
-
-        def setup
-          # Only run for the first time
-          unless File.exist? 'linode/linode.yml'
-            empty_directory 'linode'
-            empty_directory 'linode/instances'
-            template 'templates/setup/linode/linode.yml', 'linode/linode.yml'
-            say shell.set_color('Now go ahead and edit linode.yml, then run this command again!', :green, true)
-            abort
-          end
-
-          @config = YAML.load(File.read('linode/linode.yml'))
-
-          if @config['fqdn']['zone'] == 'example.com'
-            say shell.set_color('You must have your own settings in linode.yml', :red, true)
-            abort
-          end
-
-          # When route53 is specified for DNS, check if it's properly configured and if not, fail earlier.
-          setup_route53 if @config['dns'] == 'route53'
-
-          @ui = HighLine.new
-
-          @sshkey = File.read(File.expand_path(@config['root_sshkey_path'])).chomp
-          if @sshkey.match(/\n/)
-            say shell.set_color("RootSSHKey #{@sshkey.inspect} must not be multi-line! Check inside \"#{@config['root_sshkey_path']}\"", :red, true)
-            abort
-          end
-
-          # Ask environment and hostname
-          @env = ask("environment? (#{@config['environments'].join(' / ')}): ", String) {|q| q.in = @config['environments'] }.to_s
-          @host = ask('hostname? (only the last part of subdomain): ', String).to_s
-
-          @fqdn = @config['fqdn'][@env].gsub(/%{host}/, @host)
-          @label = @config['label'][@env].gsub(/%{host}/, @host)
-          @group = @config['group'][@env]
-          @api = ::Linode.new(:api_key => @config['api_key'])
-
-          # Choose a plan
-          result = @api.avail.linodeplans
-          result.each{|i| say "#{i.planid}: #{i.ram}MB, $#{i.price}" }
-          @planid = ask('which plan?: ', Integer) {|q| q.in = result.map(&:planid); q.default = result.first.planid }
-          @plan_label = result.find{|i| i.planid == @planid }.label
-
-          # Choose a datacenter
-          result = @api.avail.datacenters
-          result.each{|i| say "#{i.datacenterid}: #{i.location}" }
-          @datacenterid = ask('which datacenter?: ', Integer) {|q| q.in = result.map(&:datacenterid); q.default = result.first.datacenterid }
-          @datacenter_location = result.find{|i| i.datacenterid == @datacenterid }.location
-
-          # Choose a distribution
-          result = @api.avail.distributions
-          if @config['distributions_filter']
-            result = result.select{|i| i.label.match Regexp.new(@config['distributions_filter'], Regexp::IGNORECASE) }
-          end
-          result.each{|i| say "#{i.distributionid}: #{i.label}" }
-          @distributionid = ask('which distribution?: ', Integer) {|q| q.in = result.map(&:distributionid); q.default = result.first.distributionid }
-          @distribution_label = result.find{|i| i.distributionid == @distributionid }.label
-
-          # Choose a kernel
-          result = @api.avail.kernels
-          if @config['kernels_filter']
-            result = result.select{|i| i.label.match Regexp.new(@config['kernels_filter'], Regexp::IGNORECASE) }
-          end
-          result.each{|i| say "#{i.kernelid}: #{i.label}" }
-          @kernelid = ask('which kernel?: ', Integer) {|q| q.in = result.map(&:kernelid); q.default = result.first.kernelid }
-          @kernel_label = result.find{|i| i.kernelid == @kernelid }.label
-
-          # Choose swap size
-          @swap_size = ask('swap size in MB? (default: 256MB): ', Integer) { |q| q.default = 256 }
-
-          # Go ahead?
-          moveon = ask("Are you sure to go ahead and create #{@fqdn}? (y/n) ", String) {|q| q.in = ['y','n']}
-          exit unless moveon == 'y'
-
-          # Create
-          say "creating a new linode..."
-          result = @api.linode.create(:DatacenterID => @datacenterid, :PlanID => @planid, :PaymentTerm => @config['payment_term'])
-          @linodeid = result.linodeid
-          say "created a new instance: linodeid = #{@linodeid}"
-
-          result = @api.linode.list.select{|i| i.linodeid == @linodeid }.first
-          @totalhd = result.totalhd
-
-          # Update settings
-          say "Updating settings..."
-          result = @api.linode.update(
-            :LinodeID => @linodeid,
-            :Label => @label,
-            :lpm_displayGroup => @group,
-            # :Alert_cpu_threshold => 90,
-            # :Alert_diskio_threshold => 1000,
-            # :Alert_bwin_threshold => 5,
-            # :Alert_bwout_threshold => 5,
-            # :Alert_bwquota_threshold => 80,
-          )
-
-          # Create a root disk
-          say "Creating a root disk..."
-          result = @api.linode.disk.createfromdistribution(
-            :LinodeID => @linodeid,
-            :DistributionID => @distributionid,
-            :Label => "#{@distribution_label} Image",
-            :Size => @totalhd - @swap_size,
-            :rootPass => @config['root_pass'],
-            :rootSSHKey => @sshkey
-          )
-          @root_diskid = result.diskid
-
-          # Create a swap disk
-          say "Creating a swap disk..."
-          result = @api.linode.disk.create(
-            :LinodeID => @linodeid,
-            :Label => "#{@swap_size}MB Swap Image",
-            :Type => 'swap',
-            :Size => @swap_size
-          )
-          @swap_diskid = result.diskid
-
-          # Create a config profiile
-          say "Creating a config profile..."
-          result = @api.linode.config.create(
-            :LinodeID => @linodeid,
-            :KernelID => @kernelid,
-            :Label => "#{@distribution_label} Profile",
-            :DiskList => [ @root_diskid, @swap_diskid ].join(',')
-          )
-          @config_id = result.configid
-
-          # Add a private IP
-          say "Adding a private IP..."
-          result = @api.linode.ip.list(:LinodeID => @linodeid)
-          @public_ip = result.first.ipaddress
-          result = @api.linode.ip.addprivate(:LinodeID => @linodeid)
-          result = @api.linode.ip.list(:LinodeID => @linodeid).find{|i| i.ispublic == 0 }
-          @private_ip = result.ipaddress
-
-          # Register IP to DNS
-          case @config['dns']
-          when 'linode'
-            # Set the public IP to Linode DNS Manager
-            say "Setting the public IP to Linode DNS Manager..."
-            @domainid = @api.domain.list.find{|i| i.domain == @config['fqdn']['zone'] }.domainid
-            @api.domain.resource.create(:DomainID => @domainid, :Type => 'A', :Name => @fqdn, :Target => @public_ip)
-          when 'route53'
-            # Set the public IP to AWS Route 53
-            say "Setting the public IP to AWS Route 53..."
-            Route53::DNSRecord.new(@fqdn, "A", "300", [@public_ip], @route53_zone).create
-          end
-
-          # Boot
-          say shell.set_color("Done. Booting...", :green, true)
-          @api.linode.boot(:LinodeID => @linodeid)
-
-          hash = {
-            :linode_id => @linodeid,
-            :env => @env,
-            :host => @host,
-            :fqdn => @fqdn,
-            :label => @label,
-            :group => @group,
-            :plan_id => @planid,
-            :datacenter_id => @datacenterid,
-            :datacenter_location => @datacenter_location,
-            :distribution_id => @distributionid,
-            :distribution_label => @distribution_label,
-            :kernel_id => @kernelid,
-            :kernel_label => @kernel_label,
-            :swap_size => @swap_size,
-            :totalhd => @totalhd,
-            :root_diskid => @root_diskid,
-            :swap_diskid => @swap_diskid,
-            :config_id => @config_id,
-            :public_ip => @public_ip,
-            :private_ip => @private_ip,
-          }
-
-          File.open("linode/instances/#{@label}.yml",'w') do |file|
-            file.write YAML.dump(hash)
-          end
+      def setup
+        # Only run for the first time
+        unless File.exist? 'linode/linode.yml'
+          @cli.empty_directory 'linode'
+          @cli.empty_directory 'linode/instances'
+          @cli.template 'templates/setup/linode/linode.yml', 'linode/linode.yml'
+          exit_with 'Now go ahead and edit linode.yml, then run this command again!'
         end
 
-        def teardown(name)
-          unless File.exist?("linode/instances/#{name}.yml")
-            say shell.set_color("#{name}.yml was not found in the instances directory.", :red, true)
-            abort
-          end
-          @config = YAML.load(File.read('linode/linode.yml'))
-          setup_route53 if @config['dns'] == 'route53'
-
-          @instance = YAML.load(File.read("linode/instances/#{name}.yml"))
-          @api = ::Linode.new(:api_key => @config['api_key'])
-
-          # Shutdown first or disk deletion will fail
-          say shell.set_color("shutting down...", :green, true)
-          @api.linode.shutdown(:LinodeID => @instance[:linode_id])
-          sleep 10
-
-          # Delete the disks. It is required - http://www.linode.com/api/linode/linode%2Edelete
-          say shell.set_color("deleting root disk...", :green, true)
-          @api.linode.disk.delete(:LinodeID => @instance[:linode_id], :DiskID => @instance[:root_diskid]) rescue nil
-          say shell.set_color("deleting swap disk...", :green, true)
-          @api.linode.disk.delete(:LinodeID => @instance[:linode_id], :DiskID => @instance[:swap_diskid]) rescue nil
-          sleep 5
-
-          # Delete the instance
-          say shell.set_color("deleting linode...", :green, true)
-          @api.linode.delete(:LinodeID => @instance[:linode_id])
-
-          # Delete DNS record
-          case @config['dns']
-          when 'linode'
-            # Set the public IP to Linode DNS Manager
-            say "deleting the public IP to Linode DNS Manager..."
-            @domainid = @api.domain.list.find{|i| i.domain == @config['fqdn']['zone'] }.domainid
-            @resource = @api.domain.resource.list(:DomainID => @domainid).find{|i| i.target == @instance[:public_ip] }
-            @api.domain.resource.delete(:DomainID => @domainid, :ResourceID => @resource.resourceid)
-          when 'route53'
-            # Set the public IP to AWS Route 53
-            say "deleting the public IP to AWS Route 53..."
-            @record = @route53_zone.get_records.find{|i| i.values.first == @instance[:public_ip] }
-            @record.delete
-          end
-
-          # Remove the instance config file
-          remove_file "linode/instances/#{name}.yml"
-
-          say shell.set_color("Done.", :green, true)
+        @config = YAML.load(File.read('linode/linode.yml'))
+
+        if @config['fqdn']['zone'] == 'example.com'
+          abort_with 'You must have your own settings in linode.yml'
+        end
+
+        # When route53 is specified for DNS, check if it's properly configured and if not, fail earlier.
+        setup_route53 if @config['dns'] == 'route53'
+
+        @ui = HighLine.new
+
+        @sshkey = File.read(File.expand_path(@config['root_sshkey_path'])).chomp
+        if @sshkey.match(/\n/)
+          abort_with "RootSSHKey #{@sshkey.inspect} must not be multi-line! Check inside \"#{@config['root_sshkey_path']}\""
+        end
+
+        # Ask environment and hostname
+        @env = ask("environment? (#{@config['environments'].join(' / ')}): ", String) {|q| q.in = @config['environments'] }.to_s
+        @host = ask('hostname? (only the last part of subdomain): ', String).to_s
+
+        @fqdn = @config['fqdn'][@env].gsub(/%{host}/, @host)
+        @label = @config['label'][@env].gsub(/%{host}/, @host)
+        @group = @config['group'][@env]
+        @api = ::Linode.new(:api_key => @config['api_key'])
+
+        # Choose a plan
+        result = @api.avail.linodeplans
+        result.each{|i| say "#{i.planid}: #{i.ram}MB, $#{i.price}" }
+        @planid = ask('which plan?: ', Integer) {|q| q.in = result.map(&:planid); q.default = result.first.planid }
+        @plan_label = result.find{|i| i.planid == @planid }.label
+
+        # Choose a datacenter
+        result = @api.avail.datacenters
+        result.each{|i| say "#{i.datacenterid}: #{i.location}" }
+        @datacenterid = ask('which datacenter?: ', Integer) {|q| q.in = result.map(&:datacenterid); q.default = result.first.datacenterid }
+        @datacenter_location = result.find{|i| i.datacenterid == @datacenterid }.location
+
+        # Choose a distribution
+        result = @api.avail.distributions
+        if @config['distributions_filter']
+          result = result.select{|i| i.label.match Regexp.new(@config['distributions_filter'], Regexp::IGNORECASE) }
+        end
+        result.each{|i| say "#{i.distributionid}: #{i.label}" }
+        @distributionid = ask('which distribution?: ', Integer) {|q| q.in = result.map(&:distributionid); q.default = result.first.distributionid }
+        @distribution_label = result.find{|i| i.distributionid == @distributionid }.label
+
+        # Choose a kernel
+        result = @api.avail.kernels
+        if @config['kernels_filter']
+          result = result.select{|i| i.label.match Regexp.new(@config['kernels_filter'], Regexp::IGNORECASE) }
+        end
+        result.each{|i| say "#{i.kernelid}: #{i.label}" }
+        @kernelid = ask('which kernel?: ', Integer) {|q| q.in = result.map(&:kernelid); q.default = result.first.kernelid }
+        @kernel_label = result.find{|i| i.kernelid == @kernelid }.label
+
+        # Choose swap size
+        @swap_size = ask('swap size in MB? (default: 256MB): ', Integer) { |q| q.default = 256 }
+
+        # Go ahead?
+        moveon = ask("Are you sure to go ahead and create #{@fqdn}? (y/n) ", String) {|q| q.in = ['y','n']}
+        exit unless moveon == 'y'
+
+        # Create
+        say "creating a new linode..."
+        result = @api.linode.create(:DatacenterID => @datacenterid, :PlanID => @planid, :PaymentTerm => @config['payment_term'])
+        @linodeid = result.linodeid
+        say "created a new instance: linodeid = #{@linodeid}"
+
+        result = @api.linode.list.select{|i| i.linodeid == @linodeid }.first
+        @totalhd = result.totalhd
+
+        # Update settings
+        say "Updating settings..."
+        result = @api.linode.update(
+          :LinodeID => @linodeid,
+          :Label => @label,
+          :lpm_displayGroup => @group,
+          # :Alert_cpu_threshold => 90,
+          # :Alert_diskio_threshold => 1000,
+          # :Alert_bwin_threshold => 5,
+          # :Alert_bwout_threshold => 5,
+          # :Alert_bwquota_threshold => 80,
+        )
+
+        # Create a root disk
+        say "Creating a root disk..."
+        result = @api.linode.disk.createfromdistribution(
+          :LinodeID => @linodeid,
+          :DistributionID => @distributionid,
+          :Label => "#{@distribution_label} Image",
+          :Size => @totalhd - @swap_size,
+          :rootPass => @config['root_pass'],
+          :rootSSHKey => @sshkey
+        )
+        @root_diskid = result.diskid
+
+        # Create a swap disk
+        say "Creating a swap disk..."
+        result = @api.linode.disk.create(
+          :LinodeID => @linodeid,
+          :Label => "#{@swap_size}MB Swap Image",
+          :Type => 'swap',
+          :Size => @swap_size
+        )
+        @swap_diskid = result.diskid
+
+        # Create a config profiile
+        say "Creating a config profile..."
+        result = @api.linode.config.create(
+          :LinodeID => @linodeid,
+          :KernelID => @kernelid,
+          :Label => "#{@distribution_label} Profile",
+          :DiskList => [ @root_diskid, @swap_diskid ].join(',')
+        )
+        @config_id = result.configid
+
+        # Add a private IP
+        say "Adding a private IP..."
+        result = @api.linode.ip.list(:LinodeID => @linodeid)
+        @public_ip = result.first.ipaddress
+        result = @api.linode.ip.addprivate(:LinodeID => @linodeid)
+        result = @api.linode.ip.list(:LinodeID => @linodeid).find{|i| i.ispublic == 0 }
+        @private_ip = result.ipaddress
+
+        # Register IP to DNS
+        case @config['dns']
+        when 'linode'
+          # Set the public IP to Linode DNS Manager
+          say "Setting the public IP to Linode DNS Manager..."
+          @domainid = @api.domain.list.find{|i| i.domain == @config['fqdn']['zone'] }.domainid
+          @api.domain.resource.create(:DomainID => @domainid, :Type => 'A', :Name => @fqdn, :Target => @public_ip)
+        when 'route53'
+          # Set the public IP to AWS Route 53
+          say "Setting the public IP to AWS Route 53..."
+          Route53::DNSRecord.new(@fqdn, "A", "300", [@public_ip], @route53_zone).create
         end
 
-        def setup_route53
-          Sunzi::Dependency.load('route53')
-          route53 = Route53::Connection.new(@config['route53']['key'], @config['route53']['secret'])
-          @route53_zone = route53.get_zones.find{|i| i.name.sub(/\.$/,'') == @config['fqdn']['zone'] }
+        # Boot
+        say 'Done. Booting...'
+        @api.linode.boot(:LinodeID => @linodeid)
+
+        hash = {
+          :linode_id => @linodeid,
+          :env => @env,
+          :host => @host,
+          :fqdn => @fqdn,
+          :label => @label,
+          :group => @group,
+          :plan_id => @planid,
+          :datacenter_id => @datacenterid,
+          :datacenter_location => @datacenter_location,
+          :distribution_id => @distributionid,
+          :distribution_label => @distribution_label,
+          :kernel_id => @kernelid,
+          :kernel_label => @kernel_label,
+          :swap_size => @swap_size,
+          :totalhd => @totalhd,
+          :root_diskid => @root_diskid,
+          :swap_diskid => @swap_diskid,
+          :config_id => @config_id,
+          :public_ip => @public_ip,
+          :private_ip => @private_ip,
+        }
+
+        File.open("linode/instances/#{@label}.yml",'w') do |file|
+          file.write YAML.dump(hash)
         end
       end
+
+      def teardown(name)
+        unless File.exist?("linode/instances/#{name}.yml")
+          abort_with "#{name}.yml was not found in the instances directory."
+        end
+        @config = YAML.load(File.read('linode/linode.yml'))
+        setup_route53 if @config['dns'] == 'route53'
+
+        @instance = YAML.load(File.read("linode/instances/#{name}.yml"))
+        @api = ::Linode.new(:api_key => @config['api_key'])
+
+        # Shutdown first or disk deletion will fail
+        say 'shutting down...'
+        @api.linode.shutdown(:LinodeID => @instance[:linode_id])
+        sleep 10
+
+        # Delete the disks. It is required - http://www.linode.com/api/linode/linode%2Edelete
+        say 'deleting root disk...'
+        @api.linode.disk.delete(:LinodeID => @instance[:linode_id], :DiskID => @instance[:root_diskid]) rescue nil
+        say 'deleting swap disk...'
+        @api.linode.disk.delete(:LinodeID => @instance[:linode_id], :DiskID => @instance[:swap_diskid]) rescue nil
+        sleep 5
+
+        # Delete the instance
+        say 'deleting linode...'
+        @api.linode.delete(:LinodeID => @instance[:linode_id])
+
+        # Delete DNS record
+        case @config['dns']
+        when 'linode'
+          # Set the public IP to Linode DNS Manager
+          say "deleting the public IP to Linode DNS Manager..."
+          @domainid = @api.domain.list.find{|i| i.domain == @config['fqdn']['zone'] }.domainid
+          @resource = @api.domain.resource.list(:DomainID => @domainid).find{|i| i.target == @instance[:public_ip] }
+          @api.domain.resource.delete(:DomainID => @domainid, :ResourceID => @resource.resourceid)
+        when 'route53'
+          # Set the public IP to AWS Route 53
+          say "deleting the public IP to AWS Route 53..."
+          @record = @route53_zone.get_records.find{|i| i.values.first == @instance[:public_ip] }
+          @record.delete if @record
+        end
+
+        # Remove the instance config file
+        @cli.remove_file "linode/instances/#{name}.yml"
+
+        say 'Done.'
+      end
+
+      def setup_route53
+        Sunzi::Dependency.load('route53')
+        route53 = Route53::Connection.new(@config['route53']['key'], @config['route53']['secret'])
+        @route53_zone = route53.get_zones.find{|i| i.name.sub(/\.$/,'') == @config['fqdn']['zone'] }
+      end
     end
   end
 end

data/lib/sunzi/dependency.rb

@@ -2,18 +2,9 @@ module Sunzi
   class Dependency
     def self.all
       {
-        'linode' => {
-          :require => 'linode',
-          :version => '>= 0.7.7',
-        },
-        'highline' => {
-          :require => 'highline',
-          :version => '>= 1.6.11',
-        },
-        'route53' => {
-          :require => 'route53',
-          :version => '>= 0.2.1',
-        },
+        'linode' =>   { :require => 'linode',   :version => '>= 0.7.7' },
+        'highline' => { :require => 'highline', :version => '>= 1.6.11'},
+        'route53' =>  { :require => 'route53',  :version => '>= 0.2.1' },
       }
     end
 
@@ -22,7 +13,7 @@ module Sunzi
         gem(name, all[name][:version])
         require(all[name][:require])
       rescue LoadError
-        puts <<-EOS
+        Logger.error <<-EOS
 Dependency missing: #{name}
 To install the gem, issue the following command:
 
@@ -30,9 +21,8 @@ To install the gem, issue the following command:
 
 Please try again after installing the missing dependency.
         EOS
-        exit 1
+        abort
       end
     end
-
   end
 end

data/lib/sunzi/logger.rb

@@ -0,0 +1,17 @@
+module Sunzi
+  module Logger
+    class << self
+      def info(text)
+        puts text.bright
+      end
+
+      def success(text)
+        puts text.color(:green).bright
+      end
+
+      def error(text)
+        puts text.color(:red).bright
+      end
+    end
+  end
+end

data/lib/sunzi/utility.rb

@@ -0,0 +1,17 @@
+module Sunzi
+  module Utility
+    def abort_with(text)
+      Logger.error text
+      abort
+    end
+
+    def exit_with(text)
+      Logger.success text
+      exit
+    end
+
+    def say(text)
+      Logger.info text
+    end
+  end
+end

data/lib/sunzi/version.rb

@@ -1,3 +1,3 @@
 module Sunzi
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

data/sunzi.gemspec

@@ -20,4 +20,5 @@ Gem::Specification.new do |s|
 
   # s.add_development_dependency "rspec"
   s.add_runtime_dependency "thor"
+  s.add_runtime_dependency "rainbow"
 end

data/test/test_cli.rb

@@ -13,4 +13,10 @@ class TestCli < Test::Unit::TestCase
     assert_equal ['root', 'example.com', '22'],   @cli.parse_target('example.com')
     assert_equal ['root', '192.168.0.1', '22'],   @cli.parse_target('192.168.0.1')
   end
+
+  def test_create
+    @cli.create 'sandbox'
+    assert File.exist?('sandbox/sunzi.yml')
+    FileUtils.rm_rf 'sandbox'
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sunzi
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-02-26 00:00:00.000000000 Z
+date: 2012-02-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
-  requirement: &2156460860 !ruby/object:Gem::Requirement
+  requirement: &2164966260 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,7 +21,18 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2156460860
+  version_requirements: *2164966260
+- !ruby/object:Gem::Dependency
+  name: rainbow
+  requirement: &2164965000 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2164965000
 description: Server provisioning utility for minimalists
 email:
 - kenn.ejima@gmail.com
@@ -41,6 +52,8 @@ files:
 - lib/sunzi/cloud/ec2.rb
 - lib/sunzi/cloud/linode.rb
 - lib/sunzi/dependency.rb
+- lib/sunzi/logger.rb
+- lib/sunzi/utility.rb
 - lib/sunzi/version.rb
 - lib/templates/create/remote/install.sh
 - lib/templates/create/remote/recipes/ssh_key.sh