poke-your-api 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 19cf4f80beebc15282040b29e55cad480f9aa1d351a18fc8f072bba6285eb565
-  data.tar.gz: a2f33c5fdbc67ff3d3c1c01c058b849bb0020cadf101fe31337d7bc7661301a6
+  metadata.gz: ca92137a00063a26c6f3a762768653bc5e0681b404ae8453a3c5595fc8a44075
+  data.tar.gz: 762b370cd9f20cf0f8fbc73f9c01e77168daeaf55162060d0509020e9da7e71f
 SHA512:
-  metadata.gz: 7e7eb02dcd6d9158ab5fb7aded2c93333d9b88a4f1c5c4cc38d76b50c456767ae9adc26fe4e23cab390d4ecea06868363c2800de65b691e2612376190d6f67f3
-  data.tar.gz: 1d05adfd7df8e6c97b4fdc54aae8e3f6a7a2b7f160ba9fdf81ccc322880340e73606949e7b78d2575374d86c919babd7cc96f6a1ef210c0ae0cf888677180d6e
+  metadata.gz: 04c21eb78b1cd3d1f4b3f62e028ed02d24f7a8ea05a3fac7f88c2d108c7a45e8629ab466af52e025e58738580ae863b0733daba322cf8e601d8ce0154c6dce4e
+  data.tar.gz: 8da8bafa688a78a389a5399334cef7dbddce7705476617ab639b99982bf8dd4659869877f849ddc590bcb4ac5bc995b2b31ec14e198cca65b456e51a7e892c03

data/lib/poke/cli.rb

@@ -48,6 +48,19 @@ module Poke
       end
     end
 
+    desc 'speed', 'Run a speed test for an endpoint'
+    method_option :help, aliases: '-h', type: :boolean, desc: 'Display usage information'
+    method_option :env, aliases: '-e', type: :string, desc: 'Set target environment'
+    method_option :name, aliases: '-n', type: :string, desc: 'Find request by name'
+    def speed(*)
+      if options[:help]
+        invoke :help, ['curl']
+      else
+        require_relative 'commands/speed'
+        Poke::Commands::Speed.new(options).execute
+      end
+    end
+
     desc 'response', 'Print out last response'
     method_option :help, aliases: '-h', type: :boolean, desc: 'Display usage information'
     def response(*)

data/lib/poke/commands/speed.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require_relative '../command'
+require_relative '../group'
+require_relative '../config'
+require_relative '../paint'
+require_relative '../last_recently_used'
+
+require 'pathname'
+require 'pastel'
+
+require 'tty-command'
+require 'tty-editor'
+require 'tty-table'
+
+module Poke
+  module Commands
+    class Speed < Poke::Command
+      WRITE_OUT_FIELDS = {
+        'response_code' => ->(e) { e },
+        'time_total' => ->(e) { format('%.2fms', (e * 1000)) }
+      }.freeze
+
+      private
+
+      def run(output: $stdout, errors: $stderr)
+        if (name = @options.fetch(:name, nil))
+          request = Request.find_by_name(Config.find_request_name_by_alias(name))
+        else
+          errors << "Please define the target endpoint with -n or --name\n"
+          return
+        end
+
+        request.use!
+        request.group.use!
+
+        env = @options.fetch(:env, request.group.config.default_env)
+        raise Poke::GroupConfig::InvalidEnv unless request.group.config.valid_env?(env)
+
+        curl_command, comments = build_command(request.path)
+
+        table = TTY::Table.new(
+          [['ENV', env]] +
+          request.group.config.variables(env).to_a +
+          [['CMD', curl_command]] +
+          [['COMMENTS', comments]]
+        )
+        errors << table.render(:unicode, multiline: true, padding: [0, 1, 0, 1])
+        errors << "\n\n"
+
+
+        times = []
+        count = @options.fetch(:count, 10)
+
+        errors << "Running speed test for #{request.name} #{count} times\n\n"
+
+        count.to_i.times do
+          command = TTY::Command.new(printer: :null).run!(
+            request.group.config.variables(env),
+            curl_command
+          )
+
+          if command.failure?
+            output << Pastel.new.decorate(command.err, :red)
+          else
+            errors << Pastel.new.decorate(command.err, :green)
+            errors << "\n"
+            stats = JSON.parse(command.out)
+            times << stats['time_total']
+
+            table = TTY::Table.new(WRITE_OUT_FIELDS.map { |field, lambda| [field, lambda.call(stats[field])] }.to_a)
+            errors << table.render(:unicode, padding: [0, 1, 0, 1])
+            errors << "\n\n"
+          end
+        end
+        errors << "Run count: #{count}\nAverage time: #{format('%.2fms', (times.sum / times.size * 1000))}\n\n"
+      end
+
+      # extract
+      def build_command(path)
+        out = TTY::Command.new(printer: :null).run("cat #{path}").out
+        out << "\n" unless out[-1] == "\n"
+
+        command_lines = out.split(" \\\n").map(&:strip)
+
+        comments = command_lines.filter { |line| line.start_with?('#') }.join(" \\\n")
+        command_lines = command_lines.reject { |line| line.start_with?('#') }
+        command_lines += additional_curl_params
+
+        [command_lines.join(" \\\n  "), comments]
+      end
+
+      # rubocop:disable Style/FormatStringToken
+      def additional_curl_params
+        [
+          "-o #{Poke::Config.response_path}",
+          '-w "%{json}"'
+        ]
+      end
+      # rubocop:enable Style/FormatStringToken
+    end
+  end
+end

data/lib/poke/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Poke
-  VERSION = '0.2.2'
+  VERSION = '0.2.3'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: poke-your-api
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Jan Bator
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-02-21 00:00:00.000000000 Z
+date: 2024-10-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pastel
@@ -130,6 +130,7 @@ files:
 - lib/poke/commands/lru/cat.rb
 - lib/poke/commands/lru/reset.rb
 - lib/poke/commands/response.rb
+- lib/poke/commands/speed.rb
 - lib/poke/config.rb
 - lib/poke/group.rb
 - lib/poke/group_config.rb
@@ -158,7 +159,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.5
+rubygems_version: 3.4.1
 signing_key:
 specification_version: 4
 summary: manage curl requests