pager_judy 0.2.0

data/lib/pager_judy/cli/main_command.rb

@@ -0,0 +1,321 @@
+require "clamp"
+require "console_logger"
+require "pager_judy/api/client"
+require "pager_judy/cli/collection_behaviour"
+require "pager_judy/cli/item_behaviour"
+require "pager_judy/cli/time_filtering"
+require "pager_judy/sync"
+require "pager_judy/version"
+
+module PagerJudy
+  module CLI
+
+    class MainCommand < Clamp::Command
+
+      option "--debug", :flag, "enable debugging"
+      option "--dry-run", :flag, "enable dry-run mode"
+
+      option "--version", :flag, "display version" do
+        puts PagerJudy::VERSION
+        exit 0
+      end
+
+      option "--api-key", "KEY", "PagerDuty API key",
+             environment_variable: "PAGER_DUTY_API_KEY"
+
+      subcommand ["escalation-policy", "ep"], "Display escalation policy" do
+
+        parameter "ID", "escalation_policy ID"
+
+        include ItemBehaviour
+
+        def item
+          client.escalation_policies[id]
+        end
+
+      end
+
+      subcommand ["escalation_policies", "eps"], "Display escalation policies" do
+
+        option %w[-q --query], "FILTER", "name filter"
+
+        include CollectionBehaviour
+
+        def collection
+          client.escalation_policies.with(query: query)
+        end
+
+      end
+
+      subcommand ["extension", "ext"], "Display extension" do
+
+        parameter "ID", "extension ID"
+
+        include ItemBehaviour
+
+        def item
+          client.extensions[id]
+        end
+
+      end
+
+      subcommand ["extensions", "exts"], "Display extensions" do
+
+        option %w[-q --query], "FILTER", "name filter"
+
+        include CollectionBehaviour
+
+        def collection
+          client.extensions.with(query: query)
+        end
+
+      end
+
+      subcommand ["incidents"], "Display incidents" do
+
+        option %w[-s --status], "STATUS", "status", :multivalued => true
+
+        include CollectionBehaviour
+        include TimeFiltering
+
+        def collection
+          client.incidents.with(filters)
+        end
+
+        def filters
+          time_filters.merge("statuses[]" => status_list).select { |_,v| v }
+        end
+
+      end
+
+      subcommand ["notifications"], "Display notifications" do
+
+        include TimeFiltering
+
+        self.default_subcommand = "data"
+
+        subcommand ["data", "d"], "Full details" do
+          include CollectionBehaviour::DataSubcommand
+        end
+
+        def collection
+          client.notifications.with(time_filters)
+        end
+
+      end
+
+      subcommand "schedule", "Display schedule" do
+
+        parameter "ID", "schedule ID"
+
+        include ItemBehaviour
+
+        def item
+          client.schedules[id]
+        end
+
+      end
+
+      subcommand "schedules", "Display schedules" do
+
+        option %w[-q --query], "FILTER", "name filter"
+
+        include CollectionBehaviour
+
+        def collection
+          client.schedules.with(query: query)
+        end
+
+      end
+
+      subcommand "service", "Display service" do
+
+        option %w[--include], "TYPE", "linked objects to include", :multivalued => true
+
+        parameter "ID", "service ID"
+
+        include ItemBehaviour
+
+        def item
+          client.services[id].with(
+            "include[]" => include_list
+          )
+        end
+
+      end
+
+      subcommand "services", "Display services" do
+
+        option %w[-q --query], "FILTER", "name filter"
+        option %w[--team], "ID", "team ID", :multivalued => true
+        option %w[--include], "TYPE", "linked objects to include", :multivalued => true
+
+        include CollectionBehaviour
+
+        def collection
+          client.services.with(
+            "query" => query,
+            "team_ids[]" => team_list,
+            "include[]" => include_list
+          )
+        end
+
+      end
+
+      subcommand "team", "Display team" do
+
+        parameter "ID", "team ID"
+
+        include ItemBehaviour
+
+        def item
+          client.teams[id]
+        end
+
+      end
+
+      subcommand "teams", "Display teams" do
+
+        option %w[-q --query], "FILTER", "name filter"
+
+        include CollectionBehaviour
+
+        def collection
+          client.teams.with(query: query)
+        end
+
+      end
+
+      subcommand "user", "Display user" do
+
+        parameter "ID", "user ID"
+
+        include ItemBehaviour
+
+        def item
+          client.users[id]
+        end
+
+      end
+
+      subcommand "users", "User operations" do
+
+        include CollectionBehaviour
+
+        def collection
+          client.users
+        end
+
+      end
+
+      subcommand "vendor", "Specific vendor" do
+
+        parameter "ID", "vendor ID"
+
+        include ItemBehaviour
+
+        def item
+          client.vendors[id]
+        end
+
+      end
+
+      subcommand "vendors", "Vendor list" do
+
+        include CollectionBehaviour
+
+        def collection
+          client.vendors
+        end
+
+      end
+
+      subcommand "viz", "Generate Graphviz Dot diagram" do
+
+        def execute
+          services = client.services
+          escalation_policies = client.escalation_policies
+          puts %(digraph pagerduty {)
+          puts %(rankdir=LR;)
+          integrations = []
+          services.each do |service|
+            service.fetch('integrations').each do |integration|
+              integrations << integration
+              puts %(#{integration.fetch('id')} -> #{service.fetch('id')};)
+              puts %(#{integration.fetch('id')} [label="#{integration.fetch('summary')}",shape=box];)
+            end
+          end
+          puts same_rank(integrations)
+          services.each do |service|
+            puts %(#{service.fetch('id')} [label="#{service.fetch('name')}",shape=box,style=filled,color=lightgrey];)
+            ep = service['escalation_policy']
+            if ep
+              puts %(#{service.fetch('id')} -> #{ep.fetch('id')};)
+            end
+          end
+          puts same_rank(services)
+          escalation_targets = {}
+          escalation_policies.each do |ep|
+            puts %{#{ep.fetch('id')} [label="#{ep.fetch('name')}",shape=box,style=filled,color=lightgrey];}
+            ep.fetch('escalation_rules').each do |rule|
+              rule.fetch('targets').each do |target|
+                puts %(#{ep.fetch('id')} -> #{target.fetch('id')};)
+                escalation_targets[target.fetch('id')] = target.fetch('summary')
+              end
+            end
+          end
+          puts same_rank(escalation_policies)
+          escalation_targets.each do |id, summary|
+            puts %(#{id} [label="#{summary}",shape=box];)
+          end
+          puts %(})
+        end
+
+        private
+
+        def same_rank(things)
+          ids = things.map { |thing| thing.fetch('id') }
+          "{" + ["rank=same", *ids].join(';') + "}"
+        end
+
+      end
+
+      subcommand "configure", "Apply config" do
+
+        option "--check", :flag, "just validate the config"
+
+        parameter "SOURCE", "config file"
+
+        def execute
+          config = PagerJudy::Sync::Config.from(source)
+          return if check?
+          PagerJudy::Sync.sync(client: client, config: config)
+        end
+
+      end
+
+      def run(*args)
+        super(*args)
+      rescue PagerJudy::API::HttpError => e
+        $stderr.puts e.response.body
+        signal_error e.message
+      rescue ConfigMapper::MappingError => e
+        signal_error e.message
+      end
+
+      private
+
+      def client
+        signal_error "no --api-key provided" unless api_key
+        HTTPI.log = false
+        @client ||= PagerJudy::API::Client.new(api_key, logger: logger, dry_run: dry_run?)
+      end
+
+      def logger
+        @logger ||= ConsoleLogger.new(STDERR, debug?)
+      end
+
+    end
+
+  end
+end

data/lib/pager_judy/cli/time_filtering.rb

@@ -0,0 +1,46 @@
+require "clamp"
+require "time"
+
+module PagerJudy
+  module CLI
+
+    module TimeFiltering
+
+      DAY = 24 * 60 * 60
+
+      extend Clamp::Option::Declaration
+
+      option %w[-a --after], "DATETIME", "start date/time", default: "24 hours ago", attribute_name: :after
+      option %w[-b --before], "DATETIME", "end date/time", attribute_name: :before
+
+      protected
+
+      def time_filters
+        {
+          since: after,
+          until: before
+        }.select { |_,v| v }
+      end
+
+      private
+
+      def after=(s)
+        @after = Time.parse(s)
+      end
+
+      def before=(s)
+        @before = Time.parse(s)
+      end
+
+      def default_after
+        Time.now - DAY
+      end
+
+      def default_before
+        Time.now
+      end
+
+    end
+
+  end
+end

data/lib/pager_judy/cli.rb

@@ -0,0 +1 @@
+require "pager_judy/cli/main_command"

data/lib/pager_judy/sync/config.rb

@@ -0,0 +1,69 @@
+require "config_hound"
+require "config_mapper"
+require "config_mapper/config_struct"
+
+module PagerJudy
+  module Sync
+
+    # load and validate our configuration
+    class Config < ConfigMapper::ConfigStruct
+
+      class << self
+
+        # Load configuration.
+        #
+        # Multiple "sources" can be specified. First one wins.
+        # Sources may be a file-name, a URI, or a raw Ruby Hash.
+        #
+        def from(*config_sources)
+          config_data = data_from(*config_sources)
+          config_data.delete("var")
+          from_data(config_data)
+        end
+
+        # Load configuration data.
+        #
+        # Includes are processed; defaults are included, references are expanded.
+        #
+        def data_from(*sources)
+          ConfigHound.load(sources, expand_refs: true, include_key: "include")
+        end
+
+      end
+
+      component_dict :escalation_policies do
+
+        attribute :description
+
+      end
+
+      component_dict :services do
+
+        attribute :description
+
+        attribute :acknowledgement_timeout, default: 1800
+
+        attribute :auto_resolve_timeout, default: 14400
+
+        component :escalation_policy do
+
+          attribute :id
+
+          def to_h
+            super.merge("type" => "escalation_policy_reference")
+          end
+
+        end
+
+        component_dict :integrations do
+
+          attribute :type
+
+        end
+
+      end
+
+    end
+
+  end
+end

data/lib/pager_judy/sync/syncer.rb

@@ -0,0 +1,23 @@
+module PagerJudy
+  module Sync
+
+    class Syncer
+
+      def initialize(client:, config:)
+        @client = client
+        @config = config
+      end
+
+      attr_reader :client
+      attr_reader :config
+
+      def sync
+        config.services.each do |name, detail|
+          client.services.create_or_update_by_name(name, detail.to_h)
+        end
+      end
+
+    end
+
+  end
+end

data/lib/pager_judy/sync.rb

@@ -0,0 +1,12 @@
+require "pager_judy/sync/config"
+require "pager_judy/sync/syncer"
+
+module PagerJudy
+  module Sync
+
+    def self.sync(*args)
+      Syncer.new(*args).sync
+    end
+
+  end
+end

data/lib/pager_judy/version.rb

@@ -0,0 +1,5 @@
+module PagerJudy
+
+  VERSION = '0.2.0'
+
+end

data/pager_judy.gemspec

@@ -0,0 +1,26 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "pager_judy/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "pager_judy"
+  spec.version       = PagerJudy::VERSION
+  spec.authors       = ["Mike Williams"]
+  spec.email         = ["mdub@dogbiscuit.org"]
+
+  spec.summary       = %q{a Ruby client and CLI for PagerDuty}
+  spec.homepage      = "https://github.com/mdub/pager_judy"
+  spec.license       = "MIT"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.16"
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification
+name: pager_judy
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+platform: ruby
+authors:
+- Mike Williams
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2018-12-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+description: 
+email:
+- mdub@dogbiscuit.org
+executables:
+- pagerjudy
+extensions: []
+extra_rdoc_files: []
+files:
+- ".buildkite/pipeline.yml"
+- ".buildkite/upload-pipeline"
+- ".dockerignore"
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- CHANGES.md
+- Dockerfile
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- auto/build
+- auto/dev-environment
+- auto/test
+- docker-compose.yml
+- exe/pagerjudy
+- lib/pager_judy/api/client.rb
+- lib/pager_judy/api/collection.rb
+- lib/pager_judy/api/errors.rb
+- lib/pager_judy/api/fake_api_app.rb
+- lib/pager_judy/api/item.rb
+- lib/pager_judy/api/resource.rb
+- lib/pager_judy/cli.rb
+- lib/pager_judy/cli/collection_behaviour.rb
+- lib/pager_judy/cli/data_display.rb
+- lib/pager_judy/cli/item_behaviour.rb
+- lib/pager_judy/cli/main_command.rb
+- lib/pager_judy/cli/time_filtering.rb
+- lib/pager_judy/sync.rb
+- lib/pager_judy/sync/config.rb
+- lib/pager_judy/sync/syncer.rb
+- lib/pager_judy/version.rb
+- pager_judy.gemspec
+homepage: https://github.com/mdub/pager_judy
+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.7.6
+signing_key: 
+specification_version: 4
+summary: a Ruby client and CLI for PagerDuty
+test_files: []