X-Git-Url: https://review.fuel-infra.org/gitweb?a=blobdiff_plain;f=manifests%2Fpin.pp;h=f19df00f230f859d03674afad42a35ff4f532243;hb=d28de442663dbb6f982e8d8e28c2956171cf9810;hp=5a9dae6fafafb315e3134378334612d012727d0d;hpb=f848bac6072f99e1cd60a0cc0b84e5679c598dab;p=puppet-modules%2Fpuppetlabs-apt.git

diff --git a/manifests/pin.pp b/manifests/pin.pp
index 5a9dae6..f19df00 100644
--- a/manifests/pin.pp
+++ b/manifests/pin.pp
@@ -2,19 +2,31 @@
 # pin a release in apt, useful for unstable repositories
 
 define apt::pin(
-	$packages = '*',
-	$priority = 0
+  $ensure   = present,
+  $packages = '*',
+  $priority = 0,
+  $release  = '',
+  $origin   = ''
 ) {
 
-	include apt
-	
-	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}",
-	}
+  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 => template("apt/pin.pref.erb"),
+  }
 }