1 files changed, 18 insertions, 11 deletions
diff --git a/manifests/init.pp b/manifests/init.pp
index f2db659..80fbbd8 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -1,13 +1,20 @@
-# Basic class to manage augeas
-class augeas(
-  $version          = 'installed',
-  $rubylib_version  = 'installed'
-) {
-  $lens_dir = '/usr/share/augeas/lenses'
+# Class: augeas
+#
+# Install and configure Augeas
+#
+# Parameters:
+#   ['version']      - the desired version of Augeas
+#   ['ruby_version'] - the desired version of the Ruby bindings for Augeas
+#   ['lens_dir']     - the lens directory to use
+class augeas (
+  $version = $augeas_version,
+  $ruby_version = $augeas_ruby_version,
+  $lens_dir = $augeas::params::lens_dir,
+) inherits augeas::params {
 
-  case $::operatingsystem {
-    /RedHat|CentOS|Fedora/:   { include augeas::redhat }
-    /Debian|Ubuntu|kFreeBSD/: { include augeas::debian }
-    default:                  { include augeas::base }
-  }
+  class {'::augeas::packages': } ->
+  class {'::augeas::files': } ->
+  Class['augeas']
+
+  Package['ruby-augeas'] -> Augeas <| |>
 }