stingray-exec 0.1.0

data/.gitignore

@@ -0,0 +1,19 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+/.*.integration.env
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+/ZeusTM_*.tgz

data/.rspec

@@ -0,0 +1,2 @@
+--require spec_helper
+--color

data/.simplecov

@@ -0,0 +1,7 @@
+if ENV['COVERAGE']
+  SimpleCov.start do
+    add_filter '/spec/'
+  end
+end
+
+# vim:filetype=ruby

data/AUTHORS.md

@@ -0,0 +1,2 @@
+ - Dan Buch
+ - Jake Ritorto

data/CONTRIBUTING.md

@@ -0,0 +1,8 @@
+# Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Add yourself to the `AUTHORS.md` file
+4. Commit your changes (`git commit -v`)
+5. Push to the branch (`git push -u origin my-new-feature`)
+6. Create new Pull Request

data/Gemfile

@@ -0,0 +1,12 @@
+source 'https://rubygems.org'
+
+gemspec
+
+gem 'rake'
+
+group :test do
+  gem 'nyan-cat-formatter'
+  gem 'pry'
+  gem 'rspec'
+  gem 'simplecov'
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 ModCloth, Inc.
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,66 @@
+# `stingray-exec`
+
+Stingray Traffic Manager Control API Client
+
+## Installation
+
+Install it via `gem`:
+
+    $ gem install stingray-exec
+
+## Usage
+
+TODO: yup
+
+## Hacking
+
+Some (all?) of this may be obvious for those familiar with Ruby modern
+projects.  It'll be okay.
+
+Clone down a copy
+````` bash
+git clone git://github.com/modcloth/stingray-exec.git
+pushd stingray-exec
+`````
+
+Make sure you have `bundler`
+````` bash
+gem install bundler
+`````
+
+Pull in dependencies with `bundler`
+````` bash
+bundle install
+`````
+
+Verify basic operation with the `stingray-exec` script
+````` bash
+bundle exec bin/stingray-exec --help
+`````
+
+As you'll see in the usage text, all of the command-line flags may also
+be passed as environmental variables.  The idea here is that it's
+sometimes easier to define an environmental variable soup instead of
+having to pass lots of flags for every invocation.
+
+### Running the tests
+
+The test suite uses RSpec.  By default, all specs tagged with
+`:integration => true` are excluded.  This results in very little being
+tested.  In order to exercise the really interesting tests, you'll have
+to define a `STINGRAY_ENDOINT` environmental variable.  Some examples of
+how to do this are available in the `.local.integration.env` and
+`.vagrant.integration.env` files in the root of the project tree.
+
+````` bash
+bundle exec rspec
+`````
+
+## Examples
+
+There are some example `stingray-exec` scripts present in the
+`./examples` directory.  The `stingray-exec` script accepts a filename
+as positional argument, so using it in the shebang line works great.
+
+**WARNING**: some of the examples are destructive, so don't go pointing
+them at production systems or any such nonsense.

data/Rakefile

@@ -0,0 +1,22 @@
+require "bundler/gem_tasks"
+
+stingray_tarball_name = 'ZeusTM_91_Linux-x86_64.tgz'
+stingray_tarball_url = 'https://support.riverbed.com/download.htm' <<
+  "?filename=public/software/stingray/trafficmanager/9.1/#{stingray_tarball_name}"
+stingray_tarball_sha1sum = 'bef75c80dbf7f13572ccfefec7791083180e78e0'
+
+desc 'downloads zeus/stingray traffic manager tarball'
+task :download_stingray do
+  unless have_stingray_tarball?(stingray_tarball_name, stingray_tarball_sha1sum)
+    sh "curl -o '#{stingray_tarball_name}' '#{stingray_tarball_url}'"
+  end
+end
+
+def have_stingray_tarball?(tarball, sha1sum)
+  File.exists?(tarball) &&
+    get_stingray_tarball_sha1sum(tarball) == sha1sum
+end
+
+def get_stingray_tarball_sha1sum(tarball)
+  `openssl dgst -sha1 #{tarball}`.chomp.split.last
+end

data/bin/stingray-exec

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+require 'stingray-exec'
+exit Stingray::Exec::Cli.main

data/examples/add-pool

@@ -0,0 +1,10 @@
+#!/usr/bin/env stingray-exec
+# vim:filetype=ruby
+
+ssl_verify_mode :none
+
+# Adds a pool named by the first positional argument with nodes given from all
+# remaining arguments.  Note that we don't start at `ARGV.first` because this
+# script is evaluated from within an existing ruby process.
+pool.add_pool(ARGV[1] => ARGV[2..ARGV.length])
+puts "Added!"

data/examples/bork

@@ -0,0 +1,6 @@
+#!/usr/bin/env stingray-exec
+# vim:filetype=ruby
+
+# Running this script with DEBUG=1 should demonstrate how the
+# backtrace includes *this* filename and the correct line number.
+bork

data/examples/credentials.conf

@@ -0,0 +1,4 @@
+# comments are ignored
+   # indented ones, too!
+admin:admin
+anything after the first non-comment line is discarded, too

data/examples/delete-pool

@@ -0,0 +1,10 @@
+#!/usr/bin/env stingray-exec
+# vim:filetype=ruby
+
+ssl_verify_mode :none
+
+# Removes any number of named pools given as positional arguments.  Note that
+# we don't start at `ARGV.first` because this script is evaluated from within
+# an existing ruby process.
+pool.delete_pool(*ARGV[1..ARGV.length])
+puts "Deleted!"

data/examples/list-pools

@@ -0,0 +1,8 @@
+#!/usr/bin/env stingray-exec
+# vim:filetype=ruby
+
+ssl_verify_mode :none
+
+foreach(pool.get_pool_names[:pool_names][:item]) do |n|
+  puts n
+end

data/examples/list-virtual-servers

@@ -0,0 +1,8 @@
+#!/usr/bin/env stingray-exec
+# vim:filetype=ruby
+
+ssl_verify_mode :none
+
+foreach(virtual_server.get_virtual_server_names[:names][:item]) do |n|
+  puts n
+end

data/generate-actions.rb

@@ -0,0 +1,119 @@
+#!/usr/bin/env ruby
+
+require 'wasabi'
+
+USAGE = <<-EOU
+Usage: #{File.basename($0)} [-h|--help] <stingray-wsdl-dir>
+
+Generate a YAML blob containing all methods found in the WSDL files provided in
+the built-in Stingray docs.  This is done so that we can use Savon::Model to
+build classes with predefined actions without having to distribute the WSDL
+files ourselves, which would be uncool and probably illegal.
+EOU
+STINGRAY_WSDL_CONFIGURATIONS = %w(
+  AFM
+  AlertCallback
+  Alerting.Action
+  Alerting.EventType
+  Catalog.Aptimizer.Profile
+  Catalog.Authenticators
+  Catalog.Bandwidth
+  Catalog.JavaExtension
+  Catalog.Monitor
+  Catalog.Persistence
+  Catalog.Protection
+  Catalog.Rate
+  Catalog.Rule
+  Catalog.SLM
+  Catalog.SSL.CertificateAuthorities
+  Catalog.SSL.Certificates
+  Catalog.SSL.ClientCertificates
+  Catalog.SSL.DNSSEC
+  Conf.Extra_1_0
+  Conf.Extra
+  Diagnose_1_0
+  Diagnose
+  GLB.Service
+  GlobalSettings
+  Location
+  Pool
+  System.AccessLogs
+  System.Backups
+  System.Cache_1_0
+  System.Cache_1_1
+  System.Cache
+  System.CloudCredentials
+  System.Connections
+  System.LicenseKeys
+  System.Log
+  System.MachineInfo
+  System.Management
+  System.RequestLogs
+  System.Stats
+  System.Steelhead
+  TrafficIPGroups
+  Users
+  VirtualServer
+)
+
+class ActionsGenerator
+  def self.main(argv)
+    wsdl_root = argv.first
+    if %w(-h --help).include?(wsdl_root)
+      $stderr.puts(USAGE)
+      return 0
+    elsif wsdl_root.nil? || wsdl_root.empty?
+      $stderr.puts "ERROR: Missing required argument <stingray-wsdl-dir>"
+      $stderr.puts(USAGE)
+      return 1
+    end
+
+    return new(wsdl_root).generate!
+  rescue => e
+    $stderr.puts "ERROR: #{e.class.name} #{e.message}"
+    $stderr.puts e.backtrace.join("\n") if ENV['DEBUG']
+    return 2
+  end
+
+  def initialize(wsdl_root, out = $stdout)
+    @wsdl_root = wsdl_root
+    @out = out
+  end
+
+  def generate!
+    @out.puts generate_actions_json
+    return 0
+  rescue => e
+    $stderr.puts "ERROR: #{e.class.name} #{e.message}"
+    $stderr.puts e.backtrace.join("\n") if ENV['DEBUG']
+    return 2
+  end
+
+  private
+  def generate_actions_json
+    actions = {}
+    STINGRAY_WSDL_CONFIGURATIONS.each do |cfg|
+      actions[cfg] = generate_actions_for_config(cfg)
+    end
+
+    YAML.dump(actions)
+  end
+
+  def generate_actions_for_config(cfg)
+    doc = Wasabi.document(File.read("#{@wsdl_root}/#{cfg}.wsdl"))
+    sorted_ops = {}
+    ops = doc.operations
+    ops.keys.sort.each do |k|
+      sorted_ops[k] = ops[k]
+    end
+    sorted_ops
+  rescue => e
+    $stderr.puts "ERROR: getting operations for #{cfg}: #{e.class.name} #{e.message}"
+    $stderr.puts e.backtrace.join("\n") if ENV['DEBUG']
+    {}
+  end
+end
+
+if $0 == __FILE__
+  exit(ActionsGenerator.main(ARGV))
+end

data/lib/stingray-exec.rb

@@ -0,0 +1,5 @@
+module Stingray
+end
+
+require 'stingray/exec/version'
+require 'stingray/exec/cli'

data/lib/stingray/control_api.rb

@@ -0,0 +1,112 @@
+require 'stingray/junk'
+
+module Stingray
+  module ControlApi
+    CONFIGURATIONS = {}.tap do |c|
+      %w(
+        AFM
+        AlertCallback
+        Alerting.Action
+        Alerting.EventType
+        Catalog.Aptimizer.Profile
+        Catalog.Authenticators
+        Catalog.Bandwidth
+        Catalog.JavaExtension
+        Catalog.Monitor
+        Catalog.Persistence
+        Catalog.Protection
+        Catalog.Rate
+        Catalog.Rule
+        Catalog.SLM
+        Catalog.SSL.CertificateAuthorities
+        Catalog.SSL.Certificates
+        Catalog.SSL.ClientCertificates
+        Catalog.SSL.DNSSEC
+        Conf.Extra_1_0
+        Conf.Extra
+        Diagnose_1_0
+        Diagnose
+        GLB.Service
+        GlobalSettings
+        Location
+        Pool
+        System.AccessLogs
+        System.Backups
+        System.Cache_1_0
+        System.Cache_1_1
+        System.Cache
+        System.CloudCredentials
+        System.Connections
+        System.LicenseKeys
+        System.Log
+        System.MachineInfo
+        System.Management
+        System.RequestLogs
+        System.Stats
+        System.Steelhead
+        TrafficIPGroups
+        Users
+        VirtualServer
+      ).each do |cfg|
+        c[cfg] = {
+          :snaked => Stingray::Junk.snakify(cfg),
+          :consted => Stingray::Junk.constify(cfg),
+        }
+      end
+    end
+    CONFIGURATIONS_BY_CONST = {}.tap do |c|
+      CONFIGURATIONS.each do |k, v|
+        c[v[:consted]] = v.merge(:ns => k)
+      end
+    end
+
+    def self.const_missing(sym)
+      const = sym.to_s
+      snaked = (CONFIGURATIONS_BY_CONST[const] || {})[:snaked]
+      unless snaked
+        return super
+      end
+
+      require 'savon'
+      require 'stingray/control_api/endpoint'
+      require 'stingray/control_api/actions'
+
+      const_set(sym, Class.new(Object) do
+        extend Savon::Model
+
+        class << self
+          attr_accessor :_action_map
+        end
+        self._action_map = Stingray::ControlApi::Actions.send(:"#{snaked}_action_map")
+
+        actions(*_action_map.keys.map(&:to_sym))
+        _action_map.keys.each do |action|
+          action_str = action.to_s
+          response_key = :"#{action_str}_response"
+          define_method(action) do |*args|
+            custom_method = :"_custom_#{action_str}"
+            if respond_to?(custom_method)
+              # use a custom method defined in a *Methods module
+              return send(custom_method, *args).body[response_key]
+            else
+              # fall back to the Savon::Model version
+              return super(*args).body[response_key]
+            end
+          end
+        end
+
+        basic_auth(*Stingray::ControlApi::Endpoint.auth)
+        endpoint Stingray::ControlApi::Endpoint.full_endpoint_uri
+        namespace Stingray::ControlApi::Actions.send(:"#{snaked}_namespace")
+
+        _methods_file = File.expand_path("../control_api/#{snaked}_methods.rb", __FILE__)
+        if File.exists?(_methods_file)
+          load _methods_file
+          instance_eval do
+            include Stingray::ControlApi.const_get("#{const}Methods")
+          end
+        end
+      end)
+    end
+  end
+end