3scale_toolbox 0.17.1 → 0.18.0

data/lib/3scale_toolbox/commands/methods_command/list_command.rb

@@ -23,19 +23,11 @@ module ThreeScaleToolbox
           end
 
           def run
-            printer.print_collection service_methods
+            printer.print_collection service.methods.map(&:attrs)
           end
 
           private
 
-          def service_methods
-            @service_methods ||= service.methods service_hits_id
-          end
-
-          def service_hits_id
-            @service_hits_id ||= service.hits.fetch('id')
-          end
-
           def service
             @service ||= find_service
           end

data/lib/3scale_toolbox/commands/metrics_command/list_command.rb

@@ -23,7 +23,7 @@ module ThreeScaleToolbox
           end
 
           def run
-            printer.print_collection service.metrics
+            printer.print_collection service.metrics.map(&:attrs)
           end
 
           private

data/lib/3scale_toolbox/commands/plans_command/export/read_plan_limits_step.rb

@@ -9,7 +9,7 @@ module ThreeScaleToolbox
           # add metric system_name out of metric_id
           def call
             result[:limits] = plan.limits.map do |limit|
-              limit.tap { |l| l['metric'] = metric_info(l, 'Limit') }
+              limit.attrs.merge('metric' => metric_info(limit, 'Limit'))
             end
           end
         end

data/lib/3scale_toolbox/commands/plans_command/export/read_plan_methods_step.rb

@@ -21,7 +21,7 @@ module ThreeScaleToolbox
             filtered_limit_methods.each_with_object({}) do |elem, acc|
               # find_method should not return nil.
               # It is assumed that metric_id refers to existing element from previous steps
-              acc[elem['metric_id']] = find_method(elem['metric_id'])
+              acc[elem['metric_id']] = find_method(elem['metric_id']).attrs
             end
           end
 
@@ -33,7 +33,7 @@ module ThreeScaleToolbox
             filtered_pricing_rule_methods.each_with_object({}) do |elem, acc|
               # find_method should not return nil.
               # It is assumed that metric_id refers to existing element from previous steps
-              acc[elem['metric_id']] = find_method(elem['metric_id'])
+              acc[elem['metric_id']] = find_method(elem['metric_id']).attrs
             end
           end
 

data/lib/3scale_toolbox/commands/plans_command/export/read_plan_metrics_step.rb

@@ -21,7 +21,7 @@ module ThreeScaleToolbox
             filtered_limit_metrics.each_with_object({}) do |elem, acc|
               # find_metric should not return nil.
               # It is assumed that metric_id refers to existing element from previous steps
-              acc[elem['metric_id']] = find_metric(elem['metric_id'])
+              acc[elem['metric_id']] = find_metric(elem['metric_id']).attrs
             end
           end
 
@@ -33,7 +33,7 @@ module ThreeScaleToolbox
             filtered_pricing_rule_metrics.each_with_object({}) do |elem, acc|
               # find_metric should not return nil.
               # It is assumed that metric_id refers to existing element from previous steps
-              acc[elem['metric_id']] = find_metric(elem['metric_id'])
+              acc[elem['metric_id']] = find_metric(elem['metric_id']).attrs
             end
           end
 

data/lib/3scale_toolbox/commands/plans_command/export/read_plan_pricing_rules_step.rb

@@ -9,8 +9,7 @@ module ThreeScaleToolbox
           # add metric system_name out of metric_id
           def call
             result[:pricingrules] = plan.pricing_rules.map do |pr|
-              pr.merge('metric' => metric_info(pr, 'PricingRule'),
-                       'cost_per_unit' => pr.fetch('cost_per_unit').to_f)
+              pr.attrs.merge('metric' => metric_info(pr, 'PricingRule'), 'cost_per_unit' => pr.cost_per_unit.to_f)
             end
           end
         end

data/lib/3scale_toolbox/commands/plans_command/export/step.rb

@@ -45,22 +45,14 @@ module ThreeScaleToolbox
             context[:plan] ||= find_plan
           end
 
-          def service_metrics
-            context[:service_metrics] ||= service.metrics
-          end
-
-          def service_methods
-            context[:service_methods] ||= service.methods(service_hits['id'])
-          end
-
           def metric_info(elem, elem_name)
-            if (method = find_method(elem.fetch('metric_id')))
-              { 'type' => 'method', 'system_name' => method.fetch('system_name') }
-            elsif (metric = find_metric(elem.fetch('metric_id')))
-              { 'type' => 'metric', 'system_name' => metric.fetch('system_name') }
+            if (method = find_method(elem.metric_id))
+              { 'type' => 'method', 'system_name' => method.system_name }
+            elsif (metric = find_metric(elem.metric_id))
+              { 'type' => 'metric', 'system_name' => metric.system_name }
             else
-              raise ThreeScaleToolbox::Error, "Unexpected error. #{elem_name} #{elem['id']} " \
-                "referencing to metric id #{elem.fetch('metric_id')} which has not been found"
+              raise ThreeScaleToolbox::Error, "Unexpected error. #{elem_name} #{elem.id} " \
+                "referencing to metric id #{elem.metric_id} which has not been found"
             end
           end
 
@@ -80,15 +72,11 @@ module ThreeScaleToolbox
           end
 
           def find_metric(id)
-            service_metrics.find { |metric| metric['id'] == id }
+            service.metrics.find { |metric| metric.id == id }
           end
 
           def find_method(id)
-            service_methods.find { |method| method['id'] == id }
-          end
-
-          def service_hits
-            context[:service_hits] ||= service.hits
+            service.methods.find { |method| method.id == id }
           end
         end
       end

data/lib/3scale_toolbox/commands/plans_command/import/import_plan_limits_step.rb

@@ -11,15 +11,14 @@ module ThreeScaleToolbox
             # First, delete existing limits
             # Second, add new limits
             plan.limits.each do |limit|
-              metric_id = limit.fetch('metric_id')
-              plan.delete_limit metric_id, limit.fetch('id')
-              puts "Deleted existing plan limit: [metric: #{metric_id}, #{limit}]"
+              limit.delete()
+              puts "Deleted existing plan limit: [metric: #{limit.metric_id}, #{limit.attrs}]"
             end
 
-            resource_limits_processed.each do |limit|
-              metric_id = limit.delete('metric_id')
-              plan.create_limit(metric_id, limit)
-              puts "Created plan limit: [metric: #{metric_id}, #{limit}]"
+            resource_limits_processed.each do |limit_attrs|
+              metric_id = limit_attrs.delete('metric_id')
+              plan.create_limit(metric_id, limit_attrs)
+              puts "Created plan limit: [metric: #{metric_id}, #{limit_attrs}]"
             end
           end
 
@@ -30,7 +29,7 @@ module ThreeScaleToolbox
               metric = find_metric_by_system_name(limit.delete('metric_system_name'))
               # this ImportMetricLimitsStep step is assuming all metrics/methods have been created
               # in previous step, so finding metric should always succeed.
-              limit.merge('metric_id' => metric.fetch('id'))
+              limit.merge('metric_id' => metric.id)
             end
           end
         end

data/lib/3scale_toolbox/commands/plans_command/import/import_plan_metrics_step.rb

@@ -9,36 +9,29 @@ module ThreeScaleToolbox
           def call
             missing_metrics.each(&method(:create_metric))
             missing_methods.each(&method(:create_method))
-
-            # invalidate memoized methods and metrics
-            invalidate_service_methods
-            invalidate_service_metrics
           end
 
           private
 
           def missing_metrics
-            ThreeScaleToolbox::Helper.array_difference(resource_metrics, service_metrics) do |a, b|
-              ThreeScaleToolbox::Helper.compare_hashes(a, b, ['system_name'])
+            ThreeScaleToolbox::Helper.array_difference(resource_metrics, service.metrics) do |a, b|
+              a['system_name'] == b.system_name
             end
           end
 
           def missing_methods
-            ThreeScaleToolbox::Helper.array_difference(resource_methods, service_methods) do |a, b|
-              ThreeScaleToolbox::Helper.compare_hashes(a, b, ['system_name'])
+            ThreeScaleToolbox::Helper.array_difference(resource_methods, service.methods) do |a, b|
+              a['system_name'] == b.system_name
             end
           end
 
           def create_metric(metric_attrs)
-            metric = ThreeScaleToolbox::Entities::Metric.create(service: service,
-                                                                attrs: metric_attrs)
+            metric = ThreeScaleToolbox::Entities::Metric.create(service: service, attrs: metric_attrs)
             puts "Created metric: #{metric.attrs['system_name']}"
           end
 
           def create_method(method_attrs)
-            method = ThreeScaleToolbox::Entities::Method.create(service: service,
-                                                                parent_id: service_hits['id'],
-                                                                attrs: method_attrs)
+            method = ThreeScaleToolbox::Entities::Method.create(service: service, attrs: method_attrs)
             puts "Created method: #{method.attrs['system_name']}"
           end
         end

data/lib/3scale_toolbox/commands/plans_command/import/import_plan_pricing_rules_step.rb

@@ -10,31 +10,24 @@ module ThreeScaleToolbox
             # SET semantics
             # First, delete existing pricing rules
             # Second, add new pricing rules
-            remote_pr_processed.each do |pr|
-              metric_id = pr.fetch('metric_id')
-              plan.delete_pricing_rule metric_id, pr.fetch('id')
-              puts "Deleted existing plan pricing rule: [metric: #{metric_id}, #{pr}]"
+            plan.pricing_rules.each do |pr|
+              pr.delete()
+              puts "Deleted existing plan pricing rule: [metric: #{pr.metric_id}, #{pr.attrs}]"
             end
 
-            resource_pr_processed.each do |pr|
-              metric_id = pr.delete('metric_id')
-              plan.create_pricing_rule(metric_id, pr)
-              puts "Created plan pricing rule: [metric: #{metric_id}, #{pr}]"
+            resource_pr_processed.each do |pr_attrs|
+              metric_id = pr_attrs.delete('metric_id')
+              plan.create_pricing_rule(metric_id, pr_attrs)
+              puts "Created plan pricing rule: [metric: #{metric_id}, #{pr_attrs}]"
             end
           end
 
           private
 
-          def remote_pr_processed
-            plan.pricing_rules.map do |pr|
-              pr.merge('cost_per_unit' => pr.fetch('cost_per_unit').to_f)
-            end
-          end
-
           def resource_pr_processed
             resource_pricing_rules.map do |pr|
               metric = find_metric_by_system_name(pr.delete('metric_system_name'))
-              pr.merge('metric_id' => metric.fetch('id'),
+              pr.merge('metric_id' => metric.id,
                        'cost_per_unit' => pr.fetch('cost_per_unit').to_f)
             end
           end

data/lib/3scale_toolbox/commands/plans_command/import/step.rb

@@ -65,28 +65,8 @@ module ThreeScaleToolbox
             artifacts_resource['plan_features'] || []
           end
 
-          def service_metrics
-            context[:service_metrics] ||= service.metrics
-          end
-
-          def invalidate_service_metrics
-            context[:service_metrics] = nil
-          end
-
-          def service_hits
-            context[:service_hits] ||= service.hits
-          end
-
-          def service_methods
-            context[:service_methods] ||= service.methods(service_hits['id'])
-          end
-
           def service_metrics_and_methods
-            service_metrics + service_methods
-          end
-
-          def invalidate_service_methods
-            context[:service_methods] = nil
+            service.metrics + service.methods
           end
 
           def service_features
@@ -103,7 +83,7 @@ module ThreeScaleToolbox
           end
 
           def find_metric_by_system_name(system_name)
-            service_metrics_and_methods.find { |metric| metric['system_name'] == system_name }
+            service_metrics_and_methods.find { |metric| metric.system_name == system_name }
           end
 
           private

data/lib/3scale_toolbox/commands/plans_command/list_command.rb

@@ -23,7 +23,7 @@ module ThreeScaleToolbox
           end
 
           def run
-            printer.print_collection service.plans
+            printer.print_collection service.plans.map(&:attrs)
           end
 
           private

data/lib/3scale_toolbox/commands/plans_command/show_command.rb

@@ -5,7 +5,7 @@ module ThreeScaleToolbox
         class ShowSubcommand < Cri::CommandRunner
           include ThreeScaleToolbox::Command
 
-          FIELDS_TO_SHOW = %w[id name system_name approval_required end_user_required
+          FIELDS_TO_SHOW = %w[id name system_name approval_required
                               cost_per_month setup_fee trial_period_days].freeze
 
           def self.command

data/lib/3scale_toolbox/commands/policies_command.rb

@@ -0,0 +1,24 @@
+require '3scale_toolbox/commands/policies_command/export_command'
+require '3scale_toolbox/commands/policies_command/import_command'
+
+module ThreeScaleToolbox
+  module Commands
+    module PoliciesCommand
+      include ThreeScaleToolbox::Command
+      def self.command
+        Cri::Command.define do
+          name        'policies'
+          usage       'policies <sub-command> [options]'
+          summary     'policies super command'
+          description 'Policies commands'
+
+          run do |_opts, _args, cmd|
+            puts cmd.help
+          end
+        end
+      end
+      add_subcommand(ExportSubcommand)
+      add_subcommand(ImportSubcommand)
+    end
+  end
+end

data/lib/3scale_toolbox/commands/policies_command/export_command.rb

@@ -0,0 +1,98 @@
+module ThreeScaleToolbox
+  module Commands
+    module PoliciesCommand
+      class ExportSubcommand < Cri::CommandRunner
+        include ThreeScaleToolbox::Command
+
+        class JSONSerializer
+          def call(object)
+            JSON.pretty_generate(object)
+          end
+        end
+
+        class YAMLSerializer
+          def call(object)
+            YAML.dump(object)
+          end
+        end
+
+        class SerializerTransformer
+          def call(output_format)
+            raise unless %w[yaml json].include?(output_format)
+
+            case output_format
+            when 'yaml'
+              YAMLSerializer.new
+            when 'json'
+              JSONSerializer.new
+            end
+          end
+        end
+
+        def self.command
+          Cri::Command.define do
+            name        'export'
+            usage       'export [opts] <remote> <product>'
+            summary     'export product policy chain'
+            description 'export product policy chain'
+
+            option      :f, :file, 'Write to file instead of stdout', argument: :required
+            option      :o, :output, 'Output format. One of: json|yaml', argument: :required, transform: SerializerTransformer.new
+            param       :remote
+            param       :service_ref
+
+            runner ExportSubcommand
+          end
+        end
+
+        def run
+          select_output do |output|
+            output.write(serializer.call(product.policies))
+          end
+        end
+
+        private
+
+        def remote
+          @remote ||= threescale_client(arguments[:remote])
+        end
+
+        def product
+          @product ||= find_product
+        end
+
+        def service_ref
+          arguments[:service_ref]
+        end
+
+        def find_product
+          Entities::Service.find(remote: remote,
+                                 ref: service_ref).tap do |svc|
+            raise ThreeScaleToolbox::Error, "Product #{service_ref} does not exist" if svc.nil?
+          end
+        end
+
+        def file
+          options[:file]
+        end
+
+        def select_output
+          ios = if file
+                  File.open(file, 'w')
+                else
+                  $stdout
+                end
+          begin
+            yield(ios)
+          ensure
+            ios.close
+          end
+        end
+
+        def serializer
+          options.fetch(:output, YAMLSerializer.new)
+        end
+      end
+    end
+  end
+end

data/lib/3scale_toolbox/commands/policies_command/import_command.rb

@@ -0,0 +1,61 @@
+module ThreeScaleToolbox
+  module Commands
+    module PoliciesCommand
+      class ImportSubcommand < Cri::CommandRunner
+        include ThreeScaleToolbox::Command
+        include ThreeScaleToolbox::ResourceReader
+
+        def self.command
+          Cri::Command.define do
+            name        'import'
+            usage       'import [opts] <remote> <product>'
+            summary     'import product policy chain'
+            description 'import product policy chain'
+
+            option      :f, :file, 'Read from file', argument: :required
+            option      :u, :url, 'Read from url', argument: :required
+            param       :remote
+            param       :service_ref
+
+            runner ImportSubcommand
+          end
+        end
+
+        def run
+          res = product.update_policies('policies_config' => policies)
+          if res.is_a?(Hash) && (errors = res['errors'])
+            raise ThreeScaleToolbox::Error, "Product policies have not been imported. #{errors}"
+          end
+          if res.is_a?(Array) && (error_item = res.find { |i| i.key?('errors') })
+            raise ThreeScaleToolbox::Error, "Product policies have not been imported. #{error_item['errors']}"
+          end
+        end
+
+        private
+
+        def remote
+          @remote ||= threescale_client(arguments[:remote])
+        end
+
+        def service_ref
+          arguments[:service_ref]
+        end
+
+        def product
+          @product ||= find_product
+        end
+
+        def find_product
+          Entities::Service.find(remote: remote,
+                                 ref: service_ref).tap do |svc|
+            raise ThreeScaleToolbox::Error, "Product #{service_ref} does not exist" if svc.nil?
+          end
+        end
+
+        def policies
+          @policies ||= load_resource(options[:file] || options[:url] || '-')
+        end
+      end
+    end
+  end
+end