#
# License:: Apache License, Version 2.0
#
# 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.
#

require "chef/resource"

class Chef
  class Resource
    class OpensslX509Certificate < Chef::Resource
      require "chef/mixin/openssl_helper"
      include Chef::Mixin::OpenSSLHelper

      preview_resource true
      resource_name :openssl_x509_request

      description "Use the openssl_x509_request resource to generate PEM-formatted x509 certificates requests. If no existing key is specified, the resource will automatically generate a passwordless key with the certificate."
      introduced "14.4"

      property :path, String, name_property: true,
               description: "The optional path to write the file to if you'd like to specify it here instead of in the resource name."

      property :owner, String,
               description: "The owner of all files created by the resource."

      property :group, String,
               description: "The group of all files created by the resource."

      property :mode, [Integer, String], default: "0644",
               description: ""

      property :country, String,
               description: "Value for the C ssl field."

      property :state, String,
               description: "Value for the ST certificate field."

      property :city, String,
               description: "Value for the L certificate field."

      property :org, String,
               description: "Value for the O certificate field."

      property :org_unit, String,
               description: "Value for the OU certificate field."

      property :common_name, String,
               required: true,
               description: "Value for the CN certificate field."

      property :email, String,
               description: "Value for the email ssl field."

      property :key_file, String,
               description: "The path to a certificate key file on the filesystem. If the key_file attribute is specified, the resource will attempt to source a key from this location. If no key file is found, the resource will generate a new key file at this location. If the key_file attribute is not specified, the resource will generate a key file in the same directory as the generated certificate, with the same name as the generated certificate."

      property :key_pass, String,
               description: "The passphrase for an existing key's passphrase."

      property :key_type, String,
               equal_to: %w{rsa ec}, default: "ec",
               description: "The desired type of the generated key (rsa or ec)."

      property :key_length, Integer,
               equal_to: [1024, 2048, 4096, 8192], default: 2048,
               description: "The desired Bit Length of the generated key (if key_type is equal to 'rsa')."

      property :key_curve, String,
               equal_to: %w{secp384r1 secp521r1 prime256v1}, default: "prime256v1",
               description: "The desired curve of the generated key (if key_type is equal to 'ec'). Run openssl ecparam -list_curves to see available options."

      default_action :create

      action :create do
        description "Generate a certificate request."

        unless ::File.exist? new_resource.path
          converge_by("Create CSR #{@new_resource}") do
            file new_resource.name do
              owner new_resource.owner unless new_resource.owner.nil?
              group new_resource.group unless new_resource.group.nil?
              mode new_resource.mode
              content csr.to_pem
              action :create
            end

            file new_resource.key_file do
              mode new_resource.mode
              owner new_resource.owner unless new_resource.owner.nil?
              group new_resource.group unless new_resource.group.nil?
              content key.to_pem
              sensitive true
              action :create_if_missing
            end
          end
        end
      end

      action_class do
        def generate_key_file
          unless new_resource.key_file
            path, file = ::File.split(new_resource.path)
            filename = ::File.basename(file, ::File.extname(file))
            new_resource.key_file path + "/" + filename + ".key"
          end
          new_resource.key_file
        end

        def key
          @key ||= if priv_key_file_valid?(generate_key_file, new_resource.key_pass)
                     OpenSSL::PKey.read ::File.read(generate_key_file), new_resource.key_pass
                   elsif new_resource.key_type == "rsa"
                     gen_rsa_priv_key(new_resource.key_length)
                   else
                     gen_ec_priv_key(new_resource.key_curve)
                   end
          @key
        end

        def subject
          csr_subject = OpenSSL::X509::Name.new()
          csr_subject.add_entry("C", new_resource.country) unless new_resource.country.nil?
          csr_subject.add_entry("ST", new_resource.state) unless new_resource.state.nil?
          csr_subject.add_entry("L", new_resource.city) unless new_resource.city.nil?
          csr_subject.add_entry("O", new_resource.org) unless new_resource.org.nil?
          csr_subject.add_entry("OU", new_resource.org_unit) unless new_resource.org_unit.nil?
          csr_subject.add_entry("CN", new_resource.common_name)
          csr_subject.add_entry("emailAddress", new_resource.email) unless new_resource.email.nil?
          csr_subject
        end

        def csr
          gen_x509_request(subject, key)
        end
      end
    end
  end
end