1 files changed, 136 insertions, 0 deletions
diff --git a/modules/base/manifests/user.pp b/modules/base/manifests/user.pp
new file mode 100644
index 0000000..b1bd792
--- /dev/null
+++ b/modules/base/manifests/user.pp
@@ -0,0 +1,136 @@
+define base::user(
+  $pwhash,
+  $user=$name,
+  $group=$user,
+  $home=undef,
+  $extra_groups=[],
+  $homedir_mode='0700',
+  $shell='/usr/bin/zsh',
+  $persistent_data=undef,
+) {
+  $_home = $home ? {
+    undef => $user ? {
+      'root' => '/root',
+      default => "/home/$user",
+    },
+    default => $home,
+  }
+
+  include tozt::persistent
+
+  group { $group:
+    ensure => present;
+  }
+
+  user { $user:
+    ensure => 'present',
+    gid => $group,
+    groups => $extra_groups,
+    home => $_home,
+    shell => $shell,
+    password => $pwhash,
+    require => Group[$group];
+  }
+
+  file {
+    $_home:
+      ensure => 'directory',
+      owner => $user,
+      group => $group,
+      mode => $homedir_mode,
+      require => [
+        User[$user],
+        Group[$group],
+      ];
+    "${_home}/coding":
+      ensure => 'directory',
+      owner => $user,
+      group => $group,
+      mode => $homedir_mode,
+      require => [
+        User[$user],
+        Group[$group],
+        File[$_home],
+      ];
+  }
+
+  if $persistent_data == undef {
+    file {
+      "${_home}/.cargo":
+        ensure => 'directory',
+        owner => $user,
+        group => $group,
+        mode => $homedir_mode,
+        require => [
+          User[$user],
+          Group[$group],
+        ];
+      "${_home}/.rustup":
+        ensure => 'directory',
+        owner => $user,
+        group => $group,
+        mode => $homedir_mode,
+        require => [
+          User[$user],
+          Group[$group],
+        ];
+    }
+  }
+  else {
+    file {
+      "$persistent_data/cargo/${user}":
+        ensure => 'directory',
+        owner => $user,
+        group => $group,
+        mode => $homedir_mode,
+        require => [
+          User[$user],
+          Group[$group],
+        ];
+      "$persistent_data/rustup/${user}":
+        ensure => 'directory',
+        owner => $user,
+        group => $group,
+        mode => $homedir_mode,
+        require => [
+          User[$user],
+          Group[$group],
+        ];
+      "${_home}/.cargo":
+        ensure => link,
+        target => "$persistent_data/cargo/${user}",
+        owner => $user,
+        group => $group,
+        require => [
+          User[$user],
+          Group[$group],
+          File["${_home}"],
+        ];
+      "${_home}/.rustup":
+        ensure => link,
+        target => "$persistent_data/rustup/${user}",
+        owner => $user,
+        group => $group,
+        require => [
+          User[$user],
+          Group[$group],
+          File["${_home}"],
+        ];
+    }
+  }
+
+  rust::user { $user:
+  }
+  conf::user { $user:
+  }
+
+  if $user != 'root' {
+    sudo::user { $user:
+    }
+  }
+
+  if $shell == '/usr/bin/zsh' {
+    include zsh
+    Class['zsh'] -> User[$user]
+  }
+}