Merge pull request #332 from zacharyalexstern/fix_readme_typo

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

diff --git a/manifests/force.pp b/manifests/force.pp
index 564ed75ff4db02fabc6214b2a1c5434e6c100f61..152bb67354db6fa8960a20a942329f0cf5d7b147 100644 (file)
--- a/manifests/force.pp
+++ b/manifests/force.pp
@@ -2,15 +2,41 @@
 # force a package from a specific release
 
 define apt::force(
-       $release = 'testing',
-       $version = false
+  $release = false,
+  $version = false,
+  $timeout = 300
 ) {
 
-       exec { "/usr/bin/aptitude -y -t ${release} install ${name}":
-               unless => $version ? {
-                       false => "/usr/bin/dpkg -l | grep ${name}",
-                       default => "/usr/bin/dpkg -l | grep ${name} | grep ${version}"
-               }
-       }
+  $provider = $apt::params::provider
 
+  $version_string = $version ? {
+    false   => undef,
+    default => "=${version}",
+  }
+
+  $release_string = $release ? {
+    false   => undef,
+    default => "-t ${release}",
+  }
+
+  if $version == false {
+    if $release == false {
+      $install_check = "/usr/bin/dpkg -s ${name} | grep -q 'Status: install'"
+    } else {
+      # If installed version and candidate version differ, this check returns 1 (false).
+      $install_check = "/usr/bin/test \$(/usr/bin/apt-cache policy -t ${release} ${name} | /bin/grep -E 'Installed|Candidate' | /usr/bin/uniq -s 14 | /usr/bin/wc -l) -eq 1"
+    }
+  } else {
+    if $release == false {
+      $install_check = "/usr/bin/dpkg -s ${name} | grep -q 'Version: ${version}'"
+    } else {
+      $install_check = "/usr/bin/apt-cache policy -t ${release} ${name} | /bin/grep -q 'Installed: ${version}'"
+    }
+  }
+
+  exec { "${provider} -y ${release_string} install ${name}${version_string}":
+    unless    => $install_check,
+    logoutput => 'on_failure',
+    timeout   => $timeout,
+  }
 }