lsaws 0.1.0 → 0.2.0

data/lib/lsaws/lister.rb

@@ -9,17 +9,16 @@ module Lsaws
   class Lister
     CONFIG = YAML.load_file(File.join(Lsaws.root, "lsaws.yml"))
 
+    include Utils
+
     def initialize(options)
       @options = options
     end
 
-    def _prepare_entities(sdk, type)
+    def _prepare_entities(sdk, type, &block)
       edef = CONFIG.dig(sdk, type)
-      if edef.is_a?(String)
-        type = edef
-        edef = CONFIG[sdk][type] # should exist
-        raise Error, "unknown entity type: #{type}" unless edef
-      end
+      return _prepare_entities(sdk, edef, &block) if edef.is_a?(String) # redirect like 'default' -> 'instances'
+
       edef ||= {}
       require(edef["require"] || "aws-sdk-#{sdk}")
 
@@ -31,7 +30,8 @@ module Lsaws
       end
       params[:max_results] = @options[:max_results] if @options[:max_results]
 
-      client_class = edef["client_class"] || guess_client_class(sdk)
+      sdkp = SDKParser.new(sdk)
+      client_class = edef["client_class"] || sdkp.client_class_name
       client = Kernel.const_get(client_class).new
       method_name = edef["method"] || (client.respond_to?("describe_#{type}") ? "describe_#{type}" : "list_#{type}")
       unless client.respond_to?(method_name)
@@ -149,21 +149,8 @@ module Lsaws
       proc { |entity| entity.tags.map { |tag| "#{tag.key}=#{tag.value}" }.join(", ") }
     end
 
-    def get_sdks
-      r = []
-      Gem.path.each do |p|
-        next unless Dir.exist?(p)
-
-        r.append(*Dir[File.join(p, "gems/aws-sdk-*")].map do |gem_dir|
-          a = File.basename(gem_dir).split("-")
-          a.size == 4 ? a[2] : nil
-        end)
-      end
-      r.compact.uniq.sort - ["core"]
-    end
-
     def list_sdks
-      _list_array get_sdks
+      _list_array SDKParser.get_sdks
     end
 
     def _list_array(a)
@@ -177,47 +164,8 @@ module Lsaws
       end
     end
 
-    def guess_client_class(sdk)
-      c = Aws.constants.find { |x| x.to_s.downcase == sdk }
-      "Aws::#{c}::Client"
-    end
-
-    def _get_method_rdoc(data, method)
-      pos = data =~ /^\s+def\s+#{method}\s*\(/
-      return nil unless pos
-
-      chunk = ""
-      bs = 4096
-      until chunk["\n\n"]
-        chunk = data[pos - bs..pos]
-        bs *= 2
-      end
-      chunk[chunk.rindex("\n\n") + 2..]
-    end
-
-    def get_entity_types(sdk)
-      require "aws-sdk-#{sdk}"
-      client_class = Kernel.const_get(guess_client_class(sdk))
-      methods = client_class
-                .instance_methods
-                .find_all { |m| m =~ /^(describe|list)_.+s$/ && m !~ /(status|access)$/ }
-
-      return [] if methods.empty?
-
-      data = File.read(client_class.instance_method(methods[0]).source_location[0])
-      methods.delete_if do |m|
-        rdoc = _get_method_rdoc(data, m)
-        next unless rdoc
-
-        required_params = rdoc.scan(/^\s+# @option params \[required, (.+?)\] :(\w+)/)
-        required_params.any?
-      end
-
-      methods.map { |m| m.to_s.sub(/^(describe|list)_/, "") }.sort
-    end
-
     def list_entity_types(sdk)
-      _list_array get_entity_types sdk
+      _list_array SDKParser.new(sdk).entity_types
     end
 
     def _tabulo_guess_max_cols(rows, _cols)
@@ -273,8 +221,7 @@ module Lsaws
         end
       when :yaml
         rows = entities2hashes(sdk, type)
-        rows.each { |row| Utils._deep_transform_keys_in_object!(row, &:to_s) }
-        puts rows.to_yaml
+        puts rows.map { |row| _deep_transform_keys_in_object(row, &:to_s) }.to_yaml
       else
         warn "[!] unknown format: #{@options[:format]}"
         exit 1

data/lib/lsaws/sdk_parser.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+module Lsaws
+  class SDKParser
+    def self.get_sdks
+      r = []
+      Gem.path.each do |p|
+        next unless Dir.exist?(p)
+
+        r.append(*Dir[File.join(p, "gems/aws-sdk-*")].map do |gem_dir|
+          a = File.basename(gem_dir).split("-")
+          a.size == 4 ? a[2] : nil
+        end)
+      end
+      r.compact.uniq.sort - ["core"]
+    end
+
+    def initialize(sdk)
+      @sdk = sdk
+      require "aws-sdk-#{sdk}"
+    end
+
+    def client_class_name
+      @client_class_name ||=
+        begin
+          c = Aws.constants.find { |x| x.to_s.downcase == @sdk }
+          "Aws::#{c}::Client"
+        end
+    end
+
+    def client_class
+      @client_class ||= Kernel.const_get(client_class_name)
+    end
+
+    def entity_types
+      methods = client_class
+                .instance_methods
+                .find_all { |m| m =~ /^(describe|list)_.+s$/ && m !~ /(status|access)$/ }
+
+      return [] if methods.empty?
+
+      methods.delete_if do |m|
+        rdoc = get_method_rdoc(m)
+        next(true) unless rdoc
+
+        required_params = rdoc.scan(/^\s+# @option params \[required, (.+?)\] :(\w+)/)
+        required_params.any?
+      end
+
+      methods.map { |m| m.to_s.sub(/^(describe|list)_/, "") }.sort
+    end
+
+    def get_method_rdoc(method)
+      @source ||= File.read(client_class.instance_method(method).source_location[0])
+
+      pos = @source =~ /^\s+def\s+#{method}\s*\(/
+      return nil unless pos
+
+      chunk = ""
+      bs = 4096
+      until chunk["\n\n"]
+        chunk = @source[pos - bs..pos]
+        bs *= 2
+      end
+      chunk[chunk.rindex("\n\n") + 2..]
+    end
+  end
+end

data/lib/lsaws/utils.rb

@@ -3,7 +3,7 @@
 module Lsaws
   module Utils
     # copypasted from lib/active_support/core_ext/hash/keys.rb
-    def self._deep_transform_keys_in_object!(object, &block)
+    def _deep_transform_keys_in_object!(object, &block)
       case object
       when Hash
         object.each_key do |key|
@@ -17,5 +17,28 @@ module Lsaws
         object
       end
     end
+
+    def _deep_transform_keys_in_object(object, &block)
+      case object
+      when Hash
+        object.each_with_object(object.class.new) do |(key, value), result|
+          result[yield(key)] = _deep_transform_keys_in_object(value, &block)
+        end
+      when Array
+        object.map { |e| _deep_transform_keys_in_object(e, &block) }
+      else
+        object
+      end
+    end
+
+    # "2022-11-15 10:49:00 UTC" => "2022-11-15T10:49:00+00:00"
+    def to_iso8601(s)
+      s.gsub(/([0-9-]{10}) ([0-9:]+{8}) UTC/, '\1T\2+00:00')
+    end
+
+    # "2022-11-15T10:49:00+00:00" => "2022-11-15 10:49:00 UTC"
+    def from_iso8601(s)
+      s.gsub(/([0-9-]{10})T([0-9:]+{8})\+00:00/, '\1 \2 UTC')
+    end
   end
 end

data/lib/lsaws/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Lsaws
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/lsaws.rb

@@ -9,5 +9,7 @@ module Lsaws
 end
 
 require_relative "lsaws/version"
+require_relative "lsaws/sdk_parser"
 require_relative "lsaws/cli"
+require_relative "lsaws/utils"
 require_relative "lsaws/lister"

data/lsaws.yml

@@ -39,3 +39,5 @@ kms:
     - alias_name
     - target_key_id
     - last_updated_date
+s3:
+  default: buckets

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lsaws
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Andrey "Zed" Zaikin
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-02-09 00:00:00.000000000 Z
+date: 2023-02-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -59,6 +59,7 @@ files:
 - lib/lsaws.rb
 - lib/lsaws/cli.rb
 - lib/lsaws/lister.rb
+- lib/lsaws/sdk_parser.rb
 - lib/lsaws/utils.rb
 - lib/lsaws/version.rb
 - lsaws.gemspec