X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=3rdparty%2Fmodules%2Farchive%2Flib%2Fpuppet%2Ffunctions%2Farchive%2Fartifactory_checksum.rb;fp=3rdparty%2Fmodules%2Farchive%2Flib%2Fpuppet%2Ffunctions%2Farchive%2Fartifactory_checksum.rb;h=e5f9a38184cae0e00fe3132056015b0bb7ad7c6f;hb=ce70d6baf887ae03a2a6a7f5e73eb2e2c3dea208;hp=0000000000000000000000000000000000000000;hpb=0ba93256399fbad7ed8fabfa39c24dd47169dde3;p=mirror%2Fdsa-puppet.git

diff --git a/3rdparty/modules/archive/lib/puppet/functions/archive/artifactory_checksum.rb b/3rdparty/modules/archive/lib/puppet/functions/archive/artifactory_checksum.rb
new file mode 100644
index 000000000..e5f9a3818
--- /dev/null
+++ b/3rdparty/modules/archive/lib/puppet/functions/archive/artifactory_checksum.rb
@@ -0,0 +1,26 @@
+require 'json'
+require 'puppet_x/bodeco/util'
+
+Puppet::Functions.create_function(:'archive::artifactory_checksum') do
+  # @summary A function that returns the checksum value of an artifact stored in Artifactory
+  # @param url The URL of the artifact.
+  # @param checksum_type The checksum type.
+  #        Note the function will raise an error if you ask for sha256 but your artifactory instance doesn't have the sha256 value calculated.
+  # @return [String] Returns the checksum.
+  dispatch :artifactory_checksum do
+    param 'Stdlib::HTTPUrl', :url
+    optional_param "Enum['sha1','sha256','md5']", :checksum_type
+    return_type 'String'
+  end
+
+  def artifactory_checksum(url, checksum_type = 'sha1')
+    uri = URI(url.sub('/artifactory/', '/artifactory/api/storage/'))
+
+    response = PuppetX::Bodeco::Util.content(uri)
+    content = JSON.parse(response)
+
+    checksum = content['checksums'] && content['checksums'][checksum_type]
+    raise("Could not parse #{checksum_type} from url: #{uri}\nresponse: #{response.body}") unless checksum =~ %r{\b[0-9a-f]{5,64}\b}
+    checksum
+  end
+end