finish refactoring

refactor everything towards class parameters and also add enable
class for lastlog

1 files changed, 25 insertions, 28 deletions

diff --git a/manifests/debian.pp b/manifests/debian.pp
index 2df522f..12ec571 100644
--- a/manifests/debian.pp
+++ b/manifests/debian.pp
@@ -1,36 +1,33 @@
 class loginrecords::debian inherits loginrecords::base {
 
-    $login_defs_file = '/etc/login.defs'
-    $btmp_file       = '/var/log/btmp'
-    $utmp_file       = '/var/run/utmp'
-    $wtmp_file       = '/var/log/wtmp'
+  if $loginrecords::disable_btmp {
+    class{'loginrecords::btmp::disable': }
+  } else {
+    class{'loginrecords::btmp::enable': }
+  }
 
-    if $loginrecords::disable_btmp {
-        include loginrecords::btmp::disable
-    } else {
-        include loginrecords::btmp::enable
-    }
+  if $loginrecords::disable_faillog {
+    class{'loginrecords::faillog::disable': }
+  } else {
+    class{'loginrecords::faillog::enable': }
+  }
 
-    if $loginrecords::disable_faillog {
-        include loginrecords::faillog::disable
-    } else {
-        include loginrecords::faillog::enable
-    }
+  if $loginrecords::disable_lastlog {
+    class{'loginrecords::lastlog::disable': }
+  } else {
+    class{'loginrecords::lastlog::enable': }
+  }
 
-    if $loginrecords::disable_lastlog {
-        class{'loginrecords::lastlog::disable': }
-    }
+  if $loginrecords::disable_utmp {
+    class{'loginrecords::utmp::disable': }
+  } else {
+    class{'loginrecords::utmp::enable': }
+  }
 
-    if $loginrecords::disable_utmp {
-        include loginrecords::utmp::disable
-    } else {
-        include loginrecords::utmp::enable
-    }
-
-    if $loginrecords::disable_wtmp {
-        include loginrecords::wtmp::disable
-    } else {
-        include loginrecords::wtmp::enable
-    }
+  if $loginrecords::disable_wtmp {
+    class{'loginrecords::wtmp::disable': }
+  } else {
+    class{'loginrecords::wtmp::enable': }
+  }
 
 }