warthog 0.1.0

data/LICENSE

@@ -0,0 +1,31 @@
+/*
+ * Copyright (c) 2012 by Evernote Corporation, All rights reserved.
+ *
+ * Use of the source code and binary libraries included in this package
+ * is permitted under the following terms:
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.    
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License. You may obtain a 
+ * copy of the License at http://www.apache.org/licenses/LICENSE-2.0
+ */

data/bin/warthog

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+
+lib = File.expand_path(File.dirname(__FILE__) + '/../lib')
+$LOAD_PATH.unshift(lib) if File.directory?(lib) && !$LOAD_PATH.include?(lib)
+
+require 'rubygems'
+require 'warthog'
+require 'warthog/cli'
+
+module Warthog
+
+  ID = File.basename($PROGRAM_NAME).to_sym
+
+  app = CLI::CLI.new(ARGV)
+  app.run
+
+end

data/lib/warthog.rb

@@ -0,0 +1,3 @@
+module Warthog
+
+end

data/lib/warthog/a10/rest.rb

@@ -0,0 +1,84 @@
+require 'httparty'
+
+module Warthog; module A10
+
+  class AXDevice
+
+    class AuthenticationError < StandardError; end
+    class HTTPError < StandardError; end
+    class AXapiError < StandardError; end
+
+    include HTTParty
+
+    API_VERSIONS = %w(V2)
+
+    def initialize(hostname,username,password,options={})
+      @hostname = hostname
+      @session_id = nil
+      @api_version = nil
+
+      axapi :authenticate, :username => username, :password => password
+      raise StandardError, "unable to obtain session id" unless @session_id
+    end
+
+    def slb_service_group_all
+      output = ''
+      service_group_list = axapi 'slb.service_group.getAll'
+      service_group_list['response']['service_group_list']['service_group'].each do |service_group|
+        name = service_group['name']
+        service_group['member_list']['member'].each do |member|
+          output << "%-16s %12s:%-5s %s %s %s\n" % [name,member['server'],member['port'],member['template'],member['priority'],member['status']] if Hash === member
+        end
+      end
+      output
+    end
+
+    def slb_service_group_search(name)
+      output = ''
+      service_group = axapi 'slb.service_group.search', :name => name
+      service_group['response']['service_group']['member_list']['member'].each do |member|
+        output << "%-16s %12s:%-5s %s %s %s\n" % [name,member['server'],member['port'],member['template'],member['priority'],member['status']]
+      end
+      output
+    end
+
+    protected
+
+    def axapi(method,paramvalues={})
+      if @session_id and @api_version
+        uri = "https://#@hostname/services/rest/#@api_version/?session_id=#@session_id&method=#{method}"
+        paramvalues.each do |param,value|
+          uri << "&#{param}=#{value}"
+        end
+        r = self.class.get(uri)
+        if r.code == 200
+          if r['response']['status'] == 'ok'
+            return r
+          elsif r['response']['status'] == 'fail'
+            raise AXapiError, r['response']['error']['msg']
+          end
+        else
+          raise HTTPError, "HTTP code: #{r.code}"
+        end
+      else
+        API_VERSIONS.each do |api_version|
+         r = self.class.get("https://#@hostname/services/rest/#{api_version}/?method=#{method}&username=#{paramvalues[:username]}&password=#{paramvalues[:password]}")
+         if r.code == 200
+           if r['response']['status'] == 'ok'
+             @session_id = r['response']['session_id']
+             @api_version = api_version
+           elsif r['response']['status'] == 'fail'
+             next if r['response']['error']['code'] == '1004'
+             raise AuthenticationError, r['response']['error']['msg'] if r['response']['error']['code'] == '520486968'
+           end
+         else
+           raise HTTPError, "HTTP code: #{r.code}"
+         end
+         raise StandardError, "unable to obtain session id" unless @session_id
+        end
+      end
+    end
+
+  end
+
+end end

data/lib/warthog/about.rb

@@ -0,0 +1,5 @@
+require 'warthog/version'
+module Warthog
+  ME = :warthog
+  ABOUT = "#{ME} v#{VERSION}\nhttps://github.com/evernote/ops-#{ME}\nCopyright (c) 2012 by Evernote Corporation\nLicensed under the Apache License, Version 2.0 (http://www.apache.org/licenses/LICENSE-2.0)"
+end

data/lib/warthog/cli.rb

@@ -0,0 +1,103 @@
+require 'optparse'
+require 'etc'
+require 'warthog'
+require 'warthog/cli/action'
+require 'warthog/a10/rest'
+require 'warthog/version'
+require 'warthog/about'
+
+include Warthog::CLI::Action
+
+module Warthog; module CLI
+
+  class CLI
+
+    COMMANDS = %w(show)
+
+    def initialize(arguments)
+      @arguments = arguments
+      @whoami = File.basename($PROGRAM_NAME).to_sym
+
+      @options = { :debug => false, :username => Etc.getlogin }
+      @action = nil
+      @elesai = nil
+    end
+
+    def run
+      begin
+        parsed_options?
+        arguments_valid?
+        options_valid?
+        process_options
+        process_arguments
+        process_command
+
+      rescue => e #ArgumentError, OptionParser::MissingArgument, Senedsa::SendNsca::ConfigurationError => e
+        if @options[:debug]
+          output_message "#{e.class}: #{e.message}\n  #{e.backtrace.join("\n  ")}",1
+        else
+          output_message e.message,1
+        end
+      end
+    end
+
+    protected
+
+    def parsed_options?
+      opts = OptionParser.new
+
+      opts.banner = "Usage: #{ID} [options] <action> [options] [arguments]"
+      opts.separator ""
+      opts.separator "Actions: (<action> -h displays help for specific action)"
+      opts.separator "    show                             Displays component information"
+      opts.separator ""
+      opts.separator "General options:"
+      opts.on('-H', '--hostname HOSTNAME',              String,              "A10 Load Balancer hostname")                         { |o| @options[:hostname] = o }
+      opts.on('-l', '--login USERNAME',                 String,              "Username")                                           { |o| @options[:username] = o }
+      opts.on('-f', '--fake DIRECTORY',                 String,              "Path to directory with output")                      { |o| @options[:fake] = o }
+      opts.on('-d', '--debug',                                               "Enable debug mode")                                  { @options[:debug] = true}
+      opts.on('-a', '--about',                                               "Display #{ID} information")                          { output_message ABOUT, 0 }
+      opts.on('-V', '--version',                                             "Display #{ID} version")                              { output_message VERSION, 0 }
+      opts.on_tail('--help',                                                 "Show this message")                                  { @options[:HELP] = true }
+
+      opts.order!(@arguments)
+      output_message opts, 0 if (@arguments.size == 0 and @whoami != :check_a10) or @options[:HELP]
+
+      @action = @whoami == :check_a10 ? :check : @arguments.shift.to_sym
+      case @action
+        when :show then  @a10 = Show.new(@arguments,@options)
+        else raise OptionParser::InvalidArgument, "invalid action #@action"
+      end
+    end
+
+    def arguments_valid?
+      true
+    end
+
+    def options_valid?
+      true
+    end
+
+    def process_options
+      raise OptionParser::MissingArgument, "load balancer hostname (-H) must be specified" if @options[:hostname].nil?
+      true
+    end
+
+    def process_arguments
+      true
+    end
+
+    def process_command
+      @a10.exec
+    end
+
+    def output_message(message, exitstatus=nil)
+      m = (! exitstatus.nil? and exitstatus > 0) ? "%s: error: %s" % [ID, message] : message
+#      Syslog.open("elesai", Syslog::LOG_PID | Syslog::LOG_CONS) { |s| s.err "error: #{message}" } unless @options[:debug]
+      $stderr.write "#{m}\n" if STDIN.tty?
+      exit exitstatus unless exitstatus.nil?
+    end
+
+
+  end
+end end

data/lib/warthog/cli/action.rb

@@ -0,0 +1,4 @@
+require 'warthog/cli/action/show'
+
+module Warthog; module CLI; module Action
+end end end

data/lib/warthog/cli/action/show.rb

@@ -0,0 +1,78 @@
+module Warthog; module CLI; module Action
+
+  class Show
+
+    COMPONENTS = %w(service-group)
+
+    def initialize(arguments,options)
+
+      @options = options
+      @arguments = arguments
+
+      @component = nil
+      @subcomponent = nil
+
+      opts = OptionParser.new
+      opts.banner = "Usage: #{ID} [options] show <component> [arguments]"
+      opts.separator ""
+      opts.separator "      <component> is service-group"
+      opts.order!(@arguments)
+
+      options_valid?
+      arguments_valid?
+      process_arguments
+
+      @show = { :service_group => self.method(:show_service_group)
+      }
+    end
+
+    def exec
+      @a10slb = Warthog::A10::AXDevice.new(@options[:hostname],@options[:username],get_password)
+      @show[@component].call(@subcomponent)
+    end
+
+    def show_service_group(name=nil)
+      if name.nil?
+        printf "%s" % [@a10slb.slb_service_group_all]
+      else
+        printf "%s" % [@a10slb.slb_service_group_search(name)]
+      end
+    end
+
+    protected
+
+    def options_valid?
+      true
+    end
+
+    def arguments_valid?
+      raise ArgumentError, "missing component" if @arguments.size == 0
+      raise ArgumentError, "invalid component #{@arguments[0]}" unless COMPONENTS.include? @arguments[0]
+      true
+    end
+
+    def process_arguments
+      @component = case @arguments[0].gsub(/-/,'_').to_sym
+                     when :dummy then a10 = nil
+                     else @arguments[0].gsub(/-/,'_').to_sym
+                   end
+      @subcomponent = @arguments[1]
+    end
+
+    def get_password
+      password = nil
+      begin
+        printf "Password: "
+        STDOUT.flush
+        system "stty -echo"
+        password = $stdin.gets.chomp
+      ensure
+        system "stty echo"
+        printf "\n"
+      end
+      password
+    end
+  end
+
+end end end
+

data/lib/warthog/version.rb

@@ -0,0 +1,3 @@
+module Warthog
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification
+name: warthog
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Gerardo López-Fernádez
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: POkes A10 SLBs with a RESTick
+email: gerir@evernote.com
+executables:
+- warthog
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/warthog/a10/rest.rb
+- lib/warthog/about.rb
+- lib/warthog/cli/action/show.rb
+- lib/warthog/cli/action.rb
+- lib/warthog/cli.rb
+- lib/warthog/version.rb
+- lib/warthog.rb
+- bin/warthog
+- LICENSE
+- README.md
+homepage: https://github.com/evernote/ops-warthog
+licenses:
+- Apache License, Version 2.0
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.3.5
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: A10 REST poker
+test_files: []