vycap 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 468dd254bbfa3f5b55211230db1b4993a8ae8ad9
+  data.tar.gz: 0fd71f11753e45272bc0825d5e245e7133a6be27
+SHA512:
+  metadata.gz: 90f534638282176ea44da1ea2569ea407d5f3ca40d6394359f0c4faac449fa816b0455a302055a6a9221024c890ea4849329ed0339b622a6a71c354c40558b3e
+  data.tar.gz: 31cc09da24aa8e6461a9fbe2b324dcd563d55069afa01c577a42e9d4b83f6db9c174bf346f8c507c3f43401c097fd46458debec104bacc4b1eeb0bf99aabd61e

data/lib/vycap.rb

@@ -0,0 +1,5 @@
+require 'supply_drop'
+require File.expand_path(File.dirname(__FILE__) + "/vycap/tasks")
+require File.expand_path(File.dirname(__FILE__) + "/vycap/plugin")
+
+Capistrano.plugin :vycap_plugin, Vycap::Plugin

data/lib/vycap/plugin.rb

@@ -0,0 +1,90 @@
+require 'erb'
+require 'ostruct'
+
+module Vycap
+  module Plugin
+    def generate_config(server, options)
+      template_filename = File.join(
+        options.fetch(:template_dir),
+        options.fetch(:template)
+      )
+      output_filename = File.join(
+        options.fetch(:output_dir),
+        "config.#{server}"
+      )
+      template_filename += ".erb" unless template_filename =~ /\.erb$/
+
+      context = if options[:context].nil?
+                  OpenStruct.new
+                elsif options[:context].respond_to?(:call)
+                  OpenStruct.new(options[:context].call(server))
+                else
+                  OpenStruct.new(options[:context])
+                end
+
+      context.partials = evaluate_partials(options.fetch(:partials_dir), context)
+
+      erb = File.read(template_filename)
+      File.open(output_filename, "w") do |out|
+        out.write ERB.new(erb).result(context.instance_eval { binding })
+      end
+    end
+
+    def evaluate_partials(partials_dir, context)
+      return @vycap_partials unless @vycap_partials.nil?
+
+      @vycap_partials = {}
+
+      Dir["#{partials_dir}/*.erb"].sort.each do |file|
+        erb = File.read(file)
+
+        @vycap_partials[File.basename(file, ".erb")] = ERB.new(erb).result(context.instance_eval { binding })
+      end
+
+      @vycap_partials
+    end
+
+    def vcmd(cmd)
+      "/opt/vyatta/sbin/vyatta-cfg-cmd-wrapper #{cmd}"
+    end
+
+    def vrun(*cmds)
+      cmds = cmds.flatten
+      if cmds.last.is_a?(Hash)
+        options = cmds.pop
+      else
+        options = {}
+      end
+
+      vcmds = cmds.map do |cmd|
+        vcmd(cmd)
+      end
+
+      sudo_prefix = options.fetch(:sudo, false) ? sudo : ""
+
+      run %Q{#{sudo_prefix} vbash -c '#{vcmds.join(" && ")}'; true}, :eof => true, :shell => false do |ch, stream, data|
+        case stream
+        when :out
+          if data =~ /^enter .* password:$/i
+            password = Capistrano::CLI.password_prompt(data + " ")
+            ch.send_data(password + "\n")
+          elsif data =~ /is not valid/ || data =~ /failed/
+            puts _colorize(data, "\033[31m")
+          else
+            puts data
+          end
+        when :err then warn "[err :: #{ch[:server]}] #{data}"
+        end
+      end
+    end
+
+    def vsudo(*cmds)
+      vrun cmds, :sudo => true
+    end
+
+    # borrowed from http://www.codedrop.ca/blog/archives/200
+    def _colorize(text, color_code)
+      "#{color_code}#{text}\033[0m"
+    end
+  end
+end

data/lib/vycap/tasks.rb

@@ -0,0 +1,116 @@
+Capistrano::Configuration.instance.load do
+  set :vycap_template_dir, File.expand_path("./vycap/templates/")
+  set :vycap_partials_dir, vycap_template_dir + "/partials"
+  set :vycap_user, nil
+  set :vycap_remote_config, "/config/config.vycap"
+  set :vycap_tmp_dir, "/tmp"
+  set :vycap_template, "config"
+  set :vycap_template_context, {}
+  set :vycap_auth_dir, ""
+  set :vycap_remote_auth_dir, "/config/auth/vycap/"
+
+  namespace :vycap do
+    task :download_config_template do
+      failed_servers = roles[:vyatta].servers.map do |srv|
+        remote_file = SupplyDrop::Rsync.remote_address(vycap_user, srv.host, "/config/config.boot")
+        cmd = SupplyDrop::Rsync.command(
+          remote_file,
+          vycap_template_dir + "/config.#{srv}.erb",
+          :ssh => ssh_options
+        )
+        srv unless system cmd
+      end
+
+      raise "rsync failed on #{failed_servers.inspect}" if failed_servers.any?
+
+      roles[:vyatta].servers.map do |srv|
+        config_template = vycap_template_dir + "/config.#{srv}.erb"
+
+        original = File.read(config_template)
+        File.open(config_template, "w") do |file|
+          original.split("\n").each do |line|
+            file.puts line unless line =~ /\bhw-id\b/
+          end
+        end
+      end
+    end
+
+    namespace :upload do
+      task :default do
+        vycap.upload.config
+        vycap.upload.auth_files unless vycap_auth_dir == ""
+      end
+
+      task :config do
+        failed_servers = roles[:vyatta].servers.map do |srv|
+          vycap_plugin.generate_config(srv, {
+            :template => vycap_template,
+            :template_dir => vycap_template_dir,
+            :partials_dir => vycap_partials_dir,
+            :output_dir => vycap_tmp_dir,
+            :context => vycap_template_context
+          })
+
+          remote_file = SupplyDrop::Rsync.remote_address(vycap_user, srv.host, vycap_remote_config)
+
+          cmd = SupplyDrop::Rsync.command(
+            File.join(vycap_tmp_dir, "config.#{srv}"),
+            remote_file,
+            :ssh => ssh_options
+          )
+          srv unless system cmd
+        end
+
+        raise "rsync failed on #{failed_servers.inspect}" if failed_servers.any?
+
+      end
+
+      task :auth_files do
+        set :user, vycap_user unless vycap_user.nil?
+        failed_servers = roles[:vyatta].servers.map do |srv|
+          remote_file = SupplyDrop::Rsync.remote_address(vycap_user, srv.host, vycap_remote_auth_dir)
+          auth_dir = vycap_auth_dir
+
+          remote_file += "/" unless remote_file =~ /\/$/
+          auth_dir += "/" unless auth_dir =~ /\/$/
+
+
+          cmd = SupplyDrop::Rsync.command(
+            auth_dir,
+            remote_file,
+            :ssh => ssh_options
+          )
+          success = system cmd
+          run("chmod -R go-rw #{vycap_remote_auth_dir}")
+          srv unless success
+        end
+
+        raise "rsync failed on #{failed_servers.inspect}" if failed_servers.any?
+      end
+    end
+
+    namespace :diff do
+      task :show do
+        set :user, vycap_user unless vycap_user.nil?
+        vycap_plugin.vsudo(
+          "begin",
+          "load #{vycap_remote_config}",
+          'show | grep -C 3 -E "^[+->]"',
+          "discard",
+          "end"
+        )
+      end
+
+      task :apply do
+        set :user, vycap_user unless vycap_user.nil?
+        vycap_plugin.vsudo(
+          "begin",
+          "load #{vycap_remote_config}",
+          "commit",
+          "save",
+          "end"
+        )
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,45 @@
+--- !ruby/object:Gem::Specification
+name: vycap
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Patrick Schless
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-24 00:00:00.000000000 Z
+dependencies: []
+description: Tested with vyatta 6.5
+email: patrick@plainlystated.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ./lib/vycap.rb
+- ./lib/vycap/plugin.rb
+- ./lib/vycap/tasks.rb
+homepage: https://github.com/plainlystated/vycap
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Capistrano plugin for managing vyatta servers
+test_files: []