#!/usr/bin/env bash

# Tell build process to exit if there are any errors.
set -euo pipefail

# Copy included systemd unit files to system
SYSTEM_UNIT_INCLUDE="$CONFIG_DIRECTORY"/systemd/system
USER_UNIT_INCLUDE="$CONFIG_DIRECTORY"/systemd/user
SYSTEM_UNIT_DIR="/usr/lib/systemd/system"
USER_UNIT_DIR="/usr/lib/systemd/user"

if [[ -d "$SYSTEM_UNIT_INCLUDE" ]]; then
  if [[ -n $(find "$SYSTEM_UNIT_INCLUDE" -type f) ]]; then
    echo "Copying 'system' systemd units to system directory"
    cp -r "$SYSTEM_UNIT_INCLUDE"/* "$SYSTEM_UNIT_DIR"
  fi
fi
if [[ -d "$USER_UNIT_INCLUDE" ]]; then
  if [[ -n $(find "$USER_UNIT_INCLUDE" -type f) ]]; then
    echo "Copying 'user' systemd units to system directory"  
    cp -r "$USER_UNIT_INCLUDE"/* "$USER_UNIT_DIR"
  fi  
fi  

# Systemd units configuration (enable, disable, unmask & mask)
get_yaml_array ENABLED '.system.enabled[]' "$1"
get_yaml_array DISABLED '.system.disabled[]' "$1"
get_yaml_array UNMASKED '.system.unmasked[]' "$1"
get_yaml_array MASKED '.system.masked[]' "$1"
get_yaml_array USER_ENABLED '.user.enabled[]' "$1"
get_yaml_array USER_DISABLED '.user.disabled[]' "$1"
get_yaml_array USER_UNMASKED '.user.unmasked[]' "$1"
get_yaml_array USER_MASKED '.user.masked[]' "$1"

if [[ ${#ENABLED[@]} -gt 0 ]]; then
    for unit in "${ENABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl -f enable $unit
    done
fi
if [[ ${#DISABLED[@]} -gt 0 ]]; then
    for unit in "${DISABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl disable $unit
    done
fi
if [[ ${#UNMASKED[@]} -gt 0 ]]; then
    for unit in "${UNMASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl unmask $unit
    done
fi
if [[ ${#MASKED[@]} -gt 0 ]]; then
    for unit in "${MASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl mask $unit
    done
fi
if [[ ${#USER_ENABLED[@]} -gt 0 ]]; then
    for unit in "${USER_ENABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global -f enable $unit
    done
fi
if [[ ${#USER_DISABLED[@]} -gt 0 ]]; then
    for unit in "${USER_DISABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global disable $unit
    done
fi
if [[ ${#USER_UNMASKED[@]} -gt 0 ]]; then
    for unit in "${USER_UNMASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global unmask $unit
    done
fi
if [[ ${#USER_MASKED[@]} -gt 0 ]]; then
    for unit in "${USER_MASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global mask $unit
    done
fi