vana 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 73bd39d513e59438a74e818bc8f0104e162cad03
+  data.tar.gz: bc0eeb84d58fca4b421d3b006e74637d50852282
+SHA512:
+  metadata.gz: f24f390aa8b0dbae0b5c82920d7b8a0e760ba91f23b0f23991294ff38b493d8bb62b7378f73b2b6416e69d4612455cdddea176fe576eaffc2665825f0d621170
+  data.tar.gz: bb4636d1da809c141504ed5c43112e658cabccec4616db30e8bf144f6c080ffd6778db2cd54574eaa351625e0000dc01d62fd24abb13b5628781afafe90451d1

data/lib/vana.rb

@@ -0,0 +1,7 @@
+require 'vana/main.rb'
+
+require 'vana/element.rb'
+require 'vana/element/copy.rb'
+require 'vana/element/shell.rb'
+
+extend Vana

data/lib/vana/element.rb

@@ -0,0 +1,41 @@
+require 'colorize'
+require 'json'
+
+module Vana
+  module Element
+    # methods for every element
+    class BaseElement
+      def initialize(hosts, *args)
+        @hosts = hosts
+        @element_opts = {}
+        @opts = {}
+
+        yield self if block_given?
+        setup(*args)
+      end
+
+      def setup(*_args)
+        # nop
+      end
+
+      def action(_host, *_args)
+        # nop
+      end
+
+      def name=(name)
+        @element_opts[:name] = name
+      end
+
+      def execute(*args)
+        puts "-- #{@element_opts[:name]}"
+        @hosts.each do |host|
+          output = action(host, *args)
+          puts "#{host}: #{output.to_json}".colorize(output[:success] ? :light_green : :light_red)
+
+          # TODO: need a better place to handle exceptions and keep going with the script
+          # raise "Execution failed on '#{@element_opts[:name]}'" unless output[:success]
+        end
+      end
+    end
+  end
+end

data/lib/vana/element/copy.rb

@@ -0,0 +1,71 @@
+require 'fileutils'
+require 'net/scp'
+
+module Vana
+  # vana/main.rb
+  class Hosts
+    def copy(*args, &block)
+      Vana::Element::Copy.new(@hosts, *args, &block).execute
+    end
+  end
+
+  module Element
+    # copy element
+    class Copy < BaseElement
+      def setup(*args)
+        @opts[:src] ||= args[0]
+        @opts[:dest] ||= args[1]
+
+        @element_opts[:name] ||= "copy #{@opts[:src]} to #{@opts[:dest]}"
+      end
+
+      def src=(src)
+        @opts[:src] = src
+      end
+
+      def dest=(dest)
+        @opts[:dest] = dest
+      end
+
+      def remote_src=(remote_src)
+        @opts[:remote_src] = remote_src
+      end
+
+      def preserve_time=(preserve_time)
+        @opts[:preserve_time] = preserve_time
+      end
+
+      def action(host, *_args)
+        if host == 'localhost'
+          begin
+            FileUtils.cp_r(@opts[:src], @opts[:dest], preserve: @opts[:preserve_time])
+            success = true
+          rescue Errno::ENOENT
+            success = false
+          end
+          {
+            src: @opts[:src],
+            dest: @opts[:dest],
+            success: success
+          }
+        elsif @opts[:remote_src]
+          # TODO
+        else
+          begin
+            Net::SCP.start(host, nil) do |scp|
+              scp.upload!(@opts[:src], @opts[:dest], recursive: true, preserve: @opts[:preserve_time])
+            end
+            success = true
+          rescue Net::SCP::Error
+            success = false
+          end
+          {
+            src: @opts[:src],
+            dest: @opts[:dest],
+            success: success
+          }
+        end
+      end
+    end
+  end
+end

data/lib/vana/element/shell.rb

@@ -0,0 +1,67 @@
+require 'net/ssh'
+require 'open3'
+
+module Vana
+  # vana/main.rb
+  class Hosts
+    def shell(*args, &block)
+      Vana::Element::Shell.new(@hosts, *args, &block).execute
+    end
+  end
+
+  module Element
+    # shell element
+    class Shell < BaseElement
+      def setup(*args)
+        @opts[:cmd] ||= args.first
+        @element_opts[:name] ||= @opts[:cmd]
+      end
+
+      def cmd=(cmd)
+        @opts[:cmd] = cmd
+      end
+
+      def action(host, *_args)
+        if host == 'localhost'
+          stdout, stderr, status = Open3.capture3(@opts[:cmd])
+          {
+            cmd: @opts[:cmd],
+            stdout: stdout,
+            stderr: stderr,
+            success: status.success?,
+            status: status.exitstatus,
+            pid: status.pid
+          }
+        else
+          Net::SSH.start(host) do |ssh|
+            output = {
+              cmd: @opts[:cmd],
+              stdout: '',
+              stderr: ''
+            }
+
+            ssh.open_channel do |channel|
+              channel.exec(@opts[:cmd]) do |_ch, _success|
+                # stdout
+                channel.on_data { |_ch, data| output[:stdout] << data }
+
+                # stderr
+                channel.on_extended_data { |_ch, data| output[:stderr] << data }
+
+                # exit status
+                channel.on_request('exit-status') do |_ch, data|
+                  output[:status] = data.read_long
+                  output[:success] = output[:status].zero?
+                end
+              end
+            end
+
+            ssh.loop
+
+            output
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vana/main.rb

@@ -0,0 +1,19 @@
+# vana, approved by servers everywhere
+module Vana
+  # hosts block
+  class Hosts
+    def initialize(*a, &b)
+      @hosts = *a
+      @b = b
+    end
+
+    def execute
+      instance_eval(&@b)
+    end
+  end
+
+  def hosts(*a, &b)
+    h = Hosts.new(*a, &b)
+    h.execute
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: vana
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Brik Royster
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-04-19 00:00:00.000000000 Z
+dependencies: []
+description: dank emoji
+email: brik@butthole.tv
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/vana.rb
+- lib/vana/element.rb
+- lib/vana/element/copy.rb
+- lib/vana/element/shell.rb
+- lib/vana/main.rb
+homepage: http://butthole.tv
+licenses:
+- MIT
+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.4.8
+signing_key: 
+specification_version: 4
+summary: ansible but ruby
+test_files: []