akitaonrails-locarails 1.0.0 → 1.1.0

data/Manifest

@@ -6,6 +6,9 @@ README
 locarails.gemspec
 lib/locarails.rb
 lib/locarails/version.rb
-templates/database.locaweb.yml
-templates/deploy.rb
-templates/locaweb_backup.rb
+templates/database.locaweb.yml.erb
+templates/deploy.rb.erb
+templates/deploy.common.rb
+templates/locaweb_backup.rb
+templates/ssh_helper.rb
+templates/.gitignore

data/bin/locarails

@@ -2,6 +2,7 @@
 #
 #  Created on 2008-8-19.
 #  Copyright (c) 2008. All rights reserved.
+#  Fabio Akita - akitaonrails.com
 
 begin
   require 'rubygems'
@@ -11,15 +12,49 @@ end
 
 require 'ostruct'
 require 'optparse'
+require 'erb'
+
+# configuracoes da sua hospedagem
+config   = OpenStruct.new
+mysqlcfg = OpenStruct.new
+
+# defaults
+config.force = false
+config.mode = 'copy'
 
 OptionParser.new do |opts|
-  opts.banner = "Uso: #{File.basename($0)} [caminho]"
+  opts.banner = <<-STR
+LocaRails - configurador de Capistrano para projetos Rails
+especifico para hospedagens Linux da Locaweb
+
+Uso: #{File.basename($0)} [caminho] [opcoes]
+STR
 
   opts.on("-h", "--help", "Mostra esta tela de ajuda") do
     puts opts
     exit 0
   end
-
+  
+  opts.on("-m", "--mode=tipo", "modo de deployment (copy/git)", "Padrao: copy") do |mode|
+    config.mode = mode
+    unless "copy|git".include?(mode)
+      puts "#{mode} nao existe. Escolha copy ou git como --mode."
+      exit 0
+    end
+  end
+  
+  opts.on("-s", "--sshkey=caminho", "local da sua chave privada SSH", "Padrao: ~/.ssh/id_rsa") do |path|
+    config.ssh_key_path = path
+    unless File.exists?(path)
+      puts "Chave inexistente no caminho #{path}."
+      exit 0
+    end
+  end
+  
+  opts.on("-f", "--force", "Sobrescreve seu config/deploy.rb, caso exista.", "Cuidado: voce ira perder sua configuracao anterior") do
+    config.force = true
+  end
+  
   begin
     opts.parse!(ARGV)
   rescue OptionParser::ParseError => e
@@ -44,37 +79,40 @@ def get_input(message)
   gets.strip
 end
 
-# configuracoes da sua hospedagem
-config   = OpenStruct.new
-mysqlcfg = OpenStruct.new
-
 # configuracoes locais da sua maquina
 config.bin_path   = File.dirname(File.expand_path(__FILE__))
 config.local_path = File.expand_path(ARGV.shift)
 config.app_name   = config.local_path.split('/').last
 
-if File.exists?(File.join(config.local_path, 'config/deploy.rb'))
+if !config.force && File.exists?(File.join(config.local_path, 'config/deploy.rb'))
   abort "Voce ja tem capistrano configurado em config/deploy.rb. Configuracao abortada."
 end
 
 puts <<-STR
-==================================================
-Bem Vindos ao configurador de projetos da Locaweb
+========================================================
+Bem Vindos ao configurador de projetos da Locaweb.
+
 Vamos configurar seu projeto Rails para melhor se
-Adequar nas nossas hospedagens Linux compartilhada
+Adequar nas nossas hospedagens Linux compartilhada.
+
+O comando locarails deve ser executado *dentro* do
+diretorio do seu projeto, na sua maquina local.
+
 Para tanto precisaremos de algumas informacoes:
-==================================================
+========================================================
+
+Estrategia de instalacao via: #{config.mode}
 
-Garanta que a seguinte pasta contem sua aplicacao
-Rails: #{config.local_path}
+Garanta que a seguinte pasta contem sua aplicacao Rails:
+  #{config.local_path}
 
 STR
 
 # configuracoes inseridas manualmente pelo usuario
-config.app_name = get_input( "O nome da sua aplicacao" ) if config.app_name.nil? || config.app_name.empty?
-config.dominio  = get_input "O dominio do seu site (ex. teste.tempsite.ws)"
+config.app_name = get_input( "Nome da sua aplicacao" ) if config.app_name.nil? || config.app_name.empty?
+config.dominio  = get_input "Dominio do seu site (ex. teste.tempsite.ws)"
 config.usuario  = get_input "Seu usuario de hospedagem"
-mysqlcfg.db     = get_input "O nome do seu banco mysql"
+mysqlcfg.db     = get_input "Nome do seu banco mysql"
 mysqlcfg.user   = get_input "Seu usuario de mysql"
 mysqlcfg.pass   = get_input "Sua senha de mysql"
 mysqlcfg.host   = get_input "Seu servidor mysql (ex. mysqlxxx.locaweb.com.br)"
@@ -82,36 +120,40 @@ mysqlcfg.host   = get_input "Seu servidor mysql (ex. mysqlxxx.locaweb.com.br)"
 # forca rodar capistrano
 unless File.exists?("#{config.local_path}/Capfile")
   puts "- Executando Capistrano no seu projeto ..."
-  `capify .`
+  begin
+    `capify .`  
+  rescue Exception => e
+    puts <<-STR
+ERRO: Voce provavelmente nao tem Capistrano instalado
+Rode: sudo gem install capistrano
+
+STR
+  end
 end
 
 FileUtils.copy_file("#{config.bin_path}/../templates/locaweb_backup.rb", 
   "#{config.local_path}/config/locaweb_backup.rb")
-  
+
 File.open("#{config.local_path}/config/deploy.rb", 'w+') do |out| 
-  buffer = File.open("#{config.bin_path}/../templates/deploy.rb", 'r') do | f | 
-    f.read.gsub('seu.usuario', config.usuario).
-      gsub('seu.dominio',      config.dominio).
-      gsub('sua.aplicacao',    config.app_name).
-      gsub('seu.projeto',      config.local_path)
-  end
+  erb = ERB.new(File.read("#{config.bin_path}/../templates/deploy.rb.erb"))
+  buffer = erb.result(config.send(:binding))
+  buffer << File.read("#{config.bin_path}/../templates/deploy.common.rb")
   out.puts buffer
 end
 
 File.open("#{config.local_path}/config/database.locaweb.yml", 'w+') do |out| 
-  buffer = File.open("#{config.bin_path}/../templates/database.locaweb.yml", 'r') do | f | 
-    f.read.gsub('mysql.database', mysqlcfg.db).
-      gsub('mysql.username', mysqlcfg.user).
-      gsub('mysql.password', mysqlcfg.pass).
-      gsub('mysql.hostname', mysqlcfg.host)
-  end
-  out.puts buffer
+  erb = ERB.new(File.read("#{config.bin_path}/../templates/database.locaweb.yml.erb"))
+  out.puts erb.result(mysqlcfg.send(:binding))
 end
 
 puts <<-STR
 
-# Parabéns, voce terminou de configurar sua aplicacao Rails!
-# Para configurar sua hospedagem execute 'cap deploy:setup'.
-# Para subir seu projeto execute 'cap deploy'
+# Parabens, voce terminou de configurar sua aplicacao Rails!
+# Execute apenas uma vez 'cap deploy:setup' para configurar os
+  diretorios remotamente na sua hospedagem.
+# Em seguida execute 'cap deploy' para transportar sua aplicacao
+  para a hospedagem. Sempre que quiser atualizar sua aplicacao
+  no servidor apenas execute novamente 'cap deploy'
+  
 [finalizado!]
 STR

data/lib/locarails/version.rb

@@ -1,7 +1,7 @@
 module Locarails
   module VERSION #:nodoc:
     MAJOR = 1
-    MINOR = 0
+    MINOR = 1
     TINY  = 0
 
     STRING = [MAJOR, MINOR, TINY].join('.')

data/locarails.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = %q{locarails}
-  s.version = "1.0.0"
+  s.version = "1.1.0"
 
   s.specification_version = 2 if s.respond_to? :specification_version=
 
@@ -11,7 +11,7 @@ Gem::Specification.new do |s|
   s.description = %q{A maneira mais simples para instalar aplicacoes Rails na hospedagem Linux da Locaweb.}
   s.email = %q{fabio.akita@locaweb.com.br}
   s.executables = ["locarails"]
-  s.files = ["bin/locarails", "bin/locarails.cmd", "LICENSE", "Manifest", "README", "lib/locarails.rb", "lib/locarails/version.rb", "templates/database.locaweb.yml", "templates/deploy.rb", "templates/locaweb_backup.rb", "locarails.gemspec"]
+  s.files = ["bin/locarails", "bin/locarails.cmd", "LICENSE", "Manifest", "README", "locarails.gemspec", "lib/locarails.rb", "lib/locarails/version.rb", "templates/database.locaweb.yml.erb", "templates/deploy.rb.erb", "templates/deploy.common.rb", "templates/locaweb_backup.rb", "templates/ssh_helper.rb", "templates/.gitignore"]
   s.has_rdoc = true
   s.homepage = %q{http://www.locaweb.com.br/rails}
   s.require_paths = ["lib"]

data/templates/.gitignore

@@ -0,0 +1,22 @@
+log/*.log
+log/*.out
+log/*.pid
+tmp/**/*
+.DS_Store
+db/schema.rb
+db/schema.sql
+db/*.sqlite3*
+config/database.yml
+config/database.locaweb.yml
+public/photos/*
+public/upload/*
+public/assets/*
+.project
+.#*
+tmp/.*
+\#*
+*~
+vendor/**/**/doc/*
+rsa_key*
+coverage/
+coverage

data/templates/database.locaweb.yml.erb

@@ -0,0 +1,7 @@
+production:
+  adapter: mysql
+  encoding: utf8
+  database: <%= mysqlcfg.db %>
+  username: <%= mysqlcfg.user %>
+  password: <%= mysqlcfg.pass %>
+  host: <%= mysqlcfg.host %>

data/templates/deploy.common.rb

@@ -0,0 +1,86 @@
+# =============================================================================
+# TAREFAS - NAO MEXER A MENOS QUE SAIBA O QUE ESTA FAZENDO
+# =============================================================================
+desc "Garante que o database.yml foi corretamente configurado"
+task :before_symlink do
+  on_rollback {}
+  run "test -d #{release_path}/tmp || mkdir -m 755 #{release_path}/tmp"
+  run "test -d #{release_path}/db || mkdir -m 755 #{release_path}/db"
+  run "cp #{deploy_to}/etc/database.yml #{release_path}/config/database.yml"
+  run "cd #{release_path} && rake db:migrate RAILS_ENV=production"
+end
+
+desc "Garante que as configuracoes estao adequadas"
+task :before_setup do
+  ts = Time.now.strftime("%y%m%d%H%M%S")
+  # git folder
+  run "test -d #{git_repo} || mkdir -p -m 755 #{git_repo}"
+  run "test -d #{git_repo}/.git || cd #{git_repo} && git init"
+  git_config = File.join(File.dirname(__FILE__), "../.git/config")
+  has_git = false
+  if File.exists?(git_config) && File.read(git_config) !~ /locaweb/
+    `git remote add locaweb #{repository}`
+    `git push locaweb #{branch}`
+    has_git = true
+  end
+  
+  run "if [ -d #{deploy_to} ]; then mv #{deploy_to} #{deploy_to}-#{ts}.old ; fi"
+  run "test -d #{deploy_to} || mkdir -m 755 #{deploy_to}"
+  run "test -d #{deploy_to}/etc || mkdir -m 755 #{deploy_to}/etc"
+  run "if [ -d #{site_path} ]; then mv #{site_path} #{site_path}-#{ts}.old ; fi"
+  run "if [ -h #{site_path} ]; then mv #{site_path} #{site_path}-#{ts}.old ; fi"
+  run "ln -s #{deploy_to}/current/public #{public_html}/#{application}"
+  put File.read(File.dirname(__FILE__) + "/database.locaweb.yml"), "#{deploy_to}/etc/database.yml"
+
+  # ssh keygen
+  put File.read(File.dirname(__FILE__) + "/ssh_helper.rb"), "#{deploy_to}/etc/ssh_helper.rb"
+  run "test -f .ssh/id_rsa || ruby #{deploy_to}/etc/ssh_helper.rb /home/#{user}/.ssh/id_rsa #{domain} #{user}"
+  
+  unless has_git
+    3.times { puts "" }
+    puts "==============================================================="
+    puts "Rode os seguintes comandos depois de criar seu repositorio Git:"
+    puts ""
+    puts "  git remote add locaweb #{repository}"
+    puts "  git push locaweb #{branch}"
+    puts "==============================================================="
+    3.times { puts "" }
+  end
+end
+
+desc "Prepare the production database before migration"
+task :before_cold do
+end
+
+namespace :deploy do
+  desc "Pede restart ao servidor Passenger"
+  task :restart, :roles => :app do
+    run "touch #{deploy_to}/current/tmp/restart.txt"
+  end
+end
+
+[:start, :stop].each do |t|
+    desc "A tarefa #{t} nao eh necessaria num ambiente com Passenger"
+    task t, :roles => :app do ; end
+end
+
+namespace :log do
+  desc "tail production log files" 
+  task :tail, :roles => :app do
+    run "tail -f #{shared_path}/log/production.log" do |channel, stream, data|
+      puts  # para uma linha extra 
+      puts "#{channel[:host]}: #{data}" 
+      break if stream == :err    
+    end
+  end
+end
+
+namespace :db do
+  desc "Faz o backup remoto do banco de dados MySQL e ja faz o download"
+  task :backup, :roles => :db do
+    backup_rb ||= "#{deploy_to}/current/config/locaweb_backup.rb"
+    run "if [ -f #{backup_rb} ]; then ruby #{backup_rb} #{deploy_to} ; fi"
+    get "#{deploy_to}/etc/dump.tar.gz", "dump.tar.gz"
+    run "rm #{deploy_to}/etc/dump.tar.gz"
+  end
+end

data/templates/{deploy.rb → deploy.rb.erb}

@@ -57,11 +57,13 @@
 # =============================================================================
 # CONFIGURE OS VALORES DE ACORDO COM SUA HOSPEDAGEM
 # =============================================================================
-set :user, "seu.usuario"
-set :domain, "seu.dominio"
-set :application, "sua.aplicacao"
-set :repository, "seu.projeto"
-#ssh_options[:keys] = File.expand_path("~/.ssh/id_rsa") # apenas descomente se tiver chave
+set :user, "<%= config.usuario %>"
+set :domain, "<%= config.dominio %>"
+set :application, "<%= config.app_name %>"
+set :repository, "<%= config.local_path %>"
+<% if config.ssh_key_path %>
+ssh_options[:keys] = File.expand_path("<%= config.ssh_key_path %>") # apenas descomente se tiver chave
+<% end %>
 
 # =============================================================================
 # NAO MEXER DAQUI PARA BAIXO
@@ -77,70 +79,23 @@ set :runner, nil
 set :use_sudo, false
 set :no_release, true
 
+<% if config.mode == 'git' %>
+set :git_repo, "/home/#{user}/repo/#{application}.git"
+  
+set :scm, :git
+set :branch, 'local'
+set :deploy_via, :remote_cache
+set :git_shallow_clone, 1
+set :copy_exclude, %w(.git/* .svn/* log/* tmp/* .gitignore)
+set :remote, "akitaonrails1"
+set :scm_verbose, true
+set :keep_releases, 5
+<% else %>
 set :scm, :none # nenhum repositorio
 set :deploy_via, :copy 
 set :copy_exclude, %w(.git/* .svn/* log/* tmp/*)
+set :keep_releases, 5
+<% end %>
 
 ssh_options[:username] = user
-ssh_options[:paranoid] = false 
-
-set :backup_rb, "#{deploy_to}/current/config/locaweb_backup.rb"
-
-# =============================================================================
-# TAREFAS - NAO MEXER A MENOS QUE SAIBA O QUE ESTA FAZENDO
-# =============================================================================
-desc "Garante que o database.yml foi corretamente configurado"
-task :before_symlink do
-  on_rollback {}
-  run "test -d #{release_path}/tmp || mkdir -m 755 #{release_path}/tmp"
-  run "cp #{deploy_to}/etc/database.yml #{release_path}/config/database.yml"
-  run "cd #{release_path} && rake db:migrate RAILS_ENV=production"
-end
-
-desc "Garante que as configuracoes estao adequadas"
-task :before_setup do
-  ts = Time.now.strftime("%y%m%d%H%M%S")
-  run "if [ -d #{deploy_to} ]; then mv #{deploy_to} #{deploy_to}-#{ts}.old ; fi"
-  run "test -d #{deploy_to} || mkdir -m 755 #{deploy_to}"
-  run "test -d #{deploy_to}/etc || mkdir -m 755 #{deploy_to}/etc"
-  run "if [ -d #{site_path} ]; then mv #{site_path} #{site_path}-#{ts}.old ; fi"
-  run "if [ -h #{site_path} ]; then mv #{site_path} #{site_path}-#{ts}.old ; fi"
-  run "ln -s #{deploy_to}/current/public #{public_html}/#{application}"
-  put File.read(File.dirname(__FILE__) + "/database.locaweb.yml"), "#{deploy_to}/etc/database.yml"
-end
-
-desc "Prepare the production database before migration"
-task :before_cold do
-end
-
-namespace :deploy do
-  desc "Pede restart ao servidor Passenger"
-  task :restart, :roles => :app do
-    run "touch #{deploy_to}/current/tmp/restart.txt"
-  end
-end
-
-[:start, :stop].each do |t|
-    desc "A tarefa #{t} nao eh necessaria num ambiente com Passenger"
-    task t, :roles => :app do ; end
-end
-
-namespace :log do
-  desc "tail production log files" 
-  task :tail, :roles => :app do
-    run "tail -f #{shared_path}/log/production.log" do |channel, stream, data|
-      puts  # para uma linha extra 
-      puts "#{channel[:host]}: #{data}" 
-      break if stream == :err    
-    end
-  end
-end
-
-namespace :db do
-  desc "Faz o backup remoto do banco de dados MySQL e ja faz o download"
-  task :backup, :roles => :db do
-    run "if [ -f #{backup_rb} ]; then ruby #{backup_rb} #{deploy_to} ; fi"
-    get "#{deploy_to}/etc/dump.tar.gz", "dump.tar.gz"
-    run "rm #{deploy_to}/etc/dump.tar.gz"
-  end
-end
+ssh_options[:paranoid] = false

data/templates/ssh_helper.rb

@@ -0,0 +1,317 @@
+require 'rubygems'
+gem "net-ssh", ">= 1.1.3"
+require 'base64'
+require 'net/ssh'
+
+#--
+# =============================================================================
+# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
+# All rights reserved.
+#
+# This source file is distributed as part of the Net::SSH Secure Shell Client
+# library for Ruby. This file (and the library as a whole) may be used only as
+# allowed by either the BSD license, or the Ruby license (or, by association
+# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
+# distribution for the texts of these licenses.
+# -----------------------------------------------------------------------------
+# net-ssh website : http://net-ssh.rubyforge.org
+# project website: http://rubyforge.org/projects/net-ssh
+# =============================================================================
+#++
+
+module Net
+  module SSH
+
+    module Util
+
+      # The abstract ancestor module of both ReaderBufferImpl and
+      # WriterBufferImpl. It defines the common interface for both submodules.
+      module BufferBase
+
+        # exposes the content of the buffer
+        attr_reader :content
+
+        # the length of the buffer's content.
+        def length
+          @content.length
+        end
+
+        # returns a copy of the buffer's content.
+        def to_s
+          ( @content || "" ).dup
+        end
+
+        # Compares the contents of the two buffers.
+        def ==( buffer )
+          to_s == buffer.to_s
+        end
+
+        # Resets the buffer, making it empty.
+        def clear!
+          @content = ""
+        end
+
+        def init_BufferBase( content="" )
+          @content = content
+        end
+        private :init_BufferBase
+
+      end
+
+      # A convenience module for representing a string of encoded data. It
+      # provides an interface for easily reading and decoding the buffer.
+      module ReaderBufferImpl
+
+        # the current position of the pointer in the buffer
+        attr_reader :position
+
+        # used by derived modules and classes to perform any
+        # reader-buffer-specific initialization.
+        def init_ReaderBufferImpl
+          @position = 0
+        end
+        private :init_ReaderBufferImpl
+
+        # Appends the given text to the end of the buffer.
+        def append( text )
+          @content << text
+        end
+
+        # Returns all text from the current pointer to the end of the buffer as
+        # a new buffer as the same class as the receiver.
+        def remainder_as_buffer
+          self.class.new( @content[ @position..-1 ] )
+        end
+
+        # Reads +count+ bytes from the buffer. If +count+ is +nil+, this will
+        # return all remaining text in the buffer. This method will increment
+        # the pointer.
+        def read( count = nil )
+          count = length - @position unless count
+          return nil if @position + count > length
+
+          @position += count
+          @content[ @position-count, count ]
+        end
+
+        # Return the next 8 bytes as a 64-bit integer (in network byte order).
+        def read_int64
+          hi = read_long
+          lo = read_long
+          return ( hi << 32 ) + lo
+        end
+
+        # Return the next four bytes as a long integer (in network byte order).
+        def read_long
+          b = read( 4 ) or return nil
+          b.unpack( "N" ).first
+        end
+
+        # Read the next two bytes as a short integer (in network byte order).
+        def read_short
+          b = read( 2 ) or return nil
+          b.unpack( "n" ).first
+        end
+
+        # Read and return the next byte in the buffer.
+        def read_byte
+          b = read( 1 ) or return nil
+          b[0]
+        end
+
+        # Read and return an SSH2-encoded string. The string starts with a long
+        # integer that describes the number of bytes remaining in the string.
+        def read_string
+          length = read_long or return nil
+          read( length )
+        end
+
+        # Read a single byte and convert it into a boolean, using 'C' rules
+        # (i.e., zero is false, non-zero is true).
+        def read_bool
+          b = read( 1 ) or return nil
+          b[0] != 0
+        end
+
+        # Resets the pointer to the start of the buffer.
+        def reset!
+          @position = 0
+        end
+
+        # Returns true if the pointer is at the end of the buffer.
+        def eof?
+          @position >= length
+        end
+
+        # Resets the buffer, making it empty.
+        def clear!
+          @content = ""
+          @position = 0
+        end
+
+      end # ReaderBufferImpl
+
+      # A convenience module for writing a string of encoded data. It provides
+      # an interface for easily writing and encoding data.
+      module WriterBufferImpl
+
+        def init_WriterBufferImpl
+          # nothing
+        end
+        private :init_WriterBufferImpl
+
+        # Writes the given data literally into the string.
+        def write( *data )
+          @content << data.join
+        end
+
+        # Writes each argument to the buffer as a network-byte-order-encoded
+        # 64-bit integer (8 bytes).
+        def write_int64( *n )
+          n.each do |i|
+            hi = ( i >> 32 ) & 0xFFFFFFFF
+            lo = i & 0xFFFFFFFF
+            @content << [ hi, lo ].pack( "N2" )
+          end
+        end
+
+        # Writes each argument to the buffer as a network-byte-order-encoded
+        # long (4-byte) integer.
+        def write_long( *n )
+          @content << n.pack( "N*" )
+        end
+
+        # Writes each argument to the buffer as a network-byte-order-encoded
+        # short (2-byte) integer.
+        def write_short( *n )
+          @content << n.pack( "n*" )
+        end
+
+        # Writes each argument to the buffer as a byte.
+        def write_byte( *n )
+          @content << n.map { |c| c.chr }.join
+        end
+
+        # Writes each argument to the buffer as an SSH2-encoded string. Each
+        # string is prefixed by its length, encoded as a 4-byte long integer.
+        def write_string( *text )
+          text.each do |string|
+            write_long( string.length )
+            write( string )
+          end
+        end
+
+        # Writes each argument to the buffer as a (C-style) boolean, with 1
+        # meaning true, and 0 meaning false.
+        def write_bool( *b )
+          @content << b.map { |v| ( v ? 1 : 0 ).chr }.join
+        end
+
+        # Writes each argument to the buffer as a bignum (SSH2-style). No
+        # checking is done to ensure that the arguments are, in fact, bignums.
+        def write_bignum( *n )
+          @content << n.map { |b| b.to_ssh }.join
+        end
+
+        # Writes the given arguments to the buffer as SSH2-encoded keys.
+        def write_key( *key )
+          key.each do |k|
+            write_string( k.ssh_type )
+
+            case k.ssh_type
+              when "ssh-dss"
+                write_bignum( k.p )
+                write_bignum( k.q )
+                write_bignum( k.g )
+                write_bignum( k.pub_key )
+
+              when "ssh-rsa"
+                write_bignum( k.e )
+                write_bignum( k.n )
+
+              else
+                raise NotImplementedError,
+                  "unsupported key type '#{k.ssh_type}'"
+            end
+          end
+        end
+
+      end # class WriterBufferImpl
+
+      # A convenience class for a read-only buffer.
+      class ReaderBuffer
+        include BufferBase
+        include ReaderBufferImpl
+
+        def initialize( content )
+          init_BufferBase( content )
+          init_ReaderBufferImpl
+        end
+      end
+
+      # A convenience class for a write-only buffer.
+      class WriterBuffer
+        include BufferBase
+        include WriterBufferImpl
+
+        def initialize( content="" )
+          init_BufferBase( content )
+          init_WriterBufferImpl
+        end
+      end
+
+      # A convenience class for a read/write buffer.
+      class Buffer
+        include BufferBase
+        include ReaderBufferImpl
+        include WriterBufferImpl
+
+        def initialize( content="" )
+          init_BufferBase( content )
+          init_ReaderBufferImpl
+          init_WriterBufferImpl
+        end
+      end
+
+    end # module Util
+
+  end # module SSH
+end # module Net
+
+class SSHHelper
+  def self.generate_keys(filename)
+    authorized_file = "#{File.dirname(filename)}/authorized_keys"
+    
+    private_key = OpenSSL::PKey::RSA.new(1024)
+    public_pem = export_ssh_pubkey( private_key )
+    
+    File.open( filename, "w" ) { |file| file.write private_key.export }
+    File.open( "#{filename}.pub", "w" ) { |file| file.write public_pem }
+    File.open( authorized_file, 'w') { |file| file.write public_pem + "\n"}
+    
+    FileUtils.chmod 0755, authorized_file
+    FileUtils.chmod 0600, filename
+  end
+  
+  def self.test_connection(hostname, username)
+    # will create known_host
+    Net::SSH.start(hostname, username) { |ssh| puts ssh.exec!('hostname') }
+  end
+
+  private 
+  def self.export_ssh_pubkey( key )
+    s = key.ssh_type + " "
+
+    writer = Net::SSH::Util::WriterBuffer.new
+    writer.write_key key
+    s << Base64.encode64( writer.to_s ).strip.gsub( /[\n\r\t ]/, "" )
+    s << " #{ENV['USER']}@#{ENV['HOSTNAME']}"
+
+    s
+  end
+end
+
+if ARGV && ARGV.size == 3
+  SSHHelper.generate_keys(ARGV[0])
+  SSHHelper.test_connection(ARGV[1], ARGV[2])
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: akitaonrails-locarails
 version: !ruby/object:Gem::Version 
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors: 
 - Fabio Akita
@@ -27,12 +27,15 @@ files:
 - LICENSE
 - Manifest
 - README
+- locarails.gemspec
 - lib/locarails.rb
 - lib/locarails/version.rb
-- templates/database.locaweb.yml
-- templates/deploy.rb
+- templates/database.locaweb.yml.erb
+- templates/deploy.rb.erb
+- templates/deploy.common.rb
 - templates/locaweb_backup.rb
-- locarails.gemspec
+- templates/ssh_helper.rb
+- templates/.gitignore
 has_rdoc: true
 homepage: http://www.locaweb.com.br/rails
 post_install_message: 

data/templates/database.locaweb.yml

@@ -1,7 +0,0 @@
-production:
-  adapter: mysql
-  encoding: utf8
-  database: mysql.database
-  username: mysql.username
-  password: mysql.password
-  host: mysql.hostname