vpsb 1.0.0

data/lib/vpsb/commands/setup.rb

@@ -0,0 +1,23 @@
+module Vpsb
+  module Commands
+    class Setup
+      include AskSupport
+      attr_reader :core
+
+      def initialize(core)
+        @core = core
+      end
+
+      def execute(params)
+        unless File.exist?(core.get(:knife_rb_path))
+          return puts "You did not run init script, because i cannot find knife.rb file in bootstrap server app. ;)"
+        end
+
+        Vpsb::Tasks::DoApiAccess.new(core).call
+        Vpsb::Tasks::DropletSshKeys.new(core).call
+        Vpsb::Tasks::CreateDroplet.new(core).call
+        Vpsb::Tasks::DropletSshUpdateConfig.new(core).call
+      end
+    end
+  end
+end

data/lib/vpsb/core.rb

@@ -0,0 +1,41 @@
+module Vpsb
+  class Core
+    class MissingResourceError < StandardError; end
+    def get(name, opts={})
+      name = name.to_sym
+      return data[name] = resources[name].call(self) if opts[:force]
+      data[name] ||= get_resource(name, opts).call(self)
+    end
+
+    def present?(name)
+      !blank?(name)
+    end
+
+    def blank?(name)
+      data[name].nil?
+    end
+
+    def resources
+      @resources ||= {}
+    end
+
+    def data
+      @data ||= {}
+    end
+
+    def register_resources(*resource_list)
+      resource_list.each{ |r| register_resource(r) }
+    end
+
+    def register_resource(resource)
+      resources[resource.name] = resource
+    end
+
+    private
+
+    def get_resource(name, opts)
+      raise MissingResourceError.new("Resource '#{name}', not found") unless resources[name]
+      resources[name]
+    end
+  end
+end

data/lib/vpsb/helpers/file_replace.rb

@@ -0,0 +1,20 @@
+require 'fileutils'
+require 'etc'
+
+module Vpsb
+  module Helpers
+    class FileReplace
+      def self.replace_json(path, hash)
+        replace_text(path, hash.to_json)
+      end
+
+      def self.replace_text(path, text)
+        file_path = File.join(Dir.pwd, path)
+        FileUtils.rm(file_path) if File.exist?(file_path)
+        File.open(file_path,"w") do |f|
+          f.write(text)
+        end
+      end
+    end
+  end
+end

data/lib/vpsb/helpers/ssh.rb

@@ -0,0 +1,117 @@
+require 'fileutils'
+require 'etc'
+
+module Vpsb
+  module Helpers
+    class Ssh
+      include AskSupport
+
+      def generate_ssh(path)
+        return if path == local_ssh_path && local_ssh_pub
+        puts 'Start generating new keys'
+        system("ssh-keygen -t rsa -f #{path}/id_rsa")
+      end
+
+      def ask_for_ssh_path
+        Dir.mkdir(local_ssh_path) unless Dir.exist?(local_ssh_path)
+
+        return local_ssh_path unless local_ssh_pub # no ssh keys yet
+        return local_ssh_path if     ask_to_confirm('You have local ssh keys. Do you want to use it? If no, new keys will be generated')
+
+        begin
+          ssh_folder = Dir.home + '/.ssh/' + ask { puts "In what folder keep your keys?" }
+          can_continue = true
+
+          if Dir.exist?(ssh_folder)
+            if ask_to_confirm("Folder #{ssh_folder} already taken. Overwrite?")
+              FileUtils.rm_rf ssh_folder
+            else
+              can_continue = false
+            end
+          end
+        end until can_continue
+
+        Dir.mkdir ssh_folder
+        ssh_folder
+      end
+
+      def add_host_to_config(host, ssh_path, user)
+        make_ssh_config_copy
+        original_config = file(config_path)
+
+        unless start_place = original_config =~ /Host #{host} ##{user}/
+          File.open(config_path, 'a') { |f| f << host_config(host, ssh_path, user) }
+          return remove_config_copy
+        end
+
+        return unless ask_to_confirm("Host #{host} already defined in your ssh config file. Update?")
+
+        next_host_start_place = original_config[start_place + 1, original_config.size] =~ /Host /
+        new_config = [
+          original_config[0, start_place - 1], "\n\n",
+          host_config(host, ssh_path, user), "\n",
+          (next_host_start_place.nil? ? '' : original_config[next_host_start_place + start_place, original_config.size])
+        ].join(' ')
+
+        begin
+          File.open(config_path, 'w+') { |f| f.write(new_config) }
+        rescue => e
+          puts "Error occured when updaing ssh config. Message: #{e.message}";  restore_config_copy; puts "Original configuration restored"
+        end
+
+        remove_config_copy
+      end
+
+      def local_ssh_path
+        "#{Dir.home}/.ssh"
+      end
+
+      def local_ssh_pub
+        pub_key(local_ssh_path)
+      end
+
+      def pub_key(path)
+        path = path[-1] == '/' ? "#{path}id_rsa.pub" : "#{path}/id_rsa.pub"
+        file(path)
+      end
+
+      private
+
+      def config_path
+        "#{local_ssh_path}/config"
+      end
+
+      def config_copy_path
+        config_path + '_copy'
+      end
+
+      def file(path)
+        File.open(path, 'r') { |f| f.read } rescue nil
+      end
+
+      def host_config(host, ssh_path, user)
+          <<-EOS
+Host #{host} ##{user}
+  HostName #{host}
+  IdentityFile #{ssh_path}
+  # User #{user}
+          EOS
+      end
+
+      def make_ssh_config_copy
+        system("touch #{config_path}") unless Dir.exist?(config_path)
+        FileUtils.cp config_path, config_copy_path
+      end
+
+      def remove_config_copy
+        if File.exist?(config_copy_path)
+          FileUtils.rm config_copy_path
+        end
+      end
+
+      def restore_config_copy
+        FileUtils.cp config_copy_path, config_path
+      end
+    end
+  end
+end

data/lib/vpsb/resources/app_bags_json.rb

@@ -0,0 +1,38 @@
+module Vpsb
+  module Resources
+    class AppBagsJson < BaseResource
+      res :app_bags_json
+
+      def call(data)
+        template(data)
+      end
+
+      private
+
+      def template(data)
+        {
+          id: "webapps",
+          apps_base_dir: "/opt/www",
+          apps: [
+            {
+              name: data.get(:app_name),
+              type: "rails",
+              default_env:  "production",
+              host: data.get(:app_host),
+              try_files: "/caches/$uri.html",
+              nginx: {
+                default: true,
+                nginx_use_ssl: false
+              },
+              unicorn: {
+                unicorn_workers: data.get(:unicorn_worker_processes),
+                unicorn_timeout: data.get(:unicorn_worker_timeout)
+              },
+              app_env_variables: data.get(:app_env_variables)
+            }
+          ]
+        }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/app_env_variables.rb

@@ -0,0 +1,24 @@
+module Vpsb
+  module Resources
+    class AppEnvVariables < BaseAskResource
+      res :app_env_variables
+      question 'What are aplication env variables (default: RAILS_ENV=production)?'
+      private
+
+      def finalize(answer)
+        return { 'RAILS_ENV' => 'production' } if answer.empty?
+        answer.split(';').inject({}) do |h, i|
+          res = i.to_s.split('=')
+          key = res.shift.to_s.strip
+          next if key.empty?
+          h[key] = res.join('=').to_s.strip
+          h
+        end
+      end
+
+      def validate
+        Proc.new { true }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/app_host.rb

@@ -0,0 +1,12 @@
+module Vpsb
+  module Resources
+    class AppHost < BaseAskResource
+      res :app_host
+      # question 'What is aplication host name (example: google.pl)?'
+
+      def call(core)
+        core.get(:do_host_ip)
+      end
+    end
+  end
+end

data/lib/vpsb/resources/app_name.rb

@@ -0,0 +1,11 @@
+module Vpsb
+  module Resources
+    class AppName < BaseAskResource
+      res :app_name
+      # question 'What is aplication name (example: railsapp)?'
+      def call(core)
+        'rails'
+      end
+    end
+  end
+end

data/lib/vpsb/resources/app_nginx_use_ssl.rb

@@ -0,0 +1,12 @@
+module Vpsb
+  module Resources
+    class AppNginxUseSsl < BaseAskResource
+      res :app_nginx_use_ssl
+      question 'Application use https protocol y[es]/n[o]?'
+      private
+      def finalize(answer)
+        answer[0].to_s.downcase == 'y'
+      end
+    end
+  end
+end

data/lib/vpsb/resources/base_path.rb

@@ -0,0 +1,10 @@
+module Vpsb
+  module Resources
+    class BasePath < BaseResource
+      res :base_path
+      def call(data)
+        Dir.pwd
+      end
+    end
+  end
+end

data/lib/vpsb/resources/db_postgres_json.rb

@@ -0,0 +1,62 @@
+module Vpsb
+  module Resources
+    class DbPostgresJson < BaseResource
+      res :db_postgres_json
+
+      def call(data)
+        template(data)
+      end
+
+      private
+
+      def template(data)
+        version = data.get(:db_postgres_version)
+        {
+          name: "db",
+          description: "Database Server Role",
+          chef_type: "role",
+          json_class: "Chef::Role",
+          default_attributes: {
+            postgresql: {
+              version: version,
+              enable_pgdg_apt: true,
+              dir: "/etc/postgresql/#{version}/main",
+              config:{
+                data_directory: "/var/lib/postgresql/#{version}/main",
+                hba_file: "/etc/postgresql/#{version}/main/pg_hba.conf",
+                ident_file: "/etc/postgresql/#{version}/main/pg_ident.conf",
+                external_pid_file: "/var/run/postgresql/#{version}-main.pid",
+                ssl_cert_file: "/etc/ssl/certs/ssl-cert-snakeoil.pem",
+                ssl_key_file:"/etc/ssl/private/ssl-cert-snakeoil.key"
+              },
+              client: {
+                packages: [ "postgresql-client-#{version}"  ]
+              },
+              server:{
+                service_name: "postgresql",
+                packages:  [ "postgresql-common", "postgresql-#{version}", "postgresql-server-dev-#{version}", "libpq-dev"]
+              },
+              contrib:{
+                packages:["postgresql-contrib-#{version}"]
+              },
+              password: {
+                postgres: "postgres"
+              },
+              pg_hba: [
+                { type: "local", db: "all", user: "postgres",   addr: "",             method: "ident" },
+                { type: "local", db: "all", user: "all",        addr: "",             method: "trust" },
+                { type: "host",  db: "all", user: "all",        addr: "127.0.0.1/32", method: "trust" },
+                { type: "host",  db: "all", user: "all",        addr: "::1/128",      method: "trust" },
+                { type: "host",  db: "all", user: "postgres",   addr: "127.0.0.1/32", method: "trust" }
+              ]
+            }
+          },
+          run_list: [
+            "recipe[postgresql::server]",
+            "recipe[postgresql::client]"
+          ]
+        }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/db_postgres_pass.rb

@@ -0,0 +1,11 @@
+require 'securerandom'
+
+module Vpsb
+  module Resources
+    class DbPostgresPass < BaseAskResource
+      res :db_postgres_pass
+      question 'Postgres sql postgres user password (Default will be generated)?'
+      default  SecureRandom.hex
+    end
+  end
+end

data/lib/vpsb/resources/db_postgres_version.rb

@@ -0,0 +1,21 @@
+module Vpsb
+  module Resources
+    class DbPostgresVersion < BaseAskResource
+      res :db_postgres_version
+      question 'Postgres version (Default 9.3)?'
+      default  '9.3'
+
+      private
+
+      def validate
+        Proc.new do |answer|
+          next true if answer.strip.empty?
+          parts = answer.split('.')
+          next false if parts.length != 2
+          next false if (parts[0] + parts[1]).to_i <= 7
+          true
+        end
+      end
+    end
+  end
+end

data/lib/vpsb/resources/db_postgresql_bags_json.rb

@@ -0,0 +1,20 @@
+module Vpsb
+  module Resources
+    class DbPostgresqlBagsJson < BaseResource
+      res :db_postgresql_bags_json
+
+      def call(data)
+        template(data)
+      end
+
+      private
+
+      def template(data)
+        {
+          id: "postgresql",
+          postgresql: data.get(:db_postgres_pass)
+        }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/deploy_bags_json.rb

@@ -0,0 +1,36 @@
+module Vpsb
+  module Resources
+    class DeployBagsJson < BaseResource
+      res :deploy_bags_json
+
+      def call(data)
+        template(data)
+      end
+
+      private
+
+      def template(data)
+        {
+          id: "deploy",
+          username: "deploy",
+          ssh_keys: data.get(:deploy_ssh_keys),
+          password: data.get(:deploy_pass),
+          groups: %w{www-data deploy},
+          uid: 1001,
+          shell: "/bin/bash",
+          home: "/home/deploy",
+          comment: "deployment user",
+          manage_home: true,
+          keychain: { id_dsa: "", "id_dsa.pub" => ""},
+          to_known_hosts: [],
+          www_dir: "/opt/www",
+          env_variables_for_rvm: true,
+          global_env_variables:{
+            APPS_DIR:"/opt/www",
+            RAILS_ENV: "production"
+          }
+        }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/deploy_config.rb

@@ -0,0 +1,32 @@
+module Vpsb
+  module Resources
+    class DeployConfig < BaseResource
+      res :deploy_config
+
+      def call(data)
+        template(data)
+      end
+
+      private
+
+      def template(data)
+      user   = '#{user}'
+      domain = '#{domain}'
+<<-EOF
+set :stage,     :production
+set :rails_env, :production
+fetch(:default_env).merge!(rails_env: :production)
+
+domain = '#{data.get(:do_host_ip)}'
+user   = 'deploy'
+
+role :app, "#{user}@#{domain}"
+role :web, "#{user}@#{domain}"
+role :db,  "#{user}@#{domain}", primary: true
+
+server domain, user: user, roles: %w{db web app}
+EOF
+      end
+    end
+  end
+end

data/lib/vpsb/resources/deploy_keychain.rb

@@ -0,0 +1,19 @@
+module Vpsb
+  module Resources
+    class DeployKeychain < BaseAskResource
+      res :deploy_keychain
+      # TODO ask for ssh_keys from .ssh
+      question 'Enter id_dsa and id_dsa.pub for deploy (example: -----BEGIN RSA, ssh-rsa AB...)'
+      private
+
+      def finalize(answer)
+
+        arr = answer.split(',').join(' ').split(' ').compact.uniq
+        {
+          id_dsa: arr.shift,
+          "id_dsa.pub" => arr.shift
+        }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/deploy_pass.rb

@@ -0,0 +1,11 @@
+require 'securerandom'
+
+module Vpsb
+  module Resources
+    class DeployPass < BaseAskResource
+      res :deploy_pass
+      question 'Enter password for deploy user (Default will be generated)?'
+      default  SecureRandom.hex
+    end
+  end
+end

data/lib/vpsb/resources/deploy_ssh_keys.rb

@@ -0,0 +1,20 @@
+module Vpsb
+  module Resources
+    class DeploySshKeys < BaseAskResource
+      res :deploy_ssh_keys
+      # TODO ask for ssh_keys from .ssh
+      # question 'Enter ssh_keys(id_rsa.pub) for deploy (example: ssh-rsa AA... , ssh-rsa AB...)'
+
+      def call(core)
+        [core.get(:ssh_helper).pub_key(core.get(:do_ssh_path))]
+      end
+
+      private
+
+      # def finalize(answer)
+      #   answer.split(',').join(' ').split(' ').compact.uniq
+      # end
+
+    end
+  end
+end

data/lib/vpsb/resources/deploy_to_known_hosts.rb

@@ -0,0 +1,13 @@
+module Vpsb
+  module Resources
+    class DeployToKnownHosts < BaseAskResource
+      res :deploy_to_known_hosts
+      question 'Enter known hosts (example: bitbucket.org, github.com)?'
+      private
+
+      def finalize(answer)
+        answer.split(',').join(' ').split(' ').compact.uniq
+      end
+    end
+  end
+end

data/lib/vpsb/resources/do_api_key.rb

@@ -0,0 +1,14 @@
+module Vpsb
+  module Resources
+    class DoApiKey < BaseResource
+      include AskSupport
+      res :do_api_key
+
+      def call(core)
+        puts
+        ask { puts 'Visit https://cloud.digitalocean.com/api_access and generate API access key. Press Enter when ready to continue' }
+        ask_loop { puts 'What is your DigitalOcean api key?' }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/do_client_id.rb

@@ -0,0 +1,8 @@
+module Vpsb
+  module Resources
+    class DoClientId < BaseAskResource
+      res :do_client_id
+      question 'What is your DigitalOcean client id?'
+    end
+  end
+end

data/lib/vpsb/resources/do_droplet_ssh_key.rb

@@ -0,0 +1,15 @@
+module Vpsb
+  module Resources
+    class DoDropletSshKey < BaseResource
+      include AskSupport
+      include ShellSupport
+      res :do_droplet_ssh_key
+
+      def call(core)
+        puts "Loading list of available ssh keys"
+        in_dir(core.get(:server_app_path), 'knife digital_ocean sshkey list')
+        ask_loop { puts 'Which ssh key? Enter its id' }
+      end
+    end
+  end
+end

data/lib/vpsb/resources/do_host.rb

@@ -0,0 +1,8 @@
+module Vpsb
+  module Resources
+    class DoHost < BaseAskResource
+      res :do_host
+      question 'What will be your host name? f.e. my_site.com'
+    end
+  end
+end

data/lib/vpsb/resources/do_host_ip.rb

@@ -0,0 +1,8 @@
+module Vpsb
+  module Resources
+    class DoHostIp < BaseAskResource
+      res :do_host_ip
+      question 'What will be your host ip address? f.e. 80.240.132.140'
+    end
+  end
+end

data/lib/vpsb/resources/do_image.rb

@@ -0,0 +1,16 @@
+module Vpsb
+  module Resources
+    class DoImage < BaseResource
+      include AskSupport
+      include ShellSupport
+      res :do_image
+
+      def call(core)
+        puts "Loading list of available operating systems"
+        in_dir(core.get(:server_app_path), "knife digital_ocean image list -G | grep -i 'Ubuntu'")
+        answer = ask { puts 'Which system? Enter its id (default: 3101918)' }.strip
+        answer.empty? ? '3101918' : answer
+      end
+    end
+  end
+end

data/lib/vpsb/resources/do_region.rb

@@ -0,0 +1,16 @@
+module Vpsb
+  module Resources
+    class DoRegion < BaseResource
+      include AskSupport
+      include ShellSupport
+      res :do_region
+
+      def call(core)
+        puts "Loading list of available regions"
+        in_dir(core.get(:server_app_path), "knife digital_ocean region list")
+        answer = ask { puts 'Which region? Enter its id (default: 5)' }.strip
+        answer.empty? ? '5' : answer
+      end
+    end
+  end
+end