From f07bb135cd9d76c9507882c028b322b1b1880e76 Mon Sep 17 00:00:00 2001
From: Grant Slater <github@firefishy.com>
Date: Mon, 3 Jul 2023 17:01:59 +0100
Subject: [PATCH] Add awscli cookbook

---
 .kitchen.yml                                  |  3 +
 cookbooks/awscli/README.md                    |  3 +
 cookbooks/awscli/attributes/default.rb        |  2 +
 cookbooks/awscli/metadata.rb                  |  9 ++
 cookbooks/awscli/recipes/default.rb           | 89 +++++++++++++++++++
 .../awscli/serverspec/awscli_spec.rb          |  8 ++
 6 files changed, 114 insertions(+)
 create mode 100644 cookbooks/awscli/README.md
 create mode 100644 cookbooks/awscli/attributes/default.rb
 create mode 100644 cookbooks/awscli/metadata.rb
 create mode 100644 cookbooks/awscli/recipes/default.rb
 create mode 100644 test/integration/awscli/serverspec/awscli_spec.rb

diff --git a/.kitchen.yml b/.kitchen.yml
index e40508517..ad7fbe7c8 100644
--- a/.kitchen.yml
+++ b/.kitchen.yml
@@ -60,6 +60,9 @@ suites:
   - name: apt
     run_list:
       - recipe[apt::default]
+  - name: awscli
+    run_list:
+      - recipe[awscli::default]
   - name: backup
     run_list:
       - recipe[backup::default]
diff --git a/cookbooks/awscli/README.md b/cookbooks/awscli/README.md
new file mode 100644
index 000000000..9009059e7
--- /dev/null
+++ b/cookbooks/awscli/README.md
@@ -0,0 +1,3 @@
+# awscli Cookbook
+
+This cookbook installs and configures awscli
diff --git a/cookbooks/awscli/attributes/default.rb b/cookbooks/awscli/attributes/default.rb
new file mode 100644
index 000000000..09d356dbe
--- /dev/null
+++ b/cookbooks/awscli/attributes/default.rb
@@ -0,0 +1,2 @@
+# Set the default awscli version
+default[:awscli][:version] = "latest"
diff --git a/cookbooks/awscli/metadata.rb b/cookbooks/awscli/metadata.rb
new file mode 100644
index 000000000..56be15b3c
--- /dev/null
+++ b/cookbooks/awscli/metadata.rb
@@ -0,0 +1,9 @@
+name              "awscli"
+maintainer        "OpenStreetMap Administrators"
+maintainer_email  "admins@openstreetmap.org"
+license           "Apache-2.0"
+description       "Installs and configures awscli"
+
+version           "1.0.0"
+supports          "debian"
+supports          "ubuntu"
diff --git a/cookbooks/awscli/recipes/default.rb b/cookbooks/awscli/recipes/default.rb
new file mode 100644
index 000000000..22684864d
--- /dev/null
+++ b/cookbooks/awscli/recipes/default.rb
@@ -0,0 +1,89 @@
+#
+# Cookbook:: awscli
+# Recipe:: default
+#
+# Copyright:: 2023, OpenStreetMap Foundation
+#
+# 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
+#
+#     https://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.
+#
+
+cache_dir = Chef::Config[:file_cache_path]
+
+# Determine architecture of system for the AWS CLI download
+awscli_arch = if arm?
+                "aarch64"
+              else
+                "x86_64"
+              end
+
+awscli_version_suffix = if node[:awscli][:version] == "latest"
+                          "" # latest version does not have a suffix
+                        else
+                          "-#{node[:awscli][:version]}"
+                        end
+
+awscli_zip = "awscliv2#{awscli_version_suffix}.zip"
+
+# Ensure unpack directory is removed
+directory "#{cache_dir}/awscli" do
+  action :delete
+  recursive true
+end
+
+# Remove any existing AWS CLI zip files, unless it's the one we're downloading
+Dir.glob("#{cache_dir}/awscliv2*.zip").each do |zip|
+  next if zip == "#{cache_dir}/#{awscli_zip}"
+
+  file zip do
+    action :delete
+    backup false
+  end
+end
+
+# Download the AWS CLI zip file
+remote_file "#{cache_dir}/#{awscli_zip}" do
+  source "https://awscli.amazonaws.com/awscli-exe-linux-#{awscli_arch}#{awscli_version_suffix}.zip"
+  owner "root"
+  group "root"
+  mode "644"
+  backup false
+  ignore_failure true
+end
+
+# Extract the AWS CLI zip file
+archive_file "#{cache_dir}/#{awscli_zip}" do
+  path "#{cache_dir}/#{awscli_zip}"
+  destination "#{cache_dir}/awscli"
+  strip_components 1
+  action :nothing
+  subscribes :extract, "remote_file[#{cache_dir}/#{awscli_zip}]", :immediately
+end
+
+# Find the version of the AWS CLI we just downloaded
+# Example version string: aws-cli/2.12.6 Python/3.11.4 Linux/5.15.49-linuxkit-pr exe/aarch64.ubuntu.22 prompt/off
+# Install CLI to path: /opt/awscli/v2/current/bin/aws
+ruby_block "install-awscli" do
+  block do
+    require "fileutils"
+    awscli_version_string = shell_out("#{cache_dir}/awscli/dist/aws", "--version")
+    awscli_version = awscli_version_string.stdout.split(" ").first.split("/").last
+    FileUtils.mkdir_p("/opt/awscli/v2/#{awscli_version}/bin/", :mode => 0755)
+    FileUtils.mv("#{cache_dir}/awscli/dist", "/opt/awscli/v2/#{awscli_version}/dist", :force => true)
+    FileUtils.ln_sf("/opt/awscli/v2/#{awscli_version}/dist/aws", "/opt/awscli/v2/#{awscli_version}/bin/aws")
+    FileUtils.ln_sf("/opt/awscli/v2/#{awscli_version}/dist/aws_completer", "/opt/awscli/v2/#{awscli_version}/bin/aws_completer")
+    FileUtils.rm("/opt/awscli/v2/current") if File.exist?("/opt/awscli/v2/current")
+    FileUtils.ln_sf("/opt/awscli/v2/#{awscli_version}", "/opt/awscli/v2/current")
+  end
+  action :nothing
+  subscribes :run, "archive_file[#{cache_dir}/#{awscli_zip}]", :immediately
+end
diff --git a/test/integration/awscli/serverspec/awscli_spec.rb b/test/integration/awscli/serverspec/awscli_spec.rb
new file mode 100644
index 000000000..919363fec
--- /dev/null
+++ b/test/integration/awscli/serverspec/awscli_spec.rb
@@ -0,0 +1,8 @@
+require "serverspec"
+
+# Required by serverspec
+set :backend, :exec
+
+describe command("/opt/awscli/v2/current/bin/aws --version") do
+  its("exit_status") { should eq 0 }
+end
-- 
2.39.5