Class: Bk::Commands::Artifacts

Inherits:
Base
  • Object
show all
Defined in:
lib/bk/commands/artifacts.rb

Constant Summary collapse

BuildArtifactsQuery = 
Client.parse <<-GRAPHQL
    query($slug: ID!, $jobs_after: String) {
      build(slug: $slug) {
        number
        message
        uuid

        branch
        pipeline {
          slug
        }

        url
        pullRequest {
          id
        }

        state
        scheduledAt
        startedAt
        finishedAt
        canceledAt

        jobs(first: 500, after: $jobs_after) {
          pageInfo {
            endCursor
            hasNextPage
          }
          edges {
            node {
              __typename
              ... on JobTypeWait {
                uuid
                label
              }
              ... on JobTypeTrigger {
                uuid
                label
              }

              ... on JobTypeCommand {
                uuid
                label

                url
                exitStatus

                parallelGroupIndex
                parallelGroupTotal

                artifacts(first: 500) {
                  edges {
                    node {
                      state
                      path
                      downloadURL
                    }
                  }
                }
              }
            }
          }
        }
      }
    }

GRAPHQL

Constants included from Format

Format::HORIZONTAL_PIPE, Format::VERTICAL_PIPE

Instance Attribute Summary

Attributes inherited from Base

#spinner

Instance Method Summary collapse

Methods inherited from Base

#initialize

Methods included from Format

#annotation_colors, #build_colors, #build_header, #is_tty?, #job_colors, #pastel, #vertical_pipe

Methods included from Bk::Color

#colorize, #create_color_hash, #default_color, #error_color, #info_color, #success_color, #warning_color

Constructor Details

This class inherits a constructor from Bk::Commands::Base

Instance Method Details

#call(args: {}, url_or_slug: nil, **options) ⇒ Object 



78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/bk/commands/artifacts.rb', line 78

def call(args: {}, url_or_slug: nil, **options)
  slug = determine_slug(url_or_slug)
  unless slug
    raise ArgumentError, "Unable to figure out slug to use"
  end

  glob = options[:glob]
  download = options[:download]

  jobs_after = nil
  has_next_page = true

  while has_next_page
    result = query(BuildArtifactsQuery, variables: {slug: slug, jobs_after: jobs_after})

    build = result.data.build
    # only show the first time
    if jobs_after.nil?
      puts build_header(build)
      puts ""
    end

    jobs_after = build.jobs.page_info.end_cursor
    has_next_page = build.jobs.page_info.has_next_page

    jobs = build.jobs.edges.map(&:node)
    jobs.each do |job|
      next unless job.respond_to?(:exit_status)

      artifacts = job.artifacts.edges.map(&:node).select { |artifact| glob_matches?(glob, artifact.path) }
      next unless artifacts.any?

      color = job_colors[job.exit_status]
      header = color.call(job.label)
      if job.parallel_group_index && job.parallel_group_total
        header = "#{header} (#{job.parallel_group_index + 1}/#{job.parallel_group_total})"
      end

      puts header

      artifacts.each do |artifact|
        if download
          puts "  - #{artifact.path} (downloading to tmp/bk/[filename])"
          download_artifact(artifact)
        else
          puts "  - #{artifact.path}"
        end
      end
    end
  end
end

#download_artifact(artifact) ⇒ Object 



138
139
140
141
142
143
144
145
 
# File 'lib/bk/commands/artifacts.rb', line 138

def download_artifact(artifact)
  download_url = artifact.to_h["downloadURL"]
  redirected_response_from_aws = Net::HTTP.get_response(URI(download_url))
  artifact_response = Net::HTTP.get_response(URI(redirected_response_from_aws["location"]))
  path = Pathname.new("tmp/bk/#{artifact.path}")
  FileUtils.mkdir_p(path.dirname)
  path.write(artifact_response.body)
end

#glob_matches?(glob, path) ⇒ Boolean

Returns:

  • (Boolean) 


130
131
132
133
134
135
136
 
# File 'lib/bk/commands/artifacts.rb', line 130

def glob_matches?(glob, path)
  if glob
    File.fnmatch?(glob, path, File::FNM_PATHNAME)
  else
    true
  end
end