lm_rest 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9b62108ac0ae79610c225c5af3a0786fc20e2e8e5a6834e9eac3de9029984f25
-  data.tar.gz: 2fadc9fe2120160c4d49e9bdb77b729f235c312b9c63363a46d7dbef0a7ae0d5
+  metadata.gz: 9e0f1787b39c49b60f1431d41adde6a144152b1f2aed2b1017e7db5d1b45a75a
+  data.tar.gz: 42196790b7103ebdb125758e1e1bc3992553518384016c2d4c66eda1d55ff3fd
 SHA512:
-  metadata.gz: 33b396fea0c3e99de8fa68be8cd6e91af96e0d7b273accfcbdb5dc60c6dd993aff5a1efb1701eb69511afd2841ccd598d1b123c37913d4edc0fc31898d9897ca
-  data.tar.gz: 7f33737751f03dab145f01ec2703ea50082f963875d7a1dec27f8f8fbd89ff6678a432585eb1114d8bd67adc77fcabea6915c94d6e6ebd4b1caf71a49d57d8c3
+  metadata.gz: 2c33110c07110f9a8e82225f1cfc7950916d8aa2613e3180069ef5ded01cf71f85773fce6ee2d3910ee4d2697a6dc28b3976efe43ed73eb1fac3266bbb4dd3d8
+  data.tar.gz: ddf554ea7252633c7f9aa41a4e6a46573d3d9ac4c23b6269d2b0a2c0b32a75dbdbda044518e8923a41c7f95d4fc771c88aa04687a25ef80cce71d45a84a7ae6c

data/bin/system_category_cleaner.rb

@@ -0,0 +1,57 @@
+#!/usr/bin/env ruby
+#
+#
+# Based on a script written originally by Matt Dunham
+#
+require 'lm_rest'
+
+def usage
+  puts "USAGE:\t" + $PROGRAM_NAME + ' account id key applies_to category'
+  puts ""
+  puts "\taccount    - just the beginning of your portal name, like 'hooli'"
+  puts "\tid         - API key access id"
+  puts "\tkey        - API key access key"
+  puts "\tapplies_to - AppliesTo (in quotes on one line) matching devices you want to scrub"
+  puts "\tcategory   - The system category value you wish to remove."
+end
+
+if ARGV.length ==5 
+  @account = ARGV[0]
+  @id  = ARGV[1]
+  @key  = ARGV[2]
+  @at = ARGV[3]
+  @category = ARGV[4]
+  @lm = LMRest::APIClient.new(@account, @id, @key)
+else
+  usage
+  fail 'Bad arguments.'
+end
+
+request = {
+  currentAppliesTo: "true()",
+  needInheritProps: true,
+  originalAppliesTo: "true()",
+  type: "testAppliesTo"
+}
+
+devices = @lm.request(:post, "/functions", request)['originalMatches'].map do |device|
+  [device['id'], device['name']]
+end
+
+devices.each do |id, name|
+  puts "Fetching device id #{id}, #{name}"
+  current = @lm.request(:get, "/device/devices/#{id}/properties/system.categories", nil)
+  if (@lm.remaining.to_f / @lm.limit.to_f) * 100 <= 10
+    puts "sleeping for #{@lm.window} to avoid rate limit violation"
+    sleep @lm.window
+  end
+  if current['value'].split(',').include? @category
+    new_string = current['value'].split(',') - [@category]
+    new = current
+    new['value'] = new_string.join(",")
+    @lm.request(:put, "/device/devices/#{id}/properties/system.categories", new)
+    puts "  Successfully scrubbed!"
+  else
+    puts "  No scrubbing needed."
+  end
+end

data/lib/lm_rest/api_client.rb

@@ -17,6 +17,7 @@ module LMRest
     BASE_URL_SUFFIX = '.logicmonitor.com/santaba/rest'
 
     attr_reader :company, :api_url, :access_id
+    attr_reader :limit, :remaining, :window
 
     def initialize(company = nil, access_id =  nil, access_key = nil)
       APIClient.setup
@@ -78,15 +79,20 @@ module LMRest
 
       json_params = params.to_json
 
-      case method
-      when :get
-        response = RestClient.get(url, headers)
-      when :post
-        response = RestClient.post(url, json_params, headers)
-      when :put
-        response = RestClient.put(url, json_params, headers)
-      when :delete
-        response = RestClient.delete(url, headers: headers)
+      begin
+        case method
+        when :get
+          response = RestClient.get(url, headers)
+        when :post
+          response = RestClient.post(url, json_params, headers)
+        when :put
+          response = RestClient.put(url, json_params, headers)
+        when :delete
+          response = RestClient.delete(url, headers: headers)
+        end
+      rescue => e
+        puts e.http_body
+        raise
       end
 
       if response.code != 200
@@ -94,6 +100,9 @@ module LMRest
         raise
       end
 
+      @limit = response.headers['x_rate_limit_limit']
+      @remaining = response.headers['x_rate_limit_remaining']
+      @window = response.headers['x_rate_limit_window']
 
       JSON.parse(response.body)
     end

data/lib/lm_rest/version.rb

@@ -1,3 +1,3 @@
 module LMRest
-  VERSION = '1.0.3'
+  VERSION = '1.0.4'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lm_rest
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Michael Rodrigues
@@ -99,6 +99,7 @@ files:
 - bin/console
 - bin/ds_checker.rb
 - bin/setup
+- bin/system_category_cleaner.rb
 - lib/lm_rest.rb
 - lib/lm_rest/api_client.rb
 - lib/lm_rest/request_params.rb