# frozen_string_literal: true

# Copyright (C) 2019-2021 MongoDB Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

module Mongo
  module Crypt
    module KMS # rubocop:disable Style/Documentation
      # This error indicates that we could not obtain credential for
      # a KMS service.
      #
      # @api private
      class CredentialsNotFound < RuntimeError; end

      # Returns the base provider type from a KMS provider identifier.
      # For example, "aws:name1" returns "aws" and :aws returns "aws".
      #
      # @param [ String | Symbol ] identifier The KMS provider identifier.
      # @return [ String ] The base provider type.
      #
      # @api private
      def self.provider_base_type(identifier)
        identifier.to_s.split(':').first
      end

      # This module contains helper methods for validating KMS parameters.
      #
      # @api private
      module Validations
        # Validate if a KMS parameter is valid.
        #
        # @param [ Symbol ] key The parameter name.
        # @param [ Hash ] opts Hash should contain the parameter under the key.
        # @param [ Boolean ] required Whether the parameter is required or not.
        #   Non-required parameters can be nil.
        #
        # @return [ String | nil ] String parameter value or nil if a
        #   non-required parameter is missing.
        #
        # @raise [ ArgumentError ] If required options are missing or incorrectly
        #   formatted.
        def validate_param(key, opts, format_hint, required: true)
          value = opts.fetch(key)
          return nil if value.nil? && !required

          if value.nil?
            raise ArgumentError.new(
              "The #{key} option must be a String with at least one character; " \
              'currently have nil'
            )
          end
          unless value.is_a?(String)
            raise ArgumentError.new(
              "The #{key} option must be a String with at least one character; " \
              "currently have #{value}"
            )
          end
          if value.empty?
            raise ArgumentError.new(
              "The #{key} option must be a String with at least one character; " \
              'it is currently an empty string'
            )
          end
          value
        rescue KeyError
          if required
            raise ArgumentError.new(
              "The specified KMS provider options are invalid: #{opts}. " +
              format_hint
            )
          end
        end

        # Validate KMS TLS options.
        #
        # @param [ Hash | nil ] options TLS options to connect to KMS
        #   providers. Keys of the hash should be KSM provider names; values
        #   should be hashes of TLS connection options. The options are equivalent
        #   to TLS connection options of Mongo::Client.
        #
        # @return [ Hash ] Provided TLS options if valid.
        #
        # @raise [ ArgumentError ] If required options are missing or incorrectly
        #   formatted.
        def validate_tls_options(options)
          opts = options || {}
          opts.each do |provider, provider_opts|
            if provider_opts[:ssl] == false || opts[:tls] == false
              raise ArgumentError.new(
                "Incorrect TLS options for #{provider}: TLS is required"
              )
            end
            %i[
              ssl_verify_certificate
              ssl_verify_hostname
            ].each do |opt|
              next unless provider_opts[opt] == false

              raise ArgumentError.new(
                "Incorrect TLS options for #{provider}: " +
                'Insecure TLS options prohibited, ' +
                "#{opt} cannot be set to false for KMS"
              )
            end
          end
          opts
        end
        module_function :validate_tls_options
      end
    end
  end
end

require 'mongo/crypt/kms/credentials'
require 'mongo/crypt/kms/master_key_document'
require 'mongo/crypt/kms/aws'
require 'mongo/crypt/kms/azure'
require 'mongo/crypt/kms/gcp'
require 'mongo/crypt/kms/kmip'
require 'mongo/crypt/kms/local'
