--- /dev/null
+require 'pathname'
+require 'uri'
+require 'puppet/util'
+require 'puppet/parameter/boolean'
+
+Puppet::Type.newtype(:archive) do
+ @doc = 'Manage archive file download, extraction, and cleanup.'
+
+ ensurable do
+ desc 'whether archive file should be present/absent (default: present)'
+
+ newvalue(:present) do
+ provider.create
+ end
+
+ newvalue(:absent) do
+ provider.destroy
+ end
+
+ defaultto(:present)
+
+ # The following changes allows us to notify if the resource is being replaced
+ def is_to_s(value) # rubocop:disable Style/PredicateName
+ return "(#{resource[:checksum_type]})#{provider.archive_checksum}" if provider.archive_checksum
+ super
+ end
+
+ def should_to_s(value)
+ return "(#{resource[:checksum_type]})#{resource[:checksum]}" if provider.archive_checksum
+ super
+ end
+
+ def change_to_s(currentvalue, newvalue)
+ if currentvalue == :absent || currentvalue.nil?
+ extract = resource[:extract] == :true ? "and extracted in #{resource[:extract_path]}" : ''
+ cleanup = resource[:cleanup] == :true ? 'with cleanup' : 'without cleanup'
+
+ if provider.archive_checksum
+ "replace archive: #{provider.archive_filepath} from #{is_to_s(currentvalue)} to #{should_to_s(newvalue)}"
+ else
+ "download archive from #{resource[:source]} to #{provider.archive_filepath} #{extract} #{cleanup}"
+ end
+ elsif newvalue == :absent
+ "remove archive: #{provider.archive_filepath} "
+ else
+ super
+ end
+ rescue StandardError
+ super
+ end
+ end
+
+ newparam(:path, namevar: true) do
+ desc 'namevar, archive file fully qualified file path.'
+ validate do |value|
+ unless Puppet::Util.absolute_path? value
+ raise ArgumentError, "archive path must be absolute: #{value}"
+ end
+ end
+ end
+
+ newparam(:filename) do
+ desc 'archive file name (derived from path).'
+ end
+
+ newparam(:extract) do
+ desc 'whether archive will be extracted after download (true|false).'
+ newvalues(:true, :false)
+ defaultto(:false)
+ end
+
+ newparam(:extract_path) do
+ desc 'target folder path to extract archive.'
+ validate do |value|
+ unless Puppet::Util.absolute_path? value
+ raise ArgumentError, "archive extract_path must be absolute: #{value}"
+ end
+ end
+ end
+ newparam(:target) do
+ desc 'target folder path to extract archive. (this parameter is for camptocamp/archive compatibility)'
+ validate do |value|
+ unless Puppet::Util.absolute_path? value
+ raise ArgumentError, "archive extract_path must be absolute: #{value}"
+ end
+ end
+ munge do |val|
+ resource[:extract_path] = val
+ end
+ end
+
+ newparam(:extract_command) do
+ desc "custom extraction command ('tar xvf example.tar.gz'), also support sprintf format ('tar xvf %s') which will be processed with the filename: sprintf('tar xvf %s', filename)"
+ end
+
+ newparam(:temp_dir) do
+ desc 'Specify an alternative temporary directory to use for copying files, if unset then the operating system default will be used.'
+ validate do |value|
+ unless Puppet::Util.absolute_path?(value)
+ raise ArgumentError, "Invalid temp_dir #{value}"
+ end
+ end
+ end
+
+ newparam(:extract_flags) do
+ desc "custom extraction options, this replaces the default flags. A string such as 'xvf' for a tar file would replace the default xf flag. A hash is useful when custom flags are needed for different platforms. {'tar' => 'xzf', '7z' => 'x -aot'}."
+ defaultto(:undef)
+ end
+
+ newproperty(:creates) do
+ desc 'if file/directory exists, will not download/extract archive.'
+
+ def should_to_s(value)
+ "extracting in #{resource[:extract_path]} to create #{value}"
+ end
+ end
+
+ newparam(:cleanup) do
+ desc 'whether archive file will be removed after extraction (true|false).'
+ newvalues(:true, :false)
+ defaultto(:true)
+ end
+
+ newparam(:source) do
+ desc 'archive file source, supports puppet|http|https|ftp|file|s3 uri.'
+ validate do |value|
+ unless value =~ URI.regexp(%w[puppet http https ftp file s3]) || Puppet::Util.absolute_path?(value)
+ raise ArgumentError, "invalid source url: #{value}"
+ end
+ end
+ end
+
+ newparam(:url) do
+ desc 'archive file source, supports http|https|ftp|file uri.
+ (for camptocamp/archive compatibility)'
+ validate do |value|
+ unless value =~ URI.regexp(%w[http https file ftp])
+ raise ArgumentError, "invalid source url: #{value}"
+ end
+ end
+ munge do |val|
+ resource[:source] = val
+ end
+ end
+
+ newparam(:cookie) do
+ desc 'archive file download cookie.'
+ end
+
+ newparam(:checksum) do
+ desc 'archive file checksum (match checksum_type).'
+ newvalues(%r{\b[0-9a-f]{5,128}\b}, :true, :false, :undef, nil, '')
+ munge do |val|
+ if val.nil? || val.empty? || val == :undef
+ :false
+ elsif [:true, :false].include? val
+ resource[:checksum_verify] = val
+ else
+ val
+ end
+ end
+ end
+ newparam(:digest_string) do
+ desc 'archive file checksum (match checksum_type)
+ (this parameter is for camptocamp/archive compatibility).'
+ newvalues(%r{\b[0-9a-f]{5,128}\b})
+ munge do |val|
+ if !val.nil? && !val.empty?
+ resource[:checksum] = val
+ else
+ val
+ end
+ end
+ end
+
+ newparam(:checksum_url) do
+ desc 'archive file checksum source (instead of specifying checksum)'
+ end
+ newparam(:digest_url) do
+ desc 'archive file checksum source (instead of specifying checksum)
+ (this parameter is for camptocamp/archive compatibility)'
+ munge do |val|
+ resource[:checksum_url] = val
+ end
+ end
+
+ newparam(:checksum_type) do
+ desc 'archive file checksum type (none|md5|sha1|sha2|sha256|sha384|sha512).'
+ newvalues(:none, :md5, :sha1, :sha2, :sha256, :sha384, :sha512)
+ defaultto(:none)
+ end
+ newparam(:digest_type) do
+ desc 'archive file checksum type (none|md5|sha1|sha2|sha256|sha384|sha512)
+ (this parameter is camptocamp/archive compatibility).'
+ newvalues(:none, :md5, :sha1, :sha2, :sha256, :sha384, :sha512)
+ munge do |val|
+ resource[:checksum_type] = val
+ end
+ end
+
+ newparam(:checksum_verify) do
+ desc 'whether checksum wil be verified (true|false).'
+ newvalues(:true, :false)
+ defaultto(:true)
+ end
+
+ newparam(:username) do
+ desc 'username to download source file.'
+ end
+
+ newparam(:password) do
+ desc 'password to download source file.'
+ end
+
+ newparam(:user) do
+ desc 'extract command user (using this option will configure the archive file permission to 0644 so the user can read the file).'
+ end
+
+ newparam(:group) do
+ desc 'extract command group (using this option will configure the archive file permisison to 0644 so the user can read the file).'
+ end
+
+ newparam(:proxy_type) do
+ desc 'proxy type (none|ftp|http|https)'
+ newvalues(:none, :ftp, :http, :https)
+ end
+
+ newparam(:proxy_server) do
+ desc 'proxy address to use when accessing source'
+ end
+
+ newparam(:allow_insecure, boolean: true, parent: Puppet::Parameter::Boolean) do
+ desc 'ignore HTTPS certificate errors'
+ defaultto :false
+ end
+
+ newparam(:download_options) do
+ desc 'provider download options (affects curl, wget, and only s3 downloads for ruby provider)'
+
+ validate do |val|
+ unless val.is_a?(::String) || val.is_a?(::Array)
+ raise ArgumentError, "download_options should be String or Array: #{val}"
+ end
+ end
+
+ munge do |val|
+ case val
+ when ::String
+ [val]
+ else
+ val
+ end
+ end
+ end
+
+ autorequire(:file) do
+ [
+ Pathname.new(self[:path]).parent.to_s,
+ self[:extract_path],
+ '/root/.aws/config',
+ '/root/.aws/credentials'
+ ].compact
+ end
+
+ autorequire(:exec) do
+ ['install_aws_cli']
+ end
+
+ validate do
+ filepath = Pathname.new(self[:path])
+ self[:filename] = filepath.basename.to_s
+ if !self[:source].nil? && !self[:url].nil? && self[:source] != self[:url]
+ raise ArgumentError, "invalid parameter: url (#{self[:url]}) and source (#{self[:source]}) are mutually exclusive."
+ end
+ if !self[:checksum_url].nil? && !self[:digest_url].nil? && self[:checksum_url] != self[:digest_url]
+ raise ArgumentError, "invalid parameter: checksum_url (#{self[:checksum_url]}) and digest_url (#{self[:digest_url]}) are mutually exclusive."
+ end
+ if self[:proxy_server]
+ self[:proxy_type] ||= URI(self[:proxy_server]).scheme.to_sym
+ else
+ self[:proxy_type] = :none
+ end
+ end
+end