cloudstack-nagios 0.1.0

data/lib/cloudstack-client/commands/snapshot.rb

@@ -0,0 +1,40 @@
+module CloudstackClient
+
+	module Snapshot
+
+		##
+    # Lists snapshots.
+
+    def list_snapshots(args = {})
+      params = {
+        'command' => 'listSnapshots',
+        'isrecursive' => 'true'
+      }
+      params['name'] = args[:name] if args[:name]
+
+      if args[:project]
+        project = get_project(args[:project])
+        unless project
+          puts "Error: project #{args[:project]} not found."
+          exit 1
+        end
+        params['projectid'] = project['id']
+      end
+      if args[:account]
+        account = list_accounts({name: args[:account]}).first
+        unless account
+          puts "Error: Account #{args[:account]} not found."
+          exit 1
+        end
+        params['domainid'] = account["domainid"]
+        params['account'] = args[:account]
+      end
+      params['listall'] = args[:listall] if args[:listall]
+
+      json = send_request(params)
+      json['snapshot'] || []
+    end
+
+	end
+
+end

data/lib/cloudstack-client/commands/ssh_key_pair.rb

@@ -0,0 +1,106 @@
+module CloudstackClient
+
+	module SshKeyPair
+
+		##
+    # Lists ssh key pairs.
+    #
+
+    def list_ssh_key_pairs(args = {})
+      params = {
+          'command' => 'listSSHKeyPairs',
+          'isrecursive' => true
+      }
+      params['listall'] = true if args[:listall]
+      params['name'] = args[:name] if args[:name]
+      
+      if args[:project]
+        project = get_project(args[:project])
+        unless project
+          puts "Error: project #{args[:project]} not found."
+          exit 1
+        end
+        params['projectid'] = project['id']
+      end
+
+      if args[:account]
+        account = list_accounts({name: args[:account]}).first
+        unless account
+          puts "Error: Account #{args[:account]} not found."
+          exit 1
+        end
+        params['domainid'] = account["domainid"]
+        params['account'] = args[:account]
+      end
+      
+      json = send_request(params)
+      json['sshkeypair'] || []
+    end
+
+    ##
+    # Create ssh key pairs.
+    #
+
+    def create_ssh_key_pair(name, args = {})
+      params = {
+          'command' => 'createSSHKeyPair',
+          'name' => name
+      }
+      if args[:project]
+        project = get_project(args[:project])
+        unless project
+          puts "Error: project #{args[:project]} not found."
+          exit 1
+        end
+        params['projectid'] = project['id']
+      end
+
+      if args[:account]
+        account = list_accounts({name: args[:account]}).first
+        unless account
+          puts "Error: Account #{args[:account]} not found."
+          exit 1
+        end
+        params['domainid'] = account["domainid"]
+        params['account'] = args[:account]
+      end
+      params['publickey'] = args[:public_key] if args[:public_key]
+      
+      json = send_request(params)['keypair']
+    end
+
+    ##
+    # Delete ssh key pairs.
+    #
+
+    def delete_ssh_key_pair(name, args = {})
+      params = {
+          'command' => 'deleteSSHKeyPair',
+          'name' => name
+      }
+
+      if args[:project]
+        project = get_project(args[:project])
+        unless project
+          puts "Error: project #{args[:project]} not found."
+          exit 1
+        end
+        params['projectid'] = project['id']
+      end
+      
+      if args[:account]
+        account = list_accounts({name: args[:account]}).first
+        unless account
+          puts "Error: Account #{args[:account]} not found."
+          exit 1
+        end
+        params['domainid'] = account["domainid"]
+        params['account'] = args[:account]
+      end
+      
+      json = send_request(params)
+    end
+
+	end
+
+end

data/lib/cloudstack-client/commands/template.rb

@@ -0,0 +1,60 @@
+module CloudstackClient
+
+	module Template
+		
+		##
+    # Finds the template with the specified name.
+
+    def get_template(name)
+
+      # TODO: use name parameter
+      # listTemplates in CloudStack 2.2 doesn't seem to work
+      # when the name parameter is specified. When this is fixed,
+      # the name parameter should be added to the request.
+      params = {
+          'command' => 'listTemplates',
+          'templateFilter' => 'executable'
+      }
+      json = send_request(params)
+
+      templates = json['template']
+      if !templates then
+        return nil
+      end
+
+      templates.each { |t|
+        if t['name'] == name then
+          return t
+        end
+      }
+
+      nil
+    end
+
+    ##
+    # Lists all templates that match the specified filter.
+    #
+    # Allowable filter values are:
+    #
+    # * featured - templates that are featured and are public
+    # * self - templates that have been registered/created by the owner
+    # * self-executable - templates that have been registered/created by the owner that can be used to deploy a new VM
+    # * executable - all templates that can be used to deploy a new VM
+    # * community - templates that are public
+
+    def list_templates(args = {})
+      filter = args[:filter] || 'featured'
+      params = {
+          'command' => 'listTemplates',
+          'templateFilter' => filter
+      }
+      params['projectid'] = args[:project_id] if args[:project_id]
+      params['zoneid'] = args[:zone_id] if args[:zone_id]
+      
+      json = send_request(params)
+      json['template'] || []
+    end
+
+	end
+
+end

data/lib/cloudstack-client/commands/user.rb

@@ -0,0 +1,32 @@
+module CloudstackClient
+
+	module User
+
+		##
+    # Lists users.
+    #
+
+    def list_users(args = {})
+      params = {
+          'command' => 'listUsers',
+          'isrecursive' => true
+      }
+      params['listall'] = true if args[:listall]
+      
+      if args[:account]
+        account = list_accounts({name: args[:account]}).first
+        unless account
+          puts "Error: Account #{args[:account]} not found."
+          exit 1
+        end
+        params['domainid'] = account["domainid"]
+        params['account'] = args[:account]
+      end
+      
+      json = send_request(params)
+      json['user'] || []
+    end
+
+	end
+
+end

data/lib/cloudstack-client/commands/volume.rb

@@ -0,0 +1,20 @@
+module CloudstackClient
+
+	module Volume
+
+		##
+    # Lists all volumes.
+
+    def list_volumes(project_id = nil)
+      params = {
+          'command' => 'listVolumes',
+          'listall' => true,
+      }
+      params['projectid'] = project_id if project_id
+      json = send_request(params)
+      json['volume'] || []
+    end
+
+	end
+
+end

data/lib/cloudstack-client/commands/zone.rb

@@ -0,0 +1,57 @@
+module CloudstackClient
+
+	module Zone
+
+	  ##
+    # Finds the zone with the specified name.
+
+    def get_zone(name)
+      params = {
+          'command' => 'listZones',
+          'available' => 'true'
+      }
+      json = send_request(params)
+
+      networks = json['zone']
+      return nil unless networks
+
+      networks.each { |z|
+        if z['name'] == name then
+          return z
+        end
+      }
+
+      nil
+    end
+
+    ##
+    # Finds the default zone for your account.
+
+    def get_default_zone
+      params = {
+          'command' => 'listZones',
+          'available' => 'true'
+      }
+      json = send_request(params)
+
+      zones = json['zone']
+      return nil unless zones
+
+      zones.first
+    end
+
+    ##
+    # Lists all available zones.
+
+    def list_zones
+      params = {
+          'command' => 'listZones',
+          'available' => 'true'
+      }
+      json = send_request(params)
+      json['zone'] || []
+    end
+
+	end
+
+end

data/lib/cloudstack-client/version.rb

@@ -0,0 +1,3 @@
+module CloudstackClient
+  VERSION = "0.2.9"
+end

data/lib/cloudstack-nagios/base.rb

@@ -0,0 +1,69 @@
+require "thor"
+require "yaml"
+
+module CloudstackNagios 
+  class Base < Thor
+    include Thor::Actions
+    include CloudstackNagios::Helper
+    
+    attr_reader :config
+
+    # catch control-c and exit
+    trap("SIGINT") {
+      puts " bye"
+      exit!
+    }
+
+    # exit with return code 1 in case of a error
+    def self.exit_on_failure?
+      true
+    end
+
+    no_commands do  
+      def client(opts = {})
+        @config ||= load_configuration
+        @client ||= CloudstackClient::Connection.new(
+          @config[:url],
+          @config[:api_key],
+          @config[:secret_key],
+          opts.merge({debug: options[:debug]})
+        )
+      end
+
+      def load_configuration(config_file = options[:config], env = options[:environment])
+        unless File.exists?(config_file)
+          say "Configuration file #{config_file} not found.", :red
+          say "Please run \'cs-nagios setup\' to create one."
+          exit 1
+        end
+        begin
+          config = YAML::load(IO.read(config_file))
+        rescue
+          error "Can't load configuration from file #{config_file}."
+          exit 1
+        end
+        if env
+          config = config[env]
+          unless config
+            error "Can't find environment #{env} in configuration file."
+            exit 1
+          end
+        end
+        config
+      end
+
+      def find_project(project_name = options[:project])
+        return nil unless project_name
+        unless project = client.get_project(project_name)
+          say "Project '#{options[:project]}' not found", :red
+          exit 1
+        end
+        project
+      end
+
+      def filter_by(objects, tag_name, tag)
+        objects.select {|r| r[tag_name].downcase == tag.downcase}
+      end
+    end
+  end
+end

data/lib/cloudstack-nagios/cli.rb

@@ -0,0 +1,100 @@
+require 'erubis'
+
+module CloudstackNagios
+  class Cli < CloudstackNagios::Base
+    include Thor::Actions
+
+    package_name "cloudstack-nagios" 
+    map %w(-v --version) => :version
+
+    class_option :config,
+      default: File.join(Dir.home, '.cloudstack-nagios.yml'),
+      aliases: '-c',
+      desc: 'location of your cloudstack-nagios configuration file'
+
+    class_option :environment,
+      aliases: '-e',
+      desc: 'environment to load from the configuration file'
+
+    class_option :debug,
+      desc: 'enable debug output',
+      type: :boolean
+
+    desc "version", "outputs the cloudstack-nagios version"
+    def version
+      say "cloudstack-nagios v#{CloudstackNagios::VERSION}"
+    end
+
+    
+
+    desc "setup", "initial setup of the Cloudstack connection"
+    option :url
+    option :api_key
+    option :secret_key
+    def setup(file = options[:config])
+      config = {}
+      unless options[:url]
+        say "Configuring #{options[:environment] || 'default'} environment."
+        say "What's the URL of your Cloudstack API?", :yellow
+        say "Example: https://my-cloudstack-service/client/api/"
+        config[:url] = ask("URL:", :magenta)
+      end
+      unless options[:api_key]
+        config[:api_key] = ask("API Key:", :magenta)
+      end
+      unless options[:secret_key]
+        config[:secret_key] = ask("Secret Key:", :magenta)
+      end
+      if options[:environment]
+        config = {options[:environment] => config}
+      end
+      if File.exists? file
+        begin
+          old_config = YAML::load(IO.read(file))
+        rescue
+          error "Can't load configuration from file #{config_file}."
+          exit 1
+        end
+        say "Warning: #{file} already exists.", :red
+        exit unless yes?("Do you want to merge your settings? [y/N]", :red)
+        config = old_config.merge(config)
+      end
+      File.open(file, 'w+') {|f| f.write(config.to_yaml) }
+    end
+
+    desc "environments", "list cloudstack-nagios environments"
+    def environments(file = options[:config])
+      config = {}
+      if File.exists? file
+        begin
+          config = YAML::load(IO.read(file))
+        rescue
+          error "Can't load configuration from file #{config_file}."
+          exit 1
+        end
+        table = [%w(Name URL)]
+        table << ["default", config[:url]]
+        config.each_key do |key|
+          table << [key, config[key][:url]] unless key.class == Symbol
+        end
+        print_table table
+      end
+    end
+    map :envs => :environments
+
+    desc "command COMMAND [arg1=val1 arg2=val2...]", "run a custom api command"
+    def command(command, *args)
+      params = {'command' => command}
+      args.each do |arg|
+        arg = arg.split('=')
+        params[arg[0]] = arg[1] 
+      end
+      puts JSON.pretty_generate(client.send_request params)
+    end
+
+    require File.dirname(__FILE__) + '/commands/nagios_config.rb'
+    desc "config SUBCOMMAND ...ARGS", "Nagios configuration commands"
+    subcommand :nagios_config, NagiosConfig
+
+  end # class
+end # module