krates 1.6.0

data/lib/kontena/cli/stacks/yaml/custom_validators/affinities_validator.rb

@@ -0,0 +1,19 @@
+module Kontena::Cli::Stacks::YAML::Validations::CustomValidators
+  class AffinitiesValidator < HashValidator::Validator::Base
+    def initialize
+      super('stacks_valid_affinities')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(Array)
+        errors[key] = 'affinity must be array'
+        return
+      end
+
+      invalid_formats = value.find_all { |a| !a.match(/(?<=\!|\=)=/) }
+      if invalid_formats.count > 0
+        errors[key] = "affinity contains invalid formats: #{invalid_formats.join(', ')}"
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/custom_validators/build_validator.rb

@@ -0,0 +1,22 @@
+module Kontena::Cli::Stacks::YAML::Validations::CustomValidators
+  class BuildValidator < HashValidator::Validator::Base
+    def initialize
+      super('stacks_valid_build')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(String) || value.is_a?(Hash)
+        errors[key] = 'build must be string or hash'
+        return
+      end
+      if value.is_a?(Hash)
+        build_validation = {
+          'context' => 'string',
+          'dockerfile' => HashValidator.optional('string'),
+          'args' => HashValidator.optional(-> (value) { value.is_a?(Array) || value.is_a?(Hash) })
+        }
+        HashValidator.validator_for(build_validation).validate(key, value, build_validation, errors)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/custom_validators/certificates_validator.rb

@@ -0,0 +1,22 @@
+module Kontena::Cli::Stacks::YAML::Validations::CustomValidators
+  class CertificatesValidator < HashValidator::Validator::Base
+    def initialize
+      super('stacks_valid_certificates')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(Array)
+        errors[key] = 'certificates must be array'
+        return
+      end
+      certificate_item_validation = {
+        'subject' => 'string',
+        'name' => 'string',
+        'type' => 'string'
+      }
+      value.each do |certificate|
+        HashValidator.validator_for(certificate_item_validation).validate(key, certificate, certificate_item_validation, errors)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/custom_validators/extends_validator.rb

@@ -0,0 +1,22 @@
+module Kontena::Cli::Stacks::YAML::Validations::CustomValidators
+  class ExtendsValidator < HashValidator::Validator::Base
+    def initialize
+      super('stacks_valid_extends')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(String) || value.is_a?(Hash)
+        errors[key] = 'extends must be string or hash'
+        return
+      end
+      if value.is_a?(Hash)
+        extends_validation = {
+          'service' => 'string',
+          'file' => HashValidator.optional('string'),
+          'stack' => HashValidator.optional('string')
+        }
+        HashValidator.validator_for(extends_validation).validate(key, value, extends_validation, errors)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/custom_validators/hooks_validator.rb

@@ -0,0 +1,102 @@
+module Kontena::Cli::Stacks::YAML::Validations::CustomValidators
+  class HooksValidator < HashValidator::Validator::Base
+    def initialize
+      super('stacks_valid_hooks')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(Hash)
+        errors[key] = "must be a mapping, not #{value.class}"
+        return
+      end
+
+      value.keys.each do |hook|
+        unless %w(pre_build pre_start post_start pre_stop).include?(hook)
+          errors[key] = "invalid hook #{hook}"
+        end
+      end
+
+      if value['pre_build']
+        validate_pre_build_hooks(key, value['pre_build'], errors)
+      end
+
+      if value['pre_start']
+        validate_pre_start_hooks(key, value['pre_start'], errors)
+      end
+
+      if value['post_start']
+        validate_post_start_hooks(key, value['post_start'], errors)
+      end
+
+      if value['pre_stop']
+        validate_pre_stop_hooks(key, value['pre_stop'], errors)
+      end
+    end
+
+    def validate_pre_build_hooks(key, pre_build_hooks, errors)
+      unless pre_build_hooks.is_a?(Array)
+        errors[key] = { 'pre_build' => "must be an array" }
+        return
+      end
+      pre_build_validation = {
+        'name' => 'string',
+        'cmd' => 'string'
+      }
+      validator = HashValidator.validator_for(pre_build_validation)
+      pre_build_hooks.each do |pre_build|
+        validator.validate('hooks.pre_build', pre_build, pre_build_validation, errors)
+      end
+    end
+
+    def validate_pre_start_hooks(key, pre_start_hooks, errors)
+      unless pre_start_hooks.is_a?(Array)
+        errors[key] = { 'pre_start' => 'must be an array' }
+        return
+      end
+      pre_start_validation = {
+        'name' => 'string',
+        'instances' => (-> (value) { value.is_a?(Integer) || value == '*' }),
+        'cmd' => 'string',
+        'oneshot' => HashValidator.optional('boolean')
+      }
+      validator = HashValidator.validator_for(pre_start_validation)
+      pre_start_hooks.each do |pre_start|
+        validator.validate('hooks.pre_start', pre_start, pre_start_validation, errors)
+      end
+    end
+
+    def validate_post_start_hooks(key, post_start_hooks, errors)
+      unless post_start_hooks.is_a?(Array)
+        errors[key] = { 'post_start' => 'must be an array' }
+        return
+      end
+      post_start_validation = {
+        'name' => 'string',
+        'instances' => (-> (value) { value.is_a?(Integer) || value == '*' }),
+        'cmd' => 'string',
+        'oneshot' => HashValidator.optional('boolean')
+      }
+      validator = HashValidator.validator_for(post_start_validation)
+      post_start_hooks.each do |post_start|
+        validator.validate('hooks.post_start', post_start, post_start_validation, errors)
+      end
+    end
+
+    def validate_pre_stop_hooks(key, pre_stop_hooks, errors)
+      unless pre_stop_hooks.is_a?(Array)
+        errors[key] = { 'pre_stop' => 'must be an array' }
+        return
+      end
+      pre_stop_validation = {
+        'name' => 'string',
+        'instances' => (-> (value) { value.is_a?(Integer) || value == '*' }),
+        'cmd' => 'string',
+        'oneshot' => HashValidator.optional('boolean')
+      }
+      validator = HashValidator.validator_for(pre_stop_validation)
+      pre_stop_hooks.each do |pre_stop|
+        validator.validate('hooks.pre_stop', pre_stop, pre_stop_validation, errors)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/custom_validators/secrets_validator.rb

@@ -0,0 +1,22 @@
+module Kontena::Cli::Stacks::YAML::Validations::CustomValidators
+  class SecretsValidator < HashValidator::Validator::Base
+    def initialize
+      super('stacks_valid_secrets')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(Array)
+        errors[key] = 'secrets must be array'
+        return
+      end
+      secret_item_validation = {
+        'secret' => 'string',
+        'name' => 'string',
+        'type' => 'string'
+      }
+      value.each do |secret|
+        HashValidator.validator_for(secret_item_validation).validate(key, secret, secret_item_validation, errors)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/certificates_resolver.rb

@@ -0,0 +1,37 @@
+module Kontena::Cli::Stacks::YAML::Opto::Resolvers
+  class Certificates < ::Opto::Resolver
+    include Kontena::Cli::Common
+
+    def resolve
+      return nil unless current_master && current_grid
+      message = hint || 'Select SSL certificates'
+      certificates = get_certificates
+      if certificates.size > 0
+        prompt.multi_select(message) do |menu|
+          menu.default(*default_indexes(certificates)) if option.default
+          certificates.each do |s|
+            menu.choice s['subject']
+          end
+        end
+      end
+    end
+
+    # @return [Array<Hash>] certificates
+    def get_certificates
+      client.get("grids/#{current_grid}/certificates")['certificates']
+    rescue
+      []
+    end
+
+    # @param certificates [Array<Hash>]
+    # @return [Array<Integer>]
+    def default_indexes(certificates)
+      indexes = []
+      option.default.to_a.each do |subject|
+        index = certificates.index { |s| s['subject'] == subject }
+        indexes << index.to_i + 1 if index
+      end
+      indexes
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/prompt_resolver.rb

@@ -0,0 +1,78 @@
+require_relative '../opto'
+
+module Kontena::Cli::Stacks
+  module YAML
+    class Prompt < ::Opto::Resolver
+      include Kontena::Cli::Common
+
+      using ::Opto::Extension::HashStringOrSymbolKey
+
+      def enum?
+        option.type == 'enum'
+      end
+
+      def boolean?
+        option.type == 'boolean'
+      end
+
+      def prompt_word
+        return "Select" if enum?
+        return "Enable" if boolean?
+        "Enter"
+      end
+
+      def question_text
+        (!hint.nil? && hint != option.name) ? "#{hint} :" : "#{prompt_word} #{option.label || option.name} :"
+      end
+
+      def enum_can_be_other?
+        enum? && option.handler.options[:can_be_other] ? true : false
+      end
+
+      def enum
+        opts = option.handler.options[:options]
+        opts << { label: '(Other)', value: nil, description: '(Other)' } if enum_can_be_other?
+
+        answer = prompt.select(question_text) do |menu|
+          menu.enum ':' # makes it show numbers before values, you can press the number to select.
+          menu.default(opts.index {|opt| opt[:value] == option.default }.to_i + 1) if option.default
+          opts.each do |opt|
+            menu.choice opt[:label], opt[:value]
+          end
+        end
+
+        if answer.nil? && enum_can_be_other?
+          ask
+        else
+          answer
+        end
+      end
+
+      def bool
+        prompt.yes?(question_text, default: option.default.nil? || option.default)
+      end
+
+      def echo?
+        return true if option.handler.nil?
+        return true if option.handler.options.nil?
+        return true if option.handler.options[:echo].nil?
+        option.handler.options[:echo]
+      end
+
+      def ask
+        prompt.ask(question_text, default: option.default, echo: echo?)
+      end
+
+      def resolve
+        return nil if option.skip?
+        if enum?
+          enum
+        elsif boolean?
+          bool
+        else
+          ask
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/service_instances_resolver.rb

@@ -0,0 +1,25 @@
+module Kontena::Cli::Stacks
+  module YAML
+    class Opto::Resolvers::ServiceInstances < ::Opto::Resolver
+      include Kontena::Cli::Common
+
+      def resolve
+        return nil unless current_master && current_grid
+        require 'kontena/cli/stacks/show_command'
+        stack = client.get("stacks/#{current_grid}/#{self.stack}")
+        service = stack['services'].find { |s| s['name'] == hint }
+        if service
+          service['instances']
+        else
+          nil
+        end
+      rescue Kontena::Errors::StandardError
+        nil
+      end
+
+      def stack
+        ENV['STACK']
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/service_link_resolver.rb

@@ -0,0 +1,80 @@
+module Kontena::Cli::Stacks::YAML::Opto::Resolvers
+  class ServiceLink < ::Opto::Resolver
+    include Kontena::Cli::Common
+
+    def resolve
+      return nil unless current_master && current_grid
+      message = hint['prompt']
+      name_filter = hint['name']
+      image_filter = hint['image']
+      raise "prompt missing" unless message
+
+      services = get_services
+      services = filter_by_image(services, image_filter) if image_filter
+      services = filter_by_name(services, name_filter) if name_filter
+      return nil if services.size == 0
+      prompt.select(message) do |menu|
+        menu.default(default_index(services)) if option.default
+        menu.choice "<none>", nil unless option.required?
+        services.each do |s|
+          menu.choice service_name(s), service_link(s)
+        end
+      end
+    end
+
+    # @return [Array<Hash>]
+    def get_services
+      client.get("grids/#{current_grid}/services")['services']
+    rescue
+      []
+    end
+
+    # @param [Array<Hash>] services
+    # @return [Integer]
+    def default_index(services)
+      index = services.index {|s| service_link(s) == option.default }
+
+      if index && !option.required?
+        index + 2 # extra offset for the initial <none> option
+      elsif index
+        index + 1 # menu index starts from 1
+      else
+        0 # XXX: this just explodes?
+      end
+    end
+
+    # @param [Hash] service
+    # @return [String]
+    def service_link(service)
+      grid, stack, service = service['id'].split('/')
+      "#{stack}/#{service}"
+    end
+
+    # @param [Hash] service
+    # @return [String]
+    def service_name(service)
+      grid, stack, service = service['id'].split('/')
+      if stack == 'null'.freeze
+        service
+      else
+        "#{stack}/#{service}"
+      end
+    end
+
+    def filter_by_image(services, image)
+      services.select { |s|
+        s['image'].include?(image)
+      }
+    end
+
+    def filter_by_name(services, name)
+      services.select { |s|
+        s['name'].include?(name)
+      }
+    end
+
+    def stack
+      ENV['STACK']
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/vault_cert_prompt_resolver.rb

@@ -0,0 +1,39 @@
+module Kontena::Cli::Stacks::YAML::Opto::Resolvers
+  class VaultCertPrompt < ::Opto::Resolver
+    include Kontena::Cli::Common
+
+    def resolve
+      return nil unless current_master && current_grid
+      message = hint || 'Select SSL certs'
+      secrets = get_secrets.select{ |s|
+        s['name'].match(/(ssl|cert)/i)
+      }
+      if secrets.size > 0
+        prompt.multi_select(message) do |menu|
+          menu.default(*default_indexes(secrets)) if option.default
+          secrets.each do |s|
+            menu.choice s['name']
+          end
+        end
+      end
+    end
+
+    # @return [Array<Hash>] secrets
+    def get_secrets
+      client.get("grids/#{current_grid}/secrets")['secrets']
+    rescue
+      []
+    end
+
+    # @param [Array<Hash>] secrets
+    # @return [Array<Integer>]
+    def default_indexes(secrets)
+      indexes = []
+      option.default.to_a.each do |name|
+        index = secrets.index { |s| s['name'] == name }
+        indexes << index.to_i + 1 if index
+      end
+      indexes
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/vault_resolver.rb

@@ -0,0 +1,13 @@
+module Kontena::Cli::Stacks
+  module YAML
+    class Opto::Resolvers::Vault < ::Opto::Resolver
+      include Kontena::Cli::Common
+      def resolve
+        raise RuntimeError, "Missing or empty vault secret name" if hint.to_s.empty?
+        if current_master && current_grid
+          client.get("secrets/#{current_grid}/#{hint}")['value'] rescue nil
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto/vault_setter.rb

@@ -0,0 +1,12 @@
+module Kontena::Cli::Stacks
+  module YAML
+    class Opto::Setters::Vault < ::Opto::Setter
+      include Kontena::Cli::Common
+      def set(value)
+        if current_master && current_grid
+          client.put("secrets/#{current_grid}/#{hint}", {name: hint, value: value, upsert: true})
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/yaml/opto.rb

@@ -0,0 +1,16 @@
+module Kontena::Cli::Stacks
+  module YAML
+    module Opto
+      module Resolvers; end
+      module Setters; end
+    end
+  end
+end
+require 'opto'
+require_relative 'opto/vault_setter'
+require_relative 'opto/vault_resolver'
+require_relative 'opto/prompt_resolver'
+require_relative 'opto/service_instances_resolver'
+require_relative 'opto/vault_cert_prompt_resolver'
+require_relative 'opto/certificates_resolver'
+require_relative 'opto/service_link_resolver'