Added support for ensure option into yum::repo.

yum/manifests/init.pp

@@ -128,6 +128,9 @@ class yum::exclude {
 #
 #   $name:
 #       Repository name.
+#   $ensure:
+#       Present/enable enables repository (default), absent removes
+#       repository completely and disable disables repository.
 #   $baseurl:
 #       Base URL for this repository.
 #   $mirrorlist:
@@ -146,10 +149,15 @@ class yum::exclude {
 #     gpgkey  => "http://tmz.fedorapeople.org/repo/RPM-GPG-KEY-tmz",
 # }
 #
-define yum::repo($baseurl="", $mirrorlist="", $descr="", $gpgkey="") {
+define yum::repo($ensure="present", $baseurl="", $mirrorlist="", $descr="", $gpgkey="") {
 
     tag("bootstrap")
 
+    # can't use ensure variable inside template :(
+    $status = $ensure
+
+    if $status != "absent" {
+
         if !$baseurl and !$mirrorlist {
             fail("Either \$baseurl or \$mirrorlist needs to be defined for yum::repo")
         }
@@ -183,8 +191,13 @@ define yum::repo($baseurl="", $mirrorlist="", $descr="", $gpgkey="") {
             $descr_real = $name
         }
 
+    }
+
     file { "/etc/yum.repos.d/${name}.repo":
-        ensure  => present,
+        ensure  => $status ? {
+            "absent" => absent,
+            default  => present,
+        },
         content => template("yum/yum.repo.erb"),
         mode    => "0644",
         owner   => "root",

yum/templates/yum.repo.erb

@@ -1,6 +1,6 @@
 [<%= name %>]
 name=<%= descr_real %>
-enabled=1
+enabled=<% if status == 'disable' %>0<% else %>1<% end %>
 <% if baseurl != '' -%>
 baseurl=<%= baseurl %>
 <% else -%>