(#12522) Adjust spec test for splitting purge

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

diff --git a/manifests/init.pp b/manifests/init.pp
index 03ef116193745f4f52f842b27483eb3c79b53bc5..ceb8d5ce0bffb38a7bf1bf2fc6b74033819ae25f 100644 (file)
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -3,11 +3,16 @@
 # This module manages the initial configuration of apt.
 #
 # Parameters:
-#   Both of the parameters listed here are not required in general and were
+#   The parameters listed here are not required in general and were
 #     added for use cases related to development environments.
 #   disable_keys - disables the requirement for all packages to be signed
 #   always_apt_update - rather apt should be updated on every run (intended
 #     for development environments where package updates are frequent
+#   purge_sources_list - Accepts true or false. Defaults to false If set to
+#     true, Puppet will purge all unmanaged entries from sources.list"
+#   purge_sources_list_d - Accepts true or false. Defaults to false. If set
+#     to false, Puppet will purge all unmanaged entries from sources.list.d
+#
 # Actions:
 #
 # Requires:
@@ -15,12 +20,18 @@
 # Sample Usage:
 #  class { 'apt': }
 class apt(
-  $disable_keys = false,
-  $always_apt_update = false
+  $always_apt_update = false,
+  $disable_keys = undef,
+  $proxy_host = false,
+  $proxy_port = '8080',
+  $purge_sources_list = false,
+  $purge_sources_list_d = false
 ) {
 
   include apt::params
 
+  validate_bool($purge_sources_list, $purge_sources_list_d)
+
   $refresh_only_apt_update = $always_apt_update? {
     true => false,
     false => true
@@ -28,30 +39,55 @@ class apt(
 
   package { "python-software-properties": }
 
-       file { "sources.list":
-               ensure => present,
-               owner => root,
-               group => root,
-               mode => 644,
-       }
-    name => "${apt::params::root}/sources.list",
-
-       file { "sources.list.d":
-               ensure => directory,
-               owner => root,
-               group => root,
-       }
-    name => "${apt::params::root}/sources.list.d",
+  file { "sources.list":
+    path => "${apt::params::root}/sources.list",
+    ensure => present,
+    owner => root,
+    group => root,
+    mode => 644,
+    content => $purge_sources_list ? {
+      false =>  undef,
+      true  => "# Repos managed by puppet.\n",
+    },
+  }
+
+  file { "sources.list.d":
+    path => "${apt::params::root}/sources.list.d",
+    ensure => directory,
+    owner => root,
+    group => root,
+    purge => $purge_sources_list_d,
+    recurse => $purge_sources_list_d,
+  }
 
   exec { "apt_update":
     command => "${apt::params::provider} update",
     subscribe => [ File["sources.list"], File["sources.list.d"] ],
     refreshonly => $refresh_only_apt_update,
   }
-  if($disable_keys) {
-    exec { 'make-apt-insecure':
-      command => '/bin/echo "APT::Get::AllowUnauthenticated 1;" >> /etc/apt/apt.conf.d/99unauth',
-      creates => '/etc/apt/apt.conf.d/99unauth'
+
+  case $disable_keys {
+    true: {
+      file { "99unauth":
+        content => "APT::Get::AllowUnauthenticated 1;\n",
+        ensure  => present,
+        path    => "/etc/apt/apt.conf.d/99unauth",
+      }
+    }
+    false: {
+      file { "99unauth":
+        ensure => absent,
+        path   => "/etc/apt/apt.conf.d/99unauth",
+      }
+    }
+    undef: { } # do nothing
+    default: { fail("Valid values for disable_keys are true or false") }
+  }
+
+  if($proxy_host) {
+    file { 'configure-apt-proxy':
+      path    => '/etc/apt/apt.conf.d/proxy',
+      content => "Acquire::http::Proxy \"http://${proxy_host}:${proxy_port}\";",
     }
   }
 }