Change the structure of this module

1 files changed, 21 insertions, 15 deletions

diff --git a/manifests/classes/augeas.pp b/manifests/classes/augeas.pp
index a093905..6c6f08f 100644
--- a/manifests/classes/augeas.pp
+++ b/manifests/classes/augeas.pp
@@ -1,22 +1,28 @@
-class augeas::base {
-
+class augeas {
   case $operatingsystem {
-    RedHat: {
-      package {
-        ["augeas", "augeas-libs", "ruby-augeas"]:
-          ensure => "present";
-      }
-    }
-
-    Debian, Ubuntu: {
-      package {
-        ["augeas-lenses", "libaugeas0", "augeas-tools", "libaugeas-ruby1.8"]:
-          ensure => "present";
-      }
-    }
+    redhat: { include augeas::redhat }
+    debian: { include augeas::debian }
+    ubuntu: { include augeas::debian }
+    default: { include augeas::base }
   }
+}
 
+class augeas::base {
   file {"/usr/share/augeas/lenses/contrib":
     ensure => directory,
   }
 }
+
+class augeas::redhat inherits augeas::base {
+  package {
+    ["augeas", "augeas-libs", "ruby-augeas"]:
+      ensure => "present";
+  }
+}
+
+class augeas::debian inherits augeas::base {
+  package {
+    ["augeas-lenses", "libaugeas0", "augeas-tools", "libaugeas-ruby1.8"]:
+       ensure => "present";
+  }
+}