lots of shit

bin/helpers/user_groups.sh

@@ -0,0 +1,42 @@
+#!/usr/bin/env zsh
+
+source $HOME/dotfiles/bin/helpers/functions.sh
+
+ensure_user_groups() {
+    # Load yaml file
+    users=($(cat $DOTFILES_CONFIG | shyaml keys config.user_groups))
+
+    # For each user, ensure they are in the correct groups
+    for user in $users; do
+        # Ensure this user exists
+        if [[ ! $(id -u $user) ]]; then
+            printfe "%s\n" "red" "    - User $user does not exist"
+            continue
+        fi
+
+        ensure_user_in_groups $user
+    done
+}
+
+ensure_user_in_groups() {
+    user=$1
+    groups=($(cat $DOTFILES_CONFIG | shyaml get-values config.user_groups.$user))
+
+    printfe "%s\n" "cyan" "    - For user $user..."
+
+    # For each group, ensure the user is in it
+    for group in $groups; do
+        # Check if the group exists at all, otherwise skip
+        if [[ ! $(getent group $group) ]]; then
+            printfe "%s\n" "red" "      Group $group does not exist"
+            continue
+        fi
+
+        if [[ ! $groups == *$group* ]]; then
+            printfe "%s\n" "green" "      Adding $user to group $group"
+            sudo usermod -aG $group $user
+        else
+            printfe "%s\n" "green" "      $user is already in group $group"
+        fi
+    done
+}