standup 0.2.0

data/lib/standup.rb

@@ -0,0 +1,54 @@
+require 'settingslogic'
+require 'AWS'
+require 'aws/s3'
+require 'net/ssh'
+require 'highline'
+
+require 'standup/core_ext'
+require 'standup/railtie'
+require 'standup/settings'
+require 'standup/ec2'
+require 'standup/remoting'
+require 'standup/scripts/base'
+require 'standup/node'
+
+module Standup
+  module Scripts; end
+  
+  def self.nodes
+    Settings.nodes.keys.map{|name| Node.new name}
+  end
+  
+  def self.gem_scripts_path
+    File.expand_path('../../scripts',  __FILE__)
+  end
+  
+  def self.local_scripts_path
+    File.expand_path('config/standup',  Rails.root) rescue nil
+  end
+  
+  def self.scripts
+    unless class_variable_defined? :@@scripts
+      @@scripts = {}
+      loaded = Set.new
+      [local_scripts_path, gem_scripts_path].each do |dir|
+        Dir.foreach dir do |name|
+          next unless File.file? "#{dir}/#{name}"
+          next unless name =~ /\.rb$/
+          next if loaded.include? name
+          load "#{dir}/#{name}", true
+          loaded << name
+        end if dir && File.exists?(dir)
+      end
+    end
+    @@scripts
+  end
+  
+  def self.script &block
+    name = block.__file__.match(/([^\/]*)\.rb$/)[1]
+    script_class = Class.new(Standup::Scripts::Base, &block)
+    script_class.name = name
+    Standup::Scripts.const_set name.camelize, script_class
+    scripts[name] = script_class
+  end
+end

data/lib/tasks/standup.rake

@@ -0,0 +1,19 @@
+desc "List standup scripts"
+task :standup do
+  puts
+  bright_p "Standup scripts list:"
+  offset = Standup.scripts.keys.map(&:length).max + 2
+  Standup.scripts.each do |name, script|
+    puts "  rake standup:\e[1m#{"%-#{offset}s" % name}\e[0m #{script.description}"
+  end
+  puts
+end
+
+namespace :standup do
+  Standup.scripts.each do |name, _|
+    desc "Run script #{name} for each node"
+    task name.to_sym do
+      Standup.nodes.each {|node| node.run_script name}
+    end
+  end
+end

data/scripts/allocate_ip.rb

@@ -0,0 +1,7 @@
+Standup.script do
+  self.description = 'Request new Elasics IP to use in node config'
+  
+  def run
+    bright_p "New IP: #{Standup::EC2::ElasticIP.create.ip}" 
+  end
+end

data/scripts/basics.rb

@@ -0,0 +1,6 @@
+Standup.script do
+  def run
+    sudo 'apt-get -qq update'
+    install_packages 'build-essential libreadline5-dev mc'
+  end
+end

data/scripts/ec2.rb

@@ -0,0 +1,60 @@
+Standup.script do
+  def run
+    ensure_security_group
+      
+    open_port 22
+      
+    ensure_instance
+      
+    configure_elastic_ip
+  end
+  
+  def open_ports *ports
+    ports.map(&:to_s).each do |port|
+      rule = Standup::EC2::SecurityGroup::IPRule.new('0.0.0.0/0', 'tcp', port, port)
+      group = Standup::EC2::SecurityGroup.list[node.id_group]
+      unless group.rules.include? rule
+        bright_p "opening port #{port}"
+        group.add_rule rule
+      end
+    end
+  end
+  alias :open_port :open_ports
+  
+  protected
+  
+  def ensure_security_group
+    return if Standup::EC2::SecurityGroup.list[node.id_group]
+    
+    bright_p "creating security group #{node.id_group}"
+    Standup::EC2::SecurityGroup.create node.id_group
+  end
+  
+  def ensure_instance
+    return if instance
+    
+    bright_p "launching #{params.instance_type} instance with image #{params.image_id}"
+    inst = Standup::EC2::Instance.create params.image_id,
+                                         params.instance_type,
+                                         [Standup::EC2::SecurityGroup.list[node.id_group]]
+    puts "waiting until it's up"
+    inst.wait_until {inst.state != :running}
+  end
+  
+  def configure_elastic_ip
+    if params.elastic_ip
+      unless params.elastic_ip == instance.external_ip
+        bright_p "attaching elastic ip #{params.elastic_ip}"
+        Standup::EC2::ElasticIP.list[params.elastic_ip].attach_to instance
+        
+        puts "waiting until it's attached"
+        instance.wait_until {instance.external_ip != params.elastic_ip}
+      end
+    else
+      if ip = Standup::EC2::ElasticIP.list[instance.external_ip]
+        bright_p "detaching elastic ip #{params.elastic_ip}"
+        ip.detach
+      end
+    end
+  end
+end

data/scripts/generate/script.rb

@@ -0,0 +1,7 @@
+Standup.script do
+  self.description = 'Put description here'
+  
+  def run
+    #todo Put code here
+  end
+end

data/scripts/generate.rb

@@ -0,0 +1,14 @@
+Standup.script do
+  self.description = 'Generate script'
+  
+  def run
+    unless ENV['SCRIPT']
+      bright_p "Specify script name with SCRIPT=name argument"
+      return
+    end
+    
+    FileUtils.mkdir_p(Standup.local_scripts_path)
+    FileUtils.copy script_file('script.rb'),
+                   File.expand_path("#{ENV['SCRIPT']}.rb",  Standup.local_scripts_path)
+  end
+end

data/scripts/init/standup.yml

@@ -0,0 +1,17 @@
+aws:
+  account_id: 
+  access_key_id: 
+  secret_access_key: 
+  keypair_name: 
+  availability_zone: us-east-1a
+
+nodes:
+  main:
+    ec2:
+      image_id: ami-480df921 # Canonical Ubuntu 10.04, EBS boot
+      instance_type: m1.small
+      ssh_user: ubuntu
+      # elastic_ip: 
+    webapp:
+      github_user: 
+      github_repo: 

data/scripts/init.rb

@@ -0,0 +1,7 @@
+Standup.script do
+  self.description = 'Generate config file'
+  
+  def run
+    FileUtils.copy script_file('standup.yml'), File.expand_path('config',  Rails.root)
+  end
+end

data/scripts/localize.rb

@@ -0,0 +1,18 @@
+Standup.script do
+  self.description = 'Put specified script into project configuration'
+  
+  def run
+    unless ENV['SCRIPT']
+      bright_p "Specify script name with SCRIPT=name argument"
+      return
+    end
+    
+    FileUtils.mkdir_p(Standup.local_scripts_path)
+    [ENV['SCRIPT'], "#{ENV['SCRIPT']}.rb"].each do |path|
+      path = File.expand_path(path, Standup.gem_scripts_path)
+      if File.exists? path
+        FileUtils.cp_r path, Standup.local_scripts_path
+      end
+    end
+  end
+end

data/scripts/passenger/nginx.conf

@@ -0,0 +1,32 @@
+#user  nobody;
+worker_processes  1;
+
+events {
+    worker_connections  1024;
+}
+
+http {
+    passenger_root /usr/local/lib/ruby/gems/1.8/gems/passenger-3.0.0;
+    passenger_ruby /usr/local/bin/ruby;
+
+    include       mime.types;
+    default_type  application/octet-stream;
+
+    sendfile        on;
+
+    #keepalive_timeout  0;
+    keepalive_timeout  65;
+
+    gzip on;
+    gzip_buffers 16 8k;
+    gzip_disable "MSIE [1-6]\.";
+    gzip_proxied any;           
+    gzip_types text/plain text/css application/json application/x-javascript text/xml application/xml application/xml+rss text/javascript;
+    
+    client_max_body_size 100m;
+    
+    passenger_max_pool_size 10;
+
+    # standup place server fragments here
+    # standup place server fragments here
+}

data/scripts/passenger/upstart.conf

@@ -0,0 +1,10 @@
+description "nginx http daemon"
+
+start on runlevel [2]
+stop on runlevel [016]
+
+console output
+
+exec /opt/nginx/sbin/nginx -c /opt/nginx/conf/nginx.conf -g "daemon off;"
+
+respawn

data/scripts/passenger.rb

@@ -0,0 +1,31 @@
+Standup.script do
+  def run
+    scripts.ec2.open_port 80
+    
+    install_gem 'passenger', '3.0.0'
+    
+    unless file_exists? '/opt/nginx/sbin/nginx'
+      install_package 'libcurl4-openssl-dev'
+      sudo 'passenger-install-nginx-module --auto --auto-download --prefix=/opt/nginx'
+    end
+    
+    upload script_file('nginx.conf'),
+           :to =>'/opt/nginx/conf/nginx.conf',
+           :sudo => true
+    
+    upload script_file('upstart.conf'),
+           :to =>'/etc/init/nginx.conf',
+           :sudo => true
+    sudo 'initctl reload-configuration'
+    
+    restart_nginx
+  end
+  
+  def restart_nginx
+    if exec('initctl status nginx') =~ /running/i
+      sudo 'initctl restart nginx'
+    else
+      sudo 'initctl start nginx'
+    end
+  end
+end