sensu-plugins-stackstorm 0.0.2 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: be14a142d09aa3425b4a138764b62fbd2f3fa9ee
-  data.tar.gz: 3cbc5eea735cde517d23f385bc374aae8530a097
+  metadata.gz: 33eeae3a3a44a8091ccb1a0ed48defe122ef6bdb
+  data.tar.gz: e94b67363bf388bb9cd9954f8784f21aca3ea04a
 SHA512:
-  metadata.gz: 1a3b5e462bd30c6ab63c4a8c33e9002f28c072badf755e5ef7115ed13198bff9ad73b27229ecbc4e3eedaf90d6fd35d0cf59aa55c5dff8b2293dd87aed5670db
-  data.tar.gz: 64ad6a92f02447001988d552c97337f2ee0821bc2815c4a32ff34aea63b6859f167c91d9467cc93df870cec67f9f473c098c8756c2da5eee8cedbf9ab9fe7004
+  metadata.gz: 846c6217e64faaff9ff015f93ec46d3633cbee498d5c5bd4c8cb4d876653fb3258cb40194c4a67d262fad31aba65162f013d4abfd9c3006bb5c45e99541fed65
+  data.tar.gz: f98ccc752e68e9f90f2b80fad7c91baadd905fe601c15731b2c047196e2f3d097d6ef5708271b5cfe0cec571a8104d8213d7a3a5a29e0900cc159be4976a8316

data/bin/st2_handler.py

@@ -25,6 +25,9 @@ except ImportError:
     raise ImportError('Missing dependency "pyyaml". \
         Do ``pip install pyyaml``.')
 
+import re
+import six
+
 # ST2 configuration
 
 ST2_API_BASE_URL = None  # 'https://localhost/api/v1/'
@@ -60,6 +63,56 @@ TOKEN_AUTH_HEADER = 'X-Auth-Token'
 API_KEY_AUTH_HEADER = 'St2-Api-Key'
 
 
+class ConfigMeta(type):
+    def __getitem__(kls, key):
+        loaded_config = {}
+        for fpath in kls.get_config_files():
+            with open(fpath, 'r') as f:
+                loaded_config.update(json.loads(f.read()))
+
+        return loaded_config.get(key)
+
+    def __contains__(kls, key):
+        is_contains = False
+        for fpath in kls.get_config_files():
+            if is_contains:
+                break
+
+            with open(fpath, 'r') as f:
+                is_contains = key in json.loads(f.read())
+
+        return is_contains
+
+class ConfigBase(six.with_metaclass(ConfigMeta)):
+    @classmethod
+    def get_config_files(kls):
+        config_files = []
+        if (os.environ.get('SENSU_LOADED_TEMPFILE') and
+            os.path.exists(os.environ.get('SENSU_LOADED_TEMPFILE'))):
+            with open(os.environ.get('SENSU_LOADED_TEMPFILE'), 'r') as f:
+                config_files = f.read().split(':')
+            return config_files
+
+        elif os.environ.get('SENSU_CONFIG_FILES'):
+            return os.environ.get('SENSU_CONFIG_FILES').split(':')
+
+        else:
+            for basedir, _, files in os.walk(kls.CONFIG_DIR_BASE):
+                config_files.extend(['%s/%s' % (basedir, x) for x in files
+                                     if re.match(".*\.json$", x)])
+            return config_files
+
+    @classmethod
+    def get(kls, key, default=None):
+        if key not in kls:
+            return default
+        else:
+            return kls[key]
+
+class Config(ConfigBase):
+    CONFIG_DIR_BASE = '/etc/sensu'
+
+
 def _get_sensu_request_headers():
     b64auth = base64.b64encode(
         "%s:%s" %
@@ -293,7 +346,7 @@ def _register_with_st2(verbose=False):
         sys.exit(2)
 
 
-def _set_config_opts(config_file, verbose=False, unauthed=False, ssl_verify=False):
+def _set_config_opts(verbose=False, unauthed=False, ssl_verify=False):
     global ST2_USERNAME
     global ST2_PASSWORD
     global ST2_API_KEY
@@ -312,29 +365,33 @@ def _set_config_opts(config_file, verbose=False, unauthed=False, ssl_verify=Fals
     UNAUTHED = unauthed
     ST2_SSL_VERIFY = ssl_verify
 
-    if not os.path.exists(config_file):
-        print('Configuration file %s not found. Exiting!!!' % config_file)
-        sys.exit(1)
+    # These are default configurations
+    config = {
+        'st2_username': 'st2admin',
+        'st2_password': 'password',
+        'st2_api_key': '',
+        'st2_api_base_url': 'http://localhost:9101/v1/',
+        'st2_auth_base_url': 'http://localhost:9100/',
+    }
 
-    with open(config_file) as f:
-        config = yaml.safe_load(f)
+    config.update(Config.get('st2_handler', {}))
 
-        if verbose:
-            print('Contents of config file: %s' % config)
-
-        ST2_USERNAME = config['st2_username']
-        ST2_PASSWORD = config['st2_password']
-        ST2_API_KEY = config['st2_api_key']
-        ST2_API_BASE_URL = config['st2_api_base_url']
-        if not ST2_API_BASE_URL.endswith('/'):
-            ST2_API_BASE_URL += '/'
-        ST2_AUTH_BASE_URL = config['st2_auth_base_url']
-        if not ST2_AUTH_BASE_URL.endswith('/'):
-            ST2_AUTH_BASE_URL += '/'
-        SENSU_HOST = config.get('sensu_host', 'localhost')
-        SENSU_PORT = config.get('sensu_port', '4567')
-        SENSU_USER = config.get('sensu_user', None)
-        SENSU_PASS = config.get('sensu_pass', None)
+    if verbose:
+        print('Contents of config file: %s' % config)
+
+    ST2_USERNAME = config['st2_username']
+    ST2_PASSWORD = config['st2_password']
+    ST2_API_KEY = config['st2_api_key']
+    ST2_API_BASE_URL = config['st2_api_base_url']
+    if not ST2_API_BASE_URL.endswith('/'):
+        ST2_API_BASE_URL += '/'
+    ST2_AUTH_BASE_URL = config['st2_auth_base_url']
+    if not ST2_AUTH_BASE_URL.endswith('/'):
+        ST2_AUTH_BASE_URL += '/'
+    SENSU_HOST = Config['api'].get('host', 'localhost')
+    SENSU_PORT = Config['api'].get('port', '4567')
+    SENSU_USER = Config['api'].get('user', None)
+    SENSU_PASS = Config['api'].get('password', None)
 
     if ST2_API_KEY:
         IS_API_KEY_AUTH = True
@@ -356,17 +413,14 @@ def _set_config_opts(config_file, verbose=False, unauthed=False, ssl_verify=Fals
             sys.exit(1)
 
 
-def main(config_file, payload, verbose=False, unauthed=False, ssl_verify=False):
-    _set_config_opts(config_file=config_file, unauthed=unauthed, verbose=verbose,
-                     ssl_verify=ssl_verify)
+def main(payload, verbose=False, unauthed=False, ssl_verify=False):
+    _set_config_opts(unauthed=unauthed, verbose=verbose, ssl_verify=ssl_verify)
     _register_with_st2(verbose=verbose)
     _post_event_to_st2(payload, verbose=verbose)
 
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='StackStorm sensu event handler.')
-    parser.add_argument('config_path',
-                        help='Exchange to listen on')
     parser.add_argument('--verbose', '-v', required=False, action='store_true',
                         help='Verbose mode.')
     parser.add_argument('--unauthed', '-u', required=False, action='store_true',
@@ -376,5 +430,4 @@ if __name__ == '__main__':
                         help='Turn on SSL verification for st2 APIs.')
     args = parser.parse_args()
     payload = sys.stdin.read().strip()
-    main(config_file=args.config_path, payload=payload, verbose=args.verbose,
-         unauthed=args.unauthed, ssl_verify=args.ssl_verify)
+    main(payload=payload, verbose=args.verbose, unauthed=args.unauthed, ssl_verify=args.ssl_verify)

data/bin/st2_handler.rb

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require "sensu-plugins-stackstorm"
+
+SensuPluginsStackStorm::Command.run_python *ARGV

data/lib/sensu-plugins-stackstorm.rb

@@ -0,0 +1 @@
+require 'sensu-plugins-stackstorm/command'

data/lib/sensu-plugins-stackstorm/command.rb

@@ -0,0 +1,10 @@
+module SensuPluginsStackStorm
+  class Command
+    def self.run_python(*argv)
+      bin_dir = File.expand_path(File.dirname(__FILE__)) + '/../../bin/'
+      shell_script_path = File.join(bin_dir, File.basename($PROGRAM_NAME, '.rb') + '.py')
+  
+      exec shell_script_path, *argv
+    end
+  end
+end

data/lib/sensu-plugins-stackstorm/version.rb

@@ -0,0 +1,3 @@
+module SensuPluginsStackStorm
+  VERSION = '0.1.1'
+end

metadata

@@ -1,14 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugins-stackstorm
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.1
 platform: ruby
 authors:
+- StackStorm, Inc.
 - Hiroyasu OHYAMA
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-09-10 00:00:00.000000000 Z
+date: 2017-09-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -40,21 +41,18 @@ dependencies:
         version: '12.0'
 description: ''
 email:
+- info@stackstorm.com
 - user.localhost2000@gmail.com
 executables:
-- st2_handler.py
+- st2_handler.rb
 extensions: []
 extra_rdoc_files: []
 files:
-- ".gitignore"
-- ".travis.yml"
-- Gemfile
-- README.md
-- Rakefile
 - bin/st2_handler.py
-- etc/st2_handler.conf
-- lib/.gitkeep
-- sensu-plugins-stackstorm.gemspec
+- bin/st2_handler.rb
+- lib/sensu-plugins-stackstorm.rb
+- lib/sensu-plugins-stackstorm/command.rb
+- lib/sensu-plugins-stackstorm/version.rb
 homepage: https://github.com/userlocalhost2000/sensu-plugins-stackstorm
 licenses:
 - Apache-2.0
@@ -75,7 +73,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.11
+rubygems_version: 2.6.13
 signing_key: 
 specification_version: 4
 summary: Sensu plugins for StackStorm

data/.gitignore

@@ -1,2 +0,0 @@
-*.sw[p-z]
-*.gem

data/.travis.yml

@@ -1,34 +0,0 @@
-language: ruby
-cache:
-- bundler
-install:
-- bundle install
-rvm:
-- 2.0
-- 2.1
-- 2.2
-- 2.3.0
-- 2.4.1
-notifications:
-  email:
-    recipients:
-    - user.localhost2000@gmail.com
-    on_success: change
-    on_failure: always
-script:
-- bundle exec rake default
-- gem build sensu-plugins-stackstorm.gemspec
-- gem install sensu-plugins-stackstorm-*.gem
-deploy:
-  provider: rubygems
-  api_key:
-    secure: m6eh9IW8L2RCDzfFcnVNuWyr74ywhw+G+4WWwvTTUb7kp5cPu1dkqSeyVM8UobwKuFDwiJTh5rsGVfa2gTnPkaj7jaAWOVDAJcjs8q+oYxMob+dpe++eu3N1adbsqZC27wNH2AlfmIiuEohTHNTMYv6CTiIkr6OVkZCLeD1lLxx9+OyhNaxkb53hmPPgvADmvITK1ZKKyAKb43rCGrKV1/nQDFu+aL/+CmqlH0JLw7vYrx3uVFOmjoVmPWM5VWS1BO2xOZIJ2leInql85C4DK4K3FoqIDhAg+d1o5RqMEz9DvfLHhl2f2EWgHxGk2PyolVUu7JxIoPT1WgCQnKjZ4vOZZBpsdMLzjvnFJBHrFIp0wUTAUgDxHzCA16BKLEoZg0IMcPh+aVPd3jdVJg+RsUuhCr+5wZEsu8LB/LfKiLhSdaM70sjM6CsZXyfhrLFN809f32vGm/fFc3FOKy/ZHEDaVG2jIspcg2pwDwzaQ8wPYwsgp9q+RDpFF49zyk+5f9FjEyftcUwaZhzDb1WIvzQCKtR2dBNgooUZJKrCicyc/L409P1Q1XMeV7BFew+fXChteh4upbzYiqp1yile1EO3QDIDmHpPALh2CbWRwyCmpbn0y8ZfAE4YDs0Bd1V1rJqdsuUFJlEDw7coKrR9uPWzB66om8nlz3KQGfyva3I=
-  gem: sensu-plugins-stackstorm
-  on:
-    tags: true
-    all_branches: true
-    rvm: 2.4.1
-    repo: sensu-plugins/sensu-plugins-stackstorm
-addons:
-  code_climate:
-    repo_token: 

data/Gemfile

@@ -1,3 +0,0 @@
-source 'https://rubygems.org'
-
-gemspec

data/README.md

@@ -1,7 +0,0 @@
-# Sensu-Plugins-StackStorm
-
-# Functionality
-
-## Files
-
-* st2_handler.py

data/Rakefile

@@ -1,6 +0,0 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:spec)
-
-task :default => :spec

data/etc/st2_handler.conf

@@ -1,14 +0,0 @@
----
-# st2 credentials. you can either use username/password combo or specify an API key.
-st2_username: ""
-st2_password: ""
-st2_api_key: ""
-
-# trailing slash mandatory
-st2_api_base_url: "https://localhost/api/v1/"
-st2_auth_base_url: "https://localhost/auth/v1/"
-
-sensu_host: localhost
-sensu_port: 4567
-sensu_user: sensu
-sensu_pass: ""

data/sensu-plugins-stackstorm.gemspec

@@ -1,20 +0,0 @@
-# coding: utf-8
-
-Gem::Specification.new do |spec|
-  spec.name          = "sensu-plugins-stackstorm"
-  spec.version       = '0.0.2'
-  spec.authors       = ["Hiroyasu OHYAMA"]
-  spec.email         = ["user.localhost2000@gmail.com"]
-
-  spec.summary       = %q{Sensu plugins for StackStorm}
-  spec.description   = %q{}
-  spec.homepage      = "https://github.com/userlocalhost2000/sensu-plugins-stackstorm"
-
-  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
-  spec.bindir        = "bin"
-  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
-  spec.license       = "Apache-2.0"
-
-  spec.add_development_dependency 'rspec', '~> 3.6'
-  spec.add_development_dependency 'rake', '~> 12.0'
-end