move pin out to a template

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

diff --git a/manifests/pin.pp b/manifests/pin.pp
index 29fddfb85ae4656369c023b1e2522bf909f2ff5c..f19df00f230f859d03674afad42a35ff4f532243 100644 (file)
--- a/manifests/pin.pp
+++ b/manifests/pin.pp
@@ -5,19 +5,28 @@ define apt::pin(
   $ensure   = present,
   $packages = '*',
   $priority = 0,
-  $release  = $name
+  $release  = '',
+  $origin   = ''
 ) {
 
   include apt::params
 
   $preferences_d = $apt::params::preferences_d
 
+  if $release != '' {
+    $pin = "release a=${release}"
+  } elsif $origin != '' {
+    $pin = "origin \"${origin}\""
+  } else {
+    err("Apt::Pin needs either $release or $origin")
+  }
+
   file { "${name}.pref":
     ensure  => $ensure,
     path    => "${preferences_d}/${name}",
     owner   => root,
     group   => root,
     mode    => '0644',
-    content => "# ${name}\nPackage: ${packages}\nPin: release a=${release}\nPin-Priority: ${priority}",
+    content => template("apt/pin.pref.erb"),
   }
 }