Merge branch 'master' of https://github.com/puppetlabs/puppetlabs-apt

[puppet-modules/puppetlabs-apt.git] / manifests / pin.pp

diff --git a/manifests/pin.pp b/manifests/pin.pp
index e8b251f7c882c60a413ecafa60509fb1d1230b99..21cc3ffdefc0294c0c91c37de474971b94b26869 100644 (file)
--- a/manifests/pin.pp
+++ b/manifests/pin.pp
@@ -2,19 +2,47 @@
 # pin a release in apt, useful for unstable repositories
 
 define apt::pin(
-       $packages = '*',
-       $priority = 0
+  $ensure     = present,
+  $explanation = "${::caller_module_name}: ${name}",
+  $order      = '',
+  $packages   = '*',
+  $priority   = 0,
+  $release    = '',
+  $origin     = '',
+  $originator = '',
+  $version    = ''
 ) {
 
-       include apt
+  include apt::params
 
-       file { "${name}.pref":
-               name => "${apt::root}/preferences.d/${name}",
-               ensure => file,
-               owner => root,
-               group => root,
-               mode => 644,
-               content => "# ${name}\nPackage: ${packages}\nPin: release a=${name}\nPin-Priority: ${priority}",
-       }
+  $preferences_d = $apt::params::preferences_d
 
+  if $order != '' and !is_integer($order) {
+    fail('Only integers are allowed in the apt::pin order param')
+  }
+
+  if $release != '' {
+    $pin = "release a=${release}"
+  } elsif $origin != '' {
+    $pin = "origin \"${origin}\""
+  } elsif $originator != '' {
+    $pin = "release o=${originator}"
+  } elsif $version != '' {
+    $pin = "version ${version}"
+  } else {
+    $pin = "release a=${name}"
+  }
+
+  $path = $order ? {
+    ''      => "${preferences_d}/${name}.pref",
+    default => "${preferences_d}/${order}-${name}.pref",
+  }
+  file { "${name}.pref":
+    ensure  => $ensure,
+    path    => $path,
+    owner   => root,
+    group   => root,
+    mode    => '0644',
+    content => template('apt/pin.pref.erb'),
+  }
 }