Merge pull request #75 from ytjohn/patch-1

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

diff --git a/manifests/pin.pp b/manifests/pin.pp
index 2899fce77a568fad54f418f6943d7c5678269da0..14df14b5b3f370c8ec7bc7978f535ceb15e28548 100644 (file)
--- a/manifests/pin.pp
+++ b/manifests/pin.pp
@@ -2,20 +2,34 @@
 # pin a release in apt, useful for unstable repositories
 
 define apt::pin(
-  $packages = '*',
-  $priority = 0
+  $ensure     = present,
+  $packages   = '*',
+  $priority   = 0,
+  $release    = '',
+  $origin     = '',
+  $originator = ''
 ) {
 
   include apt::params
 
   $preferences_d = $apt::params::preferences_d
 
+  if $release != '' {
+    $pin = "release a=${release}"
+  } elsif $origin != '' {
+    $pin = "origin \"${origin}\""
+  } elsif $originator != '' {
+    $pin = "release o=${originator}"
+  } else {
+    $pin = "release a=${name}"
+  }
+
   file { "${name}.pref":
-    ensure  => file,
-    path    => "${preferences_d}/${name}",
+    ensure  => $ensure,
+    path    => "${preferences_d}/${name}.pref",
     owner   => root,
     group   => root,
     mode    => '0644',
-    content => "# ${name}\nPackage: ${packages}\nPin: release a=${name}\nPin-Priority: ${priority}",
+    content => template("apt/pin.pref.erb"),
   }
 }