buildkite_graphql_ruby 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3be58744b80556d590b40f92e5b5bfd7c8036174
-  data.tar.gz: 54b056181cb1a9fb5c35396d2b3ab2f0950ad84d
+  metadata.gz: eebb44c22e64df704093d9c619a570d3fdc0d51d
+  data.tar.gz: 9cfe4004f3693dcc7ed45772b838c44d106dc393
 SHA512:
-  metadata.gz: e08d44aaf3ded617e2dbf3542e329fcf90ce8fe36ca0a30602b47e47e9f49f576bebc245094b7b058a1bcacad6b6af53ef7bd86ed379eb59fa474c4961ffd5b3
-  data.tar.gz: 6fc7f3a022b56a8da5b718421c10bec8766f0e85e20f744ff35d1405c50add79b4dfd3920e183778e327bf1a75c71aab7f381952bcf8b0a7cccac051f3d3f28a
+  metadata.gz: affc352d9fa6b36f242317163ece7af708b7b76a6221f32b0ff7b7325c2cab598f6a47baf434fe50019e939101f67f124b6d6bc7575e79433f8d0b8e6e3b005c
+  data.tar.gz: 3385d15f403325a192339aa77627b2d977c9ba4563ddea2ac9377a5652d7092f170096f0f7ed97828dd97ce03e52fc8b31e69c63a66e1ead88b5f93cda789072

data/bin/buildkite_graphql_ruby

@@ -1,11 +1,5 @@
 #!/usr/bin/env ruby
-
-require "buildkite_graphql_ruby/cli"
-require "buildkite_graphql_ruby/query_builder"
-require "buildkite_graphql_ruby/query_runner"
-require "buildkite_graphql_ruby/commands/command_map"
-require 'pry'
-
+require "buildkite_graphql_ruby"
 options = BuildkiteGraphqlRuby::CLI.parse(args: ARGV)
 command = BuildkiteGraphqlRuby::Commands::CommandMap.get_command(command_string: options.command)
 result = command.run!(options: options)

data/lib/buildkite_graphql_ruby.rb

@@ -1,4 +1,9 @@
 require "buildkite_graphql_ruby/version"
+require "buildkite_graphql_ruby/cli"
+require "buildkite_graphql_ruby/query_builder"
+require "buildkite_graphql_ruby/query_runner"
+require "buildkite_graphql_ruby/commands/command_map"
+require 'pry'
 
 module BuildkiteGraphqlRuby
 end

data/lib/buildkite_graphql_ruby/cli.rb

@@ -26,8 +26,23 @@ module BuildkiteGraphqlRuby
           options.branch = branch.chomp
         end
 
+        opts.on("-a", "--artifact_to_pull ARTIFACT_TO_PULL",
+                "Artifact to pull for pull_artifacts command") do |artifact_to_pull|
+          options.artifact_to_pull = artifact_to_pull.chomp
+        end
+
+        opts.on("-o", "--output_artifact OUTPUT_ARTIFACT",
+                "Base name of output artifact for pull_artifacts command.  Dir is always tmp") do |output_artifact|
+          options.output_artifact = output_artifact.chomp
+        end
+
+        opts.on("-s", "--slug SLUG",
+                "Build slug to use for pull_artifacts command") do |slug|
+          options.slug = slug.chomp
+        end
+
         opts.on("-C", "--command COMMAND",
-                "What type of query to run (default and only option: branch_status") do |command|
+                "What type of query to run (default option: branch_status)") do |command|
           # TODO: Error on unaccepted query types
           options.command = command
         end

data/lib/buildkite_graphql_ruby/commands/branch_status.rb

@@ -16,7 +16,7 @@ module BuildkiteGraphqlRuby
 
         return if report_if_no_builds(total_builds, options)
 
-        builds = builds_for_branch["edges"].map{|build_response| ResultsParsers::Build.from_response(build_response) }
+        builds = builds_for_branch["edges"].map{|build_response| ResultsParsers::Build.from_response(build_response['node']) }
         
         return if report_if_no_finished_builds(builds, options)
 

data/lib/buildkite_graphql_ruby/commands/command_map.rb

@@ -2,9 +2,11 @@ module BuildkiteGraphqlRuby
   module Commands
     class CommandMap
       require "buildkite_graphql_ruby/commands/branch_status"
+      require "buildkite_graphql_ruby/commands/pull_artifacts"
 
       COMMAND_MAP = {
-        'branch_status' => BranchStatus
+        'branch_status' => BranchStatus,
+        'pull_artifacts' => PullArtifacts
       }
 
       def self.get_command(command_string:)

data/lib/buildkite_graphql_ruby/commands/pull_artifacts.rb

@@ -0,0 +1,29 @@
+module BuildkiteGraphqlRuby
+  module Commands
+    require 'rainbow'
+    require "buildkite_graphql_ruby/results_parsers/build"
+    require "buildkite_graphql_ruby/results_parsers/rspec_results"
+
+    class PullArtifacts
+      def run!(options:)
+        query = BuildkiteGraphqlRuby::QueryBuilder.new.artifacts_for_build_slug(slug: options.slug)
+        query_runner = BuildkiteGraphqlRuby::QueryRunner.new.run_query(query: query, options: options)
+      end
+
+      def report_result(result:, options:)
+        build = ResultsParsers::Build.from_response(result['data']['build'])
+        all_jobs = build.jobs
+        all_artifacts = []
+        all_jobs.each do |job|
+          all_artifacts += job.artifacts.select{|a| a.path == options.artifact_to_pull}
+        end
+
+        all_artifacts.each_with_index do |artifact, index|
+          filename = "tmp/#{index} #{options.output_artifact}"
+          puts "Writing file: #{filename}"
+          File.open(filename, 'w') { |f| f.write(artifact.download) }
+        end
+      end
+    end
+  end
+end

data/lib/buildkite_graphql_ruby/query_builder.rb

@@ -1,5 +1,34 @@
 module BuildkiteGraphqlRuby
   class QueryBuilder
+    def artifacts_for_build_slug(slug:)
+      query = <<~EOS
+        {
+          build(slug:"#{slug}") {
+            jobs(last: 50) {
+              edges {
+                node {
+                  ... on JobTypeCommand {
+                    agent {
+                      id
+                    }
+                    artifacts(first: 300) {
+                      edges {
+                        node {
+                          id
+                          path
+                          downloadURL
+                        }
+                      }
+                    }
+                  }
+                }
+              } 
+            }
+          }
+        }
+        EOS
+    end
+
     def branch_status(branch:)
       query = <<~EOS
         {

data/lib/buildkite_graphql_ruby/results_parsers/build.rb

@@ -31,18 +31,16 @@ module BuildkiteGraphqlRuby
       # NOT_RUN
       # The build wasn't run
 
-      def self.from_response(response)
-        node = response['node']
-
-        jobs = node['jobs']["edges"].select{|j| j['node'].keys.count > 0 }.map{|build_response| ResultsParsers::Job.from_response(build_response) }
+      def self.from_response(build_response)
+        jobs = build_response['jobs']["edges"].select{|j| j['node'].keys.count > 0 }.map{|build_response| ResultsParsers::Job.from_response(build_response) }
  
         new(
-          branch: node['branch'],
-          state: node['state'],
-          url: node['url'],
-          started_at: node['startedAt'] && Time.parse(node['startedAt']),
-          finished_at: node['finishedAt'] && Time.parse(node['finishedAt']),
-          pull_request: node['pullRequest'],
+          branch: build_response['branch'],
+          state: build_response['state'],
+          url: build_response['url'],
+          started_at: build_response['startedAt'] && Time.parse(build_response['startedAt']),
+          finished_at: build_response['finishedAt'] && Time.parse(build_response['finishedAt']),
+          pull_request: build_response['pullRequest'],
           jobs: jobs,
         )
       end

data/lib/buildkite_graphql_ruby/version.rb

@@ -1,3 +1,3 @@
 module BuildkiteGraphqlRuby
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: buildkite_graphql_ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Alex Evanczuk
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-12-18 00:00:00.000000000 Z
+date: 2019-03-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -80,6 +80,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-stack_explorer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.4.9.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.4.9.3
 - !ruby/object:Gem::Dependency
   name: rainbow
   requirement: !ruby/object:Gem::Requirement
@@ -111,6 +125,7 @@ files:
 - lib/buildkite_graphql_ruby/cli.rb
 - lib/buildkite_graphql_ruby/commands/branch_status.rb
 - lib/buildkite_graphql_ruby/commands/command_map.rb
+- lib/buildkite_graphql_ruby/commands/pull_artifacts.rb
 - lib/buildkite_graphql_ruby/query_builder.rb
 - lib/buildkite_graphql_ruby/query_runner.rb
 - lib/buildkite_graphql_ruby/results_parsers/artifact.rb