introduce a way to pass content for includes

1 files changed, 12 insertions, 3 deletions

diff --git a/manifests/include.pp b/manifests/include.pp
index 0ab61b7..6e5a736 100644
--- a/manifests/include.pp
+++ b/manifests/include.pp
@@ -1,11 +1,20 @@
-define squid::include(){
+define squid::include($content = undef){
   include ::squid
   require squid::includedir
 
   file{"/etc/squid/include.d/${name}":
-    source => [ "puppet:///modules/site-squid/include.d/${fqdn}/${name}",
-                "puppet:///modules/site-squid/include.d/${name}" ],
     notify => Service['squid'],
     owner => root, group => squid, mode => 0640;
   }
+
+  if !$content {
+    File["/etc/squid/include.d/${name}"]{
+      source => [ "puppet:///modules/site-squid/include.d/${fqdn}/${name}",
+                  "puppet:///modules/site-squid/include.d/${name}" ]
+    }
+  } else {
+    File["/etc/squid/include.d/${name}"]{
+      content => $content,
+    }
+  }
 }