nexus_api 1.0.0

data/lib/nexus_api/cli_commands/upload.rb

@@ -0,0 +1,126 @@
+module NexusAPI
+  class Upload < ::Thor
+    attr_accessor :api
+
+    include NexusAPI::CLIUtils
+
+    desc 'docker', 'Upload a docker image'
+    option :image,      :aliases => '-i', :desc => 'Docker image to upload', :required => true
+    option :docker_tag, :aliases => '-t', :desc => 'Docker tag', :required => true
+    def docker
+      setup
+      if_file_exists?(file: options[:image].to_s + ':' + options[:docker_tag].to_s, repository: ENV['DOCKER_PUSH_HOSTNAME']) do
+        @api.upload_docker_component(
+          image: options[:image],
+          tag: options[:docker_tag]
+        )
+      end
+    end
+
+    desc 'maven', 'Upload a maven file'
+    option :filename,    :aliases => '-f', :desc => 'Path of file', :required => true
+    option :group_id,    :aliases => '-g', :desc => 'Maven groupId for file', :required => true
+    option :artifact_id, :aliases => '-a', :desc => 'Maven artifactId for file', :required => true
+    option :version,     :aliases => '-v', :desc => 'File version', :required => true
+    option :repository,  :aliases => '-r', :desc => 'Repository name to upload file to; Overrides -e/--team-config; Required if -e not provided'
+    option :tag,         :aliases => '-t', :desc => 'Tag to add to file (tag MUST already exist!)'
+    def maven
+      return false unless repository_set?
+      set(repository: :maven_repository)
+      if_file_exists? do
+        @api.upload_maven_component(
+          filename: options[:filename],
+          group_id: options[:group_id],
+          artifact_id: options[:artifact_id],
+          version: options[:version],
+          repository: options[:repository],
+          tag: options[:tag],
+        )
+      end
+    end
+
+    desc 'npm', 'Upload a npm file'
+    option :filename,   :aliases => '-f', :desc => 'Path of file', :required => true
+    option :repository, :aliases => '-r', :desc => 'Repository name to upload file to; Overrides -e/--team-config; Required if -e not provided'
+    option :tag,        :aliases => '-t', :desc => 'Tag to add to file (tag MUST already exist!)'
+    def npm
+      return false unless repository_set?
+      set(repository: :npm_repository)
+      if_file_exists? do
+        @api.upload_npm_component(
+          filename: options[:filename],
+          repository: options[:repository],
+          tag: options[:tag],
+        )
+      end
+    end
+
+    desc 'pypi', 'Upload a pypi file'
+    option :filename,   :aliases => '-f', :desc => 'Path of file', :required => true
+    option :repository, :aliases => '-r', :desc => 'Repository name to upload file to; Overrides -e/--team-config; Required if -e not provided'
+    option :tag,        :aliases => '-t', :desc => 'Tag to add to file (tag MUST already exist!)'
+    def pypi
+      return false unless repository_set?
+      set(repository: :pypi_repository)
+      if_file_exists? do
+        @api.upload_pypi_component(
+          filename: options[:filename],
+          repository: options[:repository],
+          tag: options[:tag],
+        )
+      end
+    end
+
+    desc 'raw', 'Upload a raw file'
+    option :filename,   :aliases => '-f', :desc => 'Filename', :required => true
+    option :directory,  :aliases => '-d', :desc => 'Path to file', :required => true
+    option :repository, :aliases => '-r', :desc => 'Repository name to upload file to; Overrides -e/--team-config; Required if -e not provided'
+    option :tag,        :aliases => '-t', :desc => 'Tag to add to file (tag MUST already exist!)'
+    def raw
+      return false unless repository_set?
+      set(repository: :raw_repository)
+      if_file_exists? do
+        @api.upload_raw_component(
+          filename: options[:filename],
+          directory: options[:directory],
+          repository: options[:repository],
+          tag: options[:tag],
+        )
+      end
+    end
+
+    desc 'rubygems', 'Upload a rubygems file'
+    option :filename,   :aliases => '-f', :desc => 'Path of file', :required => true
+    option :repository, :aliases => '-r', :desc => 'Repository name to upload file to; Overrides -e/--team-config; Required if -e not provided'
+    option :tag,        :aliases => '-t', :desc => 'Tag to add to file (tag MUST already exist!)'
+    def rubygems
+      return false unless repository_set?
+      set(repository: :rubygems_repository)
+      if_file_exists? do
+        @api.upload_rubygems_component(
+          filename: options[:filename],
+          repository: options[:repository],
+          tag: options[:tag],
+        )
+      end
+    end
+
+    desc 'yum', 'Upload a yum file'
+    option :filename,   :aliases => '-f', :desc => 'Filename', :required => true
+    option :directory,  :aliases => '-d', :desc => 'Path to file', :required => true
+    option :repository, :aliases => '-r', :desc => 'Repository name to upload file to; Overrides -e/--team-config; Required if -e not provided'
+    option :tag,        :aliases => '-t', :desc => 'Tag to add to file (tag MUST already exist!)'
+    def yum
+      return false unless repository_set?
+      set(repository: :yum_repository)
+      if_file_exists? do
+        @api.upload_yum_component(
+          filename: options[:filename],
+          directory: options[:directory],
+          repository: options[:repository],
+          tag: options[:tag],
+        )
+      end
+    end
+  end
+end

data/lib/nexus_api/cli_utils.rb

@@ -0,0 +1,64 @@
+require 'dotenv'
+
+module NexusAPI
+  module CLIUtils
+    def setup
+      Dotenv.load(options[:nexus_config])
+      @api = NexusAPI::API.new(
+        username: ENV['NEXUS_USERNAME'],
+        password: ENV['NEXUS_PASSWORD'],
+        hostname: ENV['NEXUS_HOSTNAME'],
+        docker_pull_hostname: ENV['DOCKER_PULL_HOSTNAME'],
+        docker_push_hostname: ENV['DOCKER_PUSH_HOSTNAME'],
+        team_config: options[:team_config]
+      )
+    end
+
+    def print_element(action:, params:, filter:)
+      setup
+      element = @api.send(action, params)
+      puts options[:full] ? element : element[filter]
+    end
+
+    def print_paginating_set(action:, params:, filter:, proc: nil)
+      setup
+      set = Array.new.tap do |set|
+        loop do
+          params[:paginate] = true
+          set.concat(Array(@api.send(action, params)))
+          break unless @api.paginate?
+        end
+      end
+      proc = proc { set.map{ |element| element[filter] } } if proc.nil?
+      puts options[:full] ? set : proc.call(set)
+    end
+
+    def print_set(action:, filter:)
+      setup
+      set = @api.send(action)
+      puts options[:full] ? set : set.map{ |element| element[filter] }
+    end
+
+    def repository_set?
+      if options[:repository].nil? && options[:team_config].nil?
+        puts "No value provided for required option '--repository' or '--team_config' (only need 1)"
+        return false
+      end
+      true
+    end
+
+    def set(repository:)
+      setup
+      options[:repository] = @api.team_config.send(repository) if options[:repository].nil?
+    end
+
+    def if_file_exists?(file: options[:filename], repository: options[:repository])
+      begin
+        puts "Sending '#{file}' to the '#{repository}' repository in Nexus!"
+        yield
+      rescue Errno::ENOENT
+        puts "'#{file}' does not exist locally."
+      end
+    end
+  end
+end

data/lib/nexus_api/config_manager.rb

@@ -0,0 +1,53 @@
+require 'yaml'
+
+module NexusAPI
+  class ConfigManager
+    def initialize(config_path:)
+      if File.exist?(config_path)
+        @config = YAML.safe_load(File.read(config_path)) || {}
+      else
+        raise "ERROR: Specified config '#{config_path}' does not exist."
+      end
+    end
+
+    def assets_repository
+      @config['assets']
+    end
+
+    def components_repository
+      @config['components']
+    end
+
+    def search_repository
+      @config['search']
+    end
+
+    def tag_repository
+      @config['tag']
+    end
+
+    def maven_repository
+      @config['maven']
+    end
+
+    def npm_repository
+      @config['npm']
+    end
+
+    def pypi_repository
+      @config['pypi']
+    end
+
+    def raw_repository
+      @config['raw']
+    end
+
+    def rubygems_repository
+      @config['rubygems']
+    end
+
+    def yum_repository
+      @config['yum']
+    end
+  end
+end

data/lib/nexus_api/docker_manager.rb

@@ -0,0 +1,112 @@
+require 'nexus_api/docker_shell'
+
+module NexusAPI
+  class DockerManager
+
+    def initialize(docker:, options:)
+      @docker = docker
+      @username = options['username']
+      @password = options['password']
+      @pull_host = options['pull_host']
+      @push_host = options['push_host']
+    end
+
+    def download(image_name:, tag:)
+      return false unless docker_valid?
+      image_name = image_name(@pull_host, image_name, tag)
+      begin
+        image = @docker.pull_image(@username, @password, image_name)
+      rescue Docker::Error::NotFoundError => error
+        puts "ERROR: Failed to pull Docker image #{image_name}.\nDoes it exist in Nexus?"
+        return false
+      end
+      true
+    end
+
+    def upload(image_name:, tag:)
+      return false unless docker_valid?
+      begin
+        return false unless login
+        image = find_image(image_name(@pull_host, image_name, tag))
+        return false if image.nil?
+        tag(image, @push_host, image_name, tag)
+        image.push(nil, repo_tag: image_name(@push_host, image_name, tag))
+      rescue StandardError => error
+        puts "ERROR: #{error.inspect}"
+        return false
+      end
+      true
+    end
+
+    def exists?(image_name:, tag:)
+      return false unless docker_valid?
+      images = find_images(image_name(@pull_host, image_name, tag))
+      return false if images.empty?
+      true
+    end
+
+    def delete(image_name:, tag:)
+      return false unless docker_valid?
+      begin
+        image = find_image(image_name(@pull_host, image_name, tag))
+        return false if image.nil?
+        return false unless image.remove(:force => true)
+      rescue StandardError => error
+        puts "ERROR: #{error.inspect}"
+        return false
+      end
+      true
+    end
+
+    private
+
+    def docker_valid?
+      return true if @docker.validate_version!
+      puts 'ERROR: Your installed version of the Docker API is not supported by the docker-api gem!'
+      false
+    end
+
+    def image_name(host, name, tag)
+      "#{host}/#{name}:#{tag}"
+    end
+
+    def login
+      return true if @docker.authenticate!(@username, @password, @push_host)
+      puts "ERROR: Failed to authenticate to #{@push_host} as #{@username}"
+      false
+    end
+
+    def find_images(name)
+      @docker.list_images.select do |image|
+        image.info['RepoTags'].include?(name)
+      end
+    end
+
+    def valid?(images)
+      if images.empty?
+        puts 'ERROR: No matching docker images found'
+        return false
+      end
+      if images.count > 1
+        puts "ERROR: Found multiple images that match: #{images.map {|image| image.info['RepoTags']} }"
+        return false
+      end
+      true
+    end
+
+    def find_image(name)
+      images = find_images(name)
+      if valid?(images)
+        images.first
+      else
+        nil
+      end
+    end
+
+    def tag(image, host, name, tag)
+      unless image.info['RepoTags'].include?(image_name(host, name, tag))
+        image.tag('repo'=>"#{host}/#{name}", 'tag'=>tag)
+      end
+    end
+  end
+end

data/lib/nexus_api/docker_shell.rb

@@ -0,0 +1,32 @@
+require 'docker'
+
+module NexusAPI
+  class DockerShell
+    def validate_version!
+      Docker.validate_version!
+    end
+
+    def pull_image(username, password, image_name)
+      Docker::Image.create(
+        'username' => username,
+        'password' => password,
+        'fromImage' => image_name
+      )
+    rescue Docker::Error::ClientError => error
+      puts "Error: Could not pull Docker image '#{image_name}'"
+      puts error
+    end
+
+    def authenticate!(username, password, host)
+      Docker.authenticate!(
+        'username' => username,
+        'password' => password,
+        'serveraddress' => host
+      )
+    end
+
+    def list_images
+      Docker::Image.all
+    end
+  end
+end

data/lib/nexus_api/nexus_connection.rb

@@ -0,0 +1,145 @@
+require 'base64'
+require 'rest-client'
+
+module NexusAPI
+  class NexusConnection
+    VALID_RESPONSE_CODES = [200, 204].freeze
+    
+    attr_accessor :continuation_token
+
+    def initialize(username:, password:, hostname:)
+      @username = username
+      @password = password
+      @hostname = hostname
+    end
+
+    def get_response(endpoint:, paginate: false, headers: {'Content-Type' => 'application/json'})
+      response = send_get(endpoint, paginate, headers)
+      response.nil? ? Hash.new : jsonize(response)
+    end
+
+    def get(endpoint:, paginate: false, headers: {'Content-Type' => 'application/json'})
+      valid?(send_get(endpoint, paginate, headers))
+    end
+
+    def post(endpoint:, parameters: '', headers: {'Content-Type' => 'application/json'})
+      response = send_request(
+        :post, 
+        endpoint,
+        parameters: parameters,
+        headers: headers
+      )
+      valid?(response)
+    end
+
+    def put(endpoint:, parameters: '', headers: {'Content-Type' => 'application/json'})
+      response = send_request(
+        :put,
+        endpoint,
+        parameters: parameters,
+        headers: headers
+      )
+      valid?(response)
+    end
+
+    def delete(endpoint:, headers: {'Content-Type' => 'application/json'})
+      response = send_request(
+        :delete,
+        endpoint,
+        headers: headers
+      )
+      valid?(response)
+    end
+
+    def head(asset_url:)
+      catch_connection_error do
+        RestClient.head(asset_url)
+      end
+    end
+
+    def content_length(asset_url:)
+      response = head(asset_url: asset_url)
+      return -1 unless response.respond_to?(:headers)
+      response.headers[:content_length]
+    end
+
+    def download(url:)
+      catch_connection_error do
+        RestClient.get(url, authorization_header)
+      end
+    end
+
+    def paginate?
+      !@continuation_token.nil?
+    end
+
+
+    private
+
+    def valid?(response)
+      return false if response.nil?
+      VALID_RESPONSE_CODES.include?(response.code) ? true : false
+    end
+
+    def handle(error)
+      puts "ERROR: Request failed"
+      puts error.description if error.is_a?(RestClient::Response)
+    end
+
+    def catch_connection_error
+      begin
+        yield
+      rescue SocketError => error
+        return handle(error)
+      rescue RestClient::Unauthorized => error
+        return handle(error)
+      rescue RestClient::ExceptionWithResponse => error
+        return handle(error.response)
+      end
+    end
+
+    def authorization_header
+      { :Authorization => 'Basic ' + Base64.strict_encode64( "#{@username}:#{@password}" ) }
+    end
+
+    def send_request(connection_method, endpoint, parameters: '', headers: {})
+      catch_connection_error do
+        RestClient::Request.execute(
+          method: connection_method,
+          url:    "https://#{@hostname}/service/rest/v1/#{endpoint}",
+          payload: parameters,
+          headers: authorization_header.merge(headers)
+        )
+      end
+    end
+
+    def send_get(endpoint, paginate, headers)
+      # paginate answers is the user requesting pagination, paginate? answers does a continuation token exist
+      # if an empty continuation token is included in the request we'll get an ArrayIndexOutOfBoundsException
+      endpoint += "&continuationToken=#{@continuation_token}" if paginate && paginate?
+      response = send_request(
+        :get,
+        endpoint,
+        headers: headers
+      )
+    end
+
+    # That's right, nexus has inconsistent null values for its api
+    def continuation_token_for(json)
+      return nil if json['continuationToken'].nil?
+      return nil if json['continuationToken'] == 'nil'
+      json['continuationToken']
+    end
+
+    def jsonize(response)
+      json = JSON.parse(response.body)
+      if json.class == Hash
+        @continuation_token = continuation_token_for(json)
+        json = json["items"] if json["items"]
+      end
+      json
+    rescue JSON::ParserError
+      response.body
+    end
+  end
+end

data/lib/nexus_api/version.rb

@@ -0,0 +1,4 @@
+module NexusAPI
+  VERSION = '1.0.0'
+end
+