#
# Copyright:: 2016-2018, Chef Software, 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.

require_relative "../standard_platform"

module Kitchen
  module Driver
    class Aws
      class StandardPlatform
        # https://wiki.debian.org/Cloud/AmazonEC2Image
        class Debian < StandardPlatform
          StandardPlatform.platforms["debian"] = self

          # 12/13 are listed last since we default to the first item in the hash
          # and 12/13 are not released yet. When they're released move them up
          DEBIAN_CODENAMES = {
            11 => "bullseye",
            10 => "buster",
            9 => "stretch",
            8 => "jessie",
            7 => "wheezy",
            6 => "squeeze",
            12 => "bookworm",
            13 => "trixie",
          }.freeze

          # default username for this platform's ami
          # @return [String]
          def username
            "admin"
          end

          def codename
            v = version
            if v && v.size > 1
              warn("WARN: Debian version #{version} specified, but searching for #{version.to_i} instead.")
              v = v.to_i
            end
            v ? DEBIAN_CODENAMES[v.to_i] : DEBIAN_CODENAMES.values.first
          end

          def image_search
            search = {}

            # The Debian AWS owner ID changed for releases 10 and onwards
            # See https://wiki.debian.org/Amazon/EC2/HowTo/awscli
            if version.nil?
              search["owner-id"] = "136693071363"
              search["name"] = "debian-#{DEBIAN_CODENAMES.keys.first}-*"
            elsif version.to_i >= 10
              search["owner-id"] = "136693071363"
              search["name"] = "debian-#{version.to_i}-*"
            else
              search["owner-id"] = "379101102735"
              search["name"] = "debian-#{codename}-*"
            end

            search["architecture"] = architecture if architecture

            search
          end

          def self.from_image(driver, image)
            return unless /debian/i.match?(image.name)

            image.name =~ /\b(\d+|#{DEBIAN_CODENAMES.values.join("|")})\b/i
            version = (Regexp.last_match || [])[1]
            if version&.to_i&.zero?
              version = DEBIAN_CODENAMES.find do |_v, codename|
                codename == version.downcase
              end.first
            end
            new(driver, "debian", version, image.architecture)
          end
        end
      end
    end
  end
end